added nativeERC20 approve

Ethernal-Tech · Nov 28, 2023 · db9e2b6 · db9e2b6
1 parent 87002cf
commit db9e2b6
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/consensus/polybft/contracts_initializer.go b/consensus/polybft/contracts_initializer.go
@@ -37,6 +37,21 @@ func initStakeManager(polyBFTConfig PolyBFTConfig, transition *state.Transition)
 			Stake:  validator.Stake,
 			BlsKey: key.ToBigInt(),
 		}
+
+		approveFn := &contractsapi.ApproveNativeERC20Fn{
+			Spender: contracts.StakeManagerContract,
+			Amount:  validator.Stake,
+		}
+
+		input, err := approveFn.EncodeAbi()
+		if err != nil {
+			return fmt.Errorf("NativeERC20.approve params encoding failed: %w", err)
+		}
+
+		err = callContract(validator.Address, contracts.NativeERC20TokenContract, input, "NativeERC20.approve", transition)
+		if err != nil {
+			return fmt.Errorf("Error while calling contract %w", err)
+		}
 	}
 
 	initFn := &contractsapi.InitializeStakeManagerFn{

diff --git a/consensus/polybft/contractsapi/bindings-gen/main.go b/consensus/polybft/contractsapi/bindings-gen/main.go
@@ -140,6 +140,7 @@ func main() {
 			false,
 			[]string{
 				"initialize",
+				"approve",
 			},
 			[]string{},
 		},

diff --git a/consensus/polybft/contractsapi/contractsapi.go b/consensus/polybft/contractsapi/contractsapi.go