package main

import (
	"github.com/sirupsen/logrus"
	"gorm.io/driver/sqlite"
	"gorm.io/gen"
	"gorm.io/gorm"

	"git.sr.ht/~gabrielgio/img/pkg/database/sql"
	"git.sr.ht/~gabrielgio/img/pkg/ext"
)

func main() {
	cfg := gen.Config{
		OutPath:      "./pkg/database/sql/query",
		WithUnitTest: true,
		Mode:         gen.WithoutContext | gen.WithDefaultQuery | gen.WithQueryInterface, // generate mode
	}

	g := gen.NewGenerator(cfg)

	logger := logrus.New()
	logger.SetLevel(logrus.TraceLevel)
	d := sqlite.Open("test.db")
	db, err := gorm.Open(d, &gorm.Config{
		Logger: ext.Wraplog(logger.WithField("context", "sql")),
	})
	if err != nil {
		panic("failed to gen database: " + err.Error())
	}

	g.UseDB(db)

	for _, m := range []any{
		&sql.User{},
		&sql.Settings{},
		&sql.Media{},
	} {
		g.ApplyBasic(m)
	}

	g.Execute()
}