package data

import (
	"go-stock/backend/db"
	"go-stock/backend/logger"
	"gorm.io/gorm"
)

type Settings struct {
	gorm.Model
	LocalPushEnable bool   `json:"localPushEnable"`
	DingPushEnable  bool   `json:"dingPushEnable"`
	DingRobot       string `json:"dingRobot"`
}

func (receiver Settings) TableName() string {
	return "settings"
}

type SettingsApi struct {
	Config Settings
}

func NewSettingsApi(settings *Settings) *SettingsApi {
	return &SettingsApi{
		Config: *settings,
	}
}

func (s SettingsApi) UpdateConfig() string {
	count := int64(0)
	db.Dao.Model(s.Config).Count(&count)
	if count > 0 {
		db.Dao.Model(s.Config).Where("id=?", s.Config.ID).Updates(map[string]any{
			"local_push_enable": s.Config.LocalPushEnable,
			"ding_push_enable":  s.Config.DingPushEnable,
			"ding_robot":        s.Config.DingRobot,
		})
	} else {
		logger.SugaredLogger.Infof("未找到配置，创建默认配置:%+v", s.Config)
		db.Dao.Model(s.Config).Create(&Settings{
			LocalPushEnable: s.Config.LocalPushEnable,
			DingPushEnable:  s.Config.DingPushEnable,
			DingRobot:       s.Config.DingRobot,
		})
	}
	return "保存成功！"
}
func (s SettingsApi) GetConfig() *Settings {
	var settings Settings
	db.Dao.Model(&Settings{}).First(&settings)
	return &settings
}