package main

import (
	"context"
	"database/sql"
	"log"
	"log/slog"
	"os"

	"github.com/pressly/goose/v3"
	_ "modernc.org/sqlite"
)

func main() {
	if err := run(); err != nil {
		log.Fatal(err)
	}
}

func run() error {
	db, err := sql.Open("sqlite", "test.db")
	if err != nil {
		return err
	}
	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
		ReplaceAttr: func(groups []string, a slog.Attr) slog.Attr {
			if a.Key == "statement" {
				return slog.Attr{
					Key:   a.Key,
					Value: slog.StringValue("<omitted>"),
				}
			}
			if a.Key == slog.TimeKey {
				return slog.Attr{
					Key:   slog.TimeKey,
					Value: slog.StringValue(a.Value.Time().Format("2006-01-02 15:04:05")),
				}
			}
			return a
		},
	}))
	// remove "statement" from the log output for brevity
	p, err := goose.NewProvider(goose.DialectSQLite3, db, os.DirFS("testdata/migrations"),
		goose.WithSlog(logger),
		goose.WithVerbose(true),
	)
	if err != nil {
		return err
	}
	args := os.Args[1:]
	if len(args) > 0 {
		switch args[0] {
		case "up":
			_, err = p.Up(context.Background())
		case "down":
			_, err = p.Down(context.Background())
		}
	}
	return err
}