feat(paymaster): allow second execution of service

cartesi · Jul 30, 2024 · 7755228 · 7755228
1 parent 3c48bbe
commit 7755228
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 27 deletions.
diff --git a/packages/mock-verifying-paymaster/src/helpers/abi.ts b/packages/mock-verifying-paymaster/src/helpers/abi.ts
@@ -100,6 +100,19 @@ export const VERIFYING_PAYMASTER_V07_ABI = [
         ],
         stateMutability: "nonpayable",
     },
+    {
+        inputs: [],
+        name: "getDeposit",
+        outputs: [
+            {
+                internalType: "uint256",
+                name: "",
+                type: "uint256",
+            },
+        ],
+        stateMutability: "view",
+        type: "function",
+    },
     {
         type: "function",
         name: "getHash",

diff --git a/packages/mock-verifying-paymaster/src/helpers/verifyingPaymasters.ts b/packages/mock-verifying-paymaster/src/helpers/verifyingPaymasters.ts
@@ -47,32 +47,38 @@ export const setupVerifyingPaymasterV07 = async (
         chain: await getChain(),
     });
 
-    await walletClient
-        .sendTransaction({
-            to: DETERMINISTIC_DEPLOYER,
-            data,
-        })
-        .then((hash) => publicClient.waitForTransactionReceipt({ hash }))
-        .then(() => console.log("deployed VerifyingPaymaster v0.7"));
-
     const address = getContractAddress({
         opcode: "CREATE2",
         from: DETERMINISTIC_DEPLOYER,
         salt: slice(data, 0, 32),
         bytecode: slice(data, 32),
     });
 
+    if ((await publicClient.getCode({ address })) === undefined) {
+        await walletClient
+            .sendTransaction({
+                to: DETERMINISTIC_DEPLOYER,
+                data,
+            })
+            .then((hash) => publicClient.waitForTransactionReceipt({ hash }))
+            .then(() => console.log("deployed VerifyingPaymaster v0.7"));
+    }
+
     const verifyingPaymaster = getContract({
         address,
         abi: VERIFYING_PAYMASTER_V07_ABI,
         client: walletClient,
     });
 
-    await verifyingPaymaster.write
-        .deposit({
-            value: parseEther("50"),
-        })
-        .then(() => console.log("Funded VerifyingPaymaster V0.7"));
+    const requiredDeposit = parseEther("50");
+    const currentDeposit = await verifyingPaymaster.read.getDeposit();
+    if (currentDeposit < requiredDeposit) {
+        await verifyingPaymaster.write
+            .deposit({
+                value: requiredDeposit - currentDeposit,
+            })
+            .then(() => console.log("Funded VerifyingPaymaster V0.7"));
+    }
 
     return verifyingPaymaster;
 };
@@ -87,32 +93,38 @@ export const setupVerifyingPaymasterV06 = async (
         chain: await getChain(),
     });
 
-    await walletClient
-        .sendTransaction({
-            to: DETERMINISTIC_DEPLOYER,
-            data,
-        })
-        .then((hash) => publicClient.waitForTransactionReceipt({ hash }))
-        .then(() => console.log("deployed VerifyingPaymaster v0.6"));
-
     const address = getContractAddress({
         opcode: "CREATE2",
         from: DETERMINISTIC_DEPLOYER,
         salt: slice(data, 0, 32),
         bytecode: slice(data, 32),
     });
 
+    if ((await publicClient.getCode({ address })) === undefined) {
+        await walletClient
+            .sendTransaction({
+                to: DETERMINISTIC_DEPLOYER,
+                data,
+            })
+            .then((hash) => publicClient.waitForTransactionReceipt({ hash }))
+            .then(() => console.log("deployed VerifyingPaymaster v0.6"));
+    }
+
     const verifyingPaymaster = getContract({
         address,
         abi: VERIFYING_PAYMASTER_V06_ABI,
         client: walletClient,
     });
 
-    await verifyingPaymaster.write
-        .deposit({
-            value: parseEther("50"),
-        })
-        .then(() => console.log("Funded VerifyingPaymaster V0.6"));
+    const requiredDeposit = parseEther("50");
+    const currentDeposit = await verifyingPaymaster.read.getDeposit();
+    if (currentDeposit < requiredDeposit) {
+        await verifyingPaymaster.write
+            .deposit({
+                value: requiredDeposit - currentDeposit,
+            })
+            .then(() => console.log("Funded VerifyingPaymaster V0.6"));
+    }
 
     return verifyingPaymaster;
 };
diff --git a/packages/mock-verifying-paymaster/src/index.ts b/packages/mock-verifying-paymaster/src/index.ts
@@ -49,7 +49,8 @@ const main = async () => {
         return reply.code(200).send({ message: "pong" });
     });
 
-    await app.listen({ host: "0.0.0.0", port: 3000 });
+    const service = await app.listen({ host: "0.0.0.0", port: 3000 });
+    console.log(`Service ready: ${service}`);
 };
 
 main();