feat: new vote extension implementation

abci/extend_vote.go

@@ -0,0 +1,24 @@
+package vote_ext
+
+import (
+	abci "github.com/cometbft/cometbft/abci/types"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+)
+
+func (h *AbciHandler) ExtendVoteHandler() sdk.ExtendVoteHandler {
+	return func(ctx sdk.Context, req *abci.RequestExtendVote) (*abci.ResponseExtendVote, error) {
+
+		if req.GetHeight() < 3 {
+			return &abci.ResponseExtendVote{VoteExtension: []byte{}}, nil
+		}
+
+		body, err := h.constructVoteExtBody(ctx, req.GetHeight())
+		if err != nil {
+			return nil, err
+		}
+
+		bz := h.secondaryKey.SignVoteExtBody(body)
+
+		return &abci.ResponseExtendVote{VoteExtension: bz}, nil
+	}
+}

abci/handler.go

@@ -0,0 +1,30 @@
+package vote_ext
+
+import (
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
+	"github.com/node101-io/mina-signer-go/keys"
+	keyregistrykeeper "github.com/node101-io/pulsar-chain/x/keyregistry/keeper"
+	votepersistence "github.com/node101-io/pulsar-chain/x/votepersistence/keeper"
+)
+
+type SecondaryKey struct {
+	SecretKey *keys.PrivateKey
+	PublicKey *keys.PublicKey
+}
+
+type AbciHandler struct {
+	secondaryKey          SecondaryKey
+	stakingKeeper         stakingkeeper.Keeper
+	keyregistryKeeper     keyregistrykeeper.Keeper
+	votePersistenceKeeper votepersistence.Keeper
+}
+
+func NewABCIHandler(secondaryKey SecondaryKey, stakingKeeper stakingkeeper.Keeper,
+	keyregistryKeeper keyregistrykeeper.Keeper, votepersistenceKeeper votepersistence.Keeper) *AbciHandler {
+	return &AbciHandler{
+		secondaryKey:          secondaryKey,
+		stakingKeeper:         stakingKeeper,
+		keyregistryKeeper:     keyregistryKeeper,
+		votePersistenceKeeper: votepersistenceKeeper,
+	}
+}