package common

import (
	"fmt"
	"log"
	"os"
	"strings"
	"time"

	"github.com/google/uuid"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

type Database struct {
	*gorm.DB
}

var DB *gorm.DB

// Opening a database and save the reference to `Database` struct.
func InitDB() *gorm.DB {
	host := os.Getenv("DB_HOST")
	user := os.Getenv("DB_USER")
	pass := os.Getenv("DB_PASS")
	dbName := os.Getenv("DB_NAME")
	port := os.Getenv("DB_PORT")

	var sslMode string
	if os.Getenv("DB_SSL") == "TRUE" {
		sslMode = "enable"
	} else {
		sslMode = "disable"
	}

	// DB Logger config
	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		logger.Config{
			SlowThreshold:             time.Second,   // Slow SQL threshold
			LogLevel:                  logger.Silent, // Log level
			IgnoreRecordNotFoundError: true,          // Ignore ErrRecordNotFound error for logger
			Colorful:                  true,          // Disable color
		},
	)

	dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Singapore", host, user, pass, dbName, port, sslMode)

	var db *gorm.DB
	var err error

	for ok := true; ok; ok = err != nil {
		db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
			Logger: newLogger,
		})
		if err != nil {
			fmt.Println("db err: (Init) ", err)
			time.Sleep(1 * time.Second)
		}
	}

	// Setup UUID
	// CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
	db.Exec("CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\";")

	DB = db
	return DB
}

// This function will create a temporarily database for running testing cases
func TestDBInit() *gorm.DB {
	host := os.Getenv("TEST_DB_HOST")
	user := os.Getenv("TEST_DB_USER")
	pass := os.Getenv("TEST_DB_PASS")
	dbName := fmt.Sprintf("%s_%s", os.Getenv("TEST_DB_NAME"), uuid.New().String())
	dbName = strings.ReplaceAll(dbName, "-", "_")
	port := os.Getenv("TEST_DB_PORT")

	var sslMode string
	if os.Getenv("TEST_DB_SSL") == "TRUE" {
		sslMode = "enable"
	} else {
		sslMode = "disable"
	}

	// DB Logger config
	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
		logger.Config{
			SlowThreshold:             time.Second,   // Slow SQL threshold
			LogLevel:                  logger.Silent, // Log level
			IgnoreRecordNotFoundError: true,          // Ignore ErrRecordNotFound error for logger
			Colorful:                  true,          // Disable color
		},
	)

	dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Singapore", host, user, pass, "postgres", port, sslMode)

	var db *gorm.DB
	var err error

	for ok := true; ok; ok = err != nil {
		db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
			Logger: newLogger,
		})
		if err != nil {
			fmt.Println("db err: (Init) ", err)
			time.Sleep(1 * time.Second)
		}
	}

	// Create Database
	err = db.Exec(fmt.Sprintf("CREATE DATABASE %s;", dbName)).Error
	if err != nil {
		fmt.Println("db err: (Init) ", err)
	}

	// Get into testing database
	dsn = fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=Asia/Singapore", host, user, pass, dbName, port, sslMode)
	db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
		Logger: newLogger,
	})
	if err != nil {
		fmt.Println("db err: (Init) ", err)
	}

	// Setup UUID
	// CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
	db.Exec("CREATE EXTENSION IF NOT EXISTS \"uuid-ossp\";")

	DB = db
	return DB
}

func DestroyTestingDB(db *gorm.DB) {
	var dbName string
	db.Raw("SELECT current_database();").Scan(&dbName)
	db.Exec(fmt.Sprintf("DROP DATABASE %s", dbName))
}

// Using this function to get a connection, you can create your connection pool here.
func GetDB() *gorm.DB {
	return DB
}