make a should check prefund flag

pimlicolabs · Jan 30, 2025 · b69d5d4 · b69d5d4
1 parent 3d7de1f
commit b69d5d4
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/src/cli/config/bundler.ts b/src/cli/config/bundler.ts
@@ -84,7 +84,8 @@ export const bundlerArgsSchema = z.object({
             )}`
         ),
     "enable-instant-bundling-endpoint": z.boolean(),
-    "enable-experimental-7702-endpoints": z.boolean()
+    "enable-experimental-7702-endpoints": z.boolean(),
+    "should-check-prefund": z.boolean()
 })
 
 export const executorArgsSchema = z.object({

diff --git a/src/cli/config/options.ts b/src/cli/config/options.ts
@@ -135,6 +135,12 @@ export const bundlerOptions: CliCommandOptions<IBundlerArgsInput> = {
             "Should the bundler enable the pimlico_experimental_sendUserOperation7702 and pimlico_experimental_estimateUserOperationGas7702 endpoint",
         type: "boolean",
         default: false
+    },
+    "should-check-prefund": {
+        description:
+            "Should the bundler check userOp's prefund before accepting it",
+        type: "boolean",
+        default: true
     }
 }
 

diff --git a/src/rpc/rpcHandler.ts b/src/rpc/rpcHandler.ts
@@ -752,8 +752,10 @@ export class RpcHandler implements IRpcEndpoint {
             })
         }
 
+        const shouldCheckPrefund =
+            apiVersion !== "v1" || this.config.shouldCheckPrefund
         const validationResult = await this.validator.validateUserOperation({
-            shouldCheckPrefund: apiVersion !== "v1",
+            shouldCheckPrefund,
             userOperation,
             queuedUserOperations,
             entryPoint