package database

import (
	"fmt"

	"git.secnex.io/cluequest/go-sdk/config"
	"git.secnex.io/cluequest/go-sdk/models"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

type Connection struct {
	*gorm.DB
}

var Conn Connection

func NewConnection(cfg config.DatabaseConfiguration) error {
	dsn := fmt.Sprintf(
		"host=%s user=%s password=%s dbname=%s port=%s sslmode=%s",
		cfg.Host, cfg.User, cfg.Pass, cfg.DB, cfg.Port, cfg.SSL,
	)

	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
	if err != nil {
		return err
	}

	fmt.Println("✅ Connection to database established!")
	Conn = Connection{DB: db}

	// AutoMigration()
	err = Conn.AutoMigrate(&models.User{}, &models.Quiz{}, &models.Organization{})
	if err != nil {
		return err
	}

	return nil
}

func NewConnectionFromEnv() error {
	return NewConnection(config.NewDatabaseConfigurationFromEnv())
}

func AutoMigration() error {
	return Conn.AutoMigrate(&models.User{}, &models.Quiz{}, &models.Organization{})
}