fix(mu): isWallet add config to disable hb check

Author: jfrain99

servers/cu/README.md

@@ -151,6 +151,7 @@ to be retrieved from the dryrun cache.
   If a eval stream is more nonces behind than this limit, a 503 is returned.
 - `SU_ROUTER_URL`: the SU router url to default to when checking if target is a wallet
 - `HB_ROUTER_URL`: the HB router url to default to when checking if target is a wallet
+- `ENABLE_HB_WALLET_CHECK`: whether to enable HB wallet check when checking if target is a wallet
 
 ## Tests
 

servers/cu/src/domain/model.js

@@ -270,7 +270,11 @@ export const domainConfigSchema = z.object({
   /**
    * The HB router url to default to when checking if target is a wallet
    */
-  HB_ROUTER_URL: z.string()
+  HB_ROUTER_URL: z.string(),
+  /**
+   * Whether to enable HB wallet check when checking if target is a wallet
+   */
+  ENABLE_HB_WALLET_CHECK: z.boolean()
 })
 
 export const bufferSchema = z.any().refine(buffer => {

servers/mu/src/config.js

@@ -84,7 +84,8 @@ export const domainConfigSchema = z.object({
   IP_WALLET_RATE_LIMIT_INTERVAL: positiveIntSchema,
   STALE_CURSOR_RANGE: positiveIntSchema,
   SU_ROUTER_URL: z.string(),
-  HB_ROUTER_URL: z.string()
+  HB_ROUTER_URL: z.string(),
+  ENABLE_HB_WALLET_CHECK: z.boolean()
 })
 
 /**
@@ -145,7 +146,8 @@ const CONFIG_ENVS = {
     IP_WALLET_RATE_LIMIT_INTERVAL: process.env.IP_WALLET_RATE_LIMIT_INTERVAL || 1000 * 60 * 60,
     STALE_CURSOR_RANGE: process.env.STALE_CURSOR_RANGE || 1 * 24 * 60 * 60 * 1000,
     SU_ROUTER_URL: process.env.SU_ROUTER_URL || 'https://su-router.ao-testnet.xyz',
-    HB_ROUTER_URL: process.env.HB_ROUTER_URL || 'https://forward.computer'
+    HB_ROUTER_URL: process.env.HB_ROUTER_URL || 'https://forward.computer',
+    ENABLE_HB_WALLET_CHECK: process.env.ENABLE_HB_WALLET_CHECK !== 'false'
   },
   production: {
     MODE,
@@ -183,7 +185,8 @@ const CONFIG_ENVS = {
     IP_WALLET_RATE_LIMIT_INTERVAL: process.env.IP_WALLET_RATE_LIMIT_INTERVAL || 1000 * 60 * 60,
     STALE_CURSOR_RANGE: process.env.STALE_CURSOR_RANGE || 1 * 24 * 60 * 60 * 1000,
     SU_ROUTER_URL: process.env.SU_ROUTER_URL || 'https://su-router.ao-testnet.xyz',
-    HB_ROUTER_URL: process.env.HB_ROUTER_URL || 'https://forward.computer'
+    HB_ROUTER_URL: process.env.HB_ROUTER_URL || 'https://forward.computer',
+    ENABLE_HB_WALLET_CHECK: process.env.ENABLE_HB_WALLET_CHECK !== 'false'
   }
 }
 

servers/mu/src/domain/clients/gateway.js

@@ -8,6 +8,7 @@ function isWalletWith ({
   GRAPHQL_URL,
   SU_ROUTER_URL,
   HB_ROUTER_URL,
+  ENABLE_HB_WALLET_CHECK,
   logger,
   setById,
   getById
@@ -55,15 +56,19 @@ function isWalletWith ({
     }
 
     // Step 2: Check HyperBeam
-    try {
-      logger({ log: `Step 2: Checking HyperBeam for process ${id}`, logId })
-      const hyperbeamResponse = await walletFetch(`${HB_ROUTER_URL}/${id}[email protected]/info/[email protected]`)
-      if (hyperbeamResponse.status === 200) {
-        logger({ log: `Found process in HyperBeam for ${id}`, logId })
-        return setById(id, { isWallet: false }).then(() => false)
+    if (ENABLE_HB_WALLET_CHECK) {
+      try {
+        logger({ log: `Step 2: Checking HyperBeam for process ${id}`, logId })
+        const hyperbeamResponse = await walletFetch(`${HB_ROUTER_URL}/${id}[email protected]/info/[email protected]`)
+        if (hyperbeamResponse.status === 200) {
+          logger({ log: `Found process in HyperBeam for ${id}`, logId })
+          return setById(id, { isWallet: false }).then(() => false)
+        }
+      } catch (err) {
+        logger({ log: `Step 2: HyperBeam check failed for ${id}: ${err.message}`, logId })
       }
-    } catch (err) {
-      logger({ log: `Step 2: HyperBeam check failed for ${id}: ${err.message}`, logId })
+    } else {
+      logger({ log: `Step 2: Skipping HyperBeam check for ${id}`, logId })
     }
 
     // Step 3: Check Arweave

servers/mu/src/domain/index.js

@@ -96,6 +96,7 @@ export const createApis = async (ctx) => {
   const STALE_CURSOR_RANGE = ctx.STALE_CURSOR_RANGE
   const SU_ROUTER_URL = ctx.SU_ROUTER_URL
   const HB_ROUTER_URL = ctx.HB_ROUTER_URL
+  const ENABLE_HB_WALLET_CHECK = ctx.ENABLE_HB_WALLET_CHECK
 
   const logger = ctx.logger
   const fetch = ctx.fetch
@@ -233,7 +234,7 @@ export const createApis = async (ctx) => {
     fetchHyperBeamResult: cuClient.fetchHyperBeamResultWith({ fetch, HB_URL, histogram, logger: sendDataItemLogger }),
     fetchSchedulerProcess: schedulerClient.fetchSchedulerProcessWith({ getByProcess, setByProcess, fetch, histogram, logger: sendDataItemLogger }),
     crank,
-    isWallet: gatewayClient.isWalletWith({ fetch, histogram, ARWEAVE_URL, GRAPHQL_URL, SU_ROUTER_URL, HB_ROUTER_URL, logger: sendDataItemLogger }),
+    isWallet: gatewayClient.isWalletWith({ fetch, histogram, ARWEAVE_URL, GRAPHQL_URL, SU_ROUTER_URL, HB_ROUTER_URL, ENABLE_HB_WALLET_CHECK, logger: sendDataItemLogger }),
     isHyperBeamProcess: gatewayClient.isHyperBeamProcessWith({ fetch, GRAPHQL_URL, logger: sendDataItemLogger, getIsHyperBeamProcess, setIsHyperBeamProcess }),
     logger: sendDataItemLogger,
     writeDataItemArweave: uploaderClient.uploadDataItemWith({ UPLOADER_URL, logger: sendDataItemLogger, fetch, histogram }),
@@ -388,6 +389,7 @@ export const createResultApis = async (ctx) => {
   const RELAY_MAP = ctx.RELAY_MAP
   const SU_ROUTER_URL = ctx.SU_ROUTER_URL
   const HB_ROUTER_URL = ctx.HB_ROUTER_URL
+  const ENABLE_HB_WALLET_CHECK = ctx.ENABLE_HB_WALLET_CHECK
 
   const logger = ctx.logger
   const fetch = ctx.fetch
@@ -427,7 +429,7 @@ export const createResultApis = async (ctx) => {
     buildAndSign: signerClient.buildAndSignWith({ MU_WALLET, logger: processMsgLogger }),
     fetchResult: cuClient.resultWith({ fetch: fetchWithCache, histogram, CU_URL, logger: processMsgLogger }),
     fetchHyperBeamResult: cuClient.fetchHyperBeamResultWith({ fetch, HB_URL, histogram, logger: processMsgLogger }),
-    isWallet: gatewayClient.isWalletWith({ fetch, histogram, ARWEAVE_URL, GRAPHQL_URL, SU_ROUTER_URL, HB_ROUTER_URL, logger: processMsgLogger, setById, getById }),
+    isWallet: gatewayClient.isWalletWith({ fetch, histogram, ARWEAVE_URL, GRAPHQL_URL, SU_ROUTER_URL, HB_ROUTER_URL, ENABLE_HB_WALLET_CHECK, logger: processMsgLogger, setById, getById }),
     writeDataItemArweave: uploaderClient.uploadDataItemWith({ UPLOADER_URL, logger: processMsgLogger, fetch, histogram }),
     isHyperBeamProcess: gatewayClient.isHyperBeamProcessWith({ fetch, GRAPHQL_URL, logger: processMsgLogger, getIsHyperBeamProcess, setIsHyperBeamProcess }),
     RELAY_MAP,