package migrations

import (
	"context"
	"fmt"

	"github.com/uptrace/bun"
	"go.neonxp.ru/framework/pkg/middleware/session"
)

//nolint:gochecknoinits
func init() {
	Migrations.MustRegister(func(ctx context.Context, db *bun.DB) error {
		fmt.Print(" [up migration] ")

		if _, err := db.NewCreateTable().
			Model((*session.Model)(nil)).
			Exec(ctx); err != nil {
			return err
		}

		return nil
	}, func(ctx context.Context, db *bun.DB) error {
		fmt.Print(" [down migration] ")

		if _, err := db.NewDropTable().
			Model((*session.Model)(nil)).
			Exec(ctx); err != nil {
			return err
		}

		return nil
	})
}