package market

import (
	"github.com/adshao/go-binance/v2/futures"
	"github.com/rs/zerolog/log"
	"strconv"
)

func (ms MarketData) GetFuturePrice(symbol string) (float64, float64, int64, bool) {
	p, ok := ms.futureMarkPrice[symbol]
	if !ok {
		return 0, 0, 0, false
	}
	return p, ms.futureFundingRate[symbol], ms.futureNextFundingTime[symbol], true
}

func (ms MarketData) StartFutureWsMarkPrice() error {
	_, _, err := futures.WsAllMarkPriceServe(ms.futureWsMarkPriceHandler, ms.futureWsErrHandler)
	if err != nil {
		return err
	}
	return nil
}

func (ms MarketData) futureWsMarkPriceHandler(event futures.WsAllMarkPriceEvent) {
	for _, priceEvent := range event {
		price, err := strconv.ParseFloat(priceEvent.MarkPrice, 64)
		if err != nil {
			continue
		}

		fundingRate, err := strconv.ParseFloat(priceEvent.FundingRate, 64)
		if err != nil {
			continue
		}

		ms.futureMarkPrice[priceEvent.Symbol] = price
		ms.futureFundingRate[priceEvent.Symbol] = fundingRate
		ms.futureNextFundingTime[priceEvent.Symbol] = priceEvent.NextFundingTime
	}
}

func (ms MarketData) futureWsErrHandler(err error) {
	log.Debug().Err(err).Msg("Ws Error. Restart socket")
	_ = ms.StartFutureWsMarkPrice()
}

func (ms MarketData) GetAllFundRate() (map[string]float64, map[string]int64) {
	return ms.futureFundingRate, ms.futureNextFundingTime
}