claude-opencode-viewer 2.3.0 → 2.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-opencode-viewer",
-  "version": "2.3.0",
+  "version": "2.4.0",
   "description": "A unified terminal viewer for Claude Code and OpenCode with seamless switching",
   "type": "module",
   "main": "server.js",
@@ -33,6 +33,7 @@
   },
   "homepage": "https://github.com/ChrisJason121238/claude-opencode-viewer#readme",
   "dependencies": {
+    "better-sqlite3": "^11.8.1",
     "node-pty": "^1.1.0",
     "ws": "^8.19.0"
   },

package/server.js

@@ -3,10 +3,11 @@ import { createServer } from 'node:http';
 import { existsSync, createReadStream } from 'node:fs';
 import { join, dirname } from 'node:path';
 import { fileURLToPath } from 'node:url';
-import { networkInterfaces, platform, arch } from 'node:os';
+import { networkInterfaces, platform, arch, homedir } from 'node:os';
 import { chmodSync, statSync } from 'node:fs';
 import { execSync } from 'child_process';
 import { WebSocketServer } from 'ws';
+import Database from 'better-sqlite3';
 
 // 设置进程名为 claude-opencode-viewer
 process.title = 'claude-opencode-viewer';
@@ -14,6 +15,9 @@ process.title = 'claude-opencode-viewer';
 const __dirname = dirname(fileURLToPath(import.meta.url));
 const PORT = 7008;
 
+// OpenCode 数据库路径
+const OPENCODE_DB_PATH = join(homedir(), '.local/share/opencode/opencode.db');
+
 const MAX_BUFFER = 200000;
 
 let ptyModule = null;
@@ -93,7 +97,7 @@ function findCommand(cmd) {
   return cmd;
 }
 
-async function spawnProcess(mode) {
+async function spawnProcess(mode, sessionId = null) {
   const pty = await getPty();
   fixSpawnHelperPermissions();
 
@@ -108,6 +112,11 @@ async function spawnProcess(mode) {
     }
   } else {
     command = findCommand('opencode');
+    // 如果提供了 sessionId，添加 --session 参数
+    if (sessionId) {
+      args = ['--session', sessionId];
+      console.log(`[opencode] 恢复会话: ${sessionId}`);
+    }
   }
 
   const proc = pty.spawn(command, args, {
@@ -126,11 +135,18 @@ async function spawnProcess(mode) {
     dataListeners.forEach(cb => cb(data));
   });
 
-  proc.onExit(() => {
+  proc.onExit(({ exitCode }) => {
+    console.log(`[onExit] 进程退出, PID: ${proc.pid}, exitCode: ${exitCode}`);
     if (currentProcess === proc) {
       currentProcess = null;
-      exitListeners.forEach(cb => cb(0));
     }
+    if (claudeProcess === proc) {
+      claudeProcess = null;
+    }
+    if (opencodeProcess === proc) {
+      opencodeProcess = null;
+    }
+    exitListeners.forEach(cb => cb(exitCode || 0));
   });
 
   // 只在初始化时杀死旧进程，switchMode 已经处理了切换时的进程清理
@@ -214,6 +230,155 @@ function resizePty(cols, rows) {
   });
 }
 
+// 数据库访问函数
+function getOpenCodeSessions() {
+  try {
+    if (!existsSync(OPENCODE_DB_PATH)) {
+      console.log('[DB] OpenCode 数据库不存在:', OPENCODE_DB_PATH);
+      return [];
+    }
+
+    const db = new Database(OPENCODE_DB_PATH, { readonly: true });
+
+    const sessions = db.prepare(`
+      SELECT
+        s.id,
+        s.title,
+        s.directory,
+        s.time_created,
+        s.time_updated,
+        p.name as project_name
+      FROM session s
+      LEFT JOIN project p ON s.project_id = p.id
+      WHERE s.parent_id IS NULL
+        AND s.time_archived IS NULL
+      ORDER BY s.time_updated DESC
+      LIMIT 50
+    `).all();
+
+    // 为每个会话获取第一条用户消息作为预览
+    const result = sessions.map(session => {
+      // 获取第一条用户消息
+      const firstMessage = db.prepare(`
+        SELECT id
+        FROM message
+        WHERE session_id = ?
+          AND json_extract(data, '$.role') = 'user'
+        ORDER BY time_created ASC
+        LIMIT 1
+      `).get(session.id);
+
+      let preview = '';
+      if (firstMessage) {
+        // 获取该消息的文本 part
+        const textPart = db.prepare(`
+          SELECT data
+          FROM part
+          WHERE message_id = ?
+            AND json_extract(data, '$.type') = 'text'
+          LIMIT 1
+        `).get(firstMessage.id);
+
+        if (textPart) {
+          try {
+            const partData = JSON.parse(textPart.data);
+            if (partData.text) {
+              // 截取前80个字符作为预览
+              preview = partData.text.length > 80
+                ? partData.text.substring(0, 80) + '...'
+                : partData.text;
+            }
+          } catch (e) {
+            console.error('[DB] 解析 part 失败:', e.message);
+          }
+        }
+      }
+
+      return {
+        ...session,
+        preview: preview || session.title
+      };
+    });
+
+    db.close();
+    return result;
+  } catch (err) {
+    console.error('[DB] 读取会话失败:', err.message);
+    return [];
+  }
+}
+
+function getSessionMessages(sessionId) {
+  try {
+    if (!existsSync(OPENCODE_DB_PATH)) {
+      return [];
+    }
+
+    const db = new Database(OPENCODE_DB_PATH, { readonly: true });
+
+    // 获取消息列表
+    const messages = db.prepare(`
+      SELECT
+        id,
+        time_created,
+        data
+      FROM message
+      WHERE session_id = ?
+      ORDER BY time_created ASC
+    `).all(sessionId);
+
+    // 为每个消息获取其 parts
+    const result = messages.map(msg => {
+      const msgData = JSON.parse(msg.data);
+
+      // 获取该消息的所有 parts
+      const parts = db.prepare(`
+        SELECT data
+        FROM part
+        WHERE message_id = ?
+        ORDER BY time_created ASC
+      `).all(msg.id);
+
+      const parsedParts = parts.map(p => JSON.parse(p.data));
+
+      // 提取文本内容
+      let text = '';
+      let reasoning = '';
+      let toolCalls = [];
+      let toolResults = [];
+
+      for (const part of parsedParts) {
+        if (part.type === 'text' && part.text) {
+          text += part.text;
+        } else if (part.type === 'reasoning' && part.text) {
+          reasoning += part.text;
+        } else if (part.type === 'tool-call' || part.type === 'tool_call') {
+          toolCalls.push(part);
+        } else if (part.type === 'tool-result' || part.type === 'tool_result') {
+          toolResults.push(part);
+        }
+      }
+
+      return {
+        id: msg.id,
+        time_created: msg.time_created,
+        role: msgData.role,
+        text: text || undefined,
+        reasoning: reasoning || undefined,
+        toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+        toolResults: toolResults.length > 0 ? toolResults : undefined,
+        metadata: msgData
+      };
+    });
+
+    db.close();
+    return result;
+  } catch (err) {
+    console.error('[DB] 读取消息失败:', err.message);
+    return [];
+  }
+}
+
 const server = createServer((req, res) => {
   if (req.url === '/' || req.url === '/index.html') {
     res.writeHead(200, {
@@ -223,6 +388,30 @@ const server = createServer((req, res) => {
     createReadStream(join(__dirname, 'index.html')).pipe(res);
     return;
   }
+
+  // API: 获取会话列表
+  if (req.url === '/api/sessions') {
+    res.writeHead(200, {
+      'Content-Type': 'application/json',
+      'Access-Control-Allow-Origin': '*',
+    });
+    const sessions = getOpenCodeSessions();
+    res.end(JSON.stringify(sessions));
+    return;
+  }
+
+  // API: 获取会话消息
+  if (req.url?.startsWith('/api/session/')) {
+    const sessionId = req.url.split('/').pop();
+    res.writeHead(200, {
+      'Content-Type': 'application/json',
+      'Access-Control-Allow-Origin': '*',
+    });
+    const messages = getSessionMessages(sessionId);
+    res.end(JSON.stringify(messages));
+    return;
+  }
+
   res.writeHead(404, { 'Content-Type': 'text/plain' });
   res.end('Not Found');
 });
@@ -260,8 +449,20 @@ wss.on('connection', (ws, req) => {
   ws.on('message', async (raw) => {
     try {
       const msg = JSON.parse(raw);
-      
+      console.log(`[WS msg] type=${msg.type}, currentProcess=${!!currentProcess}, currentMode=${currentMode}`);
+
       if (msg.type === 'input') {
+        // 进程已退出时，自动重新启动（参考 cc-viewer 逻辑）
+        if (!currentProcess) {
+          try {
+            console.log(`[respawn] 进程已退出，自动重新启动 ${currentMode}`);
+            outputBuffer = '';
+            await spawnProcess(currentMode);
+            console.log(`[respawn] 重新启动成功, currentProcess=${!!currentProcess}`);
+          } catch (e) {
+            console.log(`[respawn] 重新启动失败: ${e.message}`);
+          }
+        }
         if (activeWs !== ws) {
           activeWs = ws;
           const mSize = getMobileSize();
@@ -293,6 +494,38 @@ wss.on('connection', (ws, req) => {
             }
           }, 100);
         }
+      } else if (msg.type === 'restore') {
+        // 恢复会话
+        if (msg.sessionId && currentMode === 'opencode') {
+          console.log(`[restore] 恢复会话: ${msg.sessionId}`);
+
+          // 杀死当前 opencode 进程
+          if (opencodeProcess) {
+            try {
+              console.log(`[restore] 杀死当前进程 PID: ${opencodeProcess.pid}`);
+              opencodeProcess.kill();
+            } catch (e) {
+              console.log('[restore] 杀死进程失败:', e.message);
+            }
+            opencodeProcess = null;
+            currentProcess = null;
+          }
+
+          // 清空输出缓冲
+          outputBuffer = '';
+
+          // 等待进程完全退出
+          await new Promise(resolve => setTimeout(resolve, 200));
+
+          // 启动新的 opencode 进程，传入 session ID
+          try {
+            await spawnProcess('opencode', msg.sessionId);
+            ws.send(JSON.stringify({ type: 'restored', sessionId: msg.sessionId }));
+          } catch (e) {
+            console.error('[restore] 启动进程失败:', e.message);
+            ws.send(JSON.stringify({ type: 'restore-error', error: e.message }));
+          }
+        }
       }
     } catch (err) {
       console.error('[WS] Error:', err.message);