feat: staking precompile

app/app.go

@@ -567,7 +567,7 @@ func New(
 		&app.FeeMarketKeeper,
 		tracer,
 		evmSs,
-		precompiles.StatefulContracts(&app.FungibleKeeper, appCodec, storetypes.TransientGasConfig()),
+		precompiles.StatefulContracts(&app.FungibleKeeper, app.StakingKeeper, appCodec, storetypes.TransientGasConfig()),
 		app.ConsensusParamsKeeper,
 		aggregateAllKeys(keys, tKeys, memKeys),
 	)

changelog.md

@@ -14,7 +14,8 @@
 * [2681](https://github.com/zeta-chain/node/pull/2681) - implement `MsgUpdateERC20CustodyPauseStatus` to pause or unpause ERC20 Custody contract (to be used for the migration process for smart contract V2)
 * [2644](https://github.com/zeta-chain/node/pull/2644) - add created_timestamp to cctx status
 * [2673](https://github.com/zeta-chain/node/pull/2673) - add relayer key importer, encryption and decryption
-* [2633](https://github.com/zeta-chain/node/pull/2633) - support for stateful precompiled contracts.
+* [2633](https://github.com/zeta-chain/node/pull/2633) - support for stateful precompiled contracts
+* [2784](https://github.com/zeta-chain/node/pull/2784) - staking precompiled contract
 
 ### Refactor
 

precompiles/precompiles.go

@@ -9,35 +9,49 @@
 	ethparams "github.com/ethereum/go-ethereum/params"
 	evmkeeper "github.com/zeta-chain/ethermint/x/evm/keeper"
 
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
 	"github.com/zeta-chain/zetacore/precompiles/prototype"
-	"github.com/zeta-chain/zetacore/x/fungible/keeper"
+	"github.com/zeta-chain/zetacore/precompiles/staking"
+	fungiblekeeper "github.com/zeta-chain/zetacore/x/fungible/keeper"
 )
 
 // EnabledStatefulContracts contains the list of all enabled stateful precompiles.
 // This is useful for listing and reading from other packages, such as BlockedAddrs() function.
 // Setting to false a contract here will disable it, not being included in the blockchain.
 var EnabledStatefulContracts = map[common.Address]bool{
 	prototype.ContractAddress: true,
+	staking.ContractAddress:   true,
 }
 
 // StatefulContracts returns all the registered precompiled contracts.
 func StatefulContracts(
-	fungibleKeeper *keeper.Keeper,
+	fungibleKeeper *fungiblekeeper.Keeper,
+	stakingKeeper *stakingkeeper.Keeper,
 	cdc codec.Codec,
 	gasConfig storetypes.GasConfig,
 ) (precompiledContracts []evmkeeper.CustomContractFn) {
 	// Initialize at 0 the custom compiled contracts and the addresses.
 	precompiledContracts = make([]evmkeeper.CustomContractFn, 0)
 
-	// Define the regular contract function.
+	// Define the prototype contract function.
 	if EnabledStatefulContracts[prototype.ContractAddress] {
 		prototypeContract := func(_ sdktypes.Context, _ ethparams.Rules) vm.PrecompiledContract {
 			return prototype.NewIPrototypeContract(fungibleKeeper, cdc, gasConfig)
 		}
 
-		// Append the regular contract to the precompiledContracts slice.
+		// Append the prototype contract to the precompiledContracts slice.
 		precompiledContracts = append(precompiledContracts, prototypeContract)
 	}
 
+	// Define the staking contract function.
+	if EnabledStatefulContracts[prototype.ContractAddress] {
+		stakingContract := func(_ sdktypes.Context, _ ethparams.Rules) vm.PrecompiledContract {
+			return staking.NewIStakingContract(stakingKeeper, cdc, gasConfig)
+		}
+
+		// Append the prototype contract to the precompiledContracts slice.
+		precompiledContracts = append(precompiledContracts, stakingContract)
+	}
+
 	return precompiledContracts
 }

precompiles/precompiles_test.go

@@ -11,7 +11,7 @@ import (
 )
 
 func Test_StatefulContracts(t *testing.T) {
-	k, ctx, _, _ := keeper.FungibleKeeper(t)
+	k, ctx, sdkk, _ := keeper.FungibleKeeper(t)
 	gasConfig := storetypes.TransientGasConfig()
 
 	var encoding ethermint.EncodingConfig
@@ -25,7 +25,7 @@ func Test_StatefulContracts(t *testing.T) {
 	}
 
 	// StatefulContracts() should return all the enabled contracts.
-	contracts := StatefulContracts(k, appCodec, gasConfig)
+	contracts := StatefulContracts(k, &sdkk.StakingKeeper, appCodec, gasConfig)
 	require.NotNil(t, contracts, "StatefulContracts() should not return a nil slice")
 	require.Len(t, contracts, expectedContracts, "StatefulContracts() should return all the enabled contracts")
 

precompiles/prototype/prototype_test.go

@@ -296,7 +296,6 @@ func Test_GetGasStabilityPoolBalance(t *testing.T) {
 	)
 
 	t.Run("should fail with invalid arguments", func(t *testing.T) {
-
 		t.Run("invalid number of arguments", func(t *testing.T) {
 			args := []interface{}{int64(1337), "second argument"}
 			_, err := contract.GetGasStabilityPoolBalance(ctx, &methodID, args)

precompiles/staking/IStaking.abi

@@ -0,0 +1,94 @@
+[
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "delegator",
+        "type": "address"
+      },
+      {
+        "internalType": "string",
+        "name": "validator",
+        "type": "string"
+      },
+      {
+        "internalType": "uint256",
+        "name": "amount",
+        "type": "uint256"
+      }
+    ],
+    "name": "delegate",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "success",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "delegator",
+        "type": "address"
+      },
+      {
+        "internalType": "string",
+        "name": "validatorSrc",
+        "type": "string"
+      },
+      {
+        "internalType": "string",
+        "name": "validatorDst",
+        "type": "string"
+      },
+      {
+        "internalType": "uint256",
+        "name": "amount",
+        "type": "uint256"
+      }
+    ],
+    "name": "redelegate",
+    "outputs": [
+      {
+        "internalType": "int64",
+        "name": "completionTime",
+        "type": "int64"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "delegator",
+        "type": "address"
+      },
+      {
+        "internalType": "string",
+        "name": "validator",
+        "type": "string"
+      },
+      {
+        "internalType": "uint256",
+        "name": "amount",
+        "type": "uint256"
+      }
+    ],
+    "name": "undelegate",
+    "outputs": [
+      {
+        "internalType": "int64",
+        "name": "completionTime",
+        "type": "int64"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  }
+]