feat(walletManager): manually record wallet usage find unlock

Signed-off-by: Pablo Maldonado <pablo@umaproject.org>

src/index.ts

@@ -151,6 +151,10 @@ app.post("/", async (req, res, next) => {
           await handleUnsupportedRequest(req, res, "No valid unlock found"); // Pass through if no unlock is found.
           return;
         }
+
+        // FindUnlock does not update wallet usage, so we need to do it manually.
+        WalletManager.getInstance().updateWalletUsage(unlock.ovalAddress, unlock.unlockerWallet, targetBlock);
+
         // Dynamically adjust refund percent so that builder nets at least configured minimum. We don't need to consider
         // refund gas costs as the builder is deducting them from refund and should not include the bundle if refund gas
         // costs exceed refund value.

src/lib/bundleUtils.ts

@@ -75,11 +75,11 @@ export const prepareUnlockTransaction = async (
     target,
   );
 
-  if (!simulate) return { ovalAddress, unlockTxHash, signedUnlockTx };
+  if (!simulate) return { ovalAddress, unlockTxHash, signedUnlockTx, unlockerWallet };
 
   const simulationResponse = await flashbotsBundleProvider.simulate([signedUnlockTx, ...backrunTxs], targetBlock);
 
-  return { ovalAddress, unlockTxHash, signedUnlockTx, simulationResponse };
+  return { ovalAddress, unlockTxHash, signedUnlockTx, simulationResponse, unlockerWallet };
 };
 
 export const getUnlockBundlesFromOvalAddresses = async (

src/lib/walletManager.ts

@@ -200,7 +200,7 @@ export class WalletManager {
     }
 
     // Update the usage statistics for a wallet
-    private async updateWalletUsage(ovalInstance: string, wallet: Wallet, targetBlock: number): Promise<void> {
+    public async updateWalletUsage(ovalInstance: string, wallet: Wallet, targetBlock: number): Promise<void> {
         const walletPubKey = await wallet.getAddress();
         const instanceUsage = this.sharedWalletUsage.get(walletPubKey) || new Map();
         const existingRecord = instanceUsage.get(targetBlock);