abci.go

package poe

import (
	"time"

	"github.com/cosmos/cosmos-sdk/telemetry"
	sdk "github.com/cosmos/cosmos-sdk/types"
	abci "github.com/tendermint/tendermint/abci/types"

	"github.com/mage-coven/gridiron/x/poe/contract"
	"github.com/mage-coven/gridiron/x/poe/keeper"
	"github.com/mage-coven/gridiron/x/poe/types"
	"github.com/mage-coven/gridiron/x/twasm"
	twasmtypes "github.com/mage-coven/gridiron/x/twasm/types"
)

type endBlockKeeper interface {
	types.Sudoer
	IteratePrivilegedContractsByType(ctx sdk.Context, privilegeType twasmtypes.PrivilegeType, cb func(prio uint8, contractAddr sdk.AccAddress) bool)
}

type abciKeeper interface {
	UpdateValidatorVotes(validatorVotes []abci.VoteInfo)
	TrackHistoricalInfo(ctx sdk.Context)
}

// EndBlocker calls the Valset contract for the validator diff.
func EndBlocker(parentCtx sdk.Context, k endBlockKeeper) []abci.ValidatorUpdate {
	defer telemetry.ModuleMeasureSince(types.ModuleName, time.Now(), telemetry.MetricKeyEndBlocker)
	logger := keeper.ModuleLogger(parentCtx)

	var diff []abci.ValidatorUpdate
	// allow validator set updates for this group only
	k.IteratePrivilegedContractsByType(parentCtx, twasmtypes.PrivilegeTypeValidatorSetUpdate, func(pos uint8, contractAddr sdk.AccAddress) bool {
		logger.Info("privileged contract callback", "type", twasmtypes.PrivilegeTypeValidatorSetUpdate.String())
		ctx, commit := parentCtx.CacheContext()
		defer twasm.RecoverToLog(logger, contractAddr)()

		var err error
		diff, err = contract.CallEndBlockWithValidatorUpdate(ctx, contractAddr, k)
		if err != nil {
			logger.Error(
				"contract callback for validator set update failed",
				"cause", err,
				"contract-address", contractAddr,
				"position", pos,
			)
			return true // stop at first contract, without commit
		}
		commit()
		if len(diff) != 0 {
			logger.Info("update validator set", "new", diff)
		}
		return true // stop at first contract
	})
	return diff
}

// BeginBlocker ABCI begin block callback
func BeginBlocker(ctx sdk.Context, k abciKeeper, b abci.RequestBeginBlock) {
	defer telemetry.ModuleMeasureSince(types.ModuleName, time.Now(), telemetry.MetricKeyBeginBlocker)

	k.UpdateValidatorVotes(b.LastCommitInfo.Votes)
	k.TrackHistoricalInfo(ctx)
}