@buda-ai/connector 0.1.5 → 0.1.6

package/dist/cli.js

@@ -2145,6 +2145,7 @@ function startConnectorRelay(config, bunnyDaemon, relayProtocol, workspaceRegist
   let pingTimer = null;
   let pongTimer = null;
   let stopped = false;
+  let suspendedForAccessRevocation = false;
   let reconnectDelayMs = RELAY_RECONNECT_INTERVAL_MS;
   let hadOpenConnection = false;
   let activeRelayProtocol = relayProtocol;
@@ -2171,7 +2172,7 @@ function startConnectorRelay(config, bunnyDaemon, relayProtocol, workspaceRegist
     return Math.max(reconnectingEstablishedSocket ? 250 : 500, Math.round(base + jitter));
   };
   const scheduleReconnect = (reason) => {
-    if (stopped || reconnectTimer) return;
+    if (stopped || suspendedForAccessRevocation || reconnectTimer) return;
     const delay = nextReconnectDelay();
     void writeLog("info", `relay reconnecting in ${delay}ms (${reason})`);
     clearTimers();
@@ -2186,7 +2187,7 @@ function startConnectorRelay(config, bunnyDaemon, relayProtocol, workspaceRegist
     }, delay);
   };
   const connect = () => {
-    if (stopped) return;
+    if (stopped || suspendedForAccessRevocation) return;
     const relayUrl = getConnectorRelayUrl(config.serverUrl, config.connectorId);
     const oauthToken = launchConfig?.oauthToken?.trim();
     const token = config.token?.trim() || getConnectorToken(launchConfig);
@@ -2250,11 +2251,14 @@ function startConnectorRelay(config, bunnyDaemon, relayProtocol, workspaceRegist
     ws.on("close", (code) => {
       detach();
       if (!shouldReconnectConnectorRelay(code)) {
-        stopped = true;
-        void writeLog(
-          "warn",
-          code === CONNECTOR_SOCKET_REPLACED_CLOSE_CODE ? "relay replaced by another process for this connector" : "relay access revoked; restore access and restart the Connector to reconnect"
-        );
+        clearTimers();
+        if (code === CONNECTOR_SOCKET_REPLACED_CLOSE_CODE) {
+          stopped = true;
+          void writeLog("warn", "relay replaced by another process for this connector");
+        } else {
+          suspendedForAccessRevocation = true;
+          void writeLog("warn", "relay access revoked; waiting for access to be restored");
+        }
         return;
       }
       void writeLog("warn", "relay disconnected");
@@ -2280,6 +2284,19 @@ function startConnectorRelay(config, bunnyDaemon, relayProtocol, workspaceRegist
         void writeLog("info", `relay protocol key updated ${nextRelayProtocol.keyId}`);
       }
     },
+    resumeAfterAccessRestore() {
+      if (stopped || !suspendedForAccessRevocation) return;
+      suspendedForAccessRevocation = false;
+      reconnectDelayMs = RELAY_RECONNECT_INTERVAL_MS;
+      hadOpenConnection = false;
+      void writeLog("info", "relay access restored; reconnecting");
+      try {
+        connect();
+      } catch (error) {
+        void writeLog("warn", `relay reconnect threw: ${formatConnectorRelayError(error)}`);
+        scheduleReconnect("access-restored");
+      }
+    },
     stop() {
       stopped = true;
       if (reconnectTimer) {
@@ -3045,6 +3062,7 @@ async function runConnectorDaemon(launchConfig) {
     const interval = setInterval(() => {
       heartbeatConnector(config, launchConfig, activeBunnyDaemon).then(({ relayProtocol: nextRelayProtocol }) => {
         relay.updateRelayProtocol(nextRelayProtocol);
+        relay.resumeAfterAccessRestore();
       }).catch((error) => {
         void writeLog(
           "warn",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@buda-ai/connector",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "private": false,
   "type": "module",
   "publishConfig": {