package migration

import (
	"fmt"
	"io"
	"os"
	"os/exec"

	"ariga.io/atlas-provider-gorm/gormschema"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
)

func loader(input string) {
	gormCfg := &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true,
			NoLowerCase:   true,
		},
	}
	stmts, err := gormschema.New("postgres", gormschema.WithConfig(gormCfg)).Load(getEntities(input)...)
	if err != nil {
		fmt.Fprintf(os.Stderr, "failed to load gorm schema: %v\n", err)
		os.Exit(1)
	}
	_, _ = io.WriteString(os.Stdout, stmts)
}

func getEntities(input string) []any {
	switch input {
	case "main":
		return getMainEntities()
	case "satusehat":
		return getSatuSehatEntities()
	}
	return nil
}

func runAtlas(args ...string) error {
	cmd := exec.Command("atlas", args...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	return cmd.Run()
}

func logMsg(msg string) {
	fmt.Fprintln(os.Stderr, msg)
}

func Migrate(input string) {
	loader(input)
	logMsg("Running atlas migrate diff...")
	if err := runAtlas("migrate", "diff", "--env", "gorm"); err != nil {
		logMsg(fmt.Sprintf("Failed to run diff: %v", err))
		return
	}

	logMsg("Running atlas migrate apply...")
	if err := runAtlas("migrate", "apply", "--env", "gorm"); err != nil {
		logMsg(fmt.Sprintf("Failed to run apply: %v", err))
		return
	}

	logMsg("Migration complete")
}