osai-agent 4.2.30 → 4.2.31

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "osai-agent",
-  "version": "4.2.30",
+  "version": "4.2.31",
   "type": "module",
   "description": "OS AI Agent - YOUR AI AGENT",
   "main": "src/index.js",

package/src/commands/run.js

@@ -11,6 +11,7 @@ import { toHttpUrl } from '../services/server-url.js';
 import { MODES } from '../utils/constants.js';
 import { printNotLoggedIn } from '../ui/terminal.js';
 import { disableMouseReporting } from '../ui/mouse-scroll.js';
+import WebSocket from 'ws';
 import { logger } from '../utils/logger.js';
 
 export const run = async ({ server: serverOverride, noConfirm = false, osOverride, coding = false, fromSession = null, local = false } = {}) => {
@@ -33,15 +34,26 @@ export const run = async ({ server: serverOverride, noConfirm = false, osOverrid
 
   if (!effectiveLocal && token && server) {
     try {
-      const res = await fetch(`${server}/auth/check-status`, {
-        headers: { 'Authorization': `Bearer ${token}` }
+      const wsUrl = server.replace('http://', 'ws://').replace('https://', 'wss://');
+      await new Promise((resolve, reject) => {
+        const ws = new WebSocket(`${wsUrl}/ws`, { handshakeTimeout: 5000 });
+        const timeout = setTimeout(() => { ws.close(); resolve(); }, 3000);
+        ws.on('open', () => ws.send(JSON.stringify({ type: 'REGISTER', token })));
+        ws.on('message', (data) => {
+          try {
+            const msg = JSON.parse(data.toString());
+            if (msg.type === 'REGISTERED') { clearTimeout(timeout); ws.close(); resolve(); }
+            else if (msg.type === 'ERROR') { clearTimeout(timeout); ws.close(); reject(new Error(msg.message)); }
+          } catch { resolve(); }
+        });
+        ws.on('error', () => { clearTimeout(timeout); resolve(); });
       });
-      const data = await res.json();
-      if (data.status === 'disabled') {
-        console.log(chalk.red(`\n  ${data.error}\n`));
+    } catch (err) {
+      if (err.message.toLowerCase().includes('disabled')) {
+        console.log(chalk.red(`\n  ${err.message}\n`));
         process.exit(1);
       }
-    } catch {}
+    }
   }
 
   const resumedMode = Object.values(MODES).includes(fromSession?.mode) ? fromSession.mode : null;