lazy-gravity 0.9.2 → 0.10.0

package/dist/services/responseMonitor.js

@@ -601,6 +601,8 @@ class ResponseMonitor {
     extractionMode;
     onProgress;
     onComplete;
+    onStructuredProgress;
+    onStructuredComplete;
     onTimeout;
     onPhaseChange;
     onProcessLog;
@@ -609,12 +611,15 @@ class ResponseMonitor {
     pollTimer = null;
     isRunning = false;
     lastText = null;
+    lastClassified = null;
     baselineText = null;
     generationStarted = false;
     currentPhase = 'waiting';
     stopGoneCount = 0;
     quotaDetected = false;
     seenProcessLogKeys = new Set();
+    currentCitedFiles = [];
+    currentFileChanges = [];
     structuredDiagLogged = false;
     lastContentContextId = null;
     // CDP disconnect handling (#48)
@@ -632,12 +637,17 @@ class ResponseMonitor {
         this.extractionMode = options.extractionMode ?? 'structured';
         this.onProgress = options.onProgress;
         this.onComplete = options.onComplete;
+        this.onStructuredProgress = options.onStructuredProgress;
+        this.onStructuredComplete = options.onStructuredComplete;
         this.onTimeout = options.onTimeout;
         this.onPhaseChange = options.onPhaseChange;
         this.onProcessLog = options.onProcessLog;
         this.initialBaselineText = options.initialBaselineText;
         this.initialSeenProcessLogKeys = options.initialSeenProcessLogKeys;
     }
+    getLastClassified() {
+        return this.lastClassified;
+    }
     /** Start monitoring */
     async start() {
         return this.initMonitoring(false);
@@ -979,6 +989,7 @@ class ResponseMonitor {
                     if (classified.diagnostics.source === 'dom-structured') {
                         currentText = classified.finalOutputText.trim() || null;
                         structuredHandledLogs = true;
+                        this.lastClassified = classified;
                         if (!this.structuredDiagLogged) {
                             this.structuredDiagLogged = true;
                             logger_1.logger.debug('[ResponseMonitor] Structured extraction OK — segments:', classified.diagnostics.segmentCounts);
@@ -987,6 +998,8 @@ class ResponseMonitor {
                         if (classified.activityLines.length > 0) {
                             this.emitNewProcessLogs(classified.activityLines);
                         }
+                        this.currentCitedFiles = classified.citedFiles || [];
+                        this.currentFileChanges = classified.fileChangesTexts || [];
                     }
                     else if (!this.structuredDiagLogged) {
                         this.structuredDiagLogged = true;
@@ -1037,6 +1050,8 @@ class ResponseMonitor {
                     await this.stop();
                     try {
                         await Promise.resolve(this.onComplete?.(''));
+                        if (this.lastClassified)
+                            await Promise.resolve(this.onStructuredComplete?.(this.lastClassified));
                     }
                     catch (error) {
                         logger_1.logger.error('[ResponseMonitor] complete callback failed:', error);
@@ -1061,7 +1076,16 @@ class ResponseMonitor {
                         this.generationStarted = true;
                     }
                 }
-                this.onProgress?.(effectiveText);
+                this.onProgress?.(effectiveText || '');
+                if (this.lastClassified)
+                    this.onStructuredProgress?.(this.lastClassified);
+            }
+            else if (this.lastClassified && this.extractionMode === 'structured') {
+                // If text didn't change but structured data did (e.g. new file changes), emit structured progress anyway
+                // To keep it simple, we just emit on structured mode if generation is active.
+                if (isGenerating && this.generationStarted) {
+                    this.onStructuredProgress?.(this.lastClassified);
+                }
             }
             // Completion: stop button gone N consecutive times
             if (!isGenerating && this.generationStarted) {
@@ -1071,7 +1095,9 @@ class ResponseMonitor {
                     this.setPhase('complete', finalText);
                     await this.stop();
                     try {
-                        await Promise.resolve(this.onComplete?.(finalText));
+                        await Promise.resolve(this.onComplete?.(finalText, this.currentCitedFiles, this.currentFileChanges));
+                        if (this.lastClassified)
+                            await Promise.resolve(this.onStructuredComplete?.(this.lastClassified));
                     }
                     catch (error) {
                         logger_1.logger.error('[ResponseMonitor] complete callback failed:', error);

package/dist/services/runCommandDetector.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.RunCommandDetector = void 0;
+const consecutiveEmptyPollGate_1 = require("../utils/consecutiveEmptyPollGate");
 const logger_1 = require("../utils/logger");
 const approvalDetector_1 = require("./approvalDetector");
 /**
@@ -25,7 +26,7 @@ const approvalDetector_1 = require("./approvalDetector");
  */
 const DETECT_RUN_COMMAND_SCRIPT = `(() => {
     const RUN_COMMAND_HEADER_PATTERNS = [
-        'run command?', 'run command', 'execute command',
+        'run command?', 'run command', 'execute command', 'command execution',
         'コマンドを実行', 'コマンド実行'
     ];
     const RUN_PATTERNS = ['run', 'accept', '実行', 'execute'];
@@ -111,8 +112,10 @@ class RunCommandDetector {
     isRunning = false;
     /** Key of the last detected dialog (for duplicate notification prevention) */
     lastDetectedKey = null;
-    /** Full RunCommandInfo from the last detection (used for clicking) */
+    /** Full RunCommandInfo from the last detection */
     lastDetectedInfo = null;
+    /** Gate for empty polls before reset */
+    emptyPollGate = new consecutiveEmptyPollGate_1.ConsecutiveEmptyPollGate(3);
     constructor(options) {
         this.cdpService = options.cdpService;
         this.pollIntervalMs = options.pollIntervalMs ?? 1500;
@@ -126,6 +129,7 @@ class RunCommandDetector {
         this.isRunning = true;
         this.lastDetectedKey = null;
         this.lastDetectedInfo = null;
+        this.emptyPollGate.reset();
         this.schedulePoll();
     }
     /** Stop monitoring. */
@@ -171,6 +175,7 @@ class RunCommandDetector {
             const result = await this.cdpService.call('Runtime.evaluate', callParams);
             const info = result?.result?.value ?? null;
             if (info) {
+                this.emptyPollGate.recordDetection();
                 // Duplicate prevention: use commandText as key
                 const key = `${info.commandText}::${info.workingDirectory}`;
                 if (key !== this.lastDetectedKey) {
@@ -180,11 +185,13 @@ class RunCommandDetector {
                 }
             }
             else {
-                const wasDetected = this.lastDetectedKey !== null;
-                this.lastDetectedKey = null;
-                this.lastDetectedInfo = null;
-                if (wasDetected && this.onResolved) {
-                    this.onResolved();
+                if (this.emptyPollGate.recordEmptyPoll()) {
+                    const wasDetected = this.lastDetectedKey !== null;
+                    this.lastDetectedKey = null;
+                    this.lastDetectedInfo = null;
+                    if (wasDetected && this.onResolved) {
+                        this.onResolved();
+                    }
                 }
             }
         }

package/dist/ui/artifactsUi.js

@@ -11,6 +11,7 @@ exports.sendArtifactsUI = sendArtifactsUI;
 exports.sendArtifactPickerUI = sendArtifactPickerUI;
 const discord_js_1 = require("discord.js");
 const artifactService_1 = require("../services/artifactService");
+const pathUtils_1 = require("../utils/pathUtils");
 // ---------------------------------------------------------------------------
 // Constants
 // ---------------------------------------------------------------------------
@@ -143,7 +144,8 @@ async function sendArtifactPickerUI(interaction, deps, edit = false, resolvedCon
             // Fallback to title matching if ID is not in DB or doesn't match a real folder
             if (!conversationId || !artifactService.listArtifacts(conversationId).length) {
                 const sessionTitle = session.displayName?.trim() ?? '';
-                const matchedId = sessionTitle ? artifactService.findConversationByTitle(sessionTitle) : null;
+                const workspaceDirName = (0, pathUtils_1.getWorkspaceDirName)(session);
+                const matchedId = sessionTitle ? artifactService.findConversationByTitle(sessionTitle, workspaceDirName) : null;
                 if (matchedId)
                     conversationId = matchedId;
             }
@@ -151,8 +153,7 @@ async function sendArtifactPickerUI(interaction, deps, edit = false, resolvedCon
             // to avoid showing artifacts from other projects/chats.
         }
         else {
-            // Final fallback: latest overall (only for non-session channels)
-            conversationId = artifactService.getLatestConversationWithArtifacts();
+            conversationId = null;
         }
     }
     if (!conversationId) {

package/dist/utils/consecutiveEmptyPollGate.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ConsecutiveEmptyPollGate = void 0;
+class ConsecutiveEmptyPollGate {
+    requiredEmptyPolls;
+    count = 0;
+    constructor(requiredEmptyPolls = 3) {
+        this.requiredEmptyPolls = requiredEmptyPolls;
+    }
+    recordDetection() {
+        this.count = 0;
+    }
+    recordEmptyPoll() {
+        this.count++;
+        return this.count >= this.requiredEmptyPolls;
+    }
+    reset() {
+        this.count = 0;
+    }
+}
+exports.ConsecutiveEmptyPollGate = ConsecutiveEmptyPollGate;

package/dist/utils/fileOpenCache.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fileOpenCache = void 0;
+/**
+ * In-memory cache to map button customIds to file URLs.
+ * Bounded to a maximum size to prevent memory leaks, and entries are removed once consumed.
+ */
+class BoundedCache {
+    max_size;
+    map;
+    constructor(max_size = 1000) {
+        this.max_size = max_size;
+        this.map = new Map();
+    }
+    set(key, value) {
+        if (this.map.size >= this.max_size) {
+            // Remove the oldest entry (first item in the map's insertion order)
+            const firstKey = this.map.keys().next().value;
+            if (firstKey !== undefined) {
+                this.map.delete(firstKey);
+            }
+        }
+        this.map.set(key, value);
+    }
+    get(key) {
+        const value = this.map.get(key);
+        if (value !== undefined) {
+            this.map.delete(key); // Remove upon consumption
+        }
+        return value;
+    }
+}
+exports.fileOpenCache = new BoundedCache(1000);

package/dist/utils/htmlToDiscordMarkdown.js

@@ -42,6 +42,9 @@ function htmlToDiscordMarkdown(html) {
     // Extract language from class="language-xxx" if present.
     // Do NOT decode entities here — let the final decodeEntities() handle them
     // after stripTags() has run, to avoid decoded < > being stripped as tags.
+    // First, Antigravity 2.0 wrapped code blocks
+    result = result.replace(/<div[^>]*class="[^"]*code-block[^"]*"[^>]*>(?:(?!<div[^>]*class="[^"]*code-block[^"]*"[^>]*>)[\s\S])*?<div[^>]*class="[^"]*code-header[^"]*"[^>]*>([\s\S]*?)<\/div>(?:(?!<div[^>]*class="[^"]*code-block[^"]*"[^>]*>)[\s\S])*?<pre[^>]*>\s*<code[^>]*>([\s\S]*?)<\/code>\s*<\/pre>(?:(?!<div[^>]*class="[^"]*code-block[^"]*"[^>]*>)[\s\S])*?<\/div>/gi, (_m, lang, content) => `\n\`\`\`${stripTags(lang).trim()}\n${content}\n\`\`\`\n`);
+    // Standard <pre><code> blocks
     result = result.replace(/<pre[^>]*>\s*<code(?:\s+class="language-([^"]*)")?[^>]*>([\s\S]*?)<\/code>\s*<\/pre>/gi, (_m, lang, content) => `\n\`\`\`${lang || ''}\n${content}\n\`\`\`\n`);
     // Handle inline <code>
     result = result.replace(/<code[^>]*>([\s\S]*?)<\/code>/gi, '`$1`');
@@ -49,8 +52,8 @@ function htmlToDiscordMarkdown(html) {
     result = result.replace(/<(?:strong|b)(?:\s[^>]*)?>((?: |\s|[^<]|<(?!\/(?:strong|b)>))*)<\/(?:strong|b)>/gi, '**$1**');
     // Handle <em> and <i>
     result = result.replace(/<(?:em|i)(?:\s[^>]*)?>((?: |\s|[^<]|<(?!\/(?:em|i)>))*)<\/(?:em|i)>/gi, '*$1*');
-    // Handle <span class="context-scope-mention"> → `text`
-    result = result.replace(/<span[^>]*class="[^"]*context-scope-mention[^"]*"[^>]*>([\s\S]*?)<\/span>/gi, (_m, text) => `\`${stripTags(text).trim()}\``);
+    // Handle <span class="context-scope-mention"> or <span class="file-chip"> → `text`
+    result = result.replace(/<span[^>]*class="[^"]*(?:context-scope-mention|file-chip)[^"]*"[^>]*>([\s\S]*?)<\/span>/gi, (_m, text) => `\`${stripTags(text).trim()}\``);
     // Handle elements with title attribute containing file paths
     // e.g. <div title="src/bot/index.ts">:54</div> → src/bot/index.ts:54
     result = result.replace(/<(?:div|span|a)[^>]*\btitle="([^"]*)"[^>]*>([\s\S]*?)<\/(?:div|span|a)>/gi, (_m, title, text) => {

package/dist/utils/pathUtils.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getAntigravityCliPath = getAntigravityCliPath;
 exports.extractProjectNameFromPath = extractProjectNameFromPath;
 exports.getAntigravityCdpHint = getAntigravityCdpHint;
+exports.getWorkspaceDirName = getWorkspaceDirName;
 /**
  * Helper to resolve the correct Antigravity CLI executable path based on the operating system
  * and environment variables.
@@ -55,3 +56,14 @@ function getAntigravityCdpHint(port = 9222) {
             return `${APP_NAME.toLowerCase()} --remote-debugging-port=${port}`;
     }
 }
+/**
+ * Extracts the workspace directory name from a chat session.
+ * @param session The chat session object
+ * @returns The base directory name, or undefined
+ */
+function getWorkspaceDirName(session) {
+    if (session && session.workspacePath) {
+        return extractProjectNameFromPath(session.workspacePath);
+    }
+    return undefined;
+}

package/dist/utils/projectResolver.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveProjectName = resolveProjectName;
+function resolveProjectName(deps, channelId, providedProjectName) {
+    if (providedProjectName) {
+        return providedProjectName;
+    }
+    const workspacePath = deps.wsHandler.getWorkspaceForChannel(channelId);
+    return workspacePath ? deps.bridge.pool.extractProjectName(workspacePath) : undefined;
+}

package/dist/utils/questionActionUtils.js

@@ -0,0 +1,25 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseQuestionCustomId = parseQuestionCustomId;
+function parseQuestionCustomId(customId, expectedPrefix) {
+    if (customId !== expectedPrefix && !customId.startsWith(expectedPrefix + ':'))
+        return null;
+    const parts = customId.split(':');
+    const result = { action: parts[0] };
+    try {
+        if (parts.length > 1) {
+            result.projectName = decodeURIComponent(parts[1]);
+        }
+        if (parts.length > 2) {
+            result.channelId = decodeURIComponent(parts[2]);
+        }
+    }
+    catch (e) {
+        // Fallback in case of malformed URI component from truncation
+        if (parts.length > 1)
+            result.projectName = parts[1];
+        if (parts.length > 2)
+            result.channelId = parts[2];
+    }
+    return result;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lazy-gravity",
-  "version": "0.9.2",
+  "version": "0.10.0",
   "description": "Control Antigravity from anywhere — a local, secure bot (Discord + Telegram) that lets you remotely operate Antigravity on your home PC from your smartphone.",
   "main": "dist/index.js",
   "bin": {