lazy-gravity 0.10.0 → 0.11.0

package/dist/services/cdpConnectionPool.js

@@ -11,6 +11,7 @@ function buildConnectionKey(projectName, accountName) {
  * Pool that manages independent CdpService instances per workspace/account pair.
  */
 class CdpConnectionPool {
+    lastActiveWorkspace = null;
     connections = new Map();
     workspaceToAccount = new Map();
     approvalDetectors = new Map();
@@ -40,6 +41,7 @@ class CdpConnectionPool {
         const key = buildConnectionKey(projectName, effectiveAccount);
         const existing = this.connections.get(key);
         if (existing && existing.isConnected()) {
+            this.lastActiveWorkspace = projectName;
             await existing.discoverAndConnectForWorkspace(workspacePath);
             return existing;
         }
@@ -50,7 +52,9 @@ class CdpConnectionPool {
         const connectPromise = this.createAndConnect(workspacePath, projectName, effectiveAccount);
         this.connectingPromises.set(key, connectPromise);
         try {
-            return await connectPromise;
+            const result = await connectPromise;
+            this.lastActiveWorkspace = projectName;
+            return result;
         }
         finally {
             this.connectingPromises.delete(key);
@@ -59,8 +63,9 @@ class CdpConnectionPool {
     }
     getConnected(projectName, accountName = 'default') {
         const effectiveAccount = this.resolveAccountName(projectName, accountName);
-        const cdp = this.connections.get(buildConnectionKey(projectName, effectiveAccount));
+        const cdp = this.connections.get(buildConnectionKey(projectName, effectiveAccount)) || null;
         if (cdp && cdp.isConnected()) {
+            this.lastActiveWorkspace = projectName;
             return cdp;
         }
         return null;

package/dist/services/cdpService.js

@@ -102,6 +102,12 @@ const WORKSPACE_STATE_SCRIPT = `(() => {
             if (isGenerating) break;
         }
     }
+    if (!isGenerating && panel) {
+        const panelText = (panel.textContent || '').trim();
+        if (/Working\.\s*$/i.test(panelText)) {
+            isGenerating = true;
+        }
+    }
 
     if (!panel) {
         return { isGenerating, sessionTitle: '', hasActiveChat: false };
@@ -462,7 +468,8 @@ class CdpService extends events_1.EventEmitter {
                 returnByValue: true,
             });
             const liveTitle = String(titleResult?.result?.value || '');
-            if (liveTitle.toLowerCase().includes(projectName.toLowerCase())) {
+            const titleParts = liveTitle.split(' - ');
+            if (titleParts[0].trim().toLowerCase() === projectName.toLowerCase()) {
                 this.currentWorkspaceName = projectName;
                 return true;
             }
@@ -509,7 +516,12 @@ class CdpService extends events_1.EventEmitter {
             logger_1.logger.debug(`  - title="${p.title}" url=${p.url}`);
         }
         // 1. Title match (fast path)
-        const titleMatch = workbenchPages.find((t) => t.title?.includes(projectName));
+        const titleMatch = workbenchPages.find((t) => {
+            if (!t.title)
+                return false;
+            const parts = t.title.split(' - ');
+            return parts[0].trim().toLowerCase() === projectName.toLowerCase();
+        });
         if (titleMatch) {
             return this.connectToPage(titleMatch, projectName);
         }
@@ -562,9 +574,8 @@ class CdpService extends events_1.EventEmitter {
                     returnByValue: true,
                 });
                 const liveTitle = String(result?.result?.value || '');
-                const normalizedLiveTitle = liveTitle.toLowerCase();
-                const normalizedProject = projectName.toLowerCase();
-                if (normalizedLiveTitle.includes(normalizedProject)) {
+                const liveParts = liveTitle.split(' - ');
+                if (liveParts[0].trim().toLowerCase() === projectName.toLowerCase()) {
                     this.currentWorkspaceName = projectName;
                     logger_1.logger.debug(`[CdpService] Probe success: detected "${projectName}"`);
                     return true;
@@ -761,7 +772,12 @@ class CdpService extends events_1.EventEmitter {
                 !t.url?.includes('workbench-jetski-agent') &&
                 t.url?.includes('workbench'));
             // Title match
-            const titleMatch = workbenchPages.find((t) => t.title?.toLowerCase().includes(projectName.toLowerCase()));
+            const titleMatch = workbenchPages.find((t) => {
+                if (!t.title)
+                    return false;
+                const parts = t.title.split(' - ');
+                return parts[0].trim().toLowerCase() === projectName.toLowerCase();
+            });
             if (titleMatch) {
                 return this.connectToPage(titleMatch, projectName);
             }
@@ -1290,9 +1306,15 @@ class CdpService extends events_1.EventEmitter {
         ].some((fragment) => message.includes(fragment));
     }
     async injectMessageCore(text, imageFilePaths) {
-        const focusResult = await this.waitForChatInputReady();
+        // Check if chat input is already available
+        let focusResult = await this.waitForChatInputReady(500);
         if (!focusResult.ok) {
-            return { ok: false, error: focusResult.error || 'Chat input field not found' };
+            // Send Cmd+L / Ctrl+L to guarantee application-level focus on the chat panel (and open it if closed)
+            await this.focusChatPanelViaShortcut();
+            focusResult = await this.waitForChatInputReady();
+            if (!focusResult.ok) {
+                return { ok: false, error: focusResult.error || 'Chat input field not found' };
+            }
         }
         // Clear any existing text in the input field before injecting.
         await this.clearInputField();
@@ -1391,6 +1413,89 @@ class CdpService extends events_1.EventEmitter {
             nativeVirtualKeyCode: 13,
         });
     }
+    /**
+     * Opens the chat panel and waits for the UI to be ready.
+     * Can be called right after the IDE launches to ensure the panel is visible.
+     */
+    async openChatPanel() {
+        if (!await this.isConnected())
+            return;
+        // Ensure IDE UI is fully loaded before sending shortcut
+        const deadline = Date.now() + 15000;
+        let workbenchReady = false;
+        while (Date.now() < deadline) {
+            try {
+                const res = await this.call('Runtime.evaluate', {
+                    expression: '!!document.querySelector(".monaco-workbench")',
+                    returnByValue: true
+                });
+                if (res?.result?.value) {
+                    workbenchReady = true;
+                    break;
+                }
+            }
+            catch {
+                // ignore
+            }
+            await new Promise(r => setTimeout(r, 500));
+        }
+        if (workbenchReady) {
+            // Check if chat panel is already open to avoid toggling it closed
+            const focusResult = await this.waitForChatInputReady(1000);
+            if (!focusResult.ok) {
+                await this.focusChatPanelViaShortcut();
+            }
+        }
+    }
+    /**
+     * Send Cmd+L / Ctrl+L to focus the chat panel via shortcut.
+     */
+    async focusChatPanelViaShortcut() {
+        const modifiers = process.platform === 'darwin' ? 4 : 2; // Meta : Ctrl
+        await this.call('Input.dispatchKeyEvent', {
+            type: 'keyDown',
+            key: 'l',
+            code: 'KeyL',
+            modifiers,
+            windowsVirtualKeyCode: 76,
+            nativeVirtualKeyCode: 76,
+        });
+        await this.call('Input.dispatchKeyEvent', {
+            type: 'keyUp',
+            key: 'l',
+            code: 'KeyL',
+            modifiers,
+            windowsVirtualKeyCode: 76,
+            nativeVirtualKeyCode: 76,
+        });
+        await new Promise(r => setTimeout(r, 100)); // wait for focus to shift
+    }
+    /**
+     * Send Cmd+W / Ctrl+W to close the active editor.
+     */
+    async closeActiveEditor() {
+        if (!this.isConnectedFlag || !this.ws) {
+            throw new Error('Not connected to CDP. Call connect() first.');
+        }
+        const modifiers = process.platform === 'darwin' ? 4 : 2; // Meta : Ctrl
+        await this.call('Input.dispatchKeyEvent', {
+            type: 'keyDown',
+            key: 'w',
+            code: 'KeyW',
+            modifiers,
+            windowsVirtualKeyCode: 87,
+            nativeVirtualKeyCode: 87,
+        });
+        await this.call('Input.dispatchKeyEvent', {
+            type: 'keyUp',
+            key: 'w',
+            code: 'KeyW',
+            modifiers,
+            windowsVirtualKeyCode: 87,
+            nativeVirtualKeyCode: 87,
+        });
+        await new Promise(r => setTimeout(r, 100));
+    }
     /**
      * Detect file input in the UI and attach the specified files.
      */

package/dist/services/chatSessionService.js

@@ -40,7 +40,7 @@ const GET_NEW_CHAT_BUTTON_SCRIPT = `(() => {
  * The title element is a div with the text-ellipsis class inside the header.
  */
 const GET_CHAT_TITLE_SCRIPT = `(() => {
-    const panel = document.querySelector('.antigravity-agent-side-panel');
+    const panel = document.querySelector('.antigravity-agent-side-panel') || document.body;
     if (!panel) return { title: '', hasActiveChat: false };
     const header = panel.querySelector('div[class*="border-b"]');
     if (!header) return { title: '', hasActiveChat: false };
@@ -58,7 +58,7 @@ const GET_CHAT_TITLE_SCRIPT = `(() => {
     return { title: title || '(Untitled)', hasActiveChat };
 })()`;
 const GET_SESSION_VIEW_STATE_SCRIPT = `(() => {
-    const panel = document.querySelector('.antigravity-agent-side-panel');
+    const panel = document.querySelector('.antigravity-agent-side-panel') || document.body;
     if (!panel) {
         return {
             panelFound: false,
@@ -169,7 +169,8 @@ const SCRAPE_PAST_CONVERSATIONS_SCRIPT = `(() => {
     // Try the visible QuickInput dialog first, then fall back to the side panel.
     const quickInputPanels = Array.from(document.querySelectorAll('div[class*="bg-quickinput-background"]'));
     const panel = quickInputPanels.find((el) => isVisible(el))
-        || document.querySelector('.antigravity-agent-side-panel');
+        || document.querySelector('.antigravity-agent-side-panel')
+        || document.body;
     if (!panel) return null;
 
     const items = [];
@@ -276,9 +277,11 @@ function buildActivateChatByTitleScript(title) {
         for (const node of nodes) {
             const text = normalize(node.textContent || '');
             if (!text) continue;
+            
+            const isShort = wanted.length < 5;
             if (text === wanted) {
                 exact.push({ node, textLength: text.length });
-            } else if (text.includes(wanted)) {
+            } else if (!isShort && text.includes(wanted)) {
                 includes.push({ node, textLength: text.length });
             } else if (wordsMatch(text, wantedRaw)) {
                 fuzzy.push({ node, textLength: text.length });
@@ -369,10 +372,12 @@ function buildActivateViaPastConversationsScript(title) {
                     if (!pattern) continue;
                     const p = normalize(pattern);
                     const pLoose = normalizeLoose(pattern);
+                    const isShort = p.length < 5;
+                    
                     if (
                         text === p ||
-                        text.includes(p) ||
-                        (pLoose && (textLoose === pLoose || textLoose.includes(pLoose)))
+                        (pLoose && textLoose === pLoose) ||
+                        (!isShort && (text.includes(p) || (pLoose && textLoose.includes(pLoose))))
                     ) {
                         matched.push({ el, score: Math.abs(text.length - pattern.length) });
                         break;
@@ -611,7 +616,8 @@ class ChatSessionService {
                 const isVisible = (el) => !!el && el instanceof HTMLElement && el.offsetParent !== null;
                 const quickInputPanels = Array.from(document.querySelectorAll('div[class*="bg-quickinput-background"]'));
                 const panel = quickInputPanels.find((el) => isVisible(el))
-                    || document.querySelector('.antigravity-agent-side-panel');
+                    || document.querySelector('.antigravity-agent-side-panel')
+                    || document.body;
                 if (!panel) return false;
                 const containers = Array.from(
                     panel.querySelectorAll('div[class*="overflow-auto"], div[class*="overflow-y-scroll"]')
@@ -1095,12 +1101,13 @@ class ChatSessionService {
     /**
      * Rename the current chat in the Antigravity UI directly by updating the DOM.
      * Note: This is a cosmetic change until Antigravity persists a rename.
+     * IDE syncing for chat renaming is strictly "best-effort".
      */
     async renameCurrentChatInUI(cdpService, newTitle) {
         try {
             const contextId = cdpService.getPrimaryContextId();
             const RENAME_SCRIPT = `(() => {
-                const panel = document.querySelector('.antigravity-agent-side-panel');
+                const panel = document.querySelector('.antigravity-agent-side-panel') || document.body;
                 if (!panel) return { ok: false, error: 'Panel not found' };
                 const header = panel.querySelector('div[class*="border-b"]');
                 const titleEl = header?.querySelector('div[class*="text-ellipsis"]');

package/dist/services/notificationSender.js

@@ -82,7 +82,7 @@ function customId(prefix, projectName, channelId) {
 // ---------------------------------------------------------------------------
 /** Build the approval notification message. */
 function buildApprovalNotification(opts) {
-    const { title, description, projectName, channelId, toolNames, extraFields, hasAlwaysAllow, alwaysAllowText } = opts;
+    const { title, description, projectName, channelId, toolNames, extraFields, hasAlwaysAllow, alwaysAllowText, walkthroughCustomId, taskCustomId } = opts;
     const richContent = (0, richContentBuilder_1.pipe)((0, richContentBuilder_1.createRichContent)(), (rc) => (0, richContentBuilder_1.withTitle)(rc, title), (rc) => (0, richContentBuilder_1.withDescription)(rc, description), (rc) => (0, richContentBuilder_1.withColor)(rc, COLOR_APPROVAL), (rc) => (0, richContentBuilder_1.addField)(rc, 'Project', projectName, true), (rc) => toolNames && toolNames.length > 0
         ? (0, richContentBuilder_1.addField)(rc, 'Tools', toolNames.join(', '), true)
         : rc, (rc) => extraFields
@@ -98,6 +98,16 @@ function buildApprovalNotification(opts) {
     const components = [
         buttonRow(...buttons),
     ];
+    const artifactButtons = [];
+    if (walkthroughCustomId) {
+        artifactButtons.push(button(walkthroughCustomId, 'Review walkthrough.md', 'success'));
+    }
+    if (taskCustomId) {
+        artifactButtons.push(button(taskCustomId, 'Review task.md', 'primary'));
+    }
+    if (artifactButtons.length > 0) {
+        components.push(buttonRow(...artifactButtons));
+    }
     return { richContent, components };
 }
 /** Build the planning mode notification message. */
@@ -107,8 +117,8 @@ function buildPlanningNotification(opts) {
         ? extraFields.reduce((acc, f) => (0, richContentBuilder_1.addField)(acc, f.name, f.value, f.inline), rc)
         : rc, (rc) => (0, richContentBuilder_1.withFooter)(rc, 'Planning mode detected'), (rc) => (0, richContentBuilder_1.withTimestamp)(rc));
     const buttons = [];
-    const openLabel = opts.openText || 'Open';
     if (opts.hasOpenButton !== false) {
+        const openLabel = opts.openText || 'Open Plan';
         buttons.push(button(customId(PLANNING_OPEN_ACTION_PREFIX, projectName, channelId), openLabel, 'primary'));
     }
     const buttonLabel = opts.proceedText || 'Proceed';

package/dist/services/planningDetector.js

@@ -247,7 +247,7 @@ class PlanningDetector {
      * @returns true if click succeeded
      */
     async clickRejectButton(buttonText) {
-        const text = buttonText ?? this.lastDetectedInfo?.rejectText ?? 'Reject All';
+        const text = buttonText || this.lastDetectedInfo?.rejectText || 'Reject All';
         return this.clickButton(text);
     }
     /**

package/dist/services/promptDispatcher.js

@@ -1,12 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PromptDispatcher = void 0;
+const logger_1 = require("../utils/logger");
 /**
  * Dispatcher that calls the existing sendPromptToAntigravity.
  * Unifies dependency injection on the caller side and simplifies event handlers.
  */
 class PromptDispatcher {
     deps;
+    activeMonitors = new Map();
     constructor(deps) {
         this.deps = deps;
     }
@@ -17,7 +19,25 @@ class PromptDispatcher {
         await this._dispatch(req, { ...req.options, resumeOnly: true });
     }
     async _dispatch(req, options) {
-        await this.deps.sendPromptImpl(this.deps.bridge, req.message, req.prompt, req.cdp, this.deps.modeService, this.deps.modelService, req.inboundImages ?? [], options);
+        const channelId = req.message.channelId;
+        const existing = this.activeMonitors.get(channelId);
+        if (existing) {
+            logger_1.logger.info(`[PromptDispatcher] Aborting previous active monitor for channel ${channelId}`);
+            await existing.stop().catch((err) => logger_1.logger.error('[PromptDispatcher] Error stopping monitor:', err));
+            this.activeMonitors.delete(channelId);
+        }
+        const wrappedOptions = {
+            ...(options || {}),
+            onMonitorCreated: (monitor) => {
+                this.activeMonitors.set(channelId, monitor);
+                options?.onMonitorCreated?.(monitor);
+            },
+            onFullCompletion: () => {
+                this.activeMonitors.delete(channelId);
+                options?.onFullCompletion?.();
+            }
+        };
+        await this.deps.sendPromptImpl(this.deps.bridge, req.message, req.prompt, req.cdp, this.deps.modeService, this.deps.modelService, req.inboundImages ?? [], wrappedOptions);
     }
 }
 exports.PromptDispatcher = PromptDispatcher;