lazy-gravity 0.2.0 → 0.4.0

package/dist/services/modelService.js

@@ -3,9 +3,10 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ModelService = exports.DEFAULT_MODEL = exports.AVAILABLE_MODELS = void 0;
 const i18n_1 = require("../utils/i18n");
 /**
- * Available LLM models
- * Aligned with models selectable in the Antigravity (Cursor fork) UI
- * Note: Models may change with Antigravity version updates
+ * Fallback model list used when CDP is not connected.
+ * NOT used for validation — CDP is the sole source of truth
+ * for available models. This list may become stale after
+ * Antigravity updates.
  */
 exports.AVAILABLE_MODELS = [
     'gemini-3.1-pro-high',
@@ -15,14 +16,20 @@ exports.AVAILABLE_MODELS = [
     'claude-opus-4.6-thinking',
     'gpt-oss-120b-medium'
 ];
-/** Default LLM model */
+/** Default LLM model (initial value before CDP connects) */
 exports.DEFAULT_MODEL = 'gemini-3-flash';
 /**
  * Service class for managing LLM models.
  * Handles model switching via the /model command.
+ *
+ * Model validation is intentionally NOT performed here.
+ * The actual model list is dynamic (fetched from CDP via
+ * cdp.getUiModels()), so setModel() accepts any string.
  */
 class ModelService {
     currentModel = exports.DEFAULT_MODEL;
+    defaultModel = null;
+    pendingSync = false;
     /**
      * Get the current LLM model
      */
@@ -30,34 +37,68 @@ class ModelService {
         return this.currentModel;
     }
     /**
-     * Switch LLM model
+     * Check if the current model is pending sync to Antigravity
+     */
+    isPendingSync() {
+        return this.pendingSync;
+    }
+    /**
+     * Mark the pending model as synced (clears pendingSync flag)
+     */
+    markSynced() {
+        this.pendingSync = false;
+    }
+    /**
+     * Switch LLM model.
+     * Accepts any model name — validation happens at the CDP layer
+     * (cdp.setUiModel) against the live model list.
+     *
      * @param modelName Model name to set (case-insensitive)
+     * @param synced Whether the model has been synced to Antigravity (default: false)
      */
-    setModel(modelName) {
+    setModel(modelName, synced = false) {
         if (!modelName || modelName.trim() === '') {
             return {
                 success: false,
-                error: (0, i18n_1.t)('⚠️ Model name not specified. Available models: ') + exports.AVAILABLE_MODELS.join(', '),
-            };
-        }
-        const normalized = modelName.trim().toLowerCase();
-        if (!exports.AVAILABLE_MODELS.includes(normalized)) {
-            return {
-                success: false,
-                error: (0, i18n_1.t)(`⚠️ Invalid model "${modelName}". Available models: ${exports.AVAILABLE_MODELS.join(', ')}`),
+                error: (0, i18n_1.t)('⚠️ Model name not specified.'),
             };
         }
-        this.currentModel = normalized;
+        this.currentModel = modelName.trim().toLowerCase();
+        this.pendingSync = !synced;
         return {
             success: true,
             model: this.currentModel,
         };
     }
     /**
-     * Get the list of available models
+     * Get the fallback list of available models.
+     * Prefer cdp.getUiModels() when CDP is connected.
      */
     getAvailableModels() {
         return exports.AVAILABLE_MODELS;
     }
+    /**
+     * Get the default model name (free-text, may not match current CDP models)
+     */
+    getDefaultModel() {
+        return this.defaultModel;
+    }
+    /**
+     * Set the default model name (free-text, persisted via DB separately)
+     * @param name Model name or null to clear
+     */
+    setDefaultModel(name) {
+        this.defaultModel = name ? name.trim() : null;
+        return { success: true, defaultModel: this.defaultModel };
+    }
+    /**
+     * Load the default model from an external source (e.g. DB).
+     * Only sets the in-memory value if not already set.
+     */
+    loadDefaultModel(name) {
+        if (this.defaultModel === null && name) {
+            this.defaultModel = name.trim();
+        }
+    }
 }
 exports.ModelService = ModelService;

package/dist/services/notificationSender.js

@@ -0,0 +1,149 @@
+"use strict";
+/**
+ * Platform-agnostic notification builders.
+ *
+ * Every exported function is **pure** — no side effects, no I/O.
+ * They return a `MessagePayload` that any platform adapter can render.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildApprovalNotification = buildApprovalNotification;
+exports.buildPlanningNotification = buildPlanningNotification;
+exports.buildErrorPopupNotification = buildErrorPopupNotification;
+exports.buildAutoApprovedNotification = buildAutoApprovedNotification;
+exports.buildResolvedOverlay = buildResolvedOverlay;
+exports.buildStatusNotification = buildStatusNotification;
+exports.buildProgressNotification = buildProgressNotification;
+const richContentBuilder_1 = require("../platform/richContentBuilder");
+// ---------------------------------------------------------------------------
+// Custom-ID prefix constants (must stay in sync with cdpBridgeManager)
+// ---------------------------------------------------------------------------
+const APPROVE_ACTION_PREFIX = 'approve_action';
+const ALWAYS_ALLOW_ACTION_PREFIX = 'always_allow_action';
+const DENY_ACTION_PREFIX = 'deny_action';
+const PLANNING_OPEN_ACTION_PREFIX = 'planning_open_action';
+const PLANNING_PROCEED_ACTION_PREFIX = 'planning_proceed_action';
+const ERROR_POPUP_DISMISS_ACTION_PREFIX = 'error_popup_dismiss_action';
+const ERROR_POPUP_COPY_DEBUG_ACTION_PREFIX = 'error_popup_copy_debug_action';
+const ERROR_POPUP_RETRY_ACTION_PREFIX = 'error_popup_retry_action';
+// ---------------------------------------------------------------------------
+// Notification colours
+// ---------------------------------------------------------------------------
+/** Warning orange — used for approval requests. */
+const COLOR_APPROVAL = 0xFFA500;
+/** Blue — used for planning / informational notifications. */
+const COLOR_PLANNING = 0x3498DB;
+/** Red — used for error notifications. */
+const COLOR_ERROR = 0xE74C3C;
+/** Green — used for success / progress notifications. */
+const COLOR_SUCCESS = 0x2ECC71;
+/** Grey — used for neutral status notifications. */
+const COLOR_NEUTRAL = 0x95A5A6;
+// ---------------------------------------------------------------------------
+// Phase → colour mapping for progress notifications
+// ---------------------------------------------------------------------------
+const PHASE_COLOURS = {
+    thinking: COLOR_PLANNING,
+    generating: COLOR_SUCCESS,
+    error: COLOR_ERROR,
+    waiting: COLOR_NEUTRAL,
+    complete: COLOR_SUCCESS,
+};
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+/** Create a single button definition. */
+function button(customId, label, style) {
+    return { type: 'button', customId, label, style };
+}
+/** Wrap one or more buttons into a component row. */
+function buttonRow(...buttons) {
+    return { components: buttons };
+}
+/**
+ * Build a colon-separated customId following the project convention:
+ *   `<prefix>:<projectName>` or `<prefix>:<projectName>:<channelId>`
+ */
+function customId(prefix, projectName, channelId) {
+    if (channelId !== null && channelId.trim().length > 0) {
+        return `${prefix}:${projectName}:${channelId}`;
+    }
+    return `${prefix}:${projectName}`;
+}
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+/** Build the approval notification message. */
+function buildApprovalNotification(opts) {
+    const { title, description, projectName, channelId, toolNames, extraFields } = 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 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')),
+    ];
+    return { richContent, components };
+}
+/** Build the planning mode notification message. */
+function buildPlanningNotification(opts) {
+    const { title, description, projectName, channelId, extraFields } = 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 components = [
+        buttonRow(button(customId(PLANNING_OPEN_ACTION_PREFIX, projectName, channelId), 'Open', 'primary'), button(customId(PLANNING_PROCEED_ACTION_PREFIX, projectName, channelId), 'Proceed', 'success')),
+    ];
+    return { richContent, components };
+}
+/** Build the error popup notification message. */
+function buildErrorPopupNotification(opts) {
+    const { title, errorMessage, projectName, channelId, extraFields } = opts;
+    const richContent = (0, richContentBuilder_1.pipe)((0, richContentBuilder_1.createRichContent)(), (rc) => (0, richContentBuilder_1.withTitle)(rc, title), (rc) => (0, richContentBuilder_1.withDescription)(rc, errorMessage), (rc) => (0, richContentBuilder_1.withColor)(rc, COLOR_ERROR), (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, 'Agent error detected'), (rc) => (0, richContentBuilder_1.withTimestamp)(rc));
+    const components = [
+        buttonRow(button(customId(ERROR_POPUP_DISMISS_ACTION_PREFIX, projectName, channelId), 'Dismiss', 'secondary'), button(customId(ERROR_POPUP_COPY_DEBUG_ACTION_PREFIX, projectName, channelId), 'Copy Debug', 'primary'), button(customId(ERROR_POPUP_RETRY_ACTION_PREFIX, projectName, channelId), 'Retry', 'success')),
+    ];
+    return { richContent, components };
+}
+/** Build an auto-approved notification (shown when auto-accept fires). */
+function buildAutoApprovedNotification(opts) {
+    const { accepted, projectName, description, approveText } = opts;
+    const richContent = (0, richContentBuilder_1.pipe)((0, richContentBuilder_1.createRichContent)(), (rc) => (0, richContentBuilder_1.withTitle)(rc, accepted ? 'Auto-approved' : 'Auto-approve failed'), (rc) => (0, richContentBuilder_1.withDescription)(rc, accepted
+        ? 'An action was automatically approved.'
+        : 'Auto-approve attempted but failed. Manual approval required.'), (rc) => (0, richContentBuilder_1.withColor)(rc, accepted ? COLOR_SUCCESS : 0xF39C12), (rc) => (0, richContentBuilder_1.addField)(rc, 'Auto-approve mode', 'ON', true), (rc) => (0, richContentBuilder_1.addField)(rc, 'Workspace', projectName, true), (rc) => (0, richContentBuilder_1.addField)(rc, 'Result', accepted ? 'Executed Always Allow/Allow' : 'Manual approval required', true), (rc) => description ? (0, richContentBuilder_1.addField)(rc, 'Action Detail', description.substring(0, 1024), false) : rc, (rc) => approveText ? (0, richContentBuilder_1.addField)(rc, 'Approved via', approveText, true) : rc, (rc) => (0, richContentBuilder_1.withTimestamp)(rc));
+    return { richContent };
+}
+/**
+ * Build a "resolved" overlay from an existing notification payload.
+ * Changes colour to grey, adds a Status field, and disables all buttons.
+ */
+function buildResolvedOverlay(original, statusText) {
+    const rc = (0, richContentBuilder_1.pipe)(original.richContent ?? (0, richContentBuilder_1.createRichContent)(), (r) => (0, richContentBuilder_1.withColor)(r, COLOR_NEUTRAL), (r) => (0, richContentBuilder_1.addField)(r, 'Status', statusText, false));
+    const disabledComponents = original.components
+        ? original.components.map((row) => ({
+            components: row.components.map((comp) => comp.type === 'button' ? { ...comp, disabled: true } : comp),
+        }))
+        : undefined;
+    return {
+        ...original,
+        richContent: rc,
+        components: disabledComponents,
+    };
+}
+/** Build a simple status embed. */
+function buildStatusNotification(opts) {
+    const { title, description, color, fields } = 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 ?? COLOR_NEUTRAL), (rc) => fields
+        ? fields.reduce((acc, f) => (0, richContentBuilder_1.addField)(acc, f.name, f.value, f.inline), rc)
+        : rc);
+    return { richContent };
+}
+/** Build a progress / phase notification (e.g. "Thinking...", "Generating..."). */
+function buildProgressNotification(opts) {
+    const { phase, projectName, detail } = opts;
+    const phaseColor = PHASE_COLOURS[phase.toLowerCase()] ?? COLOR_NEUTRAL;
+    const richContent = (0, richContentBuilder_1.pipe)((0, richContentBuilder_1.createRichContent)(), (rc) => (0, richContentBuilder_1.withTitle)(rc, phase), (rc) => (detail ? (0, richContentBuilder_1.withDescription)(rc, detail) : rc), (rc) => (0, richContentBuilder_1.withColor)(rc, phaseColor), (rc) => (projectName ? (0, richContentBuilder_1.addField)(rc, 'Project', projectName, true) : rc));
+    return { richContent };
+}

package/dist/services/responseMonitor.js

@@ -474,8 +474,7 @@ class ResponseMonitor {
         this.pollIntervalMs = options.pollIntervalMs ?? 2000;
         this.maxDurationMs = options.maxDurationMs ?? 300000;
         this.stopGoneConfirmCount = options.stopGoneConfirmCount ?? 3;
-        this.extractionMode = options.extractionMode
-            ?? (process.env.EXTRACTION_MODE === 'legacy' ? 'legacy' : 'structured');
+        this.extractionMode = options.extractionMode ?? 'structured';
         this.onProgress = options.onProgress;
         this.onComplete = options.onComplete;
         this.onTimeout = options.onTimeout;

package/dist/services/screenshotService.js

@@ -33,7 +33,7 @@ class ScreenshotService {
             if (options.captureBeyondViewport !== undefined) {
                 params.captureBeyondViewport = options.captureBeyondViewport;
             }
-            const result = await this.cdpService.call('Page.captureScreenshot', params);
+            const result = await this.cdpService.callWithRetry('Page.captureScreenshot', params);
             const base64Data = result?.data ?? '';
             if (!base64Data) {
                 return {
@@ -73,7 +73,7 @@ class ScreenshotService {
             if (options.clip) {
                 params.clip = options.clip;
             }
-            const result = await this.cdpService.call('Page.captureScreenshot', params);
+            const result = await this.cdpService.callWithRetry('Page.captureScreenshot', params);
             return result?.data ?? null;
         }
         catch (error) {

package/dist/ui/autoAcceptUi.js

@@ -1,11 +1,48 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AUTOACCEPT_BTN_REFRESH = exports.AUTOACCEPT_BTN_OFF = exports.AUTOACCEPT_BTN_ON = void 0;
+exports.buildAutoAcceptPayload = buildAutoAcceptPayload;
 exports.sendAutoAcceptUI = sendAutoAcceptUI;
 const discord_js_1 = require("discord.js");
+const richContentBuilder_1 = require("../platform/richContentBuilder");
 exports.AUTOACCEPT_BTN_ON = 'autoaccept_btn_on';
 exports.AUTOACCEPT_BTN_OFF = 'autoaccept_btn_off';
 exports.AUTOACCEPT_BTN_REFRESH = 'autoaccept_btn_refresh';
+/**
+ * Build a platform-agnostic MessagePayload for auto-accept UI.
+ */
+function buildAutoAcceptPayload(enabled) {
+    const rc = (0, richContentBuilder_1.withTimestamp)((0, richContentBuilder_1.withFooter)((0, richContentBuilder_1.withDescription)((0, richContentBuilder_1.withColor)((0, richContentBuilder_1.withTitle)((0, richContentBuilder_1.createRichContent)(), 'Auto-accept Management'), enabled ? 0x2ECC71 : 0x95A5A6), `**Current Status:** ${enabled ? 'ON' : 'OFF'}\n\n` +
+        'ON: approval dialogs are automatically allowed.\n' +
+        'OFF: approval dialogs require manual action.'), 'Use buttons below to change mode'));
+    return {
+        richContent: rc,
+        components: [
+            {
+                components: [
+                    {
+                        type: 'button',
+                        customId: exports.AUTOACCEPT_BTN_ON,
+                        label: 'Turn ON',
+                        style: enabled ? 'success' : 'secondary',
+                    },
+                    {
+                        type: 'button',
+                        customId: exports.AUTOACCEPT_BTN_OFF,
+                        label: 'Turn OFF',
+                        style: !enabled ? 'danger' : 'secondary',
+                    },
+                    {
+                        type: 'button',
+                        customId: exports.AUTOACCEPT_BTN_REFRESH,
+                        label: 'Refresh',
+                        style: 'primary',
+                    },
+                ],
+            },
+        ],
+    };
+}
 async function sendAutoAcceptUI(target, autoAcceptService) {
     const enabled = autoAcceptService.isEnabled();
     const embed = new discord_js_1.EmbedBuilder()

package/dist/ui/modeUi.js

@@ -1,8 +1,45 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildModePayload = buildModePayload;
 exports.sendModeUI = sendModeUI;
 const discord_js_1 = require("discord.js");
 const modeService_1 = require("../services/modeService");
+const richContentBuilder_1 = require("../platform/richContentBuilder");
+/**
+ * Build a platform-agnostic MessagePayload for mode selection UI.
+ * @param currentMode The current mode name
+ * @param isPending Whether the mode is pending sync to Antigravity
+ */
+function buildModePayload(currentMode, isPending = false) {
+    const pendingSuffix = isPending ? ' (pending sync)' : '';
+    const rc = (0, richContentBuilder_1.withTimestamp)((0, richContentBuilder_1.withFooter)((0, richContentBuilder_1.withDescription)((0, richContentBuilder_1.withColor)((0, richContentBuilder_1.withTitle)((0, richContentBuilder_1.createRichContent)(), 'Mode Management'), 0x57F287), `**Current Mode:** ${modeService_1.MODE_DISPLAY_NAMES[currentMode] || currentMode}${pendingSuffix}\n` +
+        `${modeService_1.MODE_DESCRIPTIONS[currentMode] || ''}\n\n` +
+        `**Available Modes (${modeService_1.AVAILABLE_MODES.length})**\n` +
+        modeService_1.AVAILABLE_MODES.map(m => {
+            const icon = m === currentMode ? '[x]' : '[ ]';
+            return `${icon} **${modeService_1.MODE_DISPLAY_NAMES[m] || m}** — ${modeService_1.MODE_DESCRIPTIONS[m] || ''}`;
+        }).join('\n')), 'Select a mode from the dropdown below'));
+    return {
+        richContent: rc,
+        components: [
+            {
+                components: [
+                    {
+                        type: 'selectMenu',
+                        customId: 'mode_select',
+                        placeholder: 'Select a mode...',
+                        options: modeService_1.AVAILABLE_MODES.map(m => ({
+                            label: modeService_1.MODE_DISPLAY_NAMES[m] || m,
+                            description: modeService_1.MODE_DESCRIPTIONS[m] || '',
+                            value: m,
+                            isDefault: m === currentMode,
+                        })),
+                    },
+                ],
+            },
+        ],
+    };
+}
 /**
  * Build and send the interactive UI for the /mode command (dropdown style)
  */
@@ -13,7 +50,7 @@ async function sendModeUI(target, modeService, deps) {
         if (cdp) {
             const liveMode = await cdp.getCurrentMode();
             if (liveMode) {
-                modeService.setMode(liveMode);
+                modeService.setMode(liveMode, true);
             }
         }
     }

package/dist/ui/modelsUi.js

@@ -1,8 +1,104 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildModelsPayload = buildModelsPayload;
 exports.buildModelsUI = buildModelsUI;
 exports.sendModelsUI = sendModelsUI;
 const discord_js_1 = require("discord.js");
+const richContentBuilder_1 = require("../platform/richContentBuilder");
+/**
+ * Build a platform-agnostic MessagePayload for model selection UI.
+ */
+function buildModelsPayload(models, currentModel, quotaData, defaultModel = null) {
+    if (models.length === 0)
+        return null;
+    function formatQuota(mName, current) {
+        if (!mName)
+            return `${current ? '[x]' : '[ ]'} Unknown`;
+        const normalize = (s) => s.toLowerCase().replace(/[\s\-_]/g, '');
+        const nName = normalize(mName);
+        const q = quotaData.find(q => {
+            const nLabel = normalize(q.label);
+            const nModel = normalize(q.model || '');
+            return nLabel === nName || nModel === nName
+                || nName.includes(nLabel) || nLabel.includes(nName)
+                || (nModel && (nName.includes(nModel) || nModel.includes(nName)));
+        });
+        if (!q || !q.quotaInfo)
+            return `${current ? '[x]' : '[ ]'} ${mName}`;
+        const rem = q.quotaInfo.remainingFraction;
+        const resetTime = q.quotaInfo.resetTime ? new Date(q.quotaInfo.resetTime) : null;
+        const diffMs = resetTime ? resetTime.getTime() - Date.now() : 0;
+        let timeStr = 'Ready';
+        if (diffMs > 0) {
+            const mins = Math.ceil(diffMs / 60000);
+            if (mins < 60)
+                timeStr = `${mins}m`;
+            else
+                timeStr = `${Math.floor(mins / 60)}h ${mins % 60}m`;
+        }
+        if (rem !== undefined && rem !== null) {
+            const percent = Math.round(rem * 100);
+            return `${current ? '[x]' : '[ ]'} ${mName} ${percent}% (${timeStr})`;
+        }
+        return `${current ? '[x]' : '[ ]'} ${mName} (${timeStr})`;
+    }
+    const currentModelFormatted = currentModel ? formatQuota(currentModel, true) : 'Unknown';
+    const defaultLine = defaultModel
+        ? `\n**Default:** ⭐ ${defaultModel}`
+        : '\n**Default:** Not set';
+    const modelLines = models.map(m => {
+        const isCurrent = m === currentModel;
+        const isDefault = defaultModel != null && m.toLowerCase() === defaultModel.toLowerCase();
+        const star = isDefault ? ' ⭐' : '';
+        return `${formatQuota(m, isCurrent)}${star}`;
+    }).join('\n');
+    const rc = (0, richContentBuilder_1.withTimestamp)((0, richContentBuilder_1.withFooter)((0, richContentBuilder_1.withDescription)((0, richContentBuilder_1.withColor)((0, richContentBuilder_1.withTitle)((0, richContentBuilder_1.createRichContent)(), 'Model Management'), 0x5865F2), `**Current Model:**\n${currentModelFormatted}${defaultLine}\n\n` +
+        `**Available Models (${models.length})**\n` +
+        modelLines), 'Latest quota information retrieved'));
+    // Use 1 button per row so model names are fully readable on Telegram.
+    // Telegram inline keyboard buttons are narrow; 5-per-row truncates names.
+    const rows = [];
+    for (const mName of models.slice(0, 24)) {
+        const safeName = mName.length > 80 ? mName.substring(0, 77) + '...' : mName;
+        const isDefault = defaultModel != null && mName.toLowerCase() === defaultModel.toLowerCase();
+        const prefix = mName === currentModel ? '✓ ' : '';
+        const suffix = isDefault ? ' ⭐' : '';
+        rows.push({
+            components: [{
+                    type: 'button',
+                    customId: `model_btn_${mName}`,
+                    label: `${prefix}${safeName}${suffix}`,
+                    style: mName === currentModel ? 'success' : 'secondary',
+                }],
+        });
+    }
+    // Default model action buttons
+    const defaultBtnRow = {
+        components: defaultModel
+            ? [{
+                    type: 'button',
+                    customId: 'model_clear_default_btn',
+                    label: 'Clear Default',
+                    style: 'danger',
+                }]
+            : [{
+                    type: 'button',
+                    customId: 'model_set_default_btn',
+                    label: 'Set Current as Default',
+                    style: 'primary',
+                }],
+    };
+    rows.push(defaultBtnRow);
+    rows.push({
+        components: [{
+                type: 'button',
+                customId: 'model_refresh_btn',
+                label: 'Refresh',
+                style: 'primary',
+            }],
+    });
+    return { richContent: rc, components: rows };
+}
 /**
  * Build the embed + button components for the models UI.
  * Returns null when CDP is unavailable or no models are found.

package/dist/ui/outputUi.js

@@ -1,10 +1,42 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.OUTPUT_BTN_PLAIN = exports.OUTPUT_BTN_EMBED = void 0;
+exports.buildOutputPayload = buildOutputPayload;
 exports.sendOutputUI = sendOutputUI;
 const discord_js_1 = require("discord.js");
+const richContentBuilder_1 = require("../platform/richContentBuilder");
 exports.OUTPUT_BTN_EMBED = 'output_btn_embed';
 exports.OUTPUT_BTN_PLAIN = 'output_btn_plain';
+/**
+ * Build a platform-agnostic MessagePayload for output format UI.
+ */
+function buildOutputPayload(currentFormat) {
+    const isEmbed = currentFormat === 'embed';
+    const rc = (0, richContentBuilder_1.withTimestamp)((0, richContentBuilder_1.withFooter)((0, richContentBuilder_1.withDescription)((0, richContentBuilder_1.withColor)((0, richContentBuilder_1.withTitle)((0, richContentBuilder_1.createRichContent)(), 'Output Format'), isEmbed ? 0x5865F2 : 0x2ECC71), `**Current Format:** ${isEmbed ? 'Embed' : 'Plain Text'}\n\n` +
+        'Embed: Rich formatting with colored borders (default).\n' +
+        'Plain Text: Simple text output, easy to copy on mobile.'), 'Use buttons below to change format'));
+    return {
+        richContent: rc,
+        components: [
+            {
+                components: [
+                    {
+                        type: 'button',
+                        customId: exports.OUTPUT_BTN_EMBED,
+                        label: 'Embed',
+                        style: isEmbed ? 'primary' : 'secondary',
+                    },
+                    {
+                        type: 'button',
+                        customId: exports.OUTPUT_BTN_PLAIN,
+                        label: 'Plain Text',
+                        style: !isEmbed ? 'success' : 'secondary',
+                    },
+                ],
+            },
+        ],
+    };
+}
 async function sendOutputUI(target, currentFormat) {
     const isEmbed = currentFormat === 'embed';
     const embed = new discord_js_1.EmbedBuilder()

package/dist/ui/projectListUi.js

@@ -3,9 +3,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ITEMS_PER_PAGE = exports.PROJECT_PAGE_PREFIX = exports.WORKSPACE_SELECT_ID = exports.PROJECT_SELECT_ID = void 0;
 exports.parseProjectPageId = parseProjectPageId;
 exports.isProjectSelectId = isProjectSelectId;
+exports.buildProjectListPayload = buildProjectListPayload;
 exports.buildProjectListUI = buildProjectListUI;
 const discord_js_1 = require("discord.js");
 const i18n_1 = require("../utils/i18n");
+const richContentBuilder_1 = require("../platform/richContentBuilder");
 /** Select menu custom ID (legacy, page 0) */
 exports.PROJECT_SELECT_ID = 'project_select';
 /** Backward compatibility: also accept old ID */
@@ -33,6 +35,59 @@ function isProjectSelectId(customId) {
         customId === exports.WORKSPACE_SELECT_ID ||
         customId.startsWith(`${exports.PROJECT_SELECT_ID}:`));
 }
+/**
+ * Build a platform-agnostic MessagePayload for project list UI.
+ */
+function buildProjectListPayload(workspaces, page = 0) {
+    const totalPages = Math.max(1, Math.ceil(workspaces.length / exports.ITEMS_PER_PAGE));
+    const safePage = Math.max(0, Math.min(page, totalPages - 1));
+    let rc = (0, richContentBuilder_1.withTimestamp)((0, richContentBuilder_1.withColor)((0, richContentBuilder_1.withTitle)((0, richContentBuilder_1.withDescription)((0, richContentBuilder_1.createRichContent)(), (0, i18n_1.t)('Select a project to auto-create a category and session channel')), 'Projects'), 0x5865F2));
+    if (workspaces.length === 0) {
+        return { richContent: rc, components: [] };
+    }
+    if (totalPages > 1) {
+        rc = (0, richContentBuilder_1.withFooter)(rc, `Page ${safePage + 1} / ${totalPages} (${workspaces.length} projects total)`);
+    }
+    const start = safePage * exports.ITEMS_PER_PAGE;
+    const end = Math.min(start + exports.ITEMS_PER_PAGE, workspaces.length);
+    const pageItems = workspaces.slice(start, end);
+    const components = [
+        {
+            components: [
+                {
+                    type: 'selectMenu',
+                    customId: `${exports.PROJECT_SELECT_ID}:${safePage}`,
+                    placeholder: (0, i18n_1.t)('Select a project...'),
+                    options: pageItems.map((ws) => ({
+                        label: ws,
+                        value: ws,
+                    })),
+                },
+            ],
+        },
+    ];
+    if (totalPages > 1) {
+        components.push({
+            components: [
+                {
+                    type: 'button',
+                    customId: `${exports.PROJECT_PAGE_PREFIX}:${Math.max(0, safePage - 1)}`,
+                    label: '\u25C0 Prev',
+                    style: 'secondary',
+                    disabled: safePage === 0,
+                },
+                {
+                    type: 'button',
+                    customId: `${exports.PROJECT_PAGE_PREFIX}:${safePage + 1}`,
+                    label: 'Next \u25B6',
+                    style: 'secondary',
+                    disabled: safePage >= totalPages - 1,
+                },
+            ],
+        });
+    }
+    return { richContent: rc, components };
+}
 /**
  * Build the project list UI with select menu and optional Prev/Next buttons.
  *

package/dist/ui/screenshotUi.js

@@ -1,8 +1,34 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.buildScreenshotPayload = buildScreenshotPayload;
 exports.handleScreenshot = handleScreenshot;
 const discord_js_1 = require("discord.js");
 const screenshotService_1 = require("../services/screenshotService");
+/**
+ * Build a platform-agnostic MessagePayload containing the screenshot.
+ * Returns a payload with the screenshot as a file attachment, or an error text.
+ */
+async function buildScreenshotPayload(cdp) {
+    if (!cdp) {
+        return { text: 'Not connected to Antigravity.' };
+    }
+    try {
+        const screenshot = new screenshotService_1.ScreenshotService({ cdpService: cdp });
+        const result = await screenshot.capture({ format: 'png' });
+        if (result.success && result.buffer) {
+            const file = {
+                name: 'screenshot.png',
+                data: result.buffer,
+                contentType: 'image/png',
+            };
+            return { files: [file] };
+        }
+        return { text: `Screenshot failed: ${result.error ?? 'Unknown error'}` };
+    }
+    catch (e) {
+        return { text: `Screenshot error: ${e.message}` };
+    }
+}
 /**
  * Capture a screenshot and send it to Discord
  */