package models

import (
	"basic.com/valib/logger.git"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"vamicro/config"
)

var db *gorm.DB
var err error

// Init creates a connection to mysql database and
// migrates any new models
func Init() {
	db, err = gorm.Open(config.DBconf.Name, "../config/system-service.db")
	if err != nil {
		logger.Debug("db open error ", err)
	}
	db.LogMode(true)
	db.AutoMigrate(
		&SysInit{},
		&Device{},
		&DeviceApp{},
		&DeviceSdk{},
		&Dictionary{},
		&Voice{},
		&LocalConfig{},
		&Cluster{},
		&Node{},
		&ResourceConfig{},
		&SysUser{},
		&SysMenu{},
		&SysRole{},
		&SysRoleMenu{},
		&SysUserRole{},
		&SysUserMenu{},
		&SysSetting{},
		&SqlSyncHis{},
		&DefHeadPic{},
		&AuthConfig{},
		&AuthDevice{},
	)

	InitSysSettingData()
	InitDefHeadPicData()
}

// GetDB ...
func GetDB() *gorm.DB {
	return db
}

func CloseDB() {
	db.Close()
}