node/network/validators_wrapper.go

// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package network

import (
	validators "github.com/luxfi/validators"
	"github.com/luxfi/crypto/bls"
	"github.com/luxfi/ids"
)

// validatorsWrapper wraps validators.Manager to match the expected interface
type validatorsWrapper struct {
	manager   validators.Manager
	callbacks []validators.SetCallbackListener
}

// NewValidatorsWrapper creates a new validators wrapper
func NewValidatorsWrapper(manager validators.Manager) *validatorsWrapper {
	return &validatorsWrapper{
		manager:   manager,
		callbacks: make([]validators.SetCallbackListener, 0),
	}
}

// GetWeight returns the weight of a validator
func (v *validatorsWrapper) GetWeight(netID ids.ID, nodeID ids.NodeID) uint64 {
	return v.manager.GetWeight(netID, nodeID)
}

// GetValidator returns validator info
func (v *validatorsWrapper) GetValidator(netID ids.ID, nodeID ids.NodeID) (validators.Validator, bool) {
	output, exists := v.manager.GetValidator(netID, nodeID)
	if !exists {
		return nil, false
	}
	// Convert GetValidatorOutput to Validator interface
	return &validatorAdapter{output: output}, true
}

// GetValidatorIDs returns all validator IDs for a net
func (v *validatorsWrapper) GetValidatorIDs(netID ids.ID) []ids.NodeID {
	validatorSet, err := v.manager.GetValidators(netID)
	if err != nil {
		return nil
	}

	validators := validatorSet.List()
	nodeIDs := make([]ids.NodeID, len(validators))
	for i, validator := range validators {
		nodeIDs[i] = validator.ID()
	}

	return nodeIDs
}

// TotalWeight returns the total weight of all validators
func (v *validatorsWrapper) TotalWeight(netID ids.ID) (uint64, error) {
	return v.manager.TotalWeight(netID)
}

// RegisterSetCallbackListener registers a callback listener
func (v *validatorsWrapper) RegisterSetCallbackListener(listener validators.SetCallbackListener) {
	v.callbacks = append(v.callbacks, listener)
}

// SetCallbackListener for validator changes
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)
}

// validatorAdapter adapts GetValidatorOutput to implement Validator interface
type validatorAdapter struct {
	output *validators.GetValidatorOutput
}

func (v *validatorAdapter) ID() ids.NodeID {
	return v.output.NodeID
}

func (v *validatorAdapter) Light() uint64 {
	return v.output.Light
}