@moxt-ai/mobius 0.0.303 → 0.0.306

package/dist/runtime.js

@@ -22687,7 +22687,7 @@ import { fetch as fetch3 } from "undici";
 // package.json
 var package_default = {
   name: "@moxt-ai/mobius",
-  version: "0.0.303",
+  version: "0.0.306",
   description: "Mobius CLI for pairing and running local AI agent services.",
   type: "module",
   bin: {
@@ -90539,13 +90539,30 @@ var RelayConnectionError = class extends Error {
 function connectRelay(url2, signal) {
   signal.throwIfAborted();
   return new Promise((resolve18, reject2) => {
+    let dispatchController;
+    const dispatcher = environmentProxyDispatcher.compose((dispatch) => (options, handler) => {
+      const trackedHandler = Object.create(handler);
+      trackedHandler.onRequestStart = (controller, context) => {
+        dispatchController = controller;
+        handler.onRequestStart?.(controller, context);
+      };
+      return dispatch(options, trackedHandler);
+    });
     const endpoint = new URL(url2);
     const headers = relayAuthorization(endpoint);
     endpoint.searchParams.delete("token");
     const socket = new WebSocket7(endpoint, {
-      dispatcher: environmentProxyDispatcher,
+      dispatcher,
       headers
     });
+    let released = false;
+    const release = (reason) => {
+      if (released) {
+        return;
+      }
+      released = true;
+      dispatchController?.abort(reason);
+    };
     let finished = false;
     const timeout = setTimeout(onTimeout, CONNECTION_TIMEOUT_MILLISECONDS);
     const cleanup = () => {
@@ -90562,6 +90579,7 @@ function connectRelay(url2, signal) {
       finished = true;
       cleanup();
       socket.close();
+      release(error62);
       reject2(error62);
     };
     const onAbort = () => {
@@ -90605,7 +90623,7 @@ function connectRelay(url2, signal) {
       }
       finished = true;
       cleanup();
-      resolve18(socket);
+      resolve18({ release, socket });
     };
     function onTimeout() {
       fail(
@@ -90780,7 +90798,8 @@ async function runRelayConnection({
 }) {
   const relayUrl = new URL(config2.relayUrl);
   relayUrl.searchParams.set("liveness", "1");
-  const socket = await connectRelay(relayUrl, signal);
+  const relay = await connectRelay(relayUrl, signal);
+  const { socket } = relay;
   diagnostics.connectionOpened(connectionEpoch);
   const connectionController = new AbortController();
   const commands = createRuntimeCommandDispatcher({
@@ -90958,6 +90977,7 @@ async function runRelayConnection({
   } finally {
     connectionController.abort();
     await commands.settle();
+    relay.release(new Error("Relay connection released"));
   }
 }