package db

import (
	"giter.top/smart/pkg/config"
	"gorm.io/gorm"
)

func NewDB(cfg *config.Config) *gorm.DB {
	driver := cfg.Data.Database.Driver
	var db *gorm.DB
	var err error
	switch driver {
	case "mysql":
		// db, err = NewMySQLDB(cfg)
	case "postgres":
		db, err = NewPgSQLDB(cfg)
	case "sqlite":
		// return NewSQLiteDB(cfg)
	default:
		panic("unsupported driver")
	}

	if err != nil {
		panic(err)
	}
	return db
}