uwonbot 1.4.0 → 1.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "uwonbot",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "description": "Uwonbot AI Assistant CLI — Your AI controls your computer",
   "main": "src/index.js",
   "bin": {

package/src/agent.js

@@ -892,16 +892,41 @@ end tell'`, { timeout: 3000 });
         console.log('');
     }
 
-    const wss = new WebSocketServer({ port });
+    const UNGATED_CMDS = new Set(['ping', 'auth_session', 'screenshot', 'screen_size']);
 
-    wss.on('connection', (ws, req) => {
-        const origin = req.headers.origin || 'unknown';
+    function attachClientHandler(ws, req) {
+        const origin = req?.headers?.origin || 'unknown';
         console.log(chalk.green(`  ✓ Client connected from ${origin}`));
 
+        let _authenticated = false;
+
         ws.on('message', async (data) => {
             const raw = data.toString();
-            let _reqId;
-            try { _reqId = JSON.parse(raw)._reqId; } catch {}
+            let _reqId, cmd;
+            try { cmd = JSON.parse(raw); _reqId = cmd._reqId; } catch {}
+
+            if (cmd?.type === 'auth_session') {
+                if (cmd.uid === uid) {
+                    _authenticated = true;
+                    console.log(chalk.green('  ✓ Client authenticated'));
+                    const result = { ok: true, authenticated: true };
+                    if (_reqId) result._reqId = _reqId;
+                    ws.send(JSON.stringify(result));
+                } else {
+                    const result = { ok: false, error: 'UID mismatch' };
+                    if (_reqId) result._reqId = _reqId;
+                    ws.send(JSON.stringify(result));
+                }
+                return;
+            }
+
+            if (!_authenticated && !UNGATED_CMDS.has(cmd?.type)) {
+                const result = { ok: false, error: 'auth_required', message: 'Authentication required. Send auth_session first.' };
+                if (_reqId) result._reqId = _reqId;
+                ws.send(JSON.stringify(result));
+                return;
+            }
+
             const result = await handleCommand(raw);
             if (_reqId) result._reqId = _reqId;
             ws.send(JSON.stringify(result));
@@ -909,12 +934,16 @@ end tell'`, { timeout: 3000 });
 
         ws.on('close', () => {
             console.log(chalk.yellow('  ○ Client disconnected'));
+            _authenticated = false;
             _assistantOpen = false;
             clearTimeout(_assistantOpenTimer);
         });
 
-        ws.send(JSON.stringify({ type: 'welcome', agent: 'uwonbot', version: '1.1.2', uid }));
-    });
+        ws.send(JSON.stringify({ type: 'welcome', agent: 'uwonbot', version: '1.5.0', uid, authRequired: true }));
+    }
+
+    const wss = new WebSocketServer({ port });
+    wss.on('connection', attachClientHandler);
 
     wss.on('error', async (err) => {
         if (err.code === 'EADDRINUSE') {
@@ -923,17 +952,7 @@ end tell'`, { timeout: 3000 });
             await new Promise(r => setTimeout(r, 2000));
             try {
                 const retry = new WebSocketServer({ port });
-                retry.on('connection', (ws, req) => {
-                    ws.on('message', async (data) => {
-                        const raw = data.toString();
-                        let _reqId;
-                        try { _reqId = JSON.parse(raw)._reqId; } catch {}
-                        const result = await handleCommand(raw);
-                        if (_reqId) result._reqId = _reqId;
-                        ws.send(JSON.stringify(result));
-                    });
-                    ws.send(JSON.stringify({ type: 'welcome', agent: 'uwonbot', version: '1.1.8', uid }));
-                });
+                retry.on('connection', (ws, req) => attachClientHandler(ws, req));
                 console.log(chalk.green(`  ✓ 재시도 성공 — ws://localhost:${port}`));
             } catch {
                 console.log(chalk.red(`  ✗ 포트 ${port}를 사용할 수 없습니다.`));