package account

import (
	"encoding/json"
	"log"
	"time"

	"git.apinb.com/quant/strategy/internal/core/trade"
	"git.apinb.com/quant/strategy/internal/models"
)

var (
	BaseAssets  string
	PlanKeyName string
	IsDebug     bool = false
	UpdHistory  bool = false
)

// 根据基本币，监控帐号可用资金变动，仓位，以及最近7天的交易情况
func NewAccountByBinance(base string, p *trade.Spec, debug, updHistory bool) {
	BaseAssets = base
	PlanKeyName = p.PlanKeyName
	IsDebug = debug
	UpdHistory = updHistory

	trade.NewAccounts()

	if p.Api.Exchange == "BINANCE" {
		//trade.InitExchange_Binance(p.BinanceFuturesClient, p.AllowSymbols, p.StrategyConf.Leverage)
		InitAccount_Binance(p)
	}

	go WatchPositions(p)
	if UpdHistory {
		// go WatchHistory(p)
	}
}

func InitAccount_Binance(p *trade.Spec) {
	var err error
	ac, err := p.BinanceClient.GetFuturesAccountBalance()
	if err != nil {
		log.Println(err)
		return
	} else {
		trade.AccountsAssets.Set(BaseAssets, ac[BaseAssets])
	}

	log.Println("WatchAccount", PlanKeyName, BaseAssets, "Account Equity:", ac[BaseAssets].AccountEquity, "Account  Available:", ac[BaseAssets].Available)

	orders, err := trade.RefreshPositions(p)
	if err != nil {
		log.Println(err)
		return
	}
	log.Println("Positions Number", len(orders))
}

func WatchAccount_Binance(p *trade.Spec) {
	for {
		var err error
		ac, err := p.BinanceClient.GetFuturesAccountBalance()
		if err != nil {
			log.Println(err)
		} else {
			trade.AccountsAssets.Set(BaseAssets, ac[BaseAssets])
			log.Println("WatchAccount", PlanKeyName, BaseAssets, "Account Equity:", ac[BaseAssets].AccountEquity, "Account  Available:", ac[BaseAssets].Available)
		}

		time.Sleep(1 * time.Hour)
	}
}

func WatchPositions(p *trade.Spec) {
	for {
		orders, err := trade.RefreshPositions(p)
		if err != nil {
			log.Println(err)
		}

		if IsDebug {
			jsonBytes, _ := json.Marshal(orders)
			log.Println("WatchPositions", string(jsonBytes))
		}

		time.Sleep(2 * time.Second)
	}
}

func WatchHistory(p *trade.Spec) {
	for {
		trade, pl := trade.RefreshHistoryTotal(p)
		models.UpdateTotalTrans(PlanKeyName, trade, pl)
		if IsDebug {
			log.Println("WatchHistory", "Trade Number", trade, "Trade RealizedPnl", pl)
		}
		time.Sleep(10 * time.Minute)
	}
}