validators

type Connector interface {
	Connected(
		ctx context.Context,
		nodeID ids.NodeID,
		nodeVersion *version.Application,
	) error
	Disconnected(ctx context.Context, nodeID ids.NodeID) error
}

type GetValidatorOutput struct {
	NodeID    ids.NodeID
	PublicKey *bls.PublicKey
	Weight    uint64
}

type Manager interface {
	fmt.Stringer

	// Add a new staker to the supernet.
	// Returns an error if:
	// - [weight] is 0
	// - [nodeID] is already in the validator set
	// If an error is returned, the set will be unmodified.
	AddStaker(supernetID ids.ID, nodeID ids.NodeID, pk *bls.PublicKey, txID ids.ID, weight uint64) error

	// AddWeight to an existing staker to the supernet.
	// Returns an error if:
	// - [weight] is 0
	// - [nodeID] is not already in the validator set
	// If an error is returned, the set will be unmodified.
	// AddWeight can result in a total weight that overflows uint64.
	// In this case no error will be returned for this call.
	// However, the next TotalWeight call will return an error.
	AddWeight(supernetID ids.ID, nodeID ids.NodeID, weight uint64) error

	// GetWeight retrieves the validator weight from the supernet.
	GetWeight(supernetID ids.ID, nodeID ids.NodeID) uint64

	// GetValidator returns the validator tied to the specified ID in supernet.
	// If the validator doesn't exist, returns false.
	GetValidator(supernetID ids.ID, nodeID ids.NodeID) (*Validator, bool)

	// GetValidatoIDs returns the validator IDs in the supernet.
	GetValidatorIDs(supernetID ids.ID) []ids.NodeID

	// SubsetWeight returns the sum of the weights of the validators in the supernet.
	// Returns err if subset weight overflows uint64.
	SubsetWeight(supernetID ids.ID, validatorIDs set.Set[ids.NodeID]) (uint64, error)

	// RemoveWeight from a staker in the supernet. If the staker's weight becomes 0, the staker
	// will be removed from the supernet set.
	// Returns an error if:
	// - [weight] is 0
	// - [nodeID] is not already in the supernet set
	// - the weight of the validator would become negative
	// If an error is returned, the set will be unmodified.
	RemoveWeight(supernetID ids.ID, nodeID ids.NodeID, weight uint64) error

	// Count returns the number of validators currently in the supernet.
	Count(supernetID ids.ID) int

	// TotalWeight returns the cumulative weight of all validators in the supernet.
	// Returns err if total weight overflows uint64.
	TotalWeight(supernetID ids.ID) (uint64, error)

	// Sample returns a collection of validatorIDs in the supernet, potentially with duplicates.
	// If sampling the requested size isn't possible, an error will be returned.
	Sample(supernetID ids.ID, size int) ([]ids.NodeID, error)

	// Map of the validators in this supernet
	GetMap(supernetID ids.ID) map[ids.NodeID]*GetValidatorOutput

	// When a validator is added, removed, or its weight changes, the listener
	// will be notified of the event.
	RegisterCallbackListener(listener ManagerCallbackListener)

	// When a validator is added, removed, or its weight changes on [supernetID],
	// the listener will be notified of the event.
	RegisterSetCallbackListener(supernetID ids.ID, listener SetCallbackListener)
}

type ManagerCallbackListener interface {
	OnValidatorAdded(supernetID ids.ID, nodeID ids.NodeID, pk *bls.PublicKey, txID ids.ID, weight uint64)
	OnValidatorRemoved(supernetID ids.ID, nodeID ids.NodeID, weight uint64)
	OnValidatorWeightChanged(supernetID ids.ID, nodeID ids.NodeID, oldWeight, newWeight uint64)
}

type MockState struct {
	// contains filtered or unexported fields
}

type MockStateMockRecorder struct {
	// contains filtered or unexported fields
}

type MockSupernetConnector struct {
	// contains filtered or unexported fields
}

type MockSupernetConnectorMockRecorder struct {
	// contains filtered or unexported fields
}

type SetCallbackListener interface {
	OnValidatorAdded(nodeID ids.NodeID, pk *bls.PublicKey, txID ids.ID, weight uint64)
	OnValidatorRemoved(nodeID ids.NodeID, weight uint64)
	OnValidatorWeightChanged(nodeID ids.NodeID, oldWeight, newWeight uint64)
}

type State interface {
	// GetMinimumHeight returns the minimum height of the block still in the
	// proposal window.
	GetMinimumHeight(context.Context) (uint64, error)
	// GetCurrentHeight returns the current height of the P-chain.
	GetCurrentHeight(context.Context) (uint64, error)

	// GetSupernetID returns the supernetID of the provided chain.
	GetSupernetID(ctx context.Context, chainID ids.ID) (ids.ID, error)

	// GetValidatorSet returns the validators of the provided supernet at the
	// requested P-chain height.
	// The returned map should not be modified.
	GetValidatorSet(
		ctx context.Context,
		height uint64,
		supernetID ids.ID,
	) (map[ids.NodeID]*GetValidatorOutput, error)
}

type SupernetConnector interface {
	ConnectedSupernet(ctx context.Context, nodeID ids.NodeID, supernetID ids.ID) error
}

type TestState struct {
	T testing.TB

	CantGetMinimumHeight,
	CantGetCurrentHeight,
	CantGetSupernetID,
	CantGetValidatorSet bool

	GetMinimumHeightF func(ctx context.Context) (uint64, error)
	GetCurrentHeightF func(ctx context.Context) (uint64, error)
	GetSupernetIDF    func(ctx context.Context, chainID ids.ID) (ids.ID, error)
	GetValidatorSetF  func(ctx context.Context, height uint64, supernetID ids.ID) (map[ids.NodeID]*GetValidatorOutput, error)
}

type Validator struct {
	NodeID    ids.NodeID
	PublicKey *bls.PublicKey
	TxID      ids.ID
	Weight    uint64
	// contains filtered or unexported fields
}