package a

import (
	"errors"
	"log"
	"strings"

	"git.apinb.com/dataset/stock/internal/impl"
	"git.apinb.com/dataset/stock/internal/models"
	"gorm.io/gorm"
)

func GetStockBasic() {
	// 参数
	params := map[string]string{
		"list_status": "L", // 上市状态 L上市 D退市 P暂停上市，默认是L
	}

	// 字段
	fields := strings.Split("ts_code,symbol,name,area,industry,fullname,enname,cnspell,market,exchange,list_date,is_hs,act_name,act_ent_type", ",")

	// 根据api 请求对应的接口
	reply, _ := TushareClient.StockBasic(params, fields)
	if reply.Code != 0 {
		log.Println("ERROR", "GetStockBasic", reply.Code, reply.Msg)
		return
	}

	for _, item := range reply.Data.Items {
		var record models.StockBasic
		data := Anys2Strings(item)

		err := impl.DBService.Where("ts_code=?", data[0]).First(&record).Error
		if errors.Is(err, gorm.ErrRecordNotFound) {
			// create
			record = models.StockBasic{
				TsCode:     data[0],
				Symbol:     data[1],
				Name:       data[2],
				Area:       data[3],
				Industry:   data[4],
				FullName:   data[5],
				EnName:     data[6],
				CnSpell:    data[7],
				Market:     data[8],
				Exchange:   data[9],
				ListDate:   data[10],
				IsHS:       data[11],
				ActName:    data[12],
				ActEntType: data[13],
			}
			impl.DBService.Create(&record)
		} else if err == nil {
			// check updated.
			upd := make(map[string]any)
			if record.Name != data[2] {
				upd["name"] = data[2]
			}
			if record.Area != data[3] {
				upd["area"] = data[3]
			}
			if record.Industry != data[4] {
				upd["industry"] = data[4]
			}
			if record.FullName != data[5] {
				upd["full_name"] = data[5]
			}
			if record.EnName != data[6] {
				upd["en_name"] = data[6]
			}
			if record.CnSpell != data[7] {
				upd["cn_spell"] = data[7]
			}
			if record.Market != data[8] {
				upd["market"] = data[8]
			}
			if record.Exchange != data[9] {
				upd["exchange"] = data[9]
			}
			if record.ListDate != data[10] {
				upd["list_date"] = data[10]
			}
			if record.IsHS != data[11] {
				upd["is_hs"] = data[11]
			}
			if record.ActName != data[12] {
				upd["act_name"] = data[12]
			}
			if record.ActEntType != data[13] {
				upd["act_ent_type"] = data[13]
			}
			if len(upd) > 0 {
				impl.DBService.Model(record).Updates(upd)
			}
		}
	}
}

func GetStockCodes() []string {
	var col []string
	impl.DBService.Model(&models.StockBasic{}).Pluck("ts_code", &col)
	return col
}