package heimdall

import (
	"context"
	"time"
)

type (
	requestTypeKey struct{}
	requestType    string

	// TODO: Uncomment once metrics are added
	// meter struct {
	// 	request map[bool]metrics.Meter // map[isSuccessful]metrics.Meter
	// 	timer   metrics.Timer
	// }
)

const (
	stateSyncRequest       requestType = "state-sync"
	spanRequest            requestType = "span"
	checkpointRequest      requestType = "checkpoint"
	checkpointCountRequest requestType = "checkpoint-count"
)

func withRequestType(ctx context.Context, reqType requestType) context.Context {
	return context.WithValue(ctx, requestTypeKey{}, reqType)
}

func getRequestType(ctx context.Context) (requestType, bool) {
	reqType, ok := ctx.Value(requestTypeKey{}).(requestType)
	return reqType, ok
}

// TODO: Uncomment once metrics are added
// var (
// 	requestMeters = map[requestType]meter{
// 		stateSyncRequest: {
// 			request: map[bool]metrics.Meter{
// 				true:  metrics.NewRegisteredMeter("client/requests/statesync/valid", nil),
// 				false: metrics.NewRegisteredMeter("client/requests/statesync/invalid", nil),
// 			},
// 			timer: metrics.NewRegisteredTimer("client/requests/statesync/duration", nil),
// 		},
// 		spanRequest: {
// 			request: map[bool]metrics.Meter{
// 				true:  metrics.NewRegisteredMeter("client/requests/span/valid", nil),
// 				false: metrics.NewRegisteredMeter("client/requests/span/invalid", nil),
// 			},
// 			timer: metrics.NewRegisteredTimer("client/requests/span/duration", nil),
// 		},
// 		checkpointRequest: {
// 			request: map[bool]metrics.Meter{
// 				true:  metrics.NewRegisteredMeter("client/requests/checkpoint/valid", nil),
// 				false: metrics.NewRegisteredMeter("client/requests/checkpoint/invalid", nil),
// 			},
// 			timer: metrics.NewRegisteredTimer("client/requests/checkpoint/duration", nil),
// 		},
// 		checkpointCountRequest: {
// 			request: map[bool]metrics.Meter{
// 				true:  metrics.NewRegisteredMeter("client/requests/checkpointcount/valid", nil),
// 				false: metrics.NewRegisteredMeter("client/requests/checkpointcount/invalid", nil),
// 			},
// 			timer: metrics.NewRegisteredTimer("client/requests/checkpointcount/duration", nil),
// 		},
// 	}
// )

// TODO: Uncomment once metrics is added
func sendMetrics(ctx context.Context, start time.Time, isSuccessful bool) {
	// reqType, ok := getRequestType(ctx)
	// if !ok {
	// 	return
	// }

	// meters, ok := requestMeters[reqType]
	// if !ok {
	// 	return
	// }

	// meters.request[isSuccessful].Mark(1)
	// meters.timer.Update(time.Since(start))
}