@polkadot-api/ws-provider 0.3.4 → 0.3.5

package/dist/node/esm/follow-enhancer.mjs

@@ -0,0 +1,66 @@
+const methods = {};
+["v1", "unstable"].forEach((version) => {
+  methods[`chainHead_${version}_follow`] = "follow";
+  methods[`chainHead_${version}_unfollow`] = "unfollow";
+});
+const followEnhancer = (base, forceDisconnect) => {
+  const prematureStops = /* @__PURE__ */ new Set();
+  const preOpId = /* @__PURE__ */ new Set();
+  const onGoing = /* @__PURE__ */ new Set();
+  const result = (onMsg) => {
+    const { send, disconnect } = base((fromProvider) => {
+      const parsed = JSON.parse(fromProvider);
+      if ("id" in parsed) {
+        const { id, result: result2 } = parsed;
+        if (preOpId.has(id)) {
+          preOpId.delete(id);
+          if (prematureStops.has(result2)) {
+            prematureStops.delete(result2);
+            return;
+          }
+          onGoing.add(result2);
+          const currentSize = onGoing.size + preOpId.size;
+          if (currentSize > 2)
+            console.warn(
+              `Too many chainHead follow subscriptions (${currentSize})`
+            );
+          else if (parsed.error) {
+            console.warn(`chainHead follow failed on the ${currentSize} sub`);
+            Promise.resolve().then(forceDisconnect);
+            return;
+          }
+        }
+      } else {
+        const { subscription, result: result2 } = parsed.params;
+        if (result2?.event === "stop") {
+          if (onGoing.has(subscription)) onGoing.delete(subscription);
+          else prematureStops.add(subscription);
+        }
+      }
+      onMsg(fromProvider);
+    });
+    return {
+      send(toProvider) {
+        const parsed = JSON.parse(toProvider);
+        const method = methods[parsed.method];
+        if (method === "follow") {
+          preOpId.add(parsed.id);
+        } else if (method === "unfollow") {
+          onGoing.delete(parsed.params[0]);
+        }
+        send(toProvider);
+      },
+      disconnect
+    };
+  };
+  return Object.assign(result, {
+    cleanup: () => {
+      prematureStops.clear();
+      preOpId.clear();
+      onGoing.clear();
+    }
+  });
+};
+
+export { followEnhancer };
+//# sourceMappingURL=follow-enhancer.mjs.map

package/dist/node/esm/follow-enhancer.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"follow-enhancer.mjs","sources":["../../../src/follow-enhancer.ts"],"sourcesContent":["import { JsonRpcProvider } from \"@polkadot-api/json-rpc-provider\"\n\nconst methods: Record<string, \"follow\" | \"unfollow\"> = {}\n;[\"v1\", \"unstable\"].forEach((version) => {\n  methods[`chainHead_${version}_follow`] = \"follow\"\n  methods[`chainHead_${version}_unfollow`] = \"unfollow\"\n})\n\nexport const followEnhancer: (\n  input: JsonRpcProvider,\n  forceDisconnect: () => void,\n) => JsonRpcProvider & {\n  cleanup: () => void\n} = (base, forceDisconnect) => {\n  const prematureStops = new Set<string>()\n  const preOpId = new Set<string>()\n  const onGoing = new Set<string>()\n\n  const result: JsonRpcProvider = (onMsg) => {\n    const { send, disconnect } = base((fromProvider) => {\n      const parsed = JSON.parse(fromProvider)\n      // it's a response\n      if (\"id\" in parsed) {\n        const { id, result } = parsed\n        if (preOpId.has(id)) {\n          preOpId.delete(id)\n          if (prematureStops.has(result)) {\n            prematureStops.delete(result)\n            return\n          }\n\n          onGoing.add(result)\n          const currentSize = onGoing.size + preOpId.size\n          if (currentSize > 2)\n            console.warn(\n              `Too many chainHead follow subscriptions (${currentSize})`,\n            )\n          else if (parsed.error) {\n            console.warn(`chainHead follow failed on the ${currentSize} sub`)\n            Promise.resolve().then(forceDisconnect)\n            return\n          }\n        }\n      } else {\n        // it's a notifiaction\n        const { subscription, result } = (parsed as any).params\n        if (result?.event === \"stop\") {\n          if (onGoing.has(subscription)) onGoing.delete(subscription)\n          else prematureStops.add(subscription)\n        }\n      }\n      onMsg(fromProvider)\n    })\n\n    return {\n      send(toProvider) {\n        const parsed = JSON.parse(toProvider)\n        const method = methods[parsed.method]\n        if (method === \"follow\") {\n          preOpId.add(parsed.id)\n        } else if (method === \"unfollow\") {\n          onGoing.delete(parsed.params[0])\n        }\n        send(toProvider)\n      },\n      disconnect,\n    }\n  }\n\n  return Object.assign(result, {\n    cleanup: () => {\n      prematureStops.clear()\n      preOpId.clear()\n      onGoing.clear()\n    },\n  })\n}\n"],"names":["result"],"mappings":"AAEA,MAAM,UAAiD,EAAC;AACvD,CAAC,IAAM,EAAA,UAAU,CAAE,CAAA,OAAA,CAAQ,CAAC,OAAY,KAAA;AACvC,EAAQ,OAAA,CAAA,CAAA,UAAA,EAAa,OAAO,CAAA,OAAA,CAAS,CAAI,GAAA,QAAA;AACzC,EAAQ,OAAA,CAAA,CAAA,UAAA,EAAa,OAAO,CAAA,SAAA,CAAW,CAAI,GAAA,UAAA;AAC7C,CAAC,CAAA;AAEY,MAAA,cAAA,GAKT,CAAC,IAAA,EAAM,eAAoB,KAAA;AAC7B,EAAM,MAAA,cAAA,uBAAqB,GAAY,EAAA;AACvC,EAAM,MAAA,OAAA,uBAAc,GAAY,EAAA;AAChC,EAAM,MAAA,OAAA,uBAAc,GAAY,EAAA;AAEhC,EAAM,MAAA,MAAA,GAA0B,CAAC,KAAU,KAAA;AACzC,IAAA,MAAM,EAAE,IAAM,EAAA,UAAA,EAAe,GAAA,IAAA,CAAK,CAAC,YAAiB,KAAA;AAClD,MAAM,MAAA,MAAA,GAAS,IAAK,CAAA,KAAA,CAAM,YAAY,CAAA;AAEtC,MAAA,IAAI,QAAQ,MAAQ,EAAA;AAClB,QAAA,MAAM,EAAE,EAAA,EAAI,MAAAA,EAAAA,OAAAA,EAAW,GAAA,MAAA;AACvB,QAAI,IAAA,OAAA,CAAQ,GAAI,CAAA,EAAE,CAAG,EAAA;AACnB,UAAA,OAAA,CAAQ,OAAO,EAAE,CAAA;AACjB,UAAI,IAAA,cAAA,CAAe,GAAIA,CAAAA,OAAM,CAAG,EAAA;AAC9B,YAAA,cAAA,CAAe,OAAOA,OAAM,CAAA;AAC5B,YAAA;AAAA;AAGF,UAAA,OAAA,CAAQ,IAAIA,OAAM,CAAA;AAClB,UAAM,MAAA,WAAA,GAAc,OAAQ,CAAA,IAAA,GAAO,OAAQ,CAAA,IAAA;AAC3C,UAAA,IAAI,WAAc,GAAA,CAAA;AAChB,YAAQ,OAAA,CAAA,IAAA;AAAA,cACN,4CAA4C,WAAW,CAAA,CAAA;AAAA,aACzD;AAAA,eAAA,IACO,OAAO,KAAO,EAAA;AACrB,YAAQ,OAAA,CAAA,IAAA,CAAK,CAAkC,+BAAA,EAAA,WAAW,CAAM,IAAA,CAAA,CAAA;AAChE,YAAQ,OAAA,CAAA,OAAA,EAAU,CAAA,IAAA,CAAK,eAAe,CAAA;AACtC,YAAA;AAAA;AACF;AACF,OACK,MAAA;AAEL,QAAA,MAAM,EAAE,YAAA,EAAc,MAAAA,EAAAA,OAAAA,KAAY,MAAe,CAAA,MAAA;AACjD,QAAIA,IAAAA,OAAAA,EAAQ,UAAU,MAAQ,EAAA;AAC5B,UAAA,IAAI,QAAQ,GAAI,CAAA,YAAY,CAAG,EAAA,OAAA,CAAQ,OAAO,YAAY,CAAA;AAAA,eACrD,cAAA,CAAe,IAAI,YAAY,CAAA;AAAA;AACtC;AAEF,MAAA,KAAA,CAAM,YAAY,CAAA;AAAA,KACnB,CAAA;AAED,IAAO,OAAA;AAAA,MACL,KAAK,UAAY,EAAA;AACf,QAAM,MAAA,MAAA,GAAS,IAAK,CAAA,KAAA,CAAM,UAAU,CAAA;AACpC,QAAM,MAAA,MAAA,GAAS,OAAQ,CAAA,MAAA,CAAO,MAAM,CAAA;AACpC,QAAA,IAAI,WAAW,QAAU,EAAA;AACvB,UAAQ,OAAA,CAAA,GAAA,CAAI,OAAO,EAAE,CAAA;AAAA,SACvB,MAAA,IAAW,WAAW,UAAY,EAAA;AAChC,UAAA,OAAA,CAAQ,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,CAAC,CAAC,CAAA;AAAA;AAEjC,QAAA,IAAA,CAAK,UAAU,CAAA;AAAA,OACjB;AAAA,MACA;AAAA,KACF;AAAA,GACF;AAEA,EAAO,OAAA,MAAA,CAAO,OAAO,MAAQ,EAAA;AAAA,IAC3B,SAAS,MAAM;AACb,MAAA,cAAA,CAAe,KAAM,EAAA;AACrB,MAAA,OAAA,CAAQ,KAAM,EAAA;AACd,MAAA,OAAA,CAAQ,KAAM,EAAA;AAAA;AAChB,GACD,CAAA;AACH;;;;"}

package/dist/node/esm/ws-provider.mjs

@@ -1,5 +1,6 @@
 import { getSyncProvider } from '@polkadot-api/json-rpc-provider-proxy';
 import { WsEvent } from './types.mjs';
+import { followEnhancer } from './follow-enhancer.mjs';
 
 const timeoutError = {
   type: WsEvent.ERROR,
@@ -33,109 +34,117 @@ const getInternalWsProvider = (WebsocketClass) => {
     let status;
     let switchTo = null;
     let disconnect = noop;
-    const result = getSyncProvider(async () => {
-      const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length];
-      switchTo = null;
-      const socket = new WebsocketClass(uri, protocols);
-      const forceSocketClose = () => {
-        try {
-          socket.addEventListener("error", noop, { once: true });
-          socket.close();
-        } catch {
-        }
-      };
-      onStatusChanged(
-        status = {
-          type: WsEvent.CONNECTING,
-          uri,
-          protocols
-        }
-      );
-      await new Promise((resolve, reject) => {
-        const onOpen = () => {
-          initialCleanup();
-          resolve();
+    let outerCleanup = noop;
+    const result = followEnhancer(
+      getSyncProvider(async () => {
+        const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length];
+        switchTo = null;
+        const socket = new WebsocketClass(uri, protocols);
+        const forceSocketClose = () => {
+          try {
+            socket.addEventListener("error", noop, { once: true });
+            socket.close();
+          } catch {
+          }
         };
-        const onError = (e) => {
-          initialCleanup();
-          if (e == null) forceSocketClose();
-          console.error(
-            `Unable to connect to ${uri}${protocols ? ", protocols: " + protocols : ""}`
-          );
-          onStatusChanged(
-            status = {
-              type: e ? WsEvent.ERROR : WsEvent.CLOSE,
-              event: e
-            }
-          );
-          setTimeout(reject, e ? 300 : 0, e);
+        onStatusChanged(
+          status = {
+            type: WsEvent.CONNECTING,
+            uri,
+            protocols
+          }
+        );
+        await new Promise((resolve, reject) => {
+          const onOpen = () => {
+            initialCleanup();
+            resolve();
+          };
+          const onError = (e) => {
+            initialCleanup();
+            if (e == null) forceSocketClose();
+            console.error(
+              `Unable to connect to ${uri}${protocols ? ", protocols: " + protocols : ""}`
+            );
+            onStatusChanged(
+              status = {
+                type: e ? WsEvent.ERROR : WsEvent.CLOSE,
+                event: e
+              }
+            );
+            setTimeout(reject, e ? 300 : 0, e);
+          };
+          const timeoutToken = timeout !== Infinity ? setTimeout(() => {
+            initialCleanup();
+            forceSocketClose();
+            onStatusChanged(status = timeoutError);
+            reject(timeoutError.event);
+          }, timeout) : void 0;
+          const initialCleanup = () => {
+            clearTimeout(timeoutToken);
+            socket.removeEventListener("error", onError);
+            socket.removeEventListener("open", onOpen);
+          };
+          socket.addEventListener("open", onOpen);
+          socket.addEventListener("error", onError);
+          disconnect = () => {
+            onError(null);
+          };
+        });
+        onStatusChanged(
+          status = {
+            type: WsEvent.CONNECTED,
+            uri,
+            protocols
+          }
+        );
+        return (onMessage, onHalt) => {
+          const _onMessage = (e) => {
+            if (typeof e.data === "string") onMessage(e.data);
+          };
+          const innerHalt = (reason) => (e) => {
+            console.warn(`WS halt (${reason})`);
+            onStatusChanged(
+              status = {
+                type: reason,
+                event: e
+              }
+            );
+            onHalt();
+          };
+          const onError = innerHalt(WsEvent.ERROR);
+          const onClose = innerHalt(WsEvent.CLOSE);
+          socket.addEventListener("message", _onMessage);
+          socket.addEventListener("error", onError);
+          socket.addEventListener("close", onClose);
+          disconnect = (withHalt) => {
+            outerCleanup();
+            disconnect = noop;
+            socket.removeEventListener("message", _onMessage);
+            socket.removeEventListener("error", onError);
+            socket.removeEventListener("close", onClose);
+            forceSocketClose();
+            if (withHalt) onClose({});
+          };
+          return {
+            send: (msg) => {
+              socket.send(msg);
+            },
+            disconnect
+          };
         };
-        const timeoutToken = timeout !== Infinity ? setTimeout(() => {
-          initialCleanup();
-          forceSocketClose();
-          onStatusChanged(status = timeoutError);
-          reject(timeoutError.event);
-        }, timeout) : void 0;
-        const initialCleanup = () => {
-          clearTimeout(timeoutToken);
-          socket.removeEventListener("error", onError);
-          socket.removeEventListener("open", onOpen);
-        };
-        socket.addEventListener("open", onOpen);
-        socket.addEventListener("error", onError);
-        disconnect = () => {
-          onError(null);
-        };
-      });
-      onStatusChanged(
-        status = {
-          type: WsEvent.CONNECTED,
-          uri,
-          protocols
-        }
-      );
-      return (onMessage, onHalt) => {
-        const _onMessage = (e) => {
-          if (typeof e.data === "string") onMessage(e.data);
-        };
-        const innerHalt = (reason) => (e) => {
-          console.warn(`WS halt (${reason})`);
-          onStatusChanged(
-            status = {
-              type: reason,
-              event: e
-            }
-          );
-          onHalt();
-        };
-        const onError = innerHalt(WsEvent.ERROR);
-        const onClose = innerHalt(WsEvent.CLOSE);
-        socket.addEventListener("message", _onMessage);
-        socket.addEventListener("error", onError);
-        socket.addEventListener("close", onClose);
-        disconnect = (withHalt) => {
-          disconnect = noop;
-          socket.removeEventListener("message", _onMessage);
-          socket.removeEventListener("error", onError);
-          socket.removeEventListener("close", onClose);
-          forceSocketClose();
-          if (withHalt) onClose({});
-        };
-        return {
-          send: (msg) => {
-            socket.send(msg);
-          },
-          disconnect
-        };
-      };
-    });
-    result.getStatus = () => status;
-    result.switch = (...args2) => {
+      }),
+      () => {
+        switchFn();
+      }
+    );
+    outerCleanup = result.cleanup;
+    delete result.cleanup;
+    const switchFn = (...args2) => {
       if (status.type === WsEvent.CLOSE) return;
       if (args2.length) switchTo = args2;
       if (status.type !== WsEvent.ERROR) disconnect(true);
     };
-    return result;
+    return Object.assign(result, { switch: switchFn, getStatus: () => status });
   };
 };
 

package/dist/node/esm/ws-provider.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"ws-provider.mjs","sources":["../../../src/ws-provider.ts"],"sourcesContent":["import { getSyncProvider } from \"@polkadot-api/json-rpc-provider-proxy\"\nimport {\n  GetWsProviderInput,\n  StatusChange,\n  WsJsonRpcProvider,\n  WsEvent,\n  WsProviderConfig,\n} from \"./types\"\n\nconst timeoutError: StatusChange = {\n  type: WsEvent.ERROR,\n  event: { type: \"timeout\" },\n}\n\nconst noop = () => {}\n\nconst mapEndpoints = (\n  endpoints: WsProviderConfig[\"endpoints\"],\n): Array<[string, string | string[]] | [string]> =>\n  endpoints.map((x) => (typeof x === \"string\" ? [x] : [x.uri, x.protocol]))\n\nexport const getInternalWsProvider = (\n  WebsocketClass: typeof WebSocket,\n): GetWsProviderInput => {\n  return (...args): WsJsonRpcProvider => {\n    let endpoints: Array<[string, string | string[]] | [string]> = []\n    let onStatusChanged: (status: StatusChange) => void = noop\n    let timeout = 3_500\n\n    const [firstArg] = args\n    if (\n      args.length === 1 &&\n      typeof firstArg === \"object\" &&\n      !Array.isArray(firstArg)\n    ) {\n      endpoints = mapEndpoints(firstArg.endpoints)\n      onStatusChanged = firstArg.onStatusChanged ?? noop\n      timeout = firstArg.timeout ?? timeout\n    } else {\n      if (typeof args[1] === \"function\")\n        onStatusChanged = args[1] as (status: StatusChange) => void\n      if (Array.isArray(firstArg)) endpoints = mapEndpoints(firstArg)\n      else {\n        endpoints = [[firstArg as string]]\n        if (args[1] && args[1] !== onStatusChanged)\n          endpoints[0][1] = args[1] as any\n        if (args[2]) onStatusChanged = args[2] as any\n      }\n    }\n\n    let idx = 0\n    let status: StatusChange\n    let switchTo: [string] | [string, string | string[]] | null = null\n    let disconnect: (withHalt?: boolean) => void = noop\n\n    const result = getSyncProvider(async () => {\n      const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length]\n      switchTo = null\n      const socket = new WebsocketClass(uri, protocols)\n      const forceSocketClose = () => {\n        try {\n          socket.addEventListener(\"error\", noop, { once: true })\n          socket.close()\n        } catch {}\n      }\n      onStatusChanged(\n        (status = {\n          type: WsEvent.CONNECTING,\n          uri,\n          protocols,\n        }),\n      )\n\n      await new Promise<void>((resolve, reject) => {\n        const onOpen = () => {\n          initialCleanup()\n          resolve()\n        }\n\n        const onError = (e: Event | null) => {\n          initialCleanup()\n          if (e == null) forceSocketClose()\n          console.error(\n            `Unable to connect to ${uri}${\n              protocols ? \", protocols: \" + protocols : \"\"\n            }`,\n          )\n          onStatusChanged(\n            (status = {\n              type: e ? WsEvent.ERROR : WsEvent.CLOSE,\n              event: e,\n            }),\n          )\n          setTimeout(reject, e ? 300 : 0, e)\n        }\n\n        const timeoutToken =\n          timeout !== Infinity\n            ? setTimeout(() => {\n                initialCleanup()\n                forceSocketClose()\n                onStatusChanged((status = timeoutError))\n                reject(timeoutError.event)\n              }, timeout)\n            : undefined\n\n        const initialCleanup = () => {\n          clearTimeout(timeoutToken)\n          socket.removeEventListener(\"error\", onError)\n          socket.removeEventListener(\"open\", onOpen)\n        }\n        socket.addEventListener(\"open\", onOpen)\n        socket.addEventListener(\"error\", onError)\n        disconnect = () => {\n          onError(null)\n        }\n      })\n\n      onStatusChanged(\n        (status = {\n          type: WsEvent.CONNECTED,\n          uri,\n          protocols,\n        }),\n      )\n\n      return (onMessage, onHalt) => {\n        const _onMessage = (e: MessageEvent) => {\n          if (typeof e.data === \"string\") onMessage(e.data)\n        }\n        const innerHalt =\n          (reason: WsEvent.CLOSE | WsEvent.ERROR) => (e: any) => {\n            console.warn(`WS halt (${reason})`)\n            onStatusChanged(\n              (status = {\n                type: reason,\n                event: e,\n              }),\n            )\n            onHalt()\n          }\n        const onError = innerHalt(WsEvent.ERROR)\n        const onClose = innerHalt(WsEvent.CLOSE)\n\n        socket.addEventListener(\"message\", _onMessage)\n        socket.addEventListener(\"error\", onError)\n        socket.addEventListener(\"close\", onClose)\n        disconnect = (withHalt) => {\n          disconnect = noop\n          socket.removeEventListener(\"message\", _onMessage)\n          socket.removeEventListener(\"error\", onError)\n          socket.removeEventListener(\"close\", onClose)\n          forceSocketClose()\n          if (withHalt) onClose({})\n        }\n\n        return {\n          send: (msg) => {\n            socket.send(msg)\n          },\n          disconnect,\n        }\n      }\n    }) as WsJsonRpcProvider\n\n    result.getStatus = () => status\n    result.switch = (...args) => {\n      if (status.type === WsEvent.CLOSE) return\n      if (args.length) switchTo = args as any\n      if (status.type !== WsEvent.ERROR) disconnect(true)\n    }\n    return result\n  }\n}\n"],"names":["args"],"mappings":";;;AASA,MAAM,YAA6B,GAAA;AAAA,EACjC,MAAM,OAAQ,CAAA,KAAA;AAAA,EACd,KAAA,EAAO,EAAE,IAAA,EAAM,SAAU;AAC3B,CAAA;AAEA,MAAM,OAAO,MAAM;AAAC,CAAA;AAEpB,MAAM,eAAe,CACnB,SAAA,KAEA,UAAU,GAAI,CAAA,CAAC,MAAO,OAAO,CAAA,KAAM,QAAW,GAAA,CAAC,CAAC,CAAI,GAAA,CAAC,EAAE,GAAK,EAAA,CAAA,CAAE,QAAQ,CAAE,CAAA;AAE7D,MAAA,qBAAA,GAAwB,CACnC,cACuB,KAAA;AACvB,EAAA,OAAO,IAAI,IAA4B,KAAA;AACrC,IAAA,IAAI,YAA2D,EAAC;AAChE,IAAA,IAAI,eAAkD,GAAA,IAAA;AACtD,IAAA,IAAI,OAAU,GAAA,IAAA;AAEd,IAAM,MAAA,CAAC,QAAQ,CAAI,GAAA,IAAA;AACnB,IACE,IAAA,IAAA,CAAK,MAAW,KAAA,CAAA,IAChB,OAAO,QAAA,KAAa,YACpB,CAAC,KAAA,CAAM,OAAQ,CAAA,QAAQ,CACvB,EAAA;AACA,MAAY,SAAA,GAAA,YAAA,CAAa,SAAS,SAAS,CAAA;AAC3C,MAAA,eAAA,GAAkB,SAAS,eAAmB,IAAA,IAAA;AAC9C,MAAA,OAAA,GAAU,SAAS,OAAW,IAAA,OAAA;AAAA,KACzB,MAAA;AACL,MAAI,IAAA,OAAO,IAAK,CAAA,CAAC,CAAM,KAAA,UAAA;AACrB,QAAA,eAAA,GAAkB,KAAK,CAAC,CAAA;AAC1B,MAAA,IAAI,MAAM,OAAQ,CAAA,QAAQ,CAAG,EAAA,SAAA,GAAY,aAAa,QAAQ,CAAA;AAAA,WACzD;AACH,QAAY,SAAA,GAAA,CAAC,CAAC,QAAkB,CAAC,CAAA;AACjC,QAAA,IAAI,IAAK,CAAA,CAAC,CAAK,IAAA,IAAA,CAAK,CAAC,CAAM,KAAA,eAAA;AACzB,UAAA,SAAA,CAAU,CAAC,CAAA,CAAE,CAAC,CAAA,GAAI,KAAK,CAAC,CAAA;AAC1B,QAAA,IAAI,IAAK,CAAA,CAAC,CAAG,EAAA,eAAA,GAAkB,KAAK,CAAC,CAAA;AAAA;AACvC;AAGF,IAAA,IAAI,GAAM,GAAA,CAAA;AACV,IAAI,IAAA,MAAA;AACJ,IAAA,IAAI,QAA0D,GAAA,IAAA;AAC9D,IAAA,IAAI,UAA2C,GAAA,IAAA;AAE/C,IAAM,MAAA,MAAA,GAAS,gBAAgB,YAAY;AACzC,MAAM,MAAA,CAAC,KAAK,SAAS,CAAA,GAAI,YAAY,SAAU,CAAA,GAAA,EAAA,GAAQ,UAAU,MAAM,CAAA;AACvE,MAAW,QAAA,GAAA,IAAA;AACX,MAAA,MAAM,MAAS,GAAA,IAAI,cAAe,CAAA,GAAA,EAAK,SAAS,CAAA;AAChD,MAAA,MAAM,mBAAmB,MAAM;AAC7B,QAAI,IAAA;AACF,UAAA,MAAA,CAAO,iBAAiB,OAAS,EAAA,IAAA,EAAM,EAAE,IAAA,EAAM,MAAM,CAAA;AACrD,UAAA,MAAA,CAAO,KAAM,EAAA;AAAA,SACP,CAAA,MAAA;AAAA;AAAC,OACX;AACA,MAAA,eAAA;AAAA,QACG,MAAS,GAAA;AAAA,UACR,MAAM,OAAQ,CAAA,UAAA;AAAA,UACd,GAAA;AAAA,UACA;AAAA;AACF,OACF;AAEA,MAAA,MAAM,IAAI,OAAA,CAAc,CAAC,OAAA,EAAS,MAAW,KAAA;AAC3C,QAAA,MAAM,SAAS,MAAM;AACnB,UAAe,cAAA,EAAA;AACf,UAAQ,OAAA,EAAA;AAAA,SACV;AAEA,QAAM,MAAA,OAAA,GAAU,CAAC,CAAoB,KAAA;AACnC,UAAe,cAAA,EAAA;AACf,UAAI,IAAA,CAAA,IAAK,MAAuB,gBAAA,EAAA;AAChC,UAAQ,OAAA,CAAA,KAAA;AAAA,YACN,wBAAwB,GAAG,CAAA,EACzB,SAAY,GAAA,eAAA,GAAkB,YAAY,EAC5C,CAAA;AAAA,WACF;AACA,UAAA,eAAA;AAAA,YACG,MAAS,GAAA;AAAA,cACR,IAAM,EAAA,CAAA,GAAI,OAAQ,CAAA,KAAA,GAAQ,OAAQ,CAAA,KAAA;AAAA,cAClC,KAAO,EAAA;AAAA;AACT,WACF;AACA,UAAA,UAAA,CAAW,MAAQ,EAAA,CAAA,GAAI,GAAM,GAAA,CAAA,EAAG,CAAC,CAAA;AAAA,SACnC;AAEA,QAAA,MAAM,YACJ,GAAA,OAAA,KAAY,QACR,GAAA,UAAA,CAAW,MAAM;AACf,UAAe,cAAA,EAAA;AACf,UAAiB,gBAAA,EAAA;AACjB,UAAA,eAAA,CAAiB,SAAS,YAAa,CAAA;AACvC,UAAA,MAAA,CAAO,aAAa,KAAK,CAAA;AAAA,SAC3B,EAAG,OAAO,CACV,GAAA,KAAA,CAAA;AAEN,QAAA,MAAM,iBAAiB,MAAM;AAC3B,UAAA,YAAA,CAAa,YAAY,CAAA;AACzB,UAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,UAAO,MAAA,CAAA,mBAAA,CAAoB,QAAQ,MAAM,CAAA;AAAA,SAC3C;AACA,QAAO,MAAA,CAAA,gBAAA,CAAiB,QAAQ,MAAM,CAAA;AACtC,QAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,QAAA,UAAA,GAAa,MAAM;AACjB,UAAA,OAAA,CAAQ,IAAI,CAAA;AAAA,SACd;AAAA,OACD,CAAA;AAED,MAAA,eAAA;AAAA,QACG,MAAS,GAAA;AAAA,UACR,MAAM,OAAQ,CAAA,SAAA;AAAA,UACd,GAAA;AAAA,UACA;AAAA;AACF,OACF;AAEA,MAAO,OAAA,CAAC,WAAW,MAAW,KAAA;AAC5B,QAAM,MAAA,UAAA,GAAa,CAAC,CAAoB,KAAA;AACtC,UAAA,IAAI,OAAO,CAAE,CAAA,IAAA,KAAS,QAAU,EAAA,SAAA,CAAU,EAAE,IAAI,CAAA;AAAA,SAClD;AACA,QAAA,MAAM,SACJ,GAAA,CAAC,MAA0C,KAAA,CAAC,CAAW,KAAA;AACrD,UAAQ,OAAA,CAAA,IAAA,CAAK,CAAY,SAAA,EAAA,MAAM,CAAG,CAAA,CAAA,CAAA;AAClC,UAAA,eAAA;AAAA,YACG,MAAS,GAAA;AAAA,cACR,IAAM,EAAA,MAAA;AAAA,cACN,KAAO,EAAA;AAAA;AACT,WACF;AACA,UAAO,MAAA,EAAA;AAAA,SACT;AACF,QAAM,MAAA,OAAA,GAAU,SAAU,CAAA,OAAA,CAAQ,KAAK,CAAA;AACvC,QAAM,MAAA,OAAA,GAAU,SAAU,CAAA,OAAA,CAAQ,KAAK,CAAA;AAEvC,QAAO,MAAA,CAAA,gBAAA,CAAiB,WAAW,UAAU,CAAA;AAC7C,QAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,QAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,QAAA,UAAA,GAAa,CAAC,QAAa,KAAA;AACzB,UAAa,UAAA,GAAA,IAAA;AACb,UAAO,MAAA,CAAA,mBAAA,CAAoB,WAAW,UAAU,CAAA;AAChD,UAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,UAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,UAAiB,gBAAA,EAAA;AACjB,UAAI,IAAA,QAAA,EAAkB,OAAA,CAAA,EAAE,CAAA;AAAA,SAC1B;AAEA,QAAO,OAAA;AAAA,UACL,IAAA,EAAM,CAAC,GAAQ,KAAA;AACb,YAAA,MAAA,CAAO,KAAK,GAAG,CAAA;AAAA,WACjB;AAAA,UACA;AAAA,SACF;AAAA,OACF;AAAA,KACD,CAAA;AAED,IAAA,MAAA,CAAO,YAAY,MAAM,MAAA;AACzB,IAAO,MAAA,CAAA,MAAA,GAAS,IAAIA,KAAS,KAAA;AAC3B,MAAI,IAAA,MAAA,CAAO,IAAS,KAAA,OAAA,CAAQ,KAAO,EAAA;AACnC,MAAIA,IAAAA,KAAAA,CAAK,QAAmBA,QAAAA,GAAAA,KAAAA;AAC5B,MAAA,IAAI,MAAO,CAAA,IAAA,KAAS,OAAQ,CAAA,KAAA,aAAkB,IAAI,CAAA;AAAA,KACpD;AACA,IAAO,OAAA,MAAA;AAAA,GACT;AACF;;;;"}
+{"version":3,"file":"ws-provider.mjs","sources":["../../../src/ws-provider.ts"],"sourcesContent":["import { getSyncProvider } from \"@polkadot-api/json-rpc-provider-proxy\"\nimport {\n  GetWsProviderInput,\n  StatusChange,\n  WsJsonRpcProvider,\n  WsEvent,\n  WsProviderConfig,\n} from \"./types\"\nimport { followEnhancer } from \"./follow-enhancer\"\n\nconst timeoutError: StatusChange = {\n  type: WsEvent.ERROR,\n  event: { type: \"timeout\" },\n}\n\nconst noop = () => {}\n\nconst mapEndpoints = (\n  endpoints: WsProviderConfig[\"endpoints\"],\n): Array<[string, string | string[]] | [string]> =>\n  endpoints.map((x) => (typeof x === \"string\" ? [x] : [x.uri, x.protocol]))\n\nexport const getInternalWsProvider = (\n  WebsocketClass: typeof WebSocket,\n): GetWsProviderInput => {\n  return (...args): WsJsonRpcProvider => {\n    let endpoints: Array<[string, string | string[]] | [string]> = []\n    let onStatusChanged: (status: StatusChange) => void = noop\n    let timeout = 3_500\n\n    const [firstArg] = args\n    if (\n      args.length === 1 &&\n      typeof firstArg === \"object\" &&\n      !Array.isArray(firstArg)\n    ) {\n      endpoints = mapEndpoints(firstArg.endpoints)\n      onStatusChanged = firstArg.onStatusChanged ?? noop\n      timeout = firstArg.timeout ?? timeout\n    } else {\n      if (typeof args[1] === \"function\")\n        onStatusChanged = args[1] as (status: StatusChange) => void\n      if (Array.isArray(firstArg)) endpoints = mapEndpoints(firstArg)\n      else {\n        endpoints = [[firstArg as string]]\n        if (args[1] && args[1] !== onStatusChanged)\n          endpoints[0][1] = args[1] as any\n        if (args[2]) onStatusChanged = args[2] as any\n      }\n    }\n\n    let idx = 0\n    let status: StatusChange\n    let switchTo: [string] | [string, string | string[]] | null = null\n    let disconnect: (withHalt?: boolean) => void = noop\n\n    let outerCleanup: () => void = noop\n    const result = followEnhancer(\n      getSyncProvider(async () => {\n        const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length]\n        switchTo = null\n        const socket = new WebsocketClass(uri, protocols)\n        const forceSocketClose = () => {\n          try {\n            socket.addEventListener(\"error\", noop, { once: true })\n            socket.close()\n          } catch {}\n        }\n        onStatusChanged(\n          (status = {\n            type: WsEvent.CONNECTING,\n            uri,\n            protocols,\n          }),\n        )\n\n        await new Promise<void>((resolve, reject) => {\n          const onOpen = () => {\n            initialCleanup()\n            resolve()\n          }\n\n          const onError = (e: Event | null) => {\n            initialCleanup()\n            if (e == null) forceSocketClose()\n            console.error(\n              `Unable to connect to ${uri}${\n                protocols ? \", protocols: \" + protocols : \"\"\n              }`,\n            )\n            onStatusChanged(\n              (status = {\n                type: e ? WsEvent.ERROR : WsEvent.CLOSE,\n                event: e,\n              }),\n            )\n            setTimeout(reject, e ? 300 : 0, e)\n          }\n\n          const timeoutToken =\n            timeout !== Infinity\n              ? setTimeout(() => {\n                  initialCleanup()\n                  forceSocketClose()\n                  onStatusChanged((status = timeoutError))\n                  reject(timeoutError.event)\n                }, timeout)\n              : undefined\n\n          const initialCleanup = () => {\n            clearTimeout(timeoutToken)\n            socket.removeEventListener(\"error\", onError)\n            socket.removeEventListener(\"open\", onOpen)\n          }\n          socket.addEventListener(\"open\", onOpen)\n          socket.addEventListener(\"error\", onError)\n          disconnect = () => {\n            onError(null)\n          }\n        })\n\n        onStatusChanged(\n          (status = {\n            type: WsEvent.CONNECTED,\n            uri,\n            protocols,\n          }),\n        )\n\n        return (onMessage, onHalt) => {\n          const _onMessage = (e: MessageEvent) => {\n            if (typeof e.data === \"string\") onMessage(e.data)\n          }\n          const innerHalt =\n            (reason: WsEvent.CLOSE | WsEvent.ERROR) => (e: any) => {\n              console.warn(`WS halt (${reason})`)\n              onStatusChanged(\n                (status = {\n                  type: reason,\n                  event: e,\n                }),\n              )\n              onHalt()\n            }\n          const onError = innerHalt(WsEvent.ERROR)\n          const onClose = innerHalt(WsEvent.CLOSE)\n\n          socket.addEventListener(\"message\", _onMessage)\n          socket.addEventListener(\"error\", onError)\n          socket.addEventListener(\"close\", onClose)\n          disconnect = (withHalt) => {\n            outerCleanup()\n            disconnect = noop\n            socket.removeEventListener(\"message\", _onMessage)\n            socket.removeEventListener(\"error\", onError)\n            socket.removeEventListener(\"close\", onClose)\n            forceSocketClose()\n            if (withHalt) onClose({})\n          }\n\n          return {\n            send: (msg) => {\n              socket.send(msg)\n            },\n            disconnect,\n          }\n        }\n      }),\n      () => {\n        switchFn()\n      },\n    )\n    outerCleanup = result.cleanup\n    delete (result as any).cleanup\n\n    const switchFn: WsJsonRpcProvider[\"switch\"] = (...args) => {\n      if (status.type === WsEvent.CLOSE) return\n      if (args.length) switchTo = args as any\n      if (status.type !== WsEvent.ERROR) disconnect(true)\n    }\n\n    return Object.assign(result, { switch: switchFn, getStatus: () => status })\n  }\n}\n"],"names":["args"],"mappings":";;;;AAUA,MAAM,YAA6B,GAAA;AAAA,EACjC,MAAM,OAAQ,CAAA,KAAA;AAAA,EACd,KAAA,EAAO,EAAE,IAAA,EAAM,SAAU;AAC3B,CAAA;AAEA,MAAM,OAAO,MAAM;AAAC,CAAA;AAEpB,MAAM,eAAe,CACnB,SAAA,KAEA,UAAU,GAAI,CAAA,CAAC,MAAO,OAAO,CAAA,KAAM,QAAW,GAAA,CAAC,CAAC,CAAI,GAAA,CAAC,EAAE,GAAK,EAAA,CAAA,CAAE,QAAQ,CAAE,CAAA;AAE7D,MAAA,qBAAA,GAAwB,CACnC,cACuB,KAAA;AACvB,EAAA,OAAO,IAAI,IAA4B,KAAA;AACrC,IAAA,IAAI,YAA2D,EAAC;AAChE,IAAA,IAAI,eAAkD,GAAA,IAAA;AACtD,IAAA,IAAI,OAAU,GAAA,IAAA;AAEd,IAAM,MAAA,CAAC,QAAQ,CAAI,GAAA,IAAA;AACnB,IACE,IAAA,IAAA,CAAK,MAAW,KAAA,CAAA,IAChB,OAAO,QAAA,KAAa,YACpB,CAAC,KAAA,CAAM,OAAQ,CAAA,QAAQ,CACvB,EAAA;AACA,MAAY,SAAA,GAAA,YAAA,CAAa,SAAS,SAAS,CAAA;AAC3C,MAAA,eAAA,GAAkB,SAAS,eAAmB,IAAA,IAAA;AAC9C,MAAA,OAAA,GAAU,SAAS,OAAW,IAAA,OAAA;AAAA,KACzB,MAAA;AACL,MAAI,IAAA,OAAO,IAAK,CAAA,CAAC,CAAM,KAAA,UAAA;AACrB,QAAA,eAAA,GAAkB,KAAK,CAAC,CAAA;AAC1B,MAAA,IAAI,MAAM,OAAQ,CAAA,QAAQ,CAAG,EAAA,SAAA,GAAY,aAAa,QAAQ,CAAA;AAAA,WACzD;AACH,QAAY,SAAA,GAAA,CAAC,CAAC,QAAkB,CAAC,CAAA;AACjC,QAAA,IAAI,IAAK,CAAA,CAAC,CAAK,IAAA,IAAA,CAAK,CAAC,CAAM,KAAA,eAAA;AACzB,UAAA,SAAA,CAAU,CAAC,CAAA,CAAE,CAAC,CAAA,GAAI,KAAK,CAAC,CAAA;AAC1B,QAAA,IAAI,IAAK,CAAA,CAAC,CAAG,EAAA,eAAA,GAAkB,KAAK,CAAC,CAAA;AAAA;AACvC;AAGF,IAAA,IAAI,GAAM,GAAA,CAAA;AACV,IAAI,IAAA,MAAA;AACJ,IAAA,IAAI,QAA0D,GAAA,IAAA;AAC9D,IAAA,IAAI,UAA2C,GAAA,IAAA;AAE/C,IAAA,IAAI,YAA2B,GAAA,IAAA;AAC/B,IAAA,MAAM,MAAS,GAAA,cAAA;AAAA,MACb,gBAAgB,YAAY;AAC1B,QAAM,MAAA,CAAC,KAAK,SAAS,CAAA,GAAI,YAAY,SAAU,CAAA,GAAA,EAAA,GAAQ,UAAU,MAAM,CAAA;AACvE,QAAW,QAAA,GAAA,IAAA;AACX,QAAA,MAAM,MAAS,GAAA,IAAI,cAAe,CAAA,GAAA,EAAK,SAAS,CAAA;AAChD,QAAA,MAAM,mBAAmB,MAAM;AAC7B,UAAI,IAAA;AACF,YAAA,MAAA,CAAO,iBAAiB,OAAS,EAAA,IAAA,EAAM,EAAE,IAAA,EAAM,MAAM,CAAA;AACrD,YAAA,MAAA,CAAO,KAAM,EAAA;AAAA,WACP,CAAA,MAAA;AAAA;AAAC,SACX;AACA,QAAA,eAAA;AAAA,UACG,MAAS,GAAA;AAAA,YACR,MAAM,OAAQ,CAAA,UAAA;AAAA,YACd,GAAA;AAAA,YACA;AAAA;AACF,SACF;AAEA,QAAA,MAAM,IAAI,OAAA,CAAc,CAAC,OAAA,EAAS,MAAW,KAAA;AAC3C,UAAA,MAAM,SAAS,MAAM;AACnB,YAAe,cAAA,EAAA;AACf,YAAQ,OAAA,EAAA;AAAA,WACV;AAEA,UAAM,MAAA,OAAA,GAAU,CAAC,CAAoB,KAAA;AACnC,YAAe,cAAA,EAAA;AACf,YAAI,IAAA,CAAA,IAAK,MAAuB,gBAAA,EAAA;AAChC,YAAQ,OAAA,CAAA,KAAA;AAAA,cACN,wBAAwB,GAAG,CAAA,EACzB,SAAY,GAAA,eAAA,GAAkB,YAAY,EAC5C,CAAA;AAAA,aACF;AACA,YAAA,eAAA;AAAA,cACG,MAAS,GAAA;AAAA,gBACR,IAAM,EAAA,CAAA,GAAI,OAAQ,CAAA,KAAA,GAAQ,OAAQ,CAAA,KAAA;AAAA,gBAClC,KAAO,EAAA;AAAA;AACT,aACF;AACA,YAAA,UAAA,CAAW,MAAQ,EAAA,CAAA,GAAI,GAAM,GAAA,CAAA,EAAG,CAAC,CAAA;AAAA,WACnC;AAEA,UAAA,MAAM,YACJ,GAAA,OAAA,KAAY,QACR,GAAA,UAAA,CAAW,MAAM;AACf,YAAe,cAAA,EAAA;AACf,YAAiB,gBAAA,EAAA;AACjB,YAAA,eAAA,CAAiB,SAAS,YAAa,CAAA;AACvC,YAAA,MAAA,CAAO,aAAa,KAAK,CAAA;AAAA,WAC3B,EAAG,OAAO,CACV,GAAA,KAAA,CAAA;AAEN,UAAA,MAAM,iBAAiB,MAAM;AAC3B,YAAA,YAAA,CAAa,YAAY,CAAA;AACzB,YAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,YAAO,MAAA,CAAA,mBAAA,CAAoB,QAAQ,MAAM,CAAA;AAAA,WAC3C;AACA,UAAO,MAAA,CAAA,gBAAA,CAAiB,QAAQ,MAAM,CAAA;AACtC,UAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,UAAA,UAAA,GAAa,MAAM;AACjB,YAAA,OAAA,CAAQ,IAAI,CAAA;AAAA,WACd;AAAA,SACD,CAAA;AAED,QAAA,eAAA;AAAA,UACG,MAAS,GAAA;AAAA,YACR,MAAM,OAAQ,CAAA,SAAA;AAAA,YACd,GAAA;AAAA,YACA;AAAA;AACF,SACF;AAEA,QAAO,OAAA,CAAC,WAAW,MAAW,KAAA;AAC5B,UAAM,MAAA,UAAA,GAAa,CAAC,CAAoB,KAAA;AACtC,YAAA,IAAI,OAAO,CAAE,CAAA,IAAA,KAAS,QAAU,EAAA,SAAA,CAAU,EAAE,IAAI,CAAA;AAAA,WAClD;AACA,UAAA,MAAM,SACJ,GAAA,CAAC,MAA0C,KAAA,CAAC,CAAW,KAAA;AACrD,YAAQ,OAAA,CAAA,IAAA,CAAK,CAAY,SAAA,EAAA,MAAM,CAAG,CAAA,CAAA,CAAA;AAClC,YAAA,eAAA;AAAA,cACG,MAAS,GAAA;AAAA,gBACR,IAAM,EAAA,MAAA;AAAA,gBACN,KAAO,EAAA;AAAA;AACT,aACF;AACA,YAAO,MAAA,EAAA;AAAA,WACT;AACF,UAAM,MAAA,OAAA,GAAU,SAAU,CAAA,OAAA,CAAQ,KAAK,CAAA;AACvC,UAAM,MAAA,OAAA,GAAU,SAAU,CAAA,OAAA,CAAQ,KAAK,CAAA;AAEvC,UAAO,MAAA,CAAA,gBAAA,CAAiB,WAAW,UAAU,CAAA;AAC7C,UAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,UAAO,MAAA,CAAA,gBAAA,CAAiB,SAAS,OAAO,CAAA;AACxC,UAAA,UAAA,GAAa,CAAC,QAAa,KAAA;AACzB,YAAa,YAAA,EAAA;AACb,YAAa,UAAA,GAAA,IAAA;AACb,YAAO,MAAA,CAAA,mBAAA,CAAoB,WAAW,UAAU,CAAA;AAChD,YAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,YAAO,MAAA,CAAA,mBAAA,CAAoB,SAAS,OAAO,CAAA;AAC3C,YAAiB,gBAAA,EAAA;AACjB,YAAI,IAAA,QAAA,EAAkB,OAAA,CAAA,EAAE,CAAA;AAAA,WAC1B;AAEA,UAAO,OAAA;AAAA,YACL,IAAA,EAAM,CAAC,GAAQ,KAAA;AACb,cAAA,MAAA,CAAO,KAAK,GAAG,CAAA;AAAA,aACjB;AAAA,YACA;AAAA,WACF;AAAA,SACF;AAAA,OACD,CAAA;AAAA,MACD,MAAM;AACJ,QAAS,QAAA,EAAA;AAAA;AACX,KACF;AACA,IAAA,YAAA,GAAe,MAAO,CAAA,OAAA;AACtB,IAAA,OAAQ,MAAe,CAAA,OAAA;AAEvB,IAAM,MAAA,QAAA,GAAwC,IAAIA,KAAS,KAAA;AACzD,MAAI,IAAA,MAAA,CAAO,IAAS,KAAA,OAAA,CAAQ,KAAO,EAAA;AACnC,MAAIA,IAAAA,KAAAA,CAAK,QAAmBA,QAAAA,GAAAA,KAAAA;AAC5B,MAAA,IAAI,MAAO,CAAA,IAAA,KAAS,OAAQ,CAAA,KAAA,aAAkB,IAAI,CAAA;AAAA,KACpD;AAEA,IAAO,OAAA,MAAA,CAAO,OAAO,MAAQ,EAAA,EAAE,QAAQ,QAAU,EAAA,SAAA,EAAW,MAAM,MAAA,EAAQ,CAAA;AAAA,GAC5E;AACF;;;;"}

package/dist/node/node.js

@@ -11,6 +11,70 @@ var WsEvent = /* @__PURE__ */ ((WsEvent2) => {
   return WsEvent2;
 })(WsEvent || {});
 
+const methods = {};
+["v1", "unstable"].forEach((version) => {
+  methods[`chainHead_${version}_follow`] = "follow";
+  methods[`chainHead_${version}_unfollow`] = "unfollow";
+});
+const followEnhancer = (base, forceDisconnect) => {
+  const prematureStops = /* @__PURE__ */ new Set();
+  const preOpId = /* @__PURE__ */ new Set();
+  const onGoing = /* @__PURE__ */ new Set();
+  const result = (onMsg) => {
+    const { send, disconnect } = base((fromProvider) => {
+      const parsed = JSON.parse(fromProvider);
+      if ("id" in parsed) {
+        const { id, result: result2 } = parsed;
+        if (preOpId.has(id)) {
+          preOpId.delete(id);
+          if (prematureStops.has(result2)) {
+            prematureStops.delete(result2);
+            return;
+          }
+          onGoing.add(result2);
+          const currentSize = onGoing.size + preOpId.size;
+          if (currentSize > 2)
+            console.warn(
+              `Too many chainHead follow subscriptions (${currentSize})`
+            );
+          else if (parsed.error) {
+            console.warn(`chainHead follow failed on the ${currentSize} sub`);
+            Promise.resolve().then(forceDisconnect);
+            return;
+          }
+        }
+      } else {
+        const { subscription, result: result2 } = parsed.params;
+        if (result2?.event === "stop") {
+          if (onGoing.has(subscription)) onGoing.delete(subscription);
+          else prematureStops.add(subscription);
+        }
+      }
+      onMsg(fromProvider);
+    });
+    return {
+      send(toProvider) {
+        const parsed = JSON.parse(toProvider);
+        const method = methods[parsed.method];
+        if (method === "follow") {
+          preOpId.add(parsed.id);
+        } else if (method === "unfollow") {
+          onGoing.delete(parsed.params[0]);
+        }
+        send(toProvider);
+      },
+      disconnect
+    };
+  };
+  return Object.assign(result, {
+    cleanup: () => {
+      prematureStops.clear();
+      preOpId.clear();
+      onGoing.clear();
+    }
+  });
+};
+
 const timeoutError = {
   type: WsEvent.ERROR,
   event: { type: "timeout" }
@@ -43,109 +107,117 @@ const getInternalWsProvider = (WebsocketClass) => {
     let status;
     let switchTo = null;
     let disconnect = noop;
-    const result = jsonRpcProviderProxy.getSyncProvider(async () => {
-      const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length];
-      switchTo = null;
-      const socket = new WebsocketClass(uri, protocols);
-      const forceSocketClose = () => {
-        try {
-          socket.addEventListener("error", noop, { once: true });
-          socket.close();
-        } catch {
-        }
-      };
-      onStatusChanged(
-        status = {
-          type: WsEvent.CONNECTING,
-          uri,
-          protocols
-        }
-      );
-      await new Promise((resolve, reject) => {
-        const onOpen = () => {
-          initialCleanup();
-          resolve();
-        };
-        const onError = (e) => {
-          initialCleanup();
-          if (e == null) forceSocketClose();
-          console.error(
-            `Unable to connect to ${uri}${protocols ? ", protocols: " + protocols : ""}`
-          );
-          onStatusChanged(
-            status = {
-              type: e ? WsEvent.ERROR : WsEvent.CLOSE,
-              event: e
-            }
-          );
-          setTimeout(reject, e ? 300 : 0, e);
-        };
-        const timeoutToken = timeout !== Infinity ? setTimeout(() => {
-          initialCleanup();
-          forceSocketClose();
-          onStatusChanged(status = timeoutError);
-          reject(timeoutError.event);
-        }, timeout) : void 0;
-        const initialCleanup = () => {
-          clearTimeout(timeoutToken);
-          socket.removeEventListener("error", onError);
-          socket.removeEventListener("open", onOpen);
-        };
-        socket.addEventListener("open", onOpen);
-        socket.addEventListener("error", onError);
-        disconnect = () => {
-          onError(null);
-        };
-      });
-      onStatusChanged(
-        status = {
-          type: WsEvent.CONNECTED,
-          uri,
-          protocols
-        }
-      );
-      return (onMessage, onHalt) => {
-        const _onMessage = (e) => {
-          if (typeof e.data === "string") onMessage(e.data);
-        };
-        const innerHalt = (reason) => (e) => {
-          console.warn(`WS halt (${reason})`);
-          onStatusChanged(
-            status = {
-              type: reason,
-              event: e
-            }
-          );
-          onHalt();
-        };
-        const onError = innerHalt(WsEvent.ERROR);
-        const onClose = innerHalt(WsEvent.CLOSE);
-        socket.addEventListener("message", _onMessage);
-        socket.addEventListener("error", onError);
-        socket.addEventListener("close", onClose);
-        disconnect = (withHalt) => {
-          disconnect = noop;
-          socket.removeEventListener("message", _onMessage);
-          socket.removeEventListener("error", onError);
-          socket.removeEventListener("close", onClose);
-          forceSocketClose();
-          if (withHalt) onClose({});
+    let outerCleanup = noop;
+    const result = followEnhancer(
+      jsonRpcProviderProxy.getSyncProvider(async () => {
+        const [uri, protocols] = switchTo || endpoints[idx++ % endpoints.length];
+        switchTo = null;
+        const socket = new WebsocketClass(uri, protocols);
+        const forceSocketClose = () => {
+          try {
+            socket.addEventListener("error", noop, { once: true });
+            socket.close();
+          } catch {
+          }
         };
-        return {
-          send: (msg) => {
-            socket.send(msg);
-          },
-          disconnect
+        onStatusChanged(
+          status = {
+            type: WsEvent.CONNECTING,
+            uri,
+            protocols
+          }
+        );
+        await new Promise((resolve, reject) => {
+          const onOpen = () => {
+            initialCleanup();
+            resolve();
+          };
+          const onError = (e) => {
+            initialCleanup();
+            if (e == null) forceSocketClose();
+            console.error(
+              `Unable to connect to ${uri}${protocols ? ", protocols: " + protocols : ""}`
+            );
+            onStatusChanged(
+              status = {
+                type: e ? WsEvent.ERROR : WsEvent.CLOSE,
+                event: e
+              }
+            );
+            setTimeout(reject, e ? 300 : 0, e);
+          };
+          const timeoutToken = timeout !== Infinity ? setTimeout(() => {
+            initialCleanup();
+            forceSocketClose();
+            onStatusChanged(status = timeoutError);
+            reject(timeoutError.event);
+          }, timeout) : void 0;
+          const initialCleanup = () => {
+            clearTimeout(timeoutToken);
+            socket.removeEventListener("error", onError);
+            socket.removeEventListener("open", onOpen);
+          };
+          socket.addEventListener("open", onOpen);
+          socket.addEventListener("error", onError);
+          disconnect = () => {
+            onError(null);
+          };
+        });
+        onStatusChanged(
+          status = {
+            type: WsEvent.CONNECTED,
+            uri,
+            protocols
+          }
+        );
+        return (onMessage, onHalt) => {
+          const _onMessage = (e) => {
+            if (typeof e.data === "string") onMessage(e.data);
+          };
+          const innerHalt = (reason) => (e) => {
+            console.warn(`WS halt (${reason})`);
+            onStatusChanged(
+              status = {
+                type: reason,
+                event: e
+              }
+            );
+            onHalt();
+          };
+          const onError = innerHalt(WsEvent.ERROR);
+          const onClose = innerHalt(WsEvent.CLOSE);
+          socket.addEventListener("message", _onMessage);
+          socket.addEventListener("error", onError);
+          socket.addEventListener("close", onClose);
+          disconnect = (withHalt) => {
+            outerCleanup();
+            disconnect = noop;
+            socket.removeEventListener("message", _onMessage);
+            socket.removeEventListener("error", onError);
+            socket.removeEventListener("close", onClose);
+            forceSocketClose();
+            if (withHalt) onClose({});
+          };
+          return {
+            send: (msg) => {
+              socket.send(msg);
+            },
+            disconnect
+          };
         };
-      };
-    });
-    result.getStatus = () => status;
-    result.switch = (...args2) => {
+      }),
+      () => {
+        switchFn();
+      }
+    );
+    outerCleanup = result.cleanup;
+    delete result.cleanup;
+    const switchFn = (...args2) => {
       if (status.type === WsEvent.CLOSE) return;
       if (args2.length) switchTo = args2;
       if (status.type !== WsEvent.ERROR) disconnect(true);
     };
-    return result;
+    return Object.assign(result, { switch: switchFn, getStatus: () => status });
   };
 };