lazy-gravity 0.9.2 → 0.10.0

package/dist/services/approvalDetector.js

@@ -3,26 +3,28 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ApprovalDetector = void 0;
 exports.buildClickScript = buildClickScript;
 const logger_1 = require("../utils/logger");
+const consecutiveEmptyPollGate_1 = require("../utils/consecutiveEmptyPollGate");
 /**
  * Approval button detection script for the Antigravity UI
  *
  * Detects allow/deny button pairs and extracts descriptions with fallbacks.
  */
 const DETECT_APPROVAL_SCRIPT = `(() => {
-    const ALLOW_ONCE_PATTERNS = ['allow once', 'allow one time', '今回のみ許可', '1回のみ許可', '一度許可'];
+    const ALLOW_ONCE_PATTERNS = ['allow once', 'allow one time', 'yes, allow this time', '今回のみ許可', '1回のみ許可', '一度許可'];
     const ALWAYS_ALLOW_PATTERNS = [
         'allow this conversation',
         'allow this chat',
         'always allow',
+        'yes, and always allow',
         '常に許可',
         'この会話を許可',
     ];
-    const ALLOW_PATTERNS = ['allow', 'permit', '許可', '承認', '確認'];
-    const DENY_PATTERNS = ['deny', '拒否', 'decline'];
+    const ALLOW_PATTERNS = ['allow', 'permit', 'accept', 'approve', '許可', '承認', '確認'];
+    const DENY_PATTERNS = ['deny', 'reject', '拒否', 'decline', 'no (tell the agent'];
 
     const normalize = (text) => (text || '').toLowerCase().replace(/\\s+/g, ' ').trim();
 
-    const allButtons = Array.from(document.querySelectorAll('button'))
+    const allButtons = Array.from(document.querySelectorAll('button, [role="button"], span.cursor-pointer, div.cursor-pointer'))
         .filter(btn => btn.offsetParent !== null);
 
     let approveBtn = allButtons.find(btn => {
@@ -45,7 +47,7 @@ const DETECT_APPROVAL_SCRIPT = `(() => {
         || approveBtn.parentElement
         || document.body;
 
-    const containerButtons = Array.from(container.querySelectorAll('button'))
+    const containerButtons = Array.from(container.querySelectorAll('button, [role="button"], span.cursor-pointer, div.cursor-pointer'))
         .filter(btn => btn.offsetParent !== null);
 
     const denyBtn = containerButtons.find(btn => {
@@ -78,14 +80,48 @@ const DETECT_APPROVAL_SCRIPT = `(() => {
 
     // 2. Parent element text (excluding button text)
     if (!description) {
-        const parent = approveBtn.parentElement?.parentElement || approveBtn.parentElement;
-        if (parent) {
-            const clone = parent.cloneNode(true);
-            const buttons = clone.querySelectorAll('button');
-            buttons.forEach(b => b.remove());
-            const parentText = (clone.textContent || '').trim();
-            if (parentText.length > 5 && parentText.length < 500) {
-                description = parentText;
+        let modal = approveBtn.closest('.notify-user-container, [role="dialog"], .modal, .dialog, .approval-container, .permission-dialog');
+        
+        if (!modal) {
+            // New Antigravity IDE uses a sticky footer with no identifying modal classes.
+            // Traverse up until we find a container that includes the file list popup (.bottom-full)
+            let p = approveBtn.parentElement;
+            while (p && p.tagName !== 'BODY') {
+                if (p.querySelector('.bottom-full')) {
+                    modal = p;
+                    break;
+                }
+                p = p.parentElement;
+            }
+            if (!modal) {
+                modal = approveBtn.parentElement?.parentElement?.parentElement || approveBtn.parentElement?.parentElement;
+            }
+        }
+
+        if (modal) {
+            const parts = [];
+            const walk = (node) => {
+                if (node.nodeType === 1) {
+                    // Skip buttons entirely
+                    if (node.tagName === 'BUTTON' || node.getAttribute('role') === 'button' || node.classList.contains('cursor-pointer')) return;
+                    
+                    const display = window.getComputedStyle(node).display;
+                    if (display === 'none') return;
+                    
+                    const isBlock = display === 'block' || display === 'flex' || node.tagName === 'DIV' || node.tagName === 'LI';
+                    if (isBlock && parts.length > 0 && parts[parts.length - 1] !== '\\n') parts.push('\\n');
+                    for (const child of node.childNodes) walk(child);
+                    if (isBlock && parts.length > 0 && parts[parts.length - 1] !== '\\n') parts.push('\\n');
+                } else if (node.nodeType === 3) {
+                    const t = node.textContent || '';
+                    if (t.trim()) parts.push(t.trim());
+                }
+            };
+            walk(modal);
+            
+            const parentText = parts.join(' ').replace(/\\n\\s*/g, '\\n').replace(/\\n{3,}/g, '\\n\\n').trim();
+            if (parentText.length > 5) {
+                description = parentText.length > 2000 ? parentText.substring(0, 2000) + '...\\n(truncated)' : parentText;
             }
         }
     }
@@ -102,11 +138,12 @@ const DETECT_APPROVAL_SCRIPT = `(() => {
  * Press the toggle on the right side of Allow Once to expand the Always Allow dropdown.
  */
 const EXPAND_ALWAYS_ALLOW_MENU_SCRIPT = `(() => {
-    const ALLOW_ONCE_PATTERNS = ['allow once', 'allow one time', '今回のみ許可', '1回のみ許可', '一度許可'];
+    const ALLOW_ONCE_PATTERNS = ['allow once', 'allow one time', 'yes, allow this time', '今回のみ許可', '1回のみ許可', '一度許可'];
     const ALWAYS_ALLOW_PATTERNS = [
         'allow this conversation',
         'allow this chat',
         'always allow',
+        'yes, and always allow',
         '常に許可',
         'この会話を許可',
     ];
@@ -181,17 +218,19 @@ function buildClickScript(buttonText) {
         const normalize = (text) => (text || '').toLowerCase().replace(/\\s+/g, ' ').trim();
         const text = ${safeText};
         const wanted = normalize(text);
-        const allButtons = Array.from(document.querySelectorAll('button'));
+        const allButtons = Array.from(document.querySelectorAll('button, [role="button"], a, [class*="btn"], [class*="button"], [class*="action"]')).reverse();
         const target = allButtons.find(btn => {
-            if (!btn.offsetParent) return false;
+            const style = window.getComputedStyle(btn);
+            if (style.display === 'none' || style.visibility === 'hidden' || btn.disabled) return false;
             const buttonText = normalize(btn.textContent || '');
             const ariaLabel = normalize(btn.getAttribute('aria-label') || '');
             return buttonText === wanted ||
                 ariaLabel === wanted ||
-                buttonText.includes(wanted) ||
-                ariaLabel.includes(wanted);
+                (buttonText.includes(wanted) && buttonText.length < wanted.length + 10) ||
+                (ariaLabel.includes(wanted) && ariaLabel.length < wanted.length + 10);
         });
         if (!target) return { ok: false, error: 'Button not found: ' + text };
+        target.scrollIntoView({ block: 'center' });
         const rect = target.getBoundingClientRect();
         const x = rect.left + rect.width / 2;
         const y = rect.top + rect.height / 2;
@@ -199,6 +238,7 @@ function buildClickScript(buttonText) {
         for (const type of ['pointerdown', 'mousedown', 'pointerup', 'mouseup', 'click']) {
             target.dispatchEvent(new PointerEvent(type, { ...eventInit, pointerId: 1 }));
         }
+        if (typeof target.click === 'function') target.click();
         return { ok: true };
     })()`;
 }
@@ -219,6 +259,8 @@ class ApprovalDetector {
     lastDetectedKey = null;
     /** Full ApprovalInfo from the last detection (used for clicking) */
     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;
@@ -234,6 +276,7 @@ class ApprovalDetector {
         this.isRunning = true;
         this.lastDetectedKey = null;
         this.lastDetectedInfo = null;
+        this.emptyPollGate.reset();
         this.schedulePoll();
     }
     /**
@@ -284,6 +327,7 @@ class ApprovalDetector {
             const result = await this.cdpService.call('Runtime.evaluate', callParams);
             const info = result?.result?.value ?? null;
             if (info) {
+                this.emptyPollGate.recordDetection();
                 // Duplicate prevention: use approveText + description combination as key
                 const key = `${info.approveText}::${info.description}`;
                 if (key !== this.lastDetectedKey) {
@@ -293,12 +337,14 @@ class ApprovalDetector {
                 }
             }
             else {
-                // Reset when buttons disappear (prepare for next approval detection)
-                const wasDetected = this.lastDetectedKey !== null;
-                this.lastDetectedKey = null;
-                this.lastDetectedInfo = null;
-                if (wasDetected && this.onResolved) {
-                    this.onResolved();
+                if (this.emptyPollGate.recordEmptyPoll()) {
+                    // Reset when buttons disappear for consecutive polls
+                    const wasDetected = this.lastDetectedKey !== null;
+                    this.lastDetectedKey = null;
+                    this.lastDetectedInfo = null;
+                    if (wasDetected && this.onResolved) {
+                        this.onResolved();
+                    }
                 }
             }
         }

package/dist/services/artifactService.js

@@ -148,11 +148,12 @@ class ArtifactService {
         }
     }
     /**
-     * Try to find a conversation UUID whose overview.txt contains the given session title.
+     * Try to find a conversation UUID whose transcript or overview contains the given session title.
      * Uses an exact match first, falling back to keyword overlap scoring.
+     * If workspaceFilter is provided, restricts matching exclusively to conversations belonging to that workspace.
      * Returns the UUID or null if not found.
      */
-    findConversationByTitle(title) {
+    findConversationByTitle(title, workspaceFilter) {
         if (!title || !title.trim())
             return null;
         const needle = title.trim().toLowerCase();
@@ -186,40 +187,58 @@ class ArtifactService {
         const uniqueNeedleWords = Array.from(new Set(cleanNeedleWords));
         // Require a stronger minimum score of 2 to prevent weak one-word matches.
         const minScore = 2;
+        const filterStr = workspaceFilter ? workspaceFilter.toLowerCase() : null;
         for (const id of sortedIds) {
-            const overviewPath = path.join(this.brainBasePath, id, '.system_generated', 'logs', 'overview.txt');
-            try {
-                if (!fs.existsSync(overviewPath))
-                    continue;
-                // Only read the first 4KB to avoid huge files
-                let fd = null;
+            let score = 0;
+            let exactMatch = false;
+            let belongsToWorkspace = !filterStr; // True if no filter, or if we prove it belongs
+            const scanFile = (filePath, readSize) => {
                 try {
-                    fd = fs.openSync(overviewPath, 'r');
-                    const buf = Buffer.alloc(4096);
-                    const bytesRead = fs.readSync(fd, buf, 0, buf.length, 0);
-                    const header = buf.slice(0, bytesRead).toString('utf-8').toLowerCase();
-                    if (header.includes(needle)) {
-                        return id; // Exact match takes precedence
-                    }
-                    // Use same Unicode-aware split for header tokens
-                    const headerTokens = new Set(header.replace(/[^\p{L}\p{N}]+/gu, ' ').split(/\s+/));
-                    let score = 0;
-                    for (const word of uniqueNeedleWords) {
-                        if (headerTokens.has(word))
-                            score++;
+                    if (!fs.existsSync(filePath))
+                        return;
+                    let fd = null;
+                    try {
+                        fd = fs.openSync(filePath, 'r');
+                        const buf = Buffer.alloc(readSize);
+                        const bytesRead = fs.readSync(fd, buf, 0, buf.length, 0);
+                        const content = buf.slice(0, bytesRead).toString('utf-8').toLowerCase();
+                        if (filterStr && content.includes(filterStr)) {
+                            belongsToWorkspace = true;
+                        }
+                        if (belongsToWorkspace) {
+                            if (content.includes(needle)) {
+                                exactMatch = true;
+                            }
+                            else {
+                                const contentTokens = new Set(content.replace(/[^\p{L}\p{N}]+/gu, ' ').split(/\s+/));
+                                let currentScore = 0;
+                                for (const word of uniqueNeedleWords) {
+                                    if (contentTokens.has(word))
+                                        currentScore++;
+                                }
+                                if (currentScore > score)
+                                    score = currentScore;
+                            }
+                        }
                     }
-                    if (score > bestScore) {
-                        bestScore = score;
-                        bestId = id;
+                    finally {
+                        if (fd !== null)
+                            fs.closeSync(fd);
                     }
                 }
-                finally {
-                    if (fd !== null)
-                        fs.closeSync(fd);
-                }
-            }
-            catch {
-                // Skip unreadable files
+                catch { /* ignore */ }
+            };
+            // 1. Check transcript.jsonl (modern)
+            scanFile(path.join(this.brainBasePath, id, '.system_generated', 'logs', 'transcript.jsonl'), 16384);
+            // 2. Check overview.txt (legacy)
+            scanFile(path.join(this.brainBasePath, id, '.system_generated', 'logs', 'overview.txt'), 4096);
+            if (!belongsToWorkspace)
+                continue;
+            if (exactMatch)
+                return id; // Exact match in the right workspace takes precedence immediately
+            if (score > bestScore) {
+                bestScore = score;
+                bestId = id;
             }
         }
         if (bestScore >= minScore) {
@@ -256,6 +275,13 @@ class ArtifactService {
         // Nothing has artifacts — return most recent anyway (caller handles empty list)
         return sorted[0]?.id ?? null;
     }
+    /**
+     * Build the filesystem path for a specific artifact file.
+     */
+    getArtifactPath(conversationId, filename) {
+        const safe = path.basename(filename);
+        return path.join(this.brainBasePath, conversationId, safe);
+    }
     /**
      * Read the markdown content of a specific artifact file.
      * Returns null if the file cannot be read.

package/dist/services/assistantDomExtractor.js

@@ -27,6 +27,12 @@ function classifyAssistantSegments(payload) {
             finalOutputText: '',
             activityLines: [],
             feedback: [],
+            planCards: [],
+            actionButtons: [],
+            fileChanges: [],
+            citations: [],
+            fileChangesTexts: [],
+            citedFiles: [],
             diagnostics: {
                 source: 'legacy-fallback',
                 segmentCounts: {},
@@ -41,8 +47,13 @@ function classifyAssistantSegments(payload) {
     const bodyTexts = [];
     const activityLines = [];
     const feedbackTexts = [];
+    const planCards = [];
+    const actionButtons = [];
+    const fileChanges = [];
+    const citations = [];
     const segmentCounts = {};
     const allFingerprints = [];
+    const fileChangesTexts = [];
     for (const seg of segments) {
         segmentCounts[seg.kind] = (segmentCounts[seg.kind] ?? 0) + 1;
         switch (seg.kind) {
@@ -65,16 +76,68 @@ function classifyAssistantSegments(payload) {
                     feedbackTexts.push(seg.text.trim());
                 }
                 break;
+            case 'plan-card':
+                if (seg.text && seg.text.trim()) {
+                    planCards.push(seg.text.trim());
+                }
+                break;
+            case 'action-button':
+                if (seg.text && seg.text.trim()) {
+                    actionButtons.push(seg.text.trim());
+                }
+                break;
+            case 'file-change':
+                if (seg.text && seg.text.trim()) {
+                    const parts = seg.text.split('|');
+                    fileChanges.push({ path: parts[0]?.trim() || '', type: parts[1]?.trim() || '' });
+                }
+                break;
+            case 'citation':
+                if (seg.text && seg.text.trim()) {
+                    citations.push(seg.text.trim());
+                }
+                break;
+            case 'file-changes':
+                if (seg.text && seg.text.trim()) {
+                    fileChangesTexts.push(seg.text.trim());
+                }
+                break;
         }
     }
     // Concatenate all body segments — the CDP script may yield multiple nodes
     // (streaming partial + final, multi-block markdown, etc.)
     const lastBody = bodyTexts.join('\n\n') || '';
     const finalOutputText = (0, htmlToDiscordMarkdown_1.htmlToDiscordMarkdown)(lastBody);
+    // Extract file:/// links from finalOutputText and preserve citations from Pass 4
+    const citedFiles = Array.from(new Set([...citations]));
+    const fileRegex = /\[.*?\]\((file:\/\/\/[^\)]+)\)/g;
+    let match;
+    while ((match = fileRegex.exec(finalOutputText)) !== null) {
+        const fileUrl = match[1];
+        if (!citedFiles.includes(fileUrl)) {
+            citedFiles.push(fileUrl);
+        }
+    }
+    // Also extract backticked inline code that looks like a filename with common extensions
+    // e.g., `implementation_plan.md`
+    const inlineCodeRegex = /`([a-zA-Z0-9_\-\.]+\.(?:md|ts|js|json|html|css|txt|csv|py|java|go|cpp|c))`[^`]?/g;
+    let inlineMatch;
+    while ((inlineMatch = inlineCodeRegex.exec(finalOutputText)) !== null) {
+        const filename = inlineMatch[1];
+        if (!citedFiles.includes(filename)) {
+            citedFiles.push(filename);
+        }
+    }
     return {
         finalOutputText,
         activityLines,
         feedback: feedbackTexts,
+        planCards,
+        actionButtons,
+        fileChanges,
+        citations,
+        fileChangesTexts,
+        citedFiles,
         diagnostics: {
             source: 'dom-structured',
             segmentCounts,
@@ -145,10 +208,17 @@ function extractAssistantSegmentsPayloadScript() {
         if (node.closest('details')) return true;
         if (node.closest('[class*="feedback"], footer')) return true;
         if (node.closest('.notify-user-container')) return true;
-        if (node.closest('[role="dialog"]')) return true;
-        if (node.closest('form')) return true;
+        // Guard against approval dialogs leaking into the output (they contain plan cards or review buttons)
+        var dialogNode = node.closest('dialog, [role="dialog"], form');
+        if (dialogNode && dialogNode.querySelector('[data-testid="plan-card"], [class*="plan-summary"], .actions-container, .review-button')) {
+            return true;
+        }
+        
+        // allow interactive tool forms (ask_question, ask_permission)
         if (node.closest('[data-message-author-role="user"], [data-message-role="user"]')) return true;
-        if (node.querySelector('textarea') || node.closest('textarea')) return true;
+        // We do not exclude nodes just because they contain a textarea (e.g. ask_question "Other" option).
+        // The main chat input is excluded by the "ask anything" check below.
+        if (node.tagName && node.tagName.toLowerCase() === 'textarea') return true;
         var text = (node.innerText || '').toLowerCase();
         if (text.includes('ask anything, @ to mention')) return true;
         if (text.includes('0 files with changes')) return true;
@@ -158,6 +228,7 @@ function extractAssistantSegmentsPayloadScript() {
     var segments = [];
     var seen = new Set();
     var bodyFound = false;
+    var latestMessageContainer = null;
 
     // Pass 1: Find assistant body — last non-excluded content node (recency first)
     var combinedSelector = selectors.join(', ');
@@ -238,6 +309,7 @@ function extractAssistantSegmentsPayloadScript() {
                 domPath: 'article:nth(' + i + ')'
             });
             bodyFound = true;
+            latestMessageContainer = node.closest('[data-message-id], .message-row, [class*="message-container"]') || node.parentNode || scope;
             break; // Only capture the single latest assistant message
         }
     }
@@ -362,6 +434,105 @@ function extractAssistantSegmentsPayloadScript() {
         }
     }
 
+    // Pass 4: Extract Antigravity 2.0 structured cards and artifacts
+    var artifactScope = latestMessageContainer || scope;
+    // 4a. Plan Cards
+    var planNodes = artifactScope.querySelectorAll('[data-testid="plan-card"], [class*="plan-summary"]');
+    for (var pi2 = 0; pi2 < planNodes.length; pi2++) {
+        var pnode = planNodes[pi2];
+        var ptext = (pnode.innerText || pnode.textContent || '').trim();
+        if (ptext) {
+            segments.push({
+                kind: 'plan-card',
+                text: ptext,
+                role: 'assistant',
+                messageIndex: 0,
+                domPath: 'plan-card:nth(' + pi2 + ')'
+            });
+        }
+    }
+
+    // 4b. Action Buttons (Open, Proceed, Review, etc.) - outside feedback footers
+    var actionBtns = artifactScope.querySelectorAll('.actions-container button, [class*="action-btn"], .review-button, [class*="review-btn"]');
+    for (var abi = 0; abi < actionBtns.length; abi++) {
+        var abtnText = (actionBtns[abi].textContent || '').trim();
+        if (!abtnText) continue;
+        var lower = abtnText.toLowerCase();
+        // Ignore approval buttons since they are handled by approvalDetector
+        if (lower.includes('accept all') || lower.includes('reject all')) continue;
+
+        segments.push({
+            kind: 'action-button',
+            text: abtnText,
+            role: 'assistant',
+            messageIndex: 0,
+            domPath: 'action-btn.' + abtnText.toLowerCase().replace(/\s+/g, '-')
+        });
+    }
+
+    // 4c. File Edits
+    var fileEdits = artifactScope.querySelectorAll('[class*="file-edit-item"], .file-edit-item, .bottom-full .cursor-pointer');
+    for (var fei = 0; fei < fileEdits.length; fei++) {
+        var editNode = fileEdits[fei];
+        if (seen.has(editNode)) continue;
+        seen.add(editNode);
+        var fpath = (editNode.querySelector('.file-path, [class*="path"]') || {}).textContent || '';
+        var ftype = (editNode.querySelector('.edit-type, [class*="type"]') || {}).textContent || 'Modified';
+        
+        if (!fpath) {
+            var spans = editNode.querySelectorAll('span.whitespace-nowrap.text-sm');
+            if (spans.length > 0) {
+                fpath = spans[0].textContent || '';
+            }
+        }
+        if (!fpath) {
+            fpath = (editNode.textContent || '').trim();
+            if (fpath.endsWith(ftype)) {
+                fpath = fpath.slice(0, -ftype.length).trim();
+            }
+        }
+        if (fpath) {
+            segments.push({
+                kind: 'file-change',
+                text: fpath.trim() + '|' + ftype.trim(),
+                role: 'assistant',
+                messageIndex: 0,
+                domPath: 'file-edit:nth(' + fei + ')'
+            });
+        }
+    }
+
+    // 4d. Citations
+    var citationNodes = artifactScope.querySelectorAll('[class*="citation"], a[href^="file://"]');
+    for (var cti = 0; cti < citationNodes.length; cti++) {
+        var cnode = citationNodes[cti];
+        var ctext = (cnode.getAttribute('href') || cnode.textContent || '').trim();
+        if (ctext) {
+            segments.push({
+                kind: 'citation',
+                text: ctext,
+                role: 'assistant',
+                messageIndex: 0,
+                domPath: 'citation:nth(' + cti + ')'
+            });
+        }
+    }
+    // 4e. File Changes Text Blocks
+    var fcbNodes = artifactScope.querySelectorAll('.file-changes-block code, [class*="file-changes"] code');
+    for (var fci = 0; fci < fcbNodes.length; fci++) {
+        var fcNode = fcbNodes[fci];
+        var fcText = (fcNode.textContent || '').trim();
+        if (fcText) {
+            segments.push({
+                kind: 'file-changes',
+                text: fcText,
+                role: 'assistant',
+                messageIndex: 0,
+                domPath: 'file-changes:nth(' + fci + ')'
+            });
+        }
+    }
+
     if (!bodyFound && segments.length === 0) return null;
 
     return {