-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #57 from initia-labs/feat/evm-tx
add evm-tx submodules and minor improvement for plain tx
- Loading branch information
Showing
13 changed files
with
4,298 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,200 @@ | ||
| package tx | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "time" | ||
|
|
||
| "cosmossdk.io/collections" | ||
| cosmoserr "cosmossdk.io/errors" | ||
| "github.com/cosmos/cosmos-sdk/codec" | ||
| codectypes "github.com/cosmos/cosmos-sdk/codec/types" | ||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| "github.com/cosmos/cosmos-sdk/types/tx" | ||
|
|
||
| abci "github.com/cometbft/cometbft/abci/types" | ||
| "github.com/cometbft/cometbft/crypto/tmhash" | ||
| coretypes "github.com/cometbft/cometbft/rpc/core/types" | ||
|
|
||
| evmtypes "github.com/initia-labs/minievm/x/evm/types" | ||
|
|
||
| "github.com/initia-labs/kvindexer/submodules/tx/types" | ||
| ) | ||
|
|
||
| func (sm EvmTxSubmodule) finalizeBlock(ctx context.Context, req abci.RequestFinalizeBlock, res abci.ResponseFinalizeBlock) error { | ||
| sm.Logger(ctx).Debug("finalizeBlock", "submodule", types.SubmoduleName, "txs", len(req.Txs), "height", req.Height) | ||
|
|
||
| if err := sm.processTxs(ctx, req, res); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (sm EvmTxSubmodule) processTxs(ctx context.Context, req abci.RequestFinalizeBlock, res abci.ResponseFinalizeBlock) error { | ||
| // key: address, value: txs slice | ||
| accTxMap := map[string][]string{} | ||
|
|
||
| txHashes := []string{} | ||
| for idx, txBytes := range req.Txs { | ||
| tx, err := parseTx(sm.cdc, txBytes) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to parse tx", "error", err, "index", idx) | ||
| continue | ||
| } | ||
|
|
||
| any, err := codectypes.NewAnyWithValue(tx) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to unpack any", "error", err, "index", idx) | ||
| continue | ||
| } | ||
|
|
||
| txHash := tmhash.Sum(txBytes) | ||
| txHashStr := fmt.Sprintf("%x", txHash) | ||
| resultTx := coretypes.ResultTx{ | ||
| Hash: txHash, | ||
| Height: req.Height, | ||
| TxResult: *res.TxResults[idx], | ||
| // No Index, Tx and Proof here. sdk.NewResponseTxResult() don't use them | ||
| } | ||
|
|
||
| txr := sdk.NewResponseResultTx(&resultTx, any, req.Time.UTC().Format(time.RFC3339)) | ||
|
|
||
| if err := sm.txMap.Set(ctx, txHashStr, *txr); err != nil { | ||
| sm.Logger(ctx).Info("failed to store tx", "error", err, "index", idx) | ||
| continue | ||
| } | ||
|
|
||
| // get addresses from the tx | ||
| addrs, err := grepAddressesFromTx(txr) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to grep addresses from tx", "error", err, "index", idx) | ||
| continue | ||
| } | ||
|
|
||
| for _, addr := range addrs { | ||
| accTxMap[addr] = uniqueAppend(accTxMap[addr], txHashStr) | ||
| } | ||
| txHashes = append(txHashes, txHashStr) | ||
| } | ||
|
|
||
| // store tx/account pair into txAccMap | ||
| for addr, txHashes := range accTxMap { | ||
| err := sm.storeAccTxs(ctx, addr, txHashes) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to store tx/account pair", "error", err, "address", addr) | ||
| } | ||
| } | ||
| return sm.storeIndices(ctx, req.Height, txHashes) | ||
| } | ||
|
|
||
| func uniqueAppend(slice []string, elem string) []string { | ||
| for _, e := range slice { | ||
| if e == elem { | ||
| return slice | ||
| } | ||
| } | ||
| return append(slice, elem) | ||
| } | ||
|
|
||
| func parseTx(cdc codec.Codec, txBytes []byte) (*tx.Tx, error) { | ||
| tx := tx.Tx{} | ||
| err := cdc.Unmarshal(txBytes, &tx) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return &tx, nil | ||
| } | ||
|
|
||
| func grepAddressesFromTx(txr *sdk.TxResponse) ([]string, error) { | ||
| grepped := []string{} | ||
| for _, event := range txr.Events { | ||
| for _, attr := range event.Attributes { | ||
| var addrs []string | ||
|
|
||
| if event.Type == evmtypes.EventTypeEVM && attr.Key == evmtypes.AttributeKeyLog { | ||
| contractAddr, err := extractAddressesFromEVMLog(attr.Value) | ||
| if err != nil { | ||
| continue | ||
| } | ||
| addrs = append(addrs, contractAddr) | ||
| } else { | ||
| addrs = findAllBech32Address(attr.Value) | ||
| addrs = append(addrs, findAllHexAddress(attr.Value)...) | ||
| } | ||
|
|
||
| for _, addr := range addrs { | ||
| accAddr, err := accAddressFromString(addr) | ||
| if err != nil { | ||
| continue | ||
| } | ||
| grepped = append(grepped, accAddr.String()) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return grepped, nil | ||
| } | ||
|
|
||
| func extractAddressesFromEVMLog(attrVal string) (string, error) { | ||
| log := evmtypes.Log{} | ||
| if err := json.Unmarshal([]byte(attrVal), &log); err != nil { | ||
| return "", err | ||
| } | ||
| return convertContractAddressToBech32(log.Address) | ||
| } | ||
|
|
||
| func (sm EvmTxSubmodule) storeAccTxs(ctx context.Context, addr string, txHashes []string) error { | ||
| if len(txHashes) == 0 { | ||
| return nil | ||
| } | ||
| acc, err := sdk.AccAddressFromBech32(addr) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to convert address", "error", err, "address", addr) | ||
| return err | ||
| } | ||
|
|
||
| seq, err := sm.accountSequenceMap.Get(ctx, acc) | ||
| if err != nil && !cosmoserr.IsOf(err, collections.ErrNotFound) { | ||
| return err | ||
| } | ||
|
|
||
| for i, txHash := range txHashes { | ||
| err = sm.txhashesByAccountMap.Set(ctx, collections.Join(acc, seq+uint64(i)), txHash) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to store tx/account pair", "error", err, "address", addr, "txhash", txHash) | ||
| continue | ||
| } | ||
|
|
||
| } | ||
| delta := seq + uint64(len(txHashes)+1) | ||
| if err = sm.accountSequenceMap.Set(ctx, acc, delta); err != nil { | ||
| sm.Logger(ctx).Info("failed to store account sequence", "error", err, "address", addr, "delta", delta) | ||
| return err | ||
| } | ||
|
|
||
| return err | ||
| } | ||
|
|
||
| func (sm EvmTxSubmodule) storeIndices(ctx context.Context, height int64, txHashes []string) error { | ||
| for i, txHash := range txHashes { | ||
| err := sm.txhashesByHeightMap.Set(ctx, collections.Join(height, uint64(i)), txHash) | ||
| if err != nil { | ||
| sm.Logger(ctx).Info("failed to store tx/height pair", "error", err, "height", height, "txhash", txHash) | ||
| continue | ||
| } | ||
|
|
||
| seq, err := sm.sequence.Next(ctx) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| err = sm.txhashesBySequenceMap.Set(ctx, seq, txHash) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.