add high level event function

zeta-chain · Sep 17, 2024 · 05d0bfb · 05d0bfb
1 parent f1e5b3a
commit 05d0bfb
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 50 deletions.
diff --git a/precompiles/bank/const.go b/precompiles/bank/const.go
@@ -7,9 +7,11 @@ const (
 	// Write methods.
 	DepositMethodName = "deposit"
 	DepositMethodGas  = 200_000
+	DepositEventName  = "Deposit"
 
 	WithdrawMethodName = "withdraw"
 	WithdrawMethodGas  = 200_000
+	WithdrawEventName  = "Withdraw"
 
 	// Read methods.
 	BalanceOfMethodName = "balanceOf"

diff --git a/precompiles/bank/logs.go b/precompiles/bank/logs.go
@@ -10,70 +10,28 @@ import (
 	"github.com/zeta-chain/node/precompiles/logs"
 )
 
-const (
-	DepositEventName  = "Deposit"
-	WithdrawEventName = "Withdraw"
-)
-
-func (c *Contract) AddDepositLog(
-	ctx sdk.Context,
-	stateDB vm.StateDB,
-	zrc20Depositor common.Address,
-	zrc20Token common.Address,
-	cosmosAddr string,
-	cosmosCoin string,
-	amount *big.Int,
-) error {
-	event := c.Abi().Events[DepositEventName]
-
-	// ZRC20, cosmos coin and depositor.
-	topics, err := logs.MakeTopics(
-		event,
-		[]interface{}{common.BytesToAddress(zrc20Depositor.Bytes())},
-		[]interface{}{common.BytesToAddress(zrc20Token.Bytes())},
-		[]interface{}{cosmosCoin},
-	)
-	if err != nil {
-		return err
-	}
-
-	// Pack cosmos address and amount as data.
-	data, err := logs.PackArguments([]logs.Argument{
-		{Type: "string", Value: cosmosAddr},
-		{Type: "uint256", Value: amount},
-	})
-	if err != nil {
-		return err
-	}
-
-	logs.AddLog(ctx, c.Address(), stateDB, topics, data)
-
-	return nil
-}
-
-func (c *Contract) AddWithdrawLog(
+func (c *Contract) addEventLog(
 	ctx sdk.Context,
 	stateDB vm.StateDB,
-	zrc20Withdrawer common.Address,
+	eventName string,
+	zrc20Addr common.Address,
 	zrc20Token common.Address,
 	cosmosAddr string,
 	cosmosCoin string,
 	amount *big.Int,
 ) error {
-	event := c.Abi().Events[WithdrawEventName]
+	event := c.Abi().Events[eventName]
 
-	// ZRC20, cosmos coin  and witgdrawer are indexed.
 	topics, err := logs.MakeTopics(
 		event,
-		[]interface{}{common.BytesToAddress(zrc20Withdrawer.Bytes())},
+		[]interface{}{common.BytesToAddress(zrc20Addr.Bytes())},
 		[]interface{}{common.BytesToAddress(zrc20Token.Bytes())},
 		[]interface{}{cosmosCoin},
 	)
 	if err != nil {
 		return err
 	}
 
-	// Pack cosmos address and amount as data.
 	data, err := logs.PackArguments([]logs.Argument{
 		{Type: "string", Value: cosmosAddr},
 		{Type: "uint256", Value: amount},

diff --git a/precompiles/bank/method_deposit.go b/precompiles/bank/method_deposit.go
@@ -148,7 +148,7 @@ func (c *Contract) deposit(
 		}
 	}
 
-	if err := c.AddDepositLog(ctx, evm.StateDB, caller, zrc20Addr, toAddr.String(), coinSet.Denoms()[0], amount); err != nil {
+	if err := c.addEventLog(ctx, evm.StateDB, DepositEventName, caller, zrc20Addr, toAddr.String(), coinSet.Denoms()[0], amount); err != nil {
 		return nil, &ptypes.ErrUnexpected{
 			When: "AddDepositLog",
 			Got:  err.Error(),

diff --git a/precompiles/bank/method_withdraw.go b/precompiles/bank/method_withdraw.go
@@ -127,7 +127,7 @@ func (c *Contract) withdraw(
 		}
 	}
 
-	if err := c.AddWithdrawLog(ctx, evm.StateDB, caller, zrc20Addr, fromAddr.String(), coinSet.Denoms()[0], amount); err != nil {
+	if err := c.addEventLog(ctx, evm.StateDB, WithdrawEventName, caller, zrc20Addr, fromAddr.String(), coinSet.Denoms()[0], amount); err != nil {
 		return nil, &ptypes.ErrUnexpected{
 			When: "AddWithdrawLog",
 			Got:  err.Error(),

diff --git a/precompiles/types/errors_test.go b/precompiles/types/errors_test.go
@@ -94,4 +94,4 @@ func Test_ErrInsufficientBalance(t *testing.T) {
 	if got != expect {
 		t.Errorf("Expected %v, got %v", expect, got)
 	}
-}
+}