Emit events on revert

zeta-chain · Oct 4, 2024 · 047c664 · 047c664
1 parent cfafa0d
commit 047c664
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 7 deletions.
diff --git a/packages/localnet/src/handleOnRevertZEVM.ts b/packages/localnet/src/handleOnRevertZEVM.ts
@@ -1,16 +1,21 @@
 import { ethers, NonceManager } from "ethers";
+import { logErr } from "./log";
 
 export const handleOnRevertZEVM = async ({
   revertOptions,
   err,
+  provider,
   tss,
   log,
+  fungibleModuleSigner,
   protocolContracts,
   deployOpts,
   exitOnError = false,
 }: {
   revertOptions: any;
   err: any;
+  provider: any;
+  fungibleModuleSigner: any;
   tss: NonceManager;
   log: (chain: "EVM" | "ZetaChain", ...messages: string[]) => void;
   protocolContracts: any;
@@ -30,18 +35,27 @@ export const handleOnRevertZEVM = async ({
     log("ZetaChain", "Gateway: calling executeRevert");
     try {
       tss.reset();
-      await protocolContracts.gatewayZEVM
-        .connect(tss)
+      const tx = await protocolContracts.gatewayZEVM
+        .connect(fungibleModuleSigner)
         .executeRevert(revertAddress, revertContext, deployOpts);
-      log("ZetaChain", "Gateway: Call onRevert success");
+      await tx.wait();
+      log("ZetaChain", "Gateway: successfully called onRevert");
+      const logs = await provider.getLogs({
+        address: revertAddress,
+        fromBlock: "latest",
+      });
+
+      logs.forEach((data: any) => {
+        log("ZetaChain", `Event from onRevert: ${JSON.stringify(data)}`);
+      });
     } catch (err) {
       const error = `Gateway: Call onRevert failed: ${err}`;
-      log("ZetaChain", error);
+      logErr("ZetaChain", error);
       if (exitOnError) throw new Error(error);
     }
   } else {
     const error = `Tx reverted without callOnRevert: ${err}`;
-    log("ZetaChain", error);
+    logErr("ZetaChain", error);
     if (exitOnError) throw new Error(error);
   }
 };
diff --git a/packages/localnet/src/handleOnZEVMCalled.ts b/packages/localnet/src/handleOnZEVMCalled.ts
@@ -9,12 +9,14 @@ export const handleOnZEVMCalled = async ({
   provider,
   protocolContracts,
   args,
+  fungibleModuleSigner,
   exitOnError = false,
 }: {
   tss: any;
   provider: ethers.JsonRpcProvider;
   protocolContracts: any;
   args: any;
+  fungibleModuleSigner: any;
   exitOnError: boolean;
 }) => {
   log("ZetaChain", "Gateway: 'Called' event emitted");
@@ -38,10 +40,13 @@ export const handleOnZEVMCalled = async ({
     });
     await executeTx.wait();
   } catch (err) {
+    logErr("EVM", `Error executing a contract: ${err}`);
     const revertOptions = args[5];
     return await handleOnRevertZEVM({
       revertOptions,
       err,
+      provider,
+      fungibleModuleSigner,
       tss,
       log,
       protocolContracts,

diff --git a/packages/localnet/src/handleOnZEVMWithdrawn.ts b/packages/localnet/src/handleOnZEVMWithdrawn.ts
@@ -10,6 +10,7 @@ export const handleOnZEVMWithdrawn = async ({
   provider,
   protocolContracts,
   args,
+  fungibleModuleSigner,
   deployer,
   foreignCoins,
   exitOnError = false,
@@ -18,6 +19,7 @@ export const handleOnZEVMWithdrawn = async ({
   provider: ethers.JsonRpcProvider;
   protocolContracts: any;
   args: any;
+  fungibleModuleSigner: any;
   deployer: any;
   foreignCoins: any[];
   exitOnError: boolean;
@@ -98,8 +100,10 @@ export const handleOnZEVMWithdrawn = async ({
     return await handleOnRevertZEVM({
       revertOptions,
       err,
+      provider,
       tss,
       log,
+      fungibleModuleSigner,
       protocolContracts,
       deployOpts,
       exitOnError,

diff --git a/packages/localnet/src/index.ts b/packages/localnet/src/index.ts
@@ -4,7 +4,6 @@ import * as Custody from "@zetachain/protocol-contracts/abi/ERC20Custody.sol/ERC
 import * as ERC1967Proxy from "@zetachain/protocol-contracts/abi/ERC1967Proxy.sol/ERC1967Proxy.json";
 import * as TestERC20 from "@zetachain/protocol-contracts/abi/TestERC20.sol/TestERC20.json";
 import * as SystemContract from "@zetachain/protocol-contracts/abi/SystemContractMock.sol/SystemContractMock.json";
-import * as ZRC20 from "@zetachain/protocol-contracts/abi/ZRC20.sol/ZRC20.json";
 import * as GatewayZEVM from "@zetachain/protocol-contracts/abi/GatewayZEVM.sol/GatewayZEVM.json";
 import * as ZetaConnectorNonNative from "@zetachain/protocol-contracts/abi/ZetaConnectorNonNative.sol/ZetaConnectorNonNative.json";
 import * as WETH9 from "@zetachain/protocol-contracts/abi/WZETA.sol/WETH9.json";
@@ -305,7 +304,14 @@ export const initLocalnet = async ({
 
   // Listen to contracts events
   protocolContracts.gatewayZEVM.on("Called", async (...args: Array<any>) => {
-    handleOnZEVMCalled({ tss, provider, protocolContracts, args, exitOnError });
+    handleOnZEVMCalled({
+      tss,
+      provider,
+      fungibleModuleSigner,
+      protocolContracts,
+      args,
+      exitOnError,
+    });
   });
 
   protocolContracts.gatewayZEVM.on("Withdrawn", async (...args: Array<any>) => {
@@ -315,6 +321,7 @@ export const initLocalnet = async ({
       protocolContracts,
       args,
       deployer,
+      fungibleModuleSigner,
       foreignCoins,
       exitOnError,
     });