thinkncollab-shell 0.0.6 → 0.0.7

package/src/core/shell.js

@@ -35,7 +35,7 @@ class ThinkNCollabShell extends EventEmitter {
             configDir:   path.join(os.homedir(), '.thinkncollab'),
             maxHistory:  1000,
             websocket: {
-                serverUrl:   options.serverUrl   || 'https://thinkncollab.com',
+                serverUrl:   options.serverUrl   || 'http://localhost:3001',
                 autoConnect: options.autoConnect || false
             },
             ...options
@@ -54,8 +54,8 @@ class ThinkNCollabShell extends EventEmitter {
         this.tcpServer     = null;
 
         this.api = new ThinkNCollabAPI({
-            apiUrl: options.apiUrl || 'https://thinkncollab.com',
-            wsUrl:  options.wsUrl  || 'https://thinkncollab.com'
+            apiUrl: options.apiUrl || 'http://localhost:3001',
+            wsUrl:  options.wsUrl  || 'http://localhost:3001'
         });
 
         this.commands  = new Map();
@@ -138,49 +138,67 @@ class ThinkNCollabShell extends EventEmitter {
 
     // ─── WebSocket → Shell + TCP bridge ──────────────────────────────────────
 
-    setupWebSocketHandlers() {
-        this.ws.on('connected',    (d) => this.pushNotification({ type: 'connected',    socketId: d.socketId }));
-        this.ws.on('disconnected', (d) => this.pushNotification({ type: 'disconnected', reason: d.reason }));
-        this.ws.on('reconnected',  ()  => this.pushNotification({ type: 'connected',    socketId: 'reconnected' }));
+setupWebSocketHandlers() {
+    this.ws.on('connected', (d) => {
+        this.api.logEvent('connected', null, { socketId: d?.socketId });
+        this.api.startHeartbeat();
+        this.pushNotification({ type: 'connected', socketId: d?.socketId });
+    });
 
-        this.ws.on('message', (d) => {
-            const myId = this.api.getUser()?._id;
-            if (myId && d.userId === myId) return;
-            this.pushNotification({ type: 'message', from: d.username, text: d.message });
-        });
+    this.ws.on('disconnected', (d) => {
+        this.api.logEvent('disconnected', null, { reason: d?.reason });
+        this.api.stopHeartbeat();
+        this.pushNotification({ type: 'disconnected', reason: d?.reason });
+    });
 
-        this.ws.on('messageHistory', (messages) => {
-            if (!messages?.length) return;
-            console.log(chalk.dim('\n  ─── Room History ───────────────────────────────'));
-            messages.forEach(msg => {
-                const from    = msg.name || msg.username || msg.sender?.name || 'Unknown';
-                const text    = msg.message || msg.content || msg.text || '';
-                const ts      = msg.timestamp || msg.createdAt;
-                const timeStr = ts ? chalk.dim(`[${new Date(ts).toLocaleTimeString()}]`) : '';
-                console.log(`  ${timeStr} ${chalk.cyan(from + ':')} ${chalk.white(text)}`);
-            });
-            console.log(chalk.dim('  ────────────────────────────────────────────────\n'));
-        });
+    this.ws.on('reconnected', () => {
+        this.api.logEvent('reconnected');
+        this.api.startHeartbeat();
+        this.pushNotification({ type: 'connected', socketId: 'reconnected' });
+    });
 
-        this.ws.on('userJoined',   (d) => this.pushNotification({ type: 'userJoined',   username: d.username }));
-        this.ws.on('userLeft',     (d) => this.pushNotification({ type: 'userLeft',     username: d.username }));
-
-        this.ws.on('notification', (d) => {
-            this.pushNotification({
-                type:       'notification',
-                level:      d.type,
-                title:      d.title,
-                message:    d.message,
-                taskTitle:  d.taskTitle,
-                assignedBy: d.assignedBy,
-            });
+    this.ws.on('message', (d) => {
+        const myId = this.api.getUser()?._id;
+        if (myId && d.userId === myId) return;
+        this.pushNotification({ type: 'message', from: d.username, text: d.message });
+    });
+
+    this.ws.on('messageHistory', (messages) => {
+        if (!messages?.length) return;
+        console.log(chalk.dim('\n  ─── Room History ───────────────────────────────'));
+        messages.forEach(msg => {
+            const from    = msg.name || msg.username || msg.sender?.name || 'Unknown';
+            const text    = msg.message || msg.content || msg.text || '';
+            const ts      = msg.timestamp || msg.createdAt;
+            const timeStr = ts ? chalk.dim(`[${new Date(ts).toLocaleTimeString()}]`) : '';
+            console.log(`  ${timeStr} ${chalk.cyan(from + ':')} ${chalk.white(text)}`);
         });
+        console.log(chalk.dim('  ────────────────────────────────────────────────\n'));
+    });
 
-        this.ws.on('error', (e) => {
-            console.log(chalk.red(`⚠️  WebSocket error: ${e.message}`));
-            this.pushNotification({ type: 'notification', level: 'error', title: 'WebSocket Error', message: e.message });
+    this.ws.on('userJoined', (d) => this.pushNotification({ type: 'userJoined', username: d.username }));
+    this.ws.on('userLeft',   (d) => this.pushNotification({ type: 'userLeft',   username: d.username }));
+
+    // room join/leave events
+    this.ws.on('joined', (roomId) => this.api.logEvent('room_join',  roomId));
+    this.ws.on('left',   (roomId) => this.api.logEvent('room_leave', roomId));
+
+    this.ws.on('notification', (d) => {
+        this.pushNotification({
+            type:       'notification',
+            level:      d.type,
+            title:      d.title,
+            message:    d.message,
+            taskTitle:  d.taskTitle,
+            assignedBy: d.assignedBy,
         });
-    }
+    });
+
+    this.ws.on('error', (e) => {
+        console.log(chalk.red(`⚠️  WebSocket error: ${e.message}`));
+        this.pushNotification({ type: 'notification', level: 'error', title: 'WebSocket Error', message: e.message });
+    });
+}
 
     // ─── Start ────────────────────────────────────────────────────────────────
 
@@ -233,11 +251,11 @@ class ThinkNCollabShell extends EventEmitter {
         try {
             cfg = loadProjectConfig();
         } catch (err) {
-            console.log(chalk.yellow(`⚠️  .tncproject: ${err.message}\n`));
+            console.log(chalk.yellow(`⚠️  tncproject: ${err.message}\n`));
             return false;
         }
 
-        console.log(chalk.dim(`  📄 .tncproject found`));
+        console.log(chalk.dim(`  📄 tncproject found`));
         console.log(chalk.dim(`     Room : ${cfg.roomId}`));
         console.log(chalk.dim(`     File : ${cfg.filePath}\n`));
 
@@ -249,7 +267,7 @@ class ThinkNCollabShell extends EventEmitter {
                 roomId:     cfg.roomId,
             });
         } catch (err) {
-            console.log(chalk.red(`❌ .tncproject auth failed: ${err.message}`));
+            console.log(chalk.red(`❌ tncproject auth failed: ${err.message}`));
             console.log(chalk.dim('   Falling back to manual login.\n'));
             return false;
         }
@@ -521,7 +539,7 @@ async execute(input) {
         try { const figlet = require('figlet'); console.log(chalk.cyan(figlet.textSync('ThinkNCollab', { font: 'Standard' }))); } catch {}
         console.log(chalk.cyan(`
 ╔══════════════════════════════════════════════════════╗
-║     ThinkNCollab Shell v0.0.5                        ║
+║     ThinkNCollab Shell v0.0.6                        ║
 ║     Type 'help' for commands                         ║
 ╚══════════════════════════════════════════════════════╝`));
     }
@@ -537,11 +555,12 @@ async execute(input) {
         try { fs.writeFileSync(this.config.historyFile, this.history.slice(-this.config.maxHistory).join('\n')); } catch {}
     }
 
-    cleanup() {
-        if (this.tcpServer) { try { this.tcpServer.close(); } catch {} }
-        if (this.ws) this.ws.disconnect();
-        this.saveHistory();
-    }
+cleanup() {
+    this.api.logout();  // session end + heartbeat stop + lastSeen update
+    if (this.tcpServer) { try { this.tcpServer.close(); } catch {} }
+    if (this.ws) this.ws.disconnect();
+    this.saveHistory();
+}
 }
 
 module.exports = ThinkNCollabShell;

package/src/core/websocket.js

@@ -10,7 +10,7 @@ class WebSocketManager extends EventEmitter {
     constructor(config = {}) {
         super();
         this.config = {
-            serverUrl:         config.serverUrl         || 'https://thinkncollab.com',
+            serverUrl:         config.serverUrl         || 'http://localhost:3001',
             reconnectAttempts: config.reconnectAttempts || 5,
             reconnectDelay:    config.reconnectDelay    || 1000,
             ...config