From c36a5424093b888ba9b1652178737b30765e37cd Mon Sep 17 00:00:00 2001 From: Denis Fadeev Date: Mon, 30 Sep 2024 16:45:17 +0300 Subject: [PATCH] exit on error --- packages/localnet/src/handleOnEVMCalled.ts | 11 +++--- packages/localnet/src/handleOnEVMDeposited.ts | 11 +++--- packages/localnet/src/handleOnRevertEVM.ts | 12 +++++-- packages/localnet/src/handleOnRevertZEVM.ts | 35 +++++++++++++------ packages/localnet/src/handleOnZEVMCalled.ts | 13 ++++--- .../localnet/src/handleOnZEVMWithdrawn.ts | 13 ++++--- packages/localnet/src/index.ts | 15 ++++++-- packages/tasks/src/localnet.ts | 13 +++---- packages/tasks/src/stop.ts | 18 ++++++++-- 9 files changed, 97 insertions(+), 44 deletions(-) diff --git a/packages/localnet/src/handleOnEVMCalled.ts b/packages/localnet/src/handleOnEVMCalled.ts index ae1ad46..1621071 100644 --- a/packages/localnet/src/handleOnEVMCalled.ts +++ b/packages/localnet/src/handleOnEVMCalled.ts @@ -12,6 +12,7 @@ export const handleOnEVMCalled = async ({ deployer, fungibleModuleSigner, foreignCoins, + exitOnError = false, }: { tss: any; provider: ethers.JsonRpcProvider; @@ -20,6 +21,7 @@ export const handleOnEVMCalled = async ({ deployer: any; fungibleModuleSigner: any; foreignCoins: any[]; + exitOnError: boolean; }) => { log("EVM", "Gateway: 'Called' event emitted"); try { @@ -54,15 +56,16 @@ export const handleOnEVMCalled = async ({ logs.forEach((data) => { log("ZetaChain", `Event from onCrossChainCall: ${JSON.stringify(data)}`); }); - } catch (e: any) { - logErr("ZetaChain", `Error executing onCrossChainCall: ${e}`); + } catch (err: any) { + logErr("ZetaChain", `Error executing onCrossChainCall: ${err}`); const revertOptions = args[3]; - await handleOnRevertEVM({ + return await handleOnRevertEVM({ revertOptions, - err: e, + err, tss, provider, protocolContracts, + exitOnError, }); } }; diff --git a/packages/localnet/src/handleOnEVMDeposited.ts b/packages/localnet/src/handleOnEVMDeposited.ts index 25c7c6a..64f8c84 100644 --- a/packages/localnet/src/handleOnEVMDeposited.ts +++ b/packages/localnet/src/handleOnEVMDeposited.ts @@ -12,6 +12,7 @@ export const handleOnEVMDeposited = async ({ deployer, fungibleModuleSigner, foreignCoins, + exitOnError = false, }: { tss: any; provider: ethers.JsonRpcProvider; @@ -20,6 +21,7 @@ export const handleOnEVMDeposited = async ({ deployer: any; fungibleModuleSigner: any; foreignCoins: any[]; + exitOnError: boolean; }) => { log("EVM", "Gateway: 'Deposited' event emitted"); try { @@ -80,15 +82,16 @@ export const handleOnEVMDeposited = async ({ await tx.wait(); log("ZetaChain", `Deposited ${amount} of ${zrc20} tokens to ${receiver}`); } - } catch (e: any) { - logErr("ZetaChain", `Error depositing: ${e}`); + } catch (err) { + logErr("ZetaChain", `Error depositing: ${err}`); const revertOptions = args[5]; - await handleOnRevertEVM({ + return await handleOnRevertEVM({ revertOptions, - err: e, + err, tss, provider, protocolContracts, + exitOnError, }); } }; diff --git a/packages/localnet/src/handleOnRevertEVM.ts b/packages/localnet/src/handleOnRevertEVM.ts index e0cd943..183fa96 100644 --- a/packages/localnet/src/handleOnRevertEVM.ts +++ b/packages/localnet/src/handleOnRevertEVM.ts @@ -8,12 +8,14 @@ export const handleOnRevertEVM = async ({ provider, tss, protocolContracts, + exitOnError = false, }: { revertOptions: any; err: any; provider: any; tss: any; protocolContracts: any; + exitOnError: boolean; }) => { const callOnRevert = revertOptions[1]; const revertAddress = revertOptions[0]; @@ -45,10 +47,14 @@ export const handleOnRevertEVM = async ({ logs.forEach((data: any) => { log("EVM", `Event from onRevert: ${JSON.stringify(data)}`); }); - } catch (e: any) { - logErr("EVM", `Gateway: Call onRevert failed: ${e}`); + } catch (err) { + const error = `Gateway: Call onRevert failed: ${err}`; + logErr("EVM", error); + if (exitOnError) throw new Error(error); } } else { - logErr("EVM", `Tx reverted without callOnRevert: ${err}`); + const error = `Tx reverted without callOnRevert: ${err}`; + logErr("EVM", error); + if (exitOnError) throw new Error(error); } }; diff --git a/packages/localnet/src/handleOnRevertZEVM.ts b/packages/localnet/src/handleOnRevertZEVM.ts index 81fa25f..5b27a1b 100644 --- a/packages/localnet/src/handleOnRevertZEVM.ts +++ b/packages/localnet/src/handleOnRevertZEVM.ts @@ -1,13 +1,22 @@ import { ethers, NonceManager } from "ethers"; -export const handleOnRevertZEVM = async ( - revertOptions: any, - err: any, - tss: NonceManager, - log: (chain: "EVM" | "ZetaChain", ...messages: string[]) => void, - protocolContracts: any, - deployOpts: any -) => { +export const handleOnRevertZEVM = async ({ + revertOptions, + err, + tss, + log, + protocolContracts, + deployOpts, + exitOnError = false, +}: { + revertOptions: any; + err: any; + tss: NonceManager; + log: (chain: "EVM" | "ZetaChain", ...messages: string[]) => void; + protocolContracts: any; + deployOpts: any; + exitOnError: boolean; +}) => { const callOnRevert = revertOptions[1]; const revertAddress = revertOptions[0]; const revertMessage = revertOptions[3]; @@ -25,10 +34,14 @@ export const handleOnRevertZEVM = async ( .connect(tss) .executeRevert(revertAddress, revertContext, deployOpts); log("ZetaChain", "Gateway: Call onRevert success"); - } catch (e) { - log("ZetaChain", `Gateway: Call onRevert failed: ${e}`); + } catch (err) { + const error = `Gateway: Call onRevert failed: ${err}`; + log("ZetaChain", error); + if (exitOnError) throw new Error(error); } } else { - log("ZetaChain", "Tx reverted without callOnRevert: ", err); + const error = `Tx reverted without callOnRevert: ${err}`; + log("ZetaChain", error); + if (exitOnError) throw new Error(error); } }; diff --git a/packages/localnet/src/handleOnZEVMCalled.ts b/packages/localnet/src/handleOnZEVMCalled.ts index 82f3728..bb41968 100644 --- a/packages/localnet/src/handleOnZEVMCalled.ts +++ b/packages/localnet/src/handleOnZEVMCalled.ts @@ -9,11 +9,13 @@ export const handleOnZEVMCalled = async ({ provider, protocolContracts, args, + exitOnError = false, }: { tss: any; provider: ethers.JsonRpcProvider; protocolContracts: any; args: any; + exitOnError: boolean; }) => { log("ZetaChain", "Gateway: 'Called' event emitted"); try { @@ -35,15 +37,16 @@ export const handleOnZEVMCalled = async ({ log("EVM", `Event from contract: ${JSON.stringify(data)}`); }); await executeTx.wait(); - } catch (e) { + } catch (err) { const revertOptions = args[5]; - await handleOnRevertZEVM( + return await handleOnRevertZEVM({ revertOptions, - e, + err, tss, log, protocolContracts, - deployOpts - ); + deployOpts, + exitOnError, + }); } }; diff --git a/packages/localnet/src/handleOnZEVMWithdrawn.ts b/packages/localnet/src/handleOnZEVMWithdrawn.ts index 1abc82b..2c61072 100644 --- a/packages/localnet/src/handleOnZEVMWithdrawn.ts +++ b/packages/localnet/src/handleOnZEVMWithdrawn.ts @@ -12,6 +12,7 @@ export const handleOnZEVMWithdrawn = async ({ args, deployer, foreignCoins, + exitOnError = false, }: { tss: any; provider: ethers.JsonRpcProvider; @@ -19,6 +20,7 @@ export const handleOnZEVMWithdrawn = async ({ args: any; deployer: any; foreignCoins: any[]; + exitOnError: boolean; }) => { log("ZetaChain", "Gateway: 'Withdrawn' event emitted"); try { @@ -74,15 +76,16 @@ export const handleOnZEVMWithdrawn = async ({ `Transferred ${amount} ERC-20 tokens from Custody to ${receiver}` ); } - } catch (e) { + } catch (err) { const revertOptions = args[9]; - await handleOnRevertZEVM( + return await handleOnRevertZEVM({ revertOptions, - e, + err, tss, log, protocolContracts, - deployOpts - ); + deployOpts, + exitOnError, + }); } }; diff --git a/packages/localnet/src/index.ts b/packages/localnet/src/index.ts index 210fe1f..1018441 100755 --- a/packages/localnet/src/index.ts +++ b/packages/localnet/src/index.ts @@ -264,7 +264,13 @@ const deployProtocolContracts = async ( }; }; -export const initLocalnet = async ({ port }: { port: number }) => { +export const initLocalnet = async ({ + port, + exitOnError, +}: { + port: number; + exitOnError: boolean; +}) => { const provider = new ethers.JsonRpcProvider(`http://127.0.0.1:${port}`); provider.pollingInterval = 100; // anvil test mnemonic @@ -299,7 +305,7 @@ export const initLocalnet = async ({ port }: { port: number }) => { // Listen to contracts events protocolContracts.gatewayZEVM.on("Called", async (...args: Array) => { - handleOnZEVMCalled({ tss, provider, protocolContracts, args }); + handleOnZEVMCalled({ tss, provider, protocolContracts, args, exitOnError }); }); protocolContracts.gatewayZEVM.on("Withdrawn", async (...args: Array) => { @@ -310,11 +316,12 @@ export const initLocalnet = async ({ port }: { port: number }) => { args, deployer, foreignCoins, + exitOnError, }); }); protocolContracts.gatewayEVM.on("Called", async (...args: Array) => { - handleOnEVMCalled({ + return await handleOnEVMCalled({ tss, provider, protocolContracts, @@ -322,6 +329,7 @@ export const initLocalnet = async ({ port }: { port: number }) => { deployer, fungibleModuleSigner, foreignCoins, + exitOnError, }); }); @@ -334,6 +342,7 @@ export const initLocalnet = async ({ port }: { port: number }) => { deployer, fungibleModuleSigner, foreignCoins, + exitOnError, }); }); diff --git a/packages/tasks/src/localnet.ts b/packages/tasks/src/localnet.ts index 5606697..ff0d83e 100644 --- a/packages/tasks/src/localnet.ts +++ b/packages/tasks/src/localnet.ts @@ -89,7 +89,10 @@ const localnet = async (args: any) => { }; try { - const addr = await initLocalnet({ port: args.port }); + const addr = await initLocalnet({ + port: args.port, + exitOnError: args.exitOnError, + }); // EVM Contract Addresses const evmHeader = "\nEVM Contract Addresses"; @@ -135,7 +138,6 @@ const localnet = async (args: any) => { const handleExit = (signal: string) => { console.log(`\nReceived ${signal}, shutting down...`); - cleanup(); process.exit(); }; @@ -144,6 +146,7 @@ const localnet = async (args: any) => { process.on("exit", () => { console.log("Process exiting..."); + cleanup(); }); if (args.stopAfterInit) { @@ -164,7 +167,5 @@ export const localnetTask = task("localnet", "Start localnet", localnet) types.string ) .addFlag("forceKill", "Force kill any process on the port without prompting") - .addFlag( - "stopAfterInit", - "Stop the localnet after successful initialization" - ); + .addFlag("stopAfterInit", "Stop the localnet after successful initialization") + .addFlag("exitOnError", "Exit with an error if revert is not handled"); diff --git a/packages/tasks/src/stop.ts b/packages/tasks/src/stop.ts index 728a17d..4d7091f 100644 --- a/packages/tasks/src/stop.ts +++ b/packages/tasks/src/stop.ts @@ -11,11 +11,23 @@ const localnetStop = async (args: any) => { } const pid = fs.readFileSync(LOCALNET_PID_FILE, "utf-8").trim(); + try { - process.kill(Number(pid)); - console.log(ansis.green(`Successfully stopped localnet (PID: ${pid})`)); + process.kill(Number(pid), 0); // check that the process is running + try { + process.kill(Number(pid)); + console.log(ansis.green(`Successfully stopped localnet (PID: ${pid})`)); + } catch (err) { + console.error(ansis.red(`Failed to stop localnet: ${err}`)); + } } catch (err) { - console.error(ansis.red(`Failed to stop localnet: ${err}`)); + console.log(ansis.yellow(`Localnet process (PID: ${pid}) is not running.`)); + try { + fs.unlinkSync(LOCALNET_PID_FILE); + console.log(ansis.green("Localnet PID file deleted.")); + } catch (err) { + console.error(ansis.red(`Failed to delete PID file: ${err}`)); + } } };