From 255ec6374cb28a6da42827e1a28d67d506be3312 Mon Sep 17 00:00:00 2001
From: mouseless <97399882+mouseless-eth@users.noreply.github.com>
Date: Sun, 26 Jan 2025 05:39:37 +0000
Subject: [PATCH] cleanup markWalletProcessed flow

---
 src/executor/executor.ts        | 4 ----
 src/executor/executorManager.ts | 8 ++++++++
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/executor/executor.ts b/src/executor/executor.ts
index d7e6450a..794d674d 100644
--- a/src/executor/executor.ts
+++ b/src/executor/executor.ts
@@ -588,7 +588,6 @@ export class Executor {
             childLogger.error(
                 "gas limit simulation encountered unexpected failure"
             )
-            this.markWalletProcessed(wallet)
             return {
                 status: "bundle_failure",
                 reason: "INTERNAL FAILURE",
@@ -600,7 +599,6 @@ export class Executor {
 
         if (opsToBundle.length === 0) {
             childLogger.warn("all ops failed simulation")
-            this.markWalletProcessed(wallet)
             return {
                 status: "bundle_failure",
                 reason: "INTERNAL FAILURE",
@@ -683,7 +681,6 @@ export class Executor {
                     { error: e },
                     "insufficient funds, not submitting transaction"
                 )
-                this.markWalletProcessed(wallet)
                 return {
                     status: "bundle_resubmit",
                     reason: InsufficientFundsError.name,
@@ -696,7 +693,6 @@ export class Executor {
                 { error: JSON.stringify(err) },
                 "error submitting bundle transaction"
             )
-            this.markWalletProcessed(wallet)
             return {
                 status: "bundle_failure",
                 reason: "INTERNAL FAILURE",
diff --git a/src/executor/executorManager.ts b/src/executor/executorManager.ts
index 2260b601..72fdc953 100644
--- a/src/executor/executorManager.ts
+++ b/src/executor/executorManager.ts
@@ -262,6 +262,14 @@ export class ExecutorManager {
                     break
             }
 
+            // Free wallet if the wallet did not make a succesful bundle tx.
+            if (
+                bundle.status === "bundle_failure" ||
+                bundle.status === "bundle_resubmit"
+            ) {
+                this.senderManager.markWalletProcessed(wallet)
+            }
+
             if (bundle.status === "bundle_resubmit") {
                 const { userOpsBundled: userOperations, reason } = bundle