package model

import (
	"aps_crm/conf"
	"aps_crm/pkg/logx"
	"aps_crm/pkg/mysqlx"
)

func Init() error {
	if err := mysqlx.Init(&conf.Conf.Mysql, logx.GetLogger()); err != nil {
		return err
	}

	if err := RegisterTables(); err != nil {
		return err
	}

	return nil
}

func RegisterTables() error {
	db := mysqlx.GetDB()
	err := db.AutoMigrate(
		User{},
		Menu{},
		JwtBlacklist{},
		Country{},
		Province{},
		City{},
		Client{},
		ClientLevel{},
		ClientOrigin{},
		ClientStatus{},
		ClientType{},
		Contact{},
		EnterpriseNature{},
		EnterpriseScale{},
		FollowRecord{},
		Industry{},
		RegisteredCapital{},
		SaleChance{},
		SalesLeads{},
		SalesSources{},
	)
	return err
}