package client

import (
	"fmt"

	"github.com/smartcontractkit/chainlink/v2/common/types"
)

const (
	NodeSelectionModeHighestHead     = "HighestHead"
	NodeSelectionModeRoundRobin      = "RoundRobin"
	NodeSelectionModeTotalDifficulty = "TotalDifficulty"
	NodeSelectionModePriorityLevel   = "PriorityLevel"
)

type NodeSelector[
	CHAIN_ID types.ID,
	HEAD Head,
	RPC NodeClient[CHAIN_ID, HEAD],
] interface {
	// Select returns a Node, or nil if none can be selected.
	// Implementation must be thread-safe.
	Select() Node[CHAIN_ID, HEAD, RPC]
	// Name returns the strategy name, e.g. "HighestHead" or "RoundRobin"
	Name() string
}

func newNodeSelector[
	CHAIN_ID types.ID,
	HEAD Head,
	RPC NodeClient[CHAIN_ID, HEAD],
](selectionMode string, nodes []Node[CHAIN_ID, HEAD, RPC]) NodeSelector[CHAIN_ID, HEAD, RPC] {
	switch selectionMode {
	case NodeSelectionModeHighestHead:
		return NewHighestHeadNodeSelector[CHAIN_ID, HEAD, RPC](nodes)
	case NodeSelectionModeRoundRobin:
		return NewRoundRobinSelector[CHAIN_ID, HEAD, RPC](nodes)
	case NodeSelectionModeTotalDifficulty:
		return NewTotalDifficultyNodeSelector[CHAIN_ID, HEAD, RPC](nodes)
	case NodeSelectionModePriorityLevel:
		return NewPriorityLevelNodeSelector[CHAIN_ID, HEAD, RPC](nodes)
	default:
		panic(fmt.Sprintf("unsupported NodeSelectionMode: %s", selectionMode))
	}
}