gemini-cli-devtools 0.1.2 → 0.1.5

package/dist/src/index.js

@@ -5,6 +5,7 @@
  */
 import http from 'node:http';
 import { EventEmitter } from 'node:events';
+import { execFile } from 'node:child_process';
 import fs from 'node:fs';
 import path from 'node:path';
 import os from 'node:os';
@@ -21,7 +22,7 @@ const isDev = (() => {
 /**
  * DevTools Viewer
  *
- * Deeply discovers and tails session logs across all projects.
+ * Receives logs via WebSocket from CLI sessions.
  */
 export class DevTools extends EventEmitter {
     static instance;
@@ -31,7 +32,6 @@ export class DevTools extends EventEmitter {
     wss = null;
     sessions = new Map();
     port = 25417;
-    watchedFiles = new Map(); // filePath -> lastSize
     constructor() {
         super();
     }
@@ -47,110 +47,6 @@ export class DevTools extends EventEmitter {
     getConsoleLogs() {
         return this.consoleLogs;
     }
-    /**
-     * Main entry for log discovery.
-     * It scans both user home and system tmp for .gemini/tmp folders.
-     */
-    setLogFile() {
-        const potentialRoots = [
-            path.join(os.homedir(), '.gemini', 'tmp'),
-            path.join(os.tmpdir(), '.gemini', 'tmp'),
-        ];
-        for (const baseDir of potentialRoots) {
-            if (fs.existsSync(baseDir)) {
-                console.log(`🔍 Scanning for logs in: ${baseDir}`);
-                this.deepDiscover(baseDir);
-                this.watchRoot(baseDir);
-            }
-        }
-    }
-    watchRoot(root) {
-        try {
-            fs.watch(root, { recursive: true }, (_event, filename) => {
-                if (filename &&
-                    filename.includes('session-') &&
-                    filename.endsWith('.jsonl')) {
-                    this.deepDiscover(root);
-                }
-            });
-        }
-        catch {
-            setInterval(() => this.deepDiscover(root), 2000);
-        }
-    }
-    deepDiscover(dir) {
-        try {
-            const items = fs.readdirSync(dir);
-            for (const item of items) {
-                const fullPath = path.join(dir, item);
-                let stats;
-                try {
-                    stats = fs.statSync(fullPath);
-                }
-                catch {
-                    continue;
-                }
-                if (stats.isDirectory()) {
-                    if (item === 'logs' || item.length > 20) {
-                        this.deepDiscover(fullPath);
-                    }
-                }
-                else if (item.startsWith('session-') && item.endsWith('.jsonl')) {
-                    if (!this.watchedFiles.has(fullPath)) {
-                        this.watchedFiles.set(fullPath, 0);
-                        this.readNewLogs(fullPath, 0);
-                    }
-                    else if (stats.size > this.watchedFiles.get(fullPath)) {
-                        this.readNewLogs(fullPath, this.watchedFiles.get(fullPath));
-                        this.watchedFiles.set(fullPath, stats.size);
-                    }
-                }
-            }
-        }
-        catch {
-            /* ignore */
-        }
-    }
-    readNewLogs(filePath, startByte) {
-        try {
-            const filename = path.basename(filePath);
-            const sessionMatch = filename.match(/session-(.*)\.jsonl/);
-            const fallbackSessionId = sessionMatch ? sessionMatch[1] : undefined;
-            const stream = fs.createReadStream(filePath, { start: startByte });
-            let buffer = '';
-            stream.on('data', (chunk) => {
-                buffer += chunk.toString();
-                const lines = buffer.split('\n');
-                buffer = lines.pop() || '';
-                for (const line of lines) {
-                    if (!line.trim())
-                        continue;
-                    try {
-                        const entry = JSON.parse(line);
-                        const sid = entry.sessionId || fallbackSessionId;
-                        if (entry.type === 'console') {
-                            this.addInternalConsoleLog(entry.payload, sid, entry.timestamp);
-                        }
-                        else if (entry.type === 'network') {
-                            this.addInternalNetworkLog(entry.payload, sid, entry.timestamp);
-                        }
-                    }
-                    catch {
-                        /* ignore */
-                    }
-                }
-                try {
-                    this.watchedFiles.set(filePath, fs.statSync(filePath).size);
-                }
-                catch {
-                    /* ignore */
-                }
-            });
-        }
-        catch {
-            /* ignore */
-        }
-    }
     addInternalConsoleLog(payload, sessionId, timestamp) {
         this.consoleLogs.push({
             ...payload,
@@ -167,20 +63,36 @@ export class DevTools extends EventEmitter {
             return;
         const existingIndex = this.logs.findIndex((l) => l.id === payload.id);
         if (existingIndex > -1) {
-            this.logs[existingIndex] = {
-                ...this.logs[existingIndex],
-                ...payload,
-                sessionId: sessionId || this.logs[existingIndex].sessionId,
-                response: payload.response
-                    ? { ...this.logs[existingIndex].response, ...payload.response }
-                    : this.logs[existingIndex].response,
-            };
+            const existing = this.logs[existingIndex];
+            // Handle chunk accumulation
+            if (payload.chunk) {
+                const chunks = existing.chunks || [];
+                chunks.push(payload.chunk);
+                console.log(`[DevTools] Received chunk ${payload.chunk.index} for request ${payload.id}, total chunks: ${chunks.length}`);
+                this.logs[existingIndex] = {
+                    ...existing,
+                    chunks,
+                    sessionId: sessionId || existing.sessionId,
+                };
+            }
+            else {
+                this.logs[existingIndex] = {
+                    ...existing,
+                    ...payload,
+                    sessionId: sessionId || existing.sessionId,
+                    chunks: existing.chunks, // Preserve chunks
+                    response: payload.response
+                        ? { ...existing.response, ...payload.response }
+                        : existing.response,
+                };
+            }
         }
         else if (payload.url) {
             this.logs.push({
                 ...payload,
                 sessionId,
                 timestamp: timestamp || Date.now(),
+                chunks: payload.chunk ? [payload.chunk] : undefined,
             });
             if (this.logs.length > 2000)
                 this.logs.shift();
@@ -262,6 +174,37 @@ export class DevTools extends EventEmitter {
                         }
                     });
                 }
+                else if (req.method === 'POST' && req.url === '/api/to-textproto') {
+                    let body = '';
+                    req.on('data', (chunk) => (body += chunk));
+                    req.on('end', () => {
+                        const tmpIn = path.join(os.tmpdir(), `req-${Date.now()}.json`);
+                        fs.writeFileSync(tmpIn, body);
+                        const script = `
+import sys
+sys.path.insert(0, '/tmp/proto_gen')
+from google.cloud.aiplatform.v1 import prediction_service_pb2
+from google.protobuf.json_format import Parse
+from google.protobuf import text_format
+import json
+
+with open(sys.argv[1]) as f:
+    msg = Parse(f.read(), prediction_service_pb2.GenerateContentRequest(), ignore_unknown_fields=True)
+print(text_format.MessageToString(msg, as_utf8=True), end='')
+`;
+                        execFile('/tmp/proto_venv/bin/python3', ['-c', script, tmpIn], { maxBuffer: 10 * 1024 * 1024 }, (err, stdout, stderr) => {
+                            fs.unlinkSync(tmpIn);
+                            if (err) {
+                                res.writeHead(500, { 'Content-Type': 'text/plain' });
+                                res.end(stderr || err.message);
+                            }
+                            else {
+                                res.writeHead(200, { 'Content-Type': 'text/plain' });
+                                res.end(stdout);
+                            }
+                        });
+                    });
+                }
                 else if (isDev) {
                     // Dev mode: proxy to Vite
                     this.proxyToVite(req, res);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gemini-cli-devtools",
-  "version": "0.1.2",
+  "version": "0.1.5",
   "type": "module",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",