package tushare

import (
	"time"

	"git.apinb.com/quant/qsdk/conv"
)

/*
TradeCal 获取交易日历

	exchange: 交易所代码，SSE 上交所，SZSE 深交所，BSE 北交所，空字符串代表所有
	start_date: 开始日期，格式:YYYYMMDD
	end_date: 结束日期，格式:YYYYMMDD
	is_open: 是否开盘，'0'休市，'1'交易，空字符串代表所有
*/
func (cli *TushareClient) TradeCal(exchange, start_date, end_date, is_open string) (*TushareRespData, error) {
	params := map[string]any{}

	if exchange != "" {
		params["exchange"] = exchange
	}
	if start_date != "" {
		params["start_date"] = start_date
	}
	if end_date != "" {
		params["end_date"] = end_date
	}
	if is_open != "" {
		params["is_open"] = is_open
	}

	req := TushareReq{
		APIName: "trade_cal",
		Params:  params,
	}

	fields := []map[string]string{
		{"exchange": "交易所代码"},
		{"cal_date": "日历日期"},
		{"is_open": "是否开盘"},
		{"pretrade_date": "上一交易日"},
	}

	return cli.Do(req, fields)
}

func (cli *TushareClient) LastTradeDay() string {
	result, err := cli.TradeCal("SSE", time.Now().AddDate(0, 0, -15).Format("20060102"), time.Now().Format("20060102"), "1")
	if err != nil {
		return ""
	}
	cal := result.Map()
	if len(cal) == 0 {
		return ""
	}
	return conv.AnyToString(cal[0]["cal_date"])
}

func (cli *TushareClient) LimitTradeDay(limit int) []string {
	if limit <= 0 {
		return []string{}
	}
	if limit > 100 {
		limit = 100
	}
	result, err := cli.TradeCal("SSE", time.Now().AddDate(0, 0, -100).Format("20060102"), time.Now().Format("20060102"), "1")
	if err != nil {
		return []string{}
	}
	cal := result.Map()
	days := []string{}
	for i := 0; i < limit; i++ {
		item := cal[i]
		days = append(days, conv.AnyToString(item["cal_date"]))
	}
	return days
}