package glassnode

import (
	"context"
	"time"

	"git.qtrade.icu/lychiyu/bbgo/pkg/datasource/glassnode/glassnodeapi"
)

type DataSource struct {
	client *glassnodeapi.RestClient
}

func New(apiKey string) *DataSource {
	client := glassnodeapi.NewRestClient()
	client.Auth(apiKey)

	return &DataSource{client: client}
}

func (d *DataSource) Query(ctx context.Context, category, metric, asset string, options QueryOptions) (glassnodeapi.DataSlice, error) {
	req := glassnodeapi.Request{
		Client:   d.client,
		Asset:    asset,
		Since:    options.Since,
		Until:    options.Until,
		Interval: options.Interval,

		Category: category,
		Metric:   metric,
	}

	resp, err := req.Do(ctx)
	if err != nil {
		return nil, err
	}

	return glassnodeapi.DataSlice(resp), nil
}

// query last futures open interest
// https://docs.glassnode.com/api/derivatives#futures-open-interest
func (d *DataSource) QueryFuturesOpenInterest(ctx context.Context, currency string) (float64, error) {
	until := time.Now()
	since := until.Add(-24 * time.Hour)

	options := QueryOptions{
		Since: &since,
		Until: &until,
	}
	resp, err := d.Query(ctx, "derivatives", "futures_open_interest_sum", currency, options)

	if err != nil {
		return 0, err
	}

	return resp.Last().Value, nil
}

// query last market cap in usd
// https://docs.glassnode.com/api/market#market-cap
func (d *DataSource) QueryMarketCapInUSD(ctx context.Context, currency string) (float64, error) {
	until := time.Now()
	since := until.Add(-24 * time.Hour)

	options := QueryOptions{
		Since: &since,
		Until: &until,
	}

	resp, err := d.Query(ctx, "market", "marketcap_usd", currency, options)

	if err != nil {
		return 0, err
	}

	return resp.Last().Value, nil
}