package tracker

import (
	"fmt"
	"gitee.com/quant1x/engine/config"
	"gitee.com/quant1x/engine/market"
	"gitee.com/quant1x/engine/models"
	"gitee.com/quant1x/engine/permissions"
	"gitee.com/quant1x/gox/concurrent"
	"gitee.com/quant1x/gox/logger"
	"gitee.com/quant1x/gox/progressbar"
	"gitee.com/quant1x/gox/tags"
	"gitee.com/quant1x/pkg/tablewriter"
	"os"
)

// ExecuteStrategy 执行策略
// Deprecated: 推荐 ExecuteStrategy
func v1ExecuteStrategy(model models.Strategy, barIndex *int) {
	fmt.Printf("策略模块: %s\n", model.Name())
	// 加载即时行情
	models.SyncAllSnapshots(barIndex)
	fmt.Println()
	// 执行策略
	allCodes := market.GetCodeList()
	count := len(allCodes)
	bar := progressbar.NewBar(*barIndex, "执行["+model.Name()+"]", count)
	results := concurrent.NewTreeMap[string, models.ResultInfo]()
	for _, securityCode := range allCodes {
		// 此处可以增加过滤规则
		model.Evaluate(securityCode, results)
		bar.Add(1)
	}
	// 输出一个换行符, 结束上一个进度条
	fmt.Println()
	table := tablewriter.NewWriter(os.Stdout)
	table.SetHeader(tags.GetHeadersByTags(models.ResultInfo{}))
	results.Each(func(key string, value models.ResultInfo) {
		table.Append(tags.GetValuesByTags(value))
	})
	table.Render()
}

// ExecuteStrategy 执行策略
func ExecuteStrategy(model models.Strategy, barIndex *int) {
	// 策略权限验证
	err := permissions.CheckPermission(model)
	if err != nil {
		logger.Error(err)
		return
	}
	tradeRule := config.GetStrategyParameterByCode(model.Code())
	if tradeRule == nil {
		fmt.Printf("strategy[%d]: trade rule not found\n", model.Code())
		return
	}
	// 加载快照数据
	models.SyncAllSnapshots(barIndex)
	// 计算市场情绪
	MarketSentiment()
	// 扫描板块
	ScanAllSectors(barIndex, model)
	// 扫描个股
	AllScan(barIndex, model)
}