package data

import (
	"nixcn-cms/config"
	"nixcn-cms/data/drivers"

	log "github.com/sirupsen/logrus"
)

var Database *drivers.DBClient

func Init() {
	// Init database
	dbType := config.Get("database.type").(string)
	exDSN := drivers.ExternalDSN{
		Host:     config.Get("database.host").(string),
		Name:     config.Get("database.name").(string),
		Username: config.Get("database.username").(string),
		Password: config.Get("database.password").(string),
	}

	if dbType != "postgres" {
		log.Fatal("[Database] Only support postgras db!")
	}

	// Conect to db
	db, err := drivers.Postgres(exDSN)
	if err != nil {
		log.Fatal("[Database] Error connecting to db!")
	}

	// Auto migrate
	err = db.DB.AutoMigrate(&User{})
	if err != nil {
		log.Error("[Database] Error migrating database: ", err)
	}

	Database = db
}