package factors

import (
	"context"
	"gitee.com/quant1x/engine/cache"
	"gitee.com/quant1x/engine/config"
	"gitee.com/quant1x/engine/datasource/dfcf"
	"gitee.com/quant1x/engine/datasource/tdxweb"
	"gitee.com/quant1x/engine/market"
	"gitee.com/quant1x/exchange"
	"gitee.com/quant1x/gotdx/securities"
	"gitee.com/quant1x/gox/api"
	"gitee.com/quant1x/gox/logger"
	"gitee.com/quant1x/num"
)

const (
	cacheL5KeyF10 = "f10"
)

// F10 证券基本面
type F10 struct {
	cache.DataSummary    `dataframe:"-"`
	Date                 string  `name:"日期" dataframe:"Date"`                      // 日期
	Code                 string  `name:"代码" dataframe:"Code"`                      // 证券代码
	SecurityName         string  `name:"名称" dataframe:"Name"`                      // 证券名称
	SubNew               bool    `name:"次新股" dataframe:"SubNew"`                   // 是否次新股
	MarginTradingTarget  bool    `name:"两融" dataframe:"MarginTradingTarget"`       // 是否两融标的
	VolUnit              int     `name:"每手" dataframe:"VolUnit"`                   // 每手单位
	DecimalPoint         int     `name:"小数点" dataframe:"DecimalPoint"`             // 小数点
	IpoDate              string  `name:"上市日期" dataframe:"IpoDate"`                 // 上市日期
	UpdateDate           string  `name:"更新日期" dataframe:"UpdateDate"`              // 更新日期
	TotalCapital         float64 `name:"总股本" dataframe:"TotalCapital"`             // 总股本
	Capital              float64 `name:"流通股本" dataframe:"Capital"`                 // 流通股本
	FreeCapital          float64 `name:"自由流通股本" dataframe:"FreeCapital"`           // 自由流通股本
	Top10Capital         float64 `name:"前十大流通股东总股本" dataframe:"Top10Capital"`      // 前十大流通股东股本
	Top10Change          float64 `name:"前十大流通股东总股本变化" dataframe:"Top10Change"`     //前十大流通股东股本变化
	ChangeCapital        float64 `name:"前十大流通股东持仓变化" dataframe:"ChangeCapital"`    // 前十大流通股东持仓变化
	IncreaseRatio        float64 `name:"当期增持比例" dataframe:"IncreaseRatio"`         // 当期增持比例
	ReductionRatio       float64 `name:"当期减持比例" dataframe:"ReductionRatio"`        // 当期减持比例
	QuarterlyYearQuarter string  `name:"季报期" dataframe:"quarterly_year_quarter"`   // 当前市场处于哪个季报期, 用于比较个股的季报数据是否存在拖延的情况
	QDate                string  `name:"新报告期" dataframe:"qdate"`                   // 最新报告期
	AnnualReportDate     string  `name:"年报披露日期" dataframe:"annual_report_date"`    // 年报披露日期
	QuarterlyReportDate  string  `name:"季报披露日期" dataframe:"quarterly_report_date"` // 最新季报披露日期
	TotalOperateIncome   float64 `name:"营业总收入" dataframe:"TotalOperateIncome"`     // 当期营业总收入
	BPS                  float64 `name:"每股净资产" dataframe:"BPS"`                    // 每股净资产
	BasicEPS             float64 `name:"每股收益" dataframe:"BasicEPS"`                // 每股收益
	DeductBasicEPS       float64 `name:"每股收益(扣除)" dataframe:"DeductBasicEPS"`      // 每股收益(扣除)
	SafetyScore          int     `name:"安全分" dataframe:"SafetyScore"`              // 通达信安全分
	Increases            int     `name:"增持" dataframe:"Increases"`                 // 公告-增持
	Reduces              int     `name:"减持" dataframe:"Reduces"`                   // 公告-减持
	Risk                 int     `name:"风险数" dataframe:"Risk"`                     // 公告-风险数
	RiskKeywords         string  `name:"风险关键词" dataframe:"RiskKeywords"`           // 公告-风险关键词
	UpdateTime           string  `name:"更新时间" dataframe:"update_time"`             // 更新时间
	State                uint64  `name:"样本状态" dataframe:"样本状态"`
}

func NewF10(date, code string) *F10 {
	summary := __mapFeatures[FeatureF10]
	v := F10{
		DataSummary:  summary,
		Date:         date,
		Code:         code,
		SecurityName: securities.GetStockName(code),
		VolUnit:      100,
		DecimalPoint: 2,
		SubNew:       market.IsSubNewStock(code),
	}
	securityInfo, ok := securities.CheckoutSecurityInfo(code)
	if ok {
		v.VolUnit = int(securityInfo.VolUnit)
		v.DecimalPoint = int(securityInfo.DecimalPoint)
		v.SecurityName = securityInfo.Name
	}
	// 检测是否两融标的
	v.MarginTradingTarget = securities.IsMarginTradingTarget(code)
	return &v
}

func (this *F10) GetDate() string {
	return this.Date
}

func (this *F10) GetSecurityCode() string {
	return this.Code
}

func (this *F10) Factory(date string, code string) Feature {
	v := NewF10(date, code)
	return v
}

func (this *F10) Init(ctx context.Context, date string) error {
	loadQuarterlyReports(this.GetDate())
	_ = ctx
	_ = date
	return nil
}

func (this *F10) FromHistory(history History) Feature {
	_ = history
	return this
}

func (this *F10) Update(code, cacheDate, featureDate string, complete bool) {
	securityCode := this.GetSecurityCode()
	logger.Warnf("update f10, code=%s", securityCode)

	// 1. 基本信息
	securityInfo := checkoutSecurityBasicInfo(securityCode, featureDate)
	_ = api.Copy(this, &securityInfo)
	// 2. 前十大流通股股东
	shareHolder := checkoutShareHolder(securityCode, featureDate)
	_ = api.Copy(this, shareHolder)
	if this.FreeCapital == 0 {
		this.FreeCapital = this.Capital
	}
	// 3. 上市公司公告
	notice := getOneNotice(securityCode, featureDate)
	_ = api.Copy(this, &notice)
	// 4. 季报
	this.QuarterlyYearQuarter = getQuarterlyYearQuarter(featureDate)
	report := getQuarterlyReportSummary(securityCode, featureDate)
	_ = api.Copy(this, &report)

	// 5. 安全分
	safetyScore := tdxweb.GetSafetyScore(securityCode)
	this.SafetyScore = safetyScore

	// 6. 年报季报披露日期
	annualReportDate, quarterlyReportDate := dfcf.NoticeDateForReport(securityCode, cacheDate)
	this.AnnualReportDate = annualReportDate
	this.QuarterlyReportDate = quarterlyReportDate

	this.UpdateTime = GetTimestamp()
	this.State |= this.Kind()
	logger.Warnf("update f10, code=%s, OK", securityCode)
	_ = complete
}

func (this *F10) Repair(code, cacheDate, featureDate string, complete bool) {
	securityCode := code

	// 1. 基本信息
	securityInfo := checkoutSecurityBasicInfo(securityCode, featureDate)
	_ = api.Copy(this, &securityInfo)
	// 2. 前十大流通股股东
	shareHolder := checkoutShareHolder(securityCode, featureDate)
	_ = api.Copy(this, shareHolder)
	if this.FreeCapital == 0 {
		this.FreeCapital = this.Capital
	}
	// 3. 上市公司公告
	notice := getOneNotice(securityCode, featureDate)
	_ = api.Copy(this, &notice)
	// 4. 季报
	report := getQuarterlyReportSummary(securityCode, featureDate)
	_ = api.Copy(this, &report)

	// 5. 安全分
	if this.SafetyScore == 0 {
		safetyScore := tdxweb.GetSafetyScore(securityCode)
		this.SafetyScore = safetyScore
	}

	// 6. 年报季报披露日期
	annualReportDate, quarterlyReportDate := dfcf.NoticeDateForReport(securityCode, cacheDate)
	this.AnnualReportDate = annualReportDate
	this.QuarterlyReportDate = quarterlyReportDate

	this.UpdateTime = GetTimestamp()
	this.State |= this.Kind()

	_ = complete
}

func (this *F10) Increase(snapshot QuoteSnapshot) Feature {
	//TODO implement me
	panic("implement me")
}

func (this *F10) ValidateSample() error {
	if this.State > 0 {
		return nil
	}
	return ErrInvalidFeatureSample
}

func (this *F10) TurnZ(v any) float64 {
	freeCapital := this.FreeCapital
	if freeCapital == 0 {
		freeCapital = this.Capital
	}
	if freeCapital == 0 {
		return 0.00
	}
	n := num.AnyToFloat64(v)
	turnoverRateZ := num.ChangeRate(freeCapital, n)
	turnoverRateZ *= 10000
	turnoverRateZ = num.Decimal(turnoverRateZ)
	return turnoverRateZ
}

// IsReportingRiskPeriod 是否财报披露前夕
func (this *F10) IsReportingRiskPeriod() bool {
	if len(this.AnnualReportDate) == 0 || len(this.QuarterlyReportDate) == 0 {
		// 如果年报和季报日期不确定, 判定为非风险期, 返回false
		// 这种情况有可能是次新股的保护期
		return false
	}
	f10Config := config.GetDataConfig().Feature.F10
	date := this.GetDate()
	ys := exchange.DateRange(date, this.AnnualReportDate)
	ly := len(ys)
	qs := exchange.DateRange(date, this.QuarterlyReportDate)
	lq := len(qs)
	if (ly > 0 && ly < f10Config.ReportingRiskPeriod) || (lq > 0 && lq < f10Config.ReportingRiskPeriod) {
		return true
	}
	return false
}