package database

import (
	"log/slog"
	"pagoda/config"

	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

var Database *gorm.DB

func Initialise() {
	var err error
	Database, err = gorm.Open(postgres.Open(config.DSN), &gorm.Config{})
	if err != nil {
		slog.Error("Failed to connect to database")
		panic(err)
	}

	slog.Info("Connected to database")

	if config.ClearAndResetDatabaseOnStart {
		if err := clearAndResetDatabase(); err != nil {
			slog.Error("Failed to clear and reset database")
			panic(err)
		}
	}

	if err := migrateSchema(); err != nil {
		slog.Error("Failed to migrate schema")
		panic(err)
	}

	slog.Info("Database initialised successfully")
}

func clearAndResetDatabase() error {
	// Drop all tables
	err := Database.Exec("DROP SCHEMA public CASCADE").Error
	if err != nil {
		slog.Error("Error dropping schema", "error", err)
		return err
	}

	// Recreate public schema
	err = Database.Exec("CREATE SCHEMA public").Error
	if err != nil {
		slog.Error("Error recreating schema", "error", err)
		return err
	}

	slog.Info("All data cleared and database reset successfully.")
	return nil
}

func migrateSchema() error {
	err := Database.AutoMigrate(&User{}, &Session{})
	if err != nil {
		slog.Error("Error migrating schema", "error", err)
		return err
	}

	slog.Info("Schema migrated successfully")
	return nil
}