package dao

import (
	"errors"
	"github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"hy2xs-admin/model/constant"
	"hy2xs-admin/model/entity"
	"time"
)

func SaveConfig(config entity.Config) (int64, error) {
	if tx := sqliteDB.Save(&config); tx.Error != nil {
		logrus.Errorf("%v", tx.Error)
		return 0, errors.New(constant.SysError)
	}
	return *config.Id, nil
}

func UpdateConfig(keys []string, updates map[string]interface{}) error {
	if len(updates) > 0 {
		updates["update_time"] = time.Now().Format("2006-01-02 15:04:05")
		if tx := sqliteDB.Model(&entity.Config{}).
			Where("key in ?", keys).
			Updates(updates); tx.Error != nil {
			logrus.Errorf("%v", tx.Error)
			return errors.New(constant.SysError)
		}
	}
	return nil
}

func GetConfig(query interface{}, args ...interface{}) (entity.Config, error) {
	var config entity.Config
	if tx := sqliteDB.Model(&entity.Config{}).
		Where(query, args...).First(&config); tx.Error != nil {
		if tx.Error == gorm.ErrRecordNotFound {
			return config, errors.New(constant.ConfigNotExist)
		}
		logrus.Errorf("%v", tx.Error)
		return config, errors.New(constant.SysError)
	}
	return config, nil
}

func ListConfig(query interface{}, args ...interface{}) ([]entity.Config, error) {
	var configs []entity.Config
	if tx := sqliteDB.Model(&entity.Config{}).
		Where(query, args...).Order("create_time desc").Find(&configs); tx.Error != nil {
		logrus.Errorf("%v", tx.Error)
		return configs, errors.New(constant.SysError)
	}
	return configs, nil
}

func UpsertConfig(configs []entity.Config) error {
	if tx := sqliteDB.Model(&entity.Config{}).Clauses(clause.OnConflict{
		Columns:   []clause.Column{{Name: "key"}},
		DoUpdates: clause.AssignmentColumns([]string{"value", "remark", "create_time", "update_time"}),
	}).Create(configs); tx.Error != nil {
		logrus.Errorf("%v", tx.Error)
		return errors.New(constant.SysError)
	}
	return nil
}