feat(halo/evmstaking2): implement Prepare

halo/evmstaking2/keeper/keeper.go

@@ -3,10 +3,16 @@ package keeper
 import (
 	"context"
 
+	"github.com/omni-network/omni/contracts/bindings"
 	"github.com/omni-network/omni/halo/evmstaking2/types"
+	"github.com/omni-network/omni/halo/genutil/evm/predeploys"
 	"github.com/omni-network/omni/lib/errors"
+	"github.com/omni-network/omni/lib/ethclient"
 	evmenginetypes "github.com/omni-network/omni/octane/evmengine/types"
 
+	"github.com/ethereum/go-ethereum"
+	"github.com/ethereum/go-ethereum/accounts/abi"
+	"github.com/ethereum/go-ethereum/accounts/abi/bind"
 	"github.com/ethereum/go-ethereum/common"
 
 	ormv1alpha1 "cosmossdk.io/api/cosmos/orm/v1alpha1"
@@ -15,14 +21,23 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 )
 
+var (
+	stakingABI           = mustGetABI(bindings.StakingMetaData)
+	createValidatorEvent = mustGetEvent(stakingABI, "CreateValidator")
+	delegateEvent        = mustGetEvent(stakingABI, "Delegate")
+)
+
 // Keeper also implements the evmenginetypes.EvmEventProcessor interface.
 type Keeper struct {
 	eventsTable     EVMEventTable
+	ethCl           ethclient.Client
+	address         common.Address
 	submissionDelay int64
 }
 
 func NewKeeper(
 	storeService store.KVStoreService,
+	ethCl ethclient.Client,
 	submissionDelay int64,
 ) (*Keeper, error) {
 	schema := &ormv1alpha1.ModuleSchemaDescriptor{SchemaFile: []*ormv1alpha1.ModuleSchemaDescriptor_FileEntry{
@@ -39,8 +54,12 @@ func NewKeeper(
 		return nil, errors.Wrap(err, "create valsync store")
 	}
 
+	address := common.HexToAddress(predeploys.Staking)
+
 	return &Keeper{
 		eventsTable:     evmstakingStore.EVMEventTable(),
+		ethCl:           ethCl,
+		address:         address,
 		submissionDelay: submissionDelay,
 	}, nil
 }
@@ -75,16 +94,38 @@ func (k *Keeper) EndBlock(ctx context.Context) error {
 }
 
 // Prepare returns all omni stake contract EVM event logs from the provided block hash.
-func (Keeper) Prepare(context.Context, common.Hash) ([]evmenginetypes.EVMEvent, error) {
-	return nil, nil
+func (k Keeper) Prepare(ctx context.Context, blockHash common.Hash) ([]evmenginetypes.EVMEvent, error) {
+	logs, err := k.ethCl.FilterLogs(ctx, ethereum.FilterQuery{
+		BlockHash: &blockHash,
+		Addresses: k.Addresses(),
+		Topics:    [][]common.Hash{{createValidatorEvent.ID, delegateEvent.ID}},
+	})
+	if err != nil {
+		return nil, errors.Wrap(err, "filter logs")
+	}
+
+	resp := make([]evmenginetypes.EVMEvent, 0, len(logs))
+	for _, l := range logs {
+		topics := make([][]byte, 0, len(l.Topics))
+		for _, t := range l.Topics {
+			topics = append(topics, t.Bytes())
+		}
+		resp = append(resp, evmenginetypes.EVMEvent{
+			Address: l.Address.Bytes(),
+			Topics:  topics,
+			Data:    l.Data,
+		})
+	}
+
+	return resp, nil
 }
 
 func (Keeper) Name() string {
 	return types.ModuleName
 }
 
-func (Keeper) Addresses() []common.Address {
-	return nil
+func (k Keeper) Addresses() []common.Address {
+	return []common.Address{k.address}
 }
 
 // Deliver processes a omni deposit log event, which must be one of:
@@ -107,3 +148,25 @@ func (k Keeper) Deliver(ctx context.Context, _ common.Hash, elog evmenginetypes.
 // parseAndDeliver parses the provided event and tries to deliver it on a state branch.
 // If the delivery fails, the error will be logged and the state branch will be discarded.
 func parseAndDeliver(context.Context, *evmenginetypes.EVMEvent) {}
+
+// mustGetABI returns the metadata's ABI as an abi.ABI type.
+// It panics on error.
+func mustGetABI(metadata *bind.MetaData) *abi.ABI {
+	abi, err := metadata.GetAbi()
+	if err != nil {
+		panic(err)
+	}
+
+	return abi
+}
+
+// mustGetEvent returns the event with the given name from the ABI.
+// It panics if the event is not found.
+func mustGetEvent(abi *abi.ABI, name string) abi.Event {
+	event, ok := abi.Events[name]
+	if !ok {
+		panic("event not found")
+	}
+
+	return event
+}

halo/evmstaking2/keeper/keeper_internal_test.go

@@ -93,7 +93,7 @@ func setupKeeper(t *testing.T, submissionDelay int64) (*Keeper, sdk.Context) {
 	ctx = ctx.WithBlockHeight(1)
 	ctx = ctx.WithChainID(netconf.Simnet.Static().OmniConsensusChainIDStr())
 
-	k, err := NewKeeper(storeSvc, submissionDelay)
+	k, err := NewKeeper(storeSvc, nil, submissionDelay)
 	require.NoError(t, err, "new keeper")
 
 	return k, ctx

halo/evmstaking2/module/module.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/omni-network/omni/halo/evmstaking2/keeper"
 	"github.com/omni-network/omni/halo/evmstaking2/types"
+	"github.com/omni-network/omni/lib/ethclient"
 
 	"cosmossdk.io/core/appmodule"
 	"cosmossdk.io/core/store"
@@ -111,6 +112,7 @@ type ModuleInputs struct {
 	depinject.In
 
 	StoreService store.KVStoreService
+	EthCl        ethclient.Client
 	Cdc          codec.Codec
 	Config       *Module
 }
@@ -125,6 +127,7 @@ type ModuleOutputs struct {
 func ProvideModule(in ModuleInputs) (ModuleOutputs, error) {
 	k, err := keeper.NewKeeper(
 		in.StoreService,
+		in.EthCl,
 		in.Config.GetDeliverInterval(),
 	)
 	if err != nil {