package rules

import (
	"gitee.com/quant1x/engine/config"
	"gitee.com/quant1x/engine/factors"
	"gitee.com/quant1x/engine/market"
	"gitee.com/quant1x/gox/api"
	"gitee.com/quant1x/gox/exception"
	"gitee.com/quant1x/gox/logger"
	"gitee.com/quant1x/gox/num"
)

func init() {
	err := RegisterFunc(KRuleF10, "基本面", ruleF10)
	if err != nil {
		logger.Fatalf("%+v", err)
	}
}

var (
	ErrF10IgnoreStock                 = exception.New(errorRuleF10+0, "忽略的个股")
	ErrF10SubNewStock                 = exception.New(errorRuleF10+1, "次新股")
	ErrF10DisableBeijingStockExchange = exception.New(errorRuleF10+2, "禁止北交所")
	ErrF10DisableChiNextBoard         = exception.New(errorRuleF10+3, "禁止创业板")
	ErrF10DisableSciTechBoard         = exception.New(errorRuleF10+4, "禁止科创板")
	ErrF10PriceRange                  = exception.New(errorRuleF10+5, "股价控制")
	ErrF10RangeOfCapital              = exception.New(errorRuleF10+6, "非流通盘范围")
	ErrF10RangeOfSafetyCode           = exception.New(errorRuleF10+7, "非安全分范围")
	ErrF10RangeOfBasicEPS             = exception.New(errorRuleF10+8, "非每股收益范围")
	ErrF10RangeOfBPS                  = exception.New(errorRuleF10+9, "非净增长范围")
	ErrF10RangeOfMarketCap            = exception.New(errorRuleF10+10, "非市值范围")
)

// RuleF10 基本面规则
func ruleF10(ruleParameter config.RuleParameter, snapshot factors.QuoteSnapshot) error {
	// 基础过滤规则, 检测F10基本面
	securityCode := snapshot.Code
	// 1. 去掉需要忽略的个股
	if market.IsNeedIgnore(securityCode) {
		return ErrF10IgnoreStock
	}
	// 2. 过滤指定的代码前缀
	if api.StartsWith(securityCode, ruleParameter.IgnoreCodes) {
		return ErrF10IgnoreStock
	}
	// 3. 去掉次新股
	if market.IsSubNewStock(securityCode) {
		return ErrF10SubNewStock
	}
	// 4. 股价控制
	if num.IsNaN(snapshot.LastClose) || !ruleParameter.Price.Validate(snapshot.LastClose) {
		return ErrF10PriceRange
	}
	// 5. F10数据
	f10 := factors.GetL5F10(securityCode)
	if f10 != nil {
		// 5.1 流通股本控制
		capital := f10.Capital / config.Billion
		if f10.Capital != 0 && !ruleParameter.Capital.Validate(capital) {
			return ErrF10RangeOfCapital
		}
		// 5.1.1 市值控制
		marketValue := f10.TotalCapital * snapshot.LastClose / config.Billion
		if !ruleParameter.MarketCap.Validate(marketValue) {
			return ErrF10RangeOfMarketCap
		}
		// 5.2 安全分太低
		if f10.SafetyScore != 0 && !ruleParameter.SafetyScore.Validate(float64(f10.SafetyScore)) {
			return ErrF10RangeOfSafetyCode
		}
		// 5.3 季报不理想
		if f10.BasicEPS != 0 && f10.BasicEPS < 0 {
			return ErrF10RangeOfBasicEPS
		}
		// 5.4 净增长小于0
		if f10.BPS != 0 && f10.BPS < 0 {
			return ErrF10RangeOfBPS
		}
		//// 5.5 处理季报有增减持数据, 两个季度前十大流通股总数对比
		//reportDate, _ := api.ParseTime(f10.UpdateDate)
		//after := reportDate.AddDate(0, 2, 0).After(time.Now())
		//// 两月内减持的剔掉, 或者减持统计超过1%
		//if after && (f10.Top10Capital < 0 || f10.ReductionRatio < -1.00) {
		//	return false
		//}
		//// 5.6. 处理上市公司公告
		//if f10.Reduce > 0 || f10.Increase > 0 || f10.Risk > 0 {
		//	return false
		//}
	}
	// 规则通过
	return nil
}