package storage

import (
	storage "git.gocasts.ir/ebhomengo/niki/adapter/storage/models"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

type DBMS struct {
	db *gorm.DB
}

func GormConnect(dsn string) (DBMS, error) {
	gorm, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		return DBMS{}, err
	}
	return DBMS{db: gorm}, nil
}

func GormAutoMigrate(db gorm.DB) error {
	// Migrate the schema
	err := db.AutoMigrate(
		storage.PiggyBank{},
	)
	if err != nil {
		panic(any("Storage auto migrate has error"))
	}
	return err
}