package utils

import (
	"context"

	"gitea.konchin.com/go2025/backend/models"
	"github.com/spf13/viper"
	"github.com/uptrace/bun"
)

func InitDB(ctx context.Context, db *bun.DB) error {
	db.RegisterModel(
		(*models.AliasImage)(nil),
		(*models.Alias)(nil),
		(*models.Image)(nil),
		(*models.Session)(nil),
	)
	if viper.GetBool("reset") {
		return db.ResetModel(ctx,
			(*models.AliasImage)(nil),
			(*models.Alias)(nil),
			(*models.Image)(nil),
			(*models.Session)(nil),
		)
	}
	return nil
}