(fix): retry when iota is unavailable

2026-07-29 02:37:18 +02:00 · 2026-07-29 02:37:18 +02:00 · 5ece90ef4b
commit 5ece90ef4b
parent 3666902ac9
3 changed files with 23 additions and 13 deletions
--- a/packages/mtp/src/context.tsx
+++ b/packages/mtp/src/context.tsx
@ -435,15 +435,21 @@ export function Provider(props: {

        const stateSync = new Promise<ProtocolMessage<"ClientStateSync">>(
          (resolve, reject) => {
+            let unsubscribeStateSync = () => {};
+            let unsubscribeNoIota = () => {};
+            const cleanupStateSync = () => {
+              clearTimeout(timeout);
+              unsubscribeStateSync();
+              unsubscribeNoIota();
+            };
            const timeout = setTimeout(() => {
-              unsubscribe();
+              cleanupStateSync();
              reject(new Error("Initial state synchronization timed out"));
            }, 120_000);
-            const unsubscribe = activeClient.subscribe(
+            unsubscribeStateSync = activeClient.subscribe(
              "ClientStateSync",
              (message) => {
-                clearTimeout(timeout);
-                unsubscribe();
+                cleanupStateSync();
                try {
                  resolve(validateResponse("ClientStateSync", message));
                } catch (error) {
@ -451,6 +457,10 @@ export function Provider(props: {
                }
              },
            );
+            unsubscribeNoIota = activeClient.subscribe("ErrorNoIota", () => {
+              cleanupStateSync();
+              reject(new Error("No Iota is currently connected"));
+            });
          },
        );
        const [, finalResponse] = await Promise.all([