package models

import (
	"strings"
	"time"

	"git.apinb.com/bsm-sdk/core/database"
	"git.apinb.com/bsm-sdk/core/utils"
	"git.apinb.com/quant/gostock/internal/impl"
	"gorm.io/gorm"
)

// StartModel 策略模型
type StratModel struct {
	gorm.Model
	StratKey      string
	Ymd           int
	Code          string
	UpDateDay     int     //上市时间，天数
	IndustryScore int     // 行业分组
	StScore       int     // 是否是ST
	GtAmount      int     // 每日交易额大于设定值
	GtPrice       int     // 最近20日交易日价格大于设定值
	RoeScore      int     // ROE 是否大于设定值
	ValRoe        float64 // ROE 值
	RecommendDesc string  // 推荐描述
	Status        int     // 状态 -1:不推荐，0:未处理， 1:推荐待指标分析， 2:待AI分析

	// macd
	MacdScore int `gorm:"default:0"`
	MacdVal   float64

	// Rsi值
	RsiScore       int `gorm:"default:0"`
	RsiValOversold int
	RsiValPrve     float64
	RsiValLast     float64

	//AI
	AiSummary      string
	AiSummary2025  string
	AiSummaryBase  string
	AiSummaryTech  string
	AiScore        int
	AiSupportLevel float64
	AiResisLevel   float64
	AiAction       string
	AiRisk         string

	TotalDesc  string
	TotalScore float64 `gorm:"default:0"`
}

func init() {
	database.AppendMigrate(&StratModel{})
}

// TableName 设置表名
func (StratModel) TableName() string {
	return "strat_model"
}

var (
	desc []string
)

func NewStratModel(key, code string, ymd int) *StratModel {
	desc = []string{}
	obj := StratModel{
		StratKey: key,
		Ymd:      ymd,
		Code:     code,
	}
	return &obj
}

func (s *StratModel) AddDesc(d string) {
	desc = append(desc, d)
}

func (s *StratModel) Save() error {
	var cnt int64
	impl.DBService.Model(&StratModel{}).Where("strat_key=? and ymd=? and code=?", s.StratKey, s.Ymd, s.Code).Count(&cnt)
	s.TotalDesc = strings.Join(desc, "||")

	if cnt == 0 {
		// create record.
		return impl.DBService.Create(s).Error
	} else {
		// update record.
		return impl.DBService.Model(&StratModel{}).Where("strat_key=? and ymd=? and code=?", s.StratKey, s.Ymd, s.Code).Updates(s).Error
	}

	return nil

}

func GetYmd() int {
	ymd := time.Now().Format("20060102")
	return utils.String2Int(ymd)
}