warp/validators/state.go

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

package validators

import (
	"context"

	"github.com/luxfi/evm/iface"
	"github.com/luxfi/geth/common"
)

// State wraps the validator state to provide special handling for the Primary Network
type State struct {
	validatorState iface.ValidatorState
	chainID        common.Hash
	chainID        common.Hash
	skipChainID    bool
}

// NewState creates a new State wrapper
func NewState(
	validatorState iface.ValidatorState,
	chainID common.Hash,
	chainID common.Hash,
	skipChainID bool,
) *State {
	return &State{
		validatorState: validatorState,
		chainID:        chainID,
		chainID:        chainID,
		skipChainID:    skipChainID,
	}
}

// GetValidatorSet implements the ValidatorState interface
func (s *State) GetValidatorSet(ctx context.Context, height uint64, chainID common.Hash) (map[common.Hash]*iface.ValidatorOutput, error) {
	return s.validatorState.GetValidatorSet(ctx, height, chainID)
}

// GetCurrentHeight implements the ValidatorState interface
func (s *State) GetCurrentHeight(ctx context.Context) (uint64, error) {
	return s.validatorState.GetCurrentHeight(ctx)
}

// GetMinimumHeight implements the ValidatorState interface
func (s *State) GetMinimumHeight(ctx context.Context) (uint64, error) {
	return s.validatorState.GetMinimumHeight(ctx)
}

// GetChainID implements the ValidatorState interface
func (s *State) GetChainID(ctx context.Context, chainID common.Hash) (common.Hash, error) {
	if s.skipChainID && chainID == s.chainID {
		return s.chainID, nil
	}
	return s.validatorState.GetChainID(ctx, chainID)
}