lazy-gravity 0.9.2 → 0.11.0

package/dist/services/notificationSender.js

@@ -6,6 +6,7 @@
  * They return a `MessagePayload` that any platform adapter can render.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.QUESTION_SKIP_ACTION_PREFIX = exports.QUESTION_SELECT_ACTION_PREFIX = exports.FEEDBACK_ACTION_PREFIX = void 0;
 exports.buildApprovalNotification = buildApprovalNotification;
 exports.buildPlanningNotification = buildPlanningNotification;
 exports.buildErrorPopupNotification = buildErrorPopupNotification;
@@ -13,6 +14,7 @@ exports.buildRunCommandNotification = buildRunCommandNotification;
 exports.buildAutoApprovedNotification = buildAutoApprovedNotification;
 exports.buildResolvedOverlay = buildResolvedOverlay;
 exports.buildStatusNotification = buildStatusNotification;
+exports.buildQuestionNotification = buildQuestionNotification;
 exports.buildProgressNotification = buildProgressNotification;
 const richContentBuilder_1 = require("../platform/richContentBuilder");
 // ---------------------------------------------------------------------------
@@ -28,6 +30,9 @@ const ERROR_POPUP_COPY_DEBUG_ACTION_PREFIX = 'error_popup_copy_debug_action';
 const ERROR_POPUP_RETRY_ACTION_PREFIX = 'error_popup_retry_action';
 const RUN_COMMAND_RUN_ACTION_PREFIX = 'run_command_run_action';
 const RUN_COMMAND_REJECT_ACTION_PREFIX = 'run_command_reject_action';
+exports.FEEDBACK_ACTION_PREFIX = 'feedback_action';
+exports.QUESTION_SELECT_ACTION_PREFIX = 'question_select_action';
+exports.QUESTION_SKIP_ACTION_PREFIX = 'question_skip_action';
 // ---------------------------------------------------------------------------
 // Notification colours
 // ---------------------------------------------------------------------------
@@ -77,15 +82,32 @@ function customId(prefix, projectName, channelId) {
 // ---------------------------------------------------------------------------
 /** Build the approval notification message. */
 function buildApprovalNotification(opts) {
-    const { title, description, projectName, channelId, toolNames, extraFields } = 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
         ? extraFields.reduce((acc, f) => (0, richContentBuilder_1.addField)(acc, f.name, f.value, f.inline), rc)
         : rc, (rc) => (0, richContentBuilder_1.withFooter)(rc, 'Approval required'), (rc) => (0, richContentBuilder_1.withTimestamp)(rc));
+    const buttons = [
+        button(customId(APPROVE_ACTION_PREFIX, projectName, channelId), 'Allow', 'success'),
+    ];
+    if (hasAlwaysAllow) {
+        buttons.push(button(customId(ALWAYS_ALLOW_ACTION_PREFIX, projectName, channelId), alwaysAllowText || 'Allow Chat', 'primary'));
+    }
+    buttons.push(button(customId(DENY_ACTION_PREFIX, projectName, channelId), 'Deny', 'danger'));
     const components = [
-        buttonRow(button(customId(APPROVE_ACTION_PREFIX, projectName, channelId), 'Allow', 'success'), button(customId(ALWAYS_ALLOW_ACTION_PREFIX, projectName, channelId), 'Allow Chat', 'primary'), button(customId(DENY_ACTION_PREFIX, projectName, channelId), 'Deny', 'danger')),
+        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. */
@@ -94,8 +116,15 @@ function buildPlanningNotification(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_PLANNING), (rc) => extraFields
         ? 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 = [];
+    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';
+    buttons.push(button(customId(PLANNING_PROCEED_ACTION_PREFIX, projectName, channelId), buttonLabel, 'success'));
     const components = [
-        buttonRow(button(customId(PLANNING_OPEN_ACTION_PREFIX, projectName, channelId), 'Open', 'primary'), button(customId(PLANNING_PROCEED_ACTION_PREFIX, projectName, channelId), 'Proceed', 'success')),
+        buttonRow(...buttons),
     ];
     return { richContent, components };
 }
@@ -158,6 +187,54 @@ function buildStatusNotification(opts) {
         : rc);
     return { richContent };
 }
+function buildQuestionNotification(params) {
+    const { title, description, projectName, channelId, options } = params;
+    const baseContent = (0, richContentBuilder_1.pipe)((0, richContentBuilder_1.createRichContent)(), (rc) => (0, richContentBuilder_1.withTitle)(rc, title), (rc) => (0, richContentBuilder_1.withColor)(rc, COLOR_APPROVAL));
+    const embed = description ? (0, richContentBuilder_1.withDescription)(baseContent, description) : baseContent;
+    const selectMenuOptions = options.map((opt, i) => ({
+        label: opt.text.substring(0, 100), // Discord max length for label is 100
+        value: i.toString(),
+    })).slice(0, 25); // Discord max options in a select menu is 25
+    // Ensure customIds fit within 100 chars without losing channelId
+    const encodeSafe = (str, maxLen) => {
+        let encoded = encodeURIComponent(str);
+        if (encoded.length <= maxLen)
+            return encoded;
+        return encoded.substring(0, maxLen).replace(/(%[0-9A-F]?)$/i, '');
+    };
+    const encodedChannel = encodeURIComponent(channelId);
+    const selectOverhead = exports.QUESTION_SELECT_ACTION_PREFIX.length + 2 + encodedChannel.length;
+    const safeSelectProjectName = encodeSafe(projectName, 100 - selectOverhead);
+    const encodedCustomId = `${exports.QUESTION_SELECT_ACTION_PREFIX}:${safeSelectProjectName}:${encodedChannel}`;
+    const skipOverhead = exports.QUESTION_SKIP_ACTION_PREFIX.length + 2 + encodedChannel.length;
+    const safeSkipProjectName = encodeSafe(projectName, 100 - skipOverhead);
+    const encodedSkipCustomId = `${exports.QUESTION_SKIP_ACTION_PREFIX}:${safeSkipProjectName}:${encodedChannel}`;
+    return {
+        richContent: embed,
+        components: [
+            {
+                components: [
+                    {
+                        type: 'selectMenu',
+                        customId: encodedCustomId,
+                        placeholder: 'Select an option...',
+                        options: selectMenuOptions,
+                    },
+                ],
+            },
+            {
+                components: [
+                    {
+                        type: 'button',
+                        customId: encodedSkipCustomId,
+                        label: 'Skip',
+                        style: 'secondary',
+                    },
+                ],
+            },
+        ],
+    };
+}
 /** Build a progress / phase notification (e.g. "Thinking...", "Generating..."). */
 function buildProgressNotification(opts) {
     const { phase, projectName, detail } = opts;

package/dist/services/planningDetector.js

@@ -10,33 +10,59 @@ const approvalDetector_1 = require("./approvalDetector");
  * and extracts plan metadata from the surrounding DOM elements.
  */
 const DETECT_PLANNING_SCRIPT = `(() => {
-    const OPEN_PATTERNS = ['open'];
+    const OPEN_PATTERNS = ['open', 'review'];
     const PROCEED_PATTERNS = ['proceed'];
+    const REJECT_PATTERNS = ['reject'];
+    const ACCEPT_PATTERNS = ['accept'];
 
     const normalize = (text) => (text || '').toLowerCase().replace(/\\s+/g, ' ').trim();
 
-    // Find the notify container that holds planning UI
-    const container = document.querySelector('.notify-user-container');
-    if (!container) return null;
+    const allButtons = Array.from(document.querySelectorAll('button, a'))
+        .filter(btn => btn.offsetParent !== null)
+        .reverse();
 
-    const allButtons = Array.from(container.querySelectorAll('button'))
-        .filter(btn => btn.offsetParent !== null);
+    let proceedBtn = null;
+    let openBtn = null;
+    let rejectBtn = null;
 
-    const openBtn = allButtons.find(btn => {
+    for (const btn of allButtons) {
         const t = normalize(btn.textContent || '');
-        return OPEN_PATTERNS.some(p => t === p || t.includes(p));
-    }) || null;
+        if (t.includes('review changes')) continue; // Ignore file change review button
 
-    const proceedBtn = allButtons.find(btn => {
-        const t = normalize(btn.textContent || '');
-        return PROCEED_PATTERNS.some(p => t === p || t.includes(p));
-    }) || null;
+        // If we hit an accept or reject button before finding a proceed button, 
+        // it means the most recent state is an Approval, not Planning.
+        if (!proceedBtn && (ACCEPT_PATTERNS.some(p => t === p || t.includes(p)) || REJECT_PATTERNS.some(p => t === p || t.includes(p)))) {
+            return null;
+        }
+
+        if (!proceedBtn && PROCEED_PATTERNS.some(p => t === p || t.includes(p))) {
+            proceedBtn = btn;
+            continue;
+        }
+
+        if (!openBtn && OPEN_PATTERNS.some(p => t === p || t.includes(p))) {
+            openBtn = btn;
+        }
+
+        if (!rejectBtn && REJECT_PATTERNS.some(p => t === p || t.includes(p))) {
+            rejectBtn = btn;
+        }
+        
+        if (proceedBtn && openBtn) break;
+    }
 
-    // Both buttons must exist for this to be a planning UI
-    if (!openBtn || !proceedBtn) return null;
+    // Proceed button must exist for this to be a planning UI
+    if (!proceedBtn) return null;
+    
+    const container = proceedBtn.closest('.notify-user-container') || proceedBtn.closest('div[class*="rounded-lg"]') || proceedBtn.parentElement?.parentElement;
+    if (!container) return null;
 
-    const openText = (openBtn.textContent || '').trim();
+    // openBtn was already extracted in the loop above
+
+    const hasOpenButton = openBtn !== null;
+    const openText = openBtn ? (openBtn.textContent || '').trim() : 'Open';
     const proceedText = (proceedBtn.textContent || '').trim();
+    const rejectText = rejectBtn ? (rejectBtn.textContent || '').trim() : '';
 
     // Extract plan title from .inline-flex.break-all
     const titleEl = container.querySelector('span.inline-flex.break-all, .inline-flex.break-all');
@@ -67,7 +93,7 @@ const DETECT_PLANNING_SCRIPT = `(() => {
         description = parts.join(' ').slice(0, 500);
     }
 
-    return { openText, proceedText, planTitle, planSummary, description };
+    return { openText, proceedText, planTitle, planSummary, description, rejectText, hasOpenButton };
 })()`;
 /**
  * Extract plan content displayed after clicking Open.
@@ -158,8 +184,12 @@ class PlanningDetector {
     lastDetectedInfo = null;
     /** Timestamp of last notification (for cooldown-based dedup) */
     lastNotifiedAt = 0;
+    /** Number of consecutive polls without detecting buttons */
+    emptyPollCount = 0;
     /** Cooldown period in ms to suppress duplicate notifications */
     static COOLDOWN_MS = 5000;
+    /** Number of consecutive empty polls required before resetting state */
+    static REQUIRED_EMPTY_POLLS = 3;
     constructor(options) {
         this.cdpService = options.cdpService;
         this.pollIntervalMs = options.pollIntervalMs ?? 2000;
@@ -174,6 +204,7 @@ class PlanningDetector {
         this.lastDetectedKey = null;
         this.lastDetectedInfo = null;
         this.lastNotifiedAt = 0;
+        this.emptyPollCount = 0;
         this.schedulePoll();
     }
     /** Stop monitoring. */
@@ -210,6 +241,15 @@ class PlanningDetector {
         const text = buttonText ?? this.lastDetectedInfo?.proceedText ?? 'Proceed';
         return this.clickButton(text);
     }
+    /**
+     * Click the Reject button via CDP.
+     * @param buttonText Text of the button to click (default: detected rejectText or "Reject All")
+     * @returns true if click succeeded
+     */
+    async clickRejectButton(buttonText) {
+        const text = buttonText || this.lastDetectedInfo?.rejectText || 'Reject All';
+        return this.clickButton(text);
+    }
     /**
      * Extract plan content from the DOM after Open has been clicked.
      * @returns Plan content text or null if not found
@@ -255,8 +295,9 @@ class PlanningDetector {
             const result = await this.cdpService.call('Runtime.evaluate', callParams);
             const info = result?.result?.value ?? null;
             if (info) {
-                // Duplicate prevention: use button text pair as key (stable across DOM redraws)
-                const key = `${info.openText}::${info.proceedText}`;
+                this.emptyPollCount = 0;
+                // Duplicate prevention: use plan title and content as key (stable across DOM redraws and button text updates)
+                const key = `${info.planTitle}::${info.planSummary}::${(info.description || '').substring(0, 50)}`;
                 const now = Date.now();
                 const withinCooldown = (now - this.lastNotifiedAt) < PlanningDetector.COOLDOWN_MS;
                 if (key !== this.lastDetectedKey && !withinCooldown) {
@@ -271,12 +312,15 @@ class PlanningDetector {
                 }
             }
             else {
-                // Reset when buttons disappear (prepare for next planning detection)
-                const wasDetected = this.lastDetectedKey !== null;
-                this.lastDetectedKey = null;
-                this.lastDetectedInfo = null;
-                if (wasDetected && this.onResolved) {
-                    this.onResolved();
+                this.emptyPollCount++;
+                if (this.emptyPollCount >= PlanningDetector.REQUIRED_EMPTY_POLLS) {
+                    // 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/promptDispatcher.js

@@ -1,17 +1,43 @@
 "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;
     }
     async send(req) {
-        await this.deps.sendPromptImpl(this.deps.bridge, req.message, req.prompt, req.cdp, this.deps.modeService, this.deps.modelService, req.inboundImages ?? [], req.options);
+        await this._dispatch(req, req.options);
+    }
+    async resume(req) {
+        await this._dispatch(req, { ...req.options, resumeOnly: true });
+    }
+    async _dispatch(req, 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;