package extend

import (
	"fmt"
	"github.com/injoyai/tdx/protocol"
	"time"
)

func DoIncomes(ks Klines, startAt time.Time, days ...int) Incomes {
	year, month, day := startAt.Date()
	start := time.Date(year, month, day, 15, 0, 0, 0, startAt.Location()).Unix()
	for i, v := range ks {
		if v.Date >= start {
			ks = ks[i:]
			break
		}
	}

	ls := Incomes{}

	for _, v := range days {
		if v < len(ks) {
			x := ks[v]
			ls = append(ls, &Income{
				Offset: v,
				Time:   time.Unix(x.Date, 0),
				Source: protocol.K{
					Open:  ks[0].Open,
					High:  ks[0].High,
					Low:   ks[0].Low,
					Close: ks[0].Close,
				},
				Current: protocol.K{
					Open:  x.Open,
					High:  x.High,
					Low:   x.Low,
					Close: x.Close,
				},
			})
		}
	}

	return ls
}

type Incomes []*Income

type Income struct {
	Offset  int        //偏移量
	Time    time.Time  //时间
	Source  protocol.K //源
	Current protocol.K //当前
}

func (this *Income) String() string {
	return fmt.Sprintf("偏移: %d, 时间: %s, 涨幅: %.1f%%", this.Offset, this.Time.Format(time.DateOnly), this.RiseRate()*100)
}

func (this *Income) Rise() protocol.Price {
	return this.Current.Close - this.Source.Close
}

func (this *Income) RiseRate() float64 {
	return this.Rise().Float64() / this.Source.Close.Float64()
}