@helmisatria/mcp-chrome-bridge 1.0.30

package/dist/agent/chat-service.js

@@ -0,0 +1,439 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentChatService = void 0;
+const node_crypto_1 = require("node:crypto");
+const project_service_1 = require("./project-service");
+const message_service_1 = require("./message-service");
+const session_service_1 = require("./session-service");
+const attachment_service_1 = require("./attachment-service");
+/**
+ * AgentChatService coordinates incoming /agent/chat requests and delegates to engines.
+ *
+ * 中文说明：该服务负责会话级调度，不关心具体 CLI/SDK 实现细节。
+ * 通过 Engine 接口实现依赖倒置，后续替换或新增引擎时无需修改 HTTP 路由层。
+ */
+class AgentChatService {
+    constructor(options) {
+        this.engines = new Map();
+        /**
+         * Registry of currently running executions, keyed by requestId.
+         */
+        this.runningExecutions = new Map();
+        this.streamManager = options.streamManager;
+        for (const engine of options.engines) {
+            this.engines.set(engine.name, engine);
+        }
+        if (options.defaultEngineName && this.engines.has(options.defaultEngineName)) {
+            this.defaultEngineName = options.defaultEngineName;
+        }
+        else {
+            // Fallback to first registered engine to avoid hard-coding 'claude' here.
+            const firstEngine = options.engines[0];
+            if (!firstEngine) {
+                throw new Error('AgentChatService requires at least one engine');
+            }
+            this.defaultEngineName = firstEngine.name;
+        }
+    }
+    async handleAct(sessionId, payload) {
+        var _a, _b, _c;
+        const trimmed = (_a = payload.instruction) === null || _a === void 0 ? void 0 : _a.trim();
+        if (!trimmed) {
+            throw new Error('instruction is required');
+        }
+        const requestId = payload.requestId || (0, node_crypto_1.randomUUID)();
+        let projectId = payload.projectId;
+        // Normalize empty string to undefined
+        const rawDbSessionId = typeof payload.dbSessionId === 'string' ? payload.dbSessionId.trim() : '';
+        const dbSessionId = rawDbSessionId || undefined;
+        // Load session from database if dbSessionId is provided
+        let dbSession;
+        if (dbSessionId) {
+            dbSession = await (0, session_service_1.getSession)(dbSessionId);
+            if (!dbSession) {
+                throw new Error(`Session not found for id: ${dbSessionId}`);
+            }
+            // Validate project association
+            if (projectId && dbSession.projectId !== projectId) {
+                throw new Error(`Session ${dbSessionId} does not belong to project: ${projectId}`);
+            }
+            // Use session's project if not explicitly provided
+            if (!projectId) {
+                projectId = dbSession.projectId;
+            }
+        }
+        // Project is required - workspace path must come from project system
+        if (!projectId) {
+            throw new Error('projectId is required. Please select or create a project first.');
+        }
+        const project = await (0, project_service_1.getProject)(projectId);
+        if (!project) {
+            throw new Error(`Project not found for id: ${projectId}`);
+        }
+        const projectRoot = project.rootPath;
+        const projectPreferredCli = project.preferredCli;
+        const projectSelectedModel = project.selectedModel;
+        const projectUseCcr = project.useCcr;
+        // Legacy fallback: if caller does not use sessions table, use project-level resume id
+        let resumeClaudeSessionId;
+        if (!dbSessionId) {
+            resumeClaudeSessionId = project.activeClaudeSessionId;
+        }
+        // Resolve engine name - session binding takes precedence
+        let engineName;
+        if (dbSession) {
+            engineName = dbSession.engineName;
+            // Validate cliPreference matches session engine
+            if (payload.cliPreference && payload.cliPreference !== engineName) {
+                throw new Error(`cliPreference (${payload.cliPreference}) does not match session.engineName (${engineName})`);
+            }
+        }
+        else {
+            engineName = this.resolveEngineName(payload.cliPreference, projectPreferredCli);
+        }
+        const engine = this.engines.get(engineName);
+        if (!engine) {
+            throw new Error(`No agent engine registered for ${engineName}`);
+        }
+        // Model priority: request > session > project
+        const effectiveModel = ((_b = payload.model) === null || _b === void 0 ? void 0 : _b.trim()) || (dbSession === null || dbSession === void 0 ? void 0 : dbSession.model) || projectSelectedModel;
+        // For Claude engine with session, use session's engineSessionId for resume
+        if (dbSession && engineName === 'claude') {
+            resumeClaudeSessionId = dbSession.engineSessionId;
+        }
+        const now = new Date().toISOString();
+        const userMessageId = (0, node_crypto_1.randomUUID)();
+        // Process and persist image attachments
+        const savedAttachments = [];
+        let attachmentMetadata;
+        let resolvedImagePaths;
+        if (projectId && payload.attachments && payload.attachments.length > 0) {
+            const imageAttachments = payload.attachments.filter((a) => a.type === 'image');
+            if (imageAttachments.length > 0) {
+                try {
+                    console.error(`[AgentChatService] Saving ${imageAttachments.length} image attachment(s) for project ${projectId}`);
+                    for (let i = 0; i < imageAttachments.length; i++) {
+                        const attachment = imageAttachments[i];
+                        const saved = await attachment_service_1.attachmentService.saveAttachment({
+                            projectId,
+                            messageId: userMessageId,
+                            attachment,
+                            index: i,
+                        });
+                        savedAttachments.push(saved);
+                    }
+                    // Build metadata array for message persistence
+                    attachmentMetadata = savedAttachments.map((s) => s.metadata);
+                    // Build paths array for engine consumption
+                    resolvedImagePaths = savedAttachments.map((s) => s.absolutePath);
+                    console.error(`[AgentChatService] Saved ${savedAttachments.length} attachment(s): ${resolvedImagePaths.join(', ')}`);
+                }
+                catch (error) {
+                    console.error('[AgentChatService] Failed to save attachments:', error);
+                    // Continue without attachments - don't fail the entire request
+                }
+            }
+        }
+        // Build metadata object for user message
+        // Include attachments, clientMeta, and displayText if present
+        let userMessageMetadata;
+        const hasAttachments = attachmentMetadata && attachmentMetadata.length > 0;
+        const hasClientMeta = payload.clientMeta !== undefined;
+        const hasDisplayText = payload.displayText !== undefined;
+        if (hasAttachments || hasClientMeta || hasDisplayText) {
+            userMessageMetadata = {};
+            if (hasAttachments) {
+                userMessageMetadata.attachments = attachmentMetadata;
+            }
+            if (hasClientMeta) {
+                userMessageMetadata.clientMeta = payload.clientMeta;
+            }
+            if (hasDisplayText) {
+                userMessageMetadata.displayText = payload.displayText;
+            }
+        }
+        // Emit a canonical user message into the stream so UI can render from server events only.
+        const userMessage = {
+            id: userMessageId,
+            sessionId,
+            role: 'user',
+            content: trimmed,
+            messageType: 'chat',
+            cliSource: engineName,
+            requestId,
+            isStreaming: false,
+            isFinal: true,
+            createdAt: now,
+            metadata: userMessageMetadata,
+        };
+        this.streamManager.publish({ type: 'message', data: userMessage });
+        if (projectId) {
+            // Persist user message into project history for later reload.
+            try {
+                await (0, project_service_1.touchProjectActivity)(projectId);
+                // Update session activity timestamp so it appears at top of session list
+                if (dbSessionId) {
+                    await (0, session_service_1.touchSessionActivity)(dbSessionId);
+                }
+                await (0, message_service_1.createMessage)({
+                    projectId,
+                    role: 'user',
+                    messageType: 'chat',
+                    content: trimmed,
+                    sessionId,
+                    cliSource: engineName,
+                    requestId,
+                    id: userMessage.id,
+                    createdAt: userMessage.createdAt,
+                    metadata: userMessageMetadata,
+                });
+            }
+            catch (error) {
+                console.error('[AgentChatService] Failed to persist user message:', error);
+            }
+        }
+        this.streamManager.publish({
+            type: 'status',
+            data: {
+                sessionId,
+                status: 'starting',
+                requestId,
+                message: 'Agent request accepted',
+            },
+        });
+        const ctx = {
+            emit: (event) => {
+                var _a;
+                this.streamManager.publish(event);
+                if (!projectId) {
+                    return;
+                }
+                if (event.type === 'message') {
+                    const msg = event.data;
+                    if (!msg)
+                        return;
+                    // Only persist final snapshots; streaming deltas are transient.
+                    if (msg.isStreaming && !msg.isFinal) {
+                        return;
+                    }
+                    // User messages are already handled above.
+                    if (msg.role === 'user') {
+                        return;
+                    }
+                    const content = (_a = msg.content) === null || _a === void 0 ? void 0 : _a.trim();
+                    if (!content) {
+                        return;
+                    }
+                    void (0, message_service_1.createMessage)({
+                        projectId,
+                        role: msg.role,
+                        messageType: msg.messageType,
+                        content,
+                        metadata: msg.metadata,
+                        sessionId: msg.sessionId,
+                        conversationId: undefined,
+                        cliSource: msg.cliSource,
+                        requestId: msg.requestId,
+                        id: msg.id,
+                        createdAt: msg.createdAt,
+                    }).catch((error) => {
+                        console.error('[AgentChatService] Failed to persist agent message:', error);
+                    });
+                }
+            },
+            // Callback to persist Claude session ID when SDK returns system/init message
+            // Prefer session-level persistence over project-level
+            persistClaudeSessionId: dbSessionId
+                ? async (claudeSessionId) => {
+                    await (0, session_service_1.updateEngineSessionId)(dbSessionId, claudeSessionId);
+                }
+                : projectId
+                    ? async (claudeSessionId) => {
+                        await (0, project_service_1.updateProjectClaudeSessionId)(projectId, claudeSessionId);
+                    }
+                    : undefined,
+            // Callback to persist management info from system:init message
+            // Only available when using session-level persistence
+            persistManagementInfo: dbSessionId
+                ? async (info) => {
+                    await (0, session_service_1.updateManagementInfo)(dbSessionId, info);
+                }
+                : undefined,
+        };
+        const engineOptions = {
+            sessionId,
+            instruction: trimmed,
+            model: effectiveModel,
+            projectRoot,
+            requestId,
+            // Pass original attachments (for fallback) and resolved paths (preferred)
+            attachments: payload.attachments,
+            resolvedImagePaths,
+            projectId,
+            dbSessionId,
+            // Session-level configuration for ClaudeEngine
+            permissionMode: dbSession === null || dbSession === void 0 ? void 0 : dbSession.permissionMode,
+            allowDangerouslySkipPermissions: dbSession === null || dbSession === void 0 ? void 0 : dbSession.allowDangerouslySkipPermissions,
+            systemPromptConfig: dbSession === null || dbSession === void 0 ? void 0 : dbSession.systemPromptConfig,
+            optionsConfig: dbSession === null || dbSession === void 0 ? void 0 : dbSession.optionsConfig,
+            // Pass Claude session ID for session resumption (ClaudeEngine only)
+            resumeClaudeSessionId: engineName === 'claude' ? resumeClaudeSessionId : undefined,
+            // Pass useCcr flag for Claude Code Router support (ClaudeEngine only)
+            useCcr: engineName === 'claude' ? projectUseCcr : undefined,
+            // Pass Codex-specific configuration (CodexEngine only)
+            codexConfig: engineName === 'codex' ? (_c = dbSession === null || dbSession === void 0 ? void 0 : dbSession.optionsConfig) === null || _c === void 0 ? void 0 : _c.codexConfig : undefined,
+        };
+        // Create abort controller for cancellation support
+        const abortController = new AbortController();
+        // Register execution in the running executions registry
+        this.runningExecutions.set(requestId, {
+            requestId,
+            sessionId,
+            engineName,
+            abortController,
+            startedAt: new Date(),
+        });
+        // Fire-and-forget execution to keep HTTP handler fast.
+        void this.runEngine(engine, engineOptions, ctx, sessionId, requestId, abortController);
+        return { requestId };
+    }
+    /**
+     * Cancel a running execution by requestId.
+     * Returns true if the execution was found and cancelled, false otherwise.
+     */
+    cancelExecution(requestId) {
+        const execution = this.runningExecutions.get(requestId);
+        if (!execution) {
+            return false;
+        }
+        // Abort the execution
+        execution.abortController.abort();
+        // Emit cancelled status
+        this.streamManager.publish({
+            type: 'status',
+            data: {
+                sessionId: execution.sessionId,
+                status: 'cancelled',
+                requestId,
+                message: 'Execution cancelled by user',
+            },
+        });
+        // Remove from registry
+        this.runningExecutions.delete(requestId);
+        return true;
+    }
+    /**
+     * Cancel all running executions for a session.
+     * Returns the number of executions cancelled.
+     */
+    cancelSessionExecutions(sessionId) {
+        let cancelled = 0;
+        for (const [requestId, execution] of this.runningExecutions) {
+            if (execution.sessionId === sessionId) {
+                execution.abortController.abort();
+                this.runningExecutions.delete(requestId);
+                cancelled++;
+            }
+        }
+        if (cancelled > 0) {
+            this.streamManager.publish({
+                type: 'status',
+                data: {
+                    sessionId,
+                    status: 'cancelled',
+                    message: `Cancelled ${cancelled} running execution(s)`,
+                },
+            });
+        }
+        return cancelled;
+    }
+    /**
+     * Get list of running executions for diagnostics.
+     */
+    getRunningExecutions() {
+        return Array.from(this.runningExecutions.values());
+    }
+    resolveEngineName(preference, projectPreferredCli) {
+        if (preference && this.engines.has(preference)) {
+            return preference;
+        }
+        if (projectPreferredCli && this.engines.has(projectPreferredCli)) {
+            return projectPreferredCli;
+        }
+        return this.defaultEngineName;
+    }
+    async runEngine(engine, options, ctx, sessionId, requestId, abortController) {
+        try {
+            // Check if already aborted before starting
+            if (abortController.signal.aborted) {
+                return;
+            }
+            this.streamManager.publish({
+                type: 'status',
+                data: {
+                    sessionId,
+                    status: 'running',
+                    requestId,
+                },
+            });
+            // Pass abort signal to engine
+            const optionsWithSignal = {
+                ...options,
+                signal: abortController.signal,
+            };
+            await engine.initializeAndRun(optionsWithSignal, ctx);
+            // Only emit completed if not aborted
+            if (!abortController.signal.aborted) {
+                this.streamManager.publish({
+                    type: 'status',
+                    data: {
+                        sessionId,
+                        status: 'completed',
+                        requestId,
+                    },
+                });
+            }
+        }
+        catch (error) {
+            // Check if this was an abort error
+            if (abortController.signal.aborted) {
+                // Already handled by cancelExecution, just return
+                return;
+            }
+            const message = error instanceof Error ? error.message : String(error);
+            this.streamManager.publish({
+                type: 'error',
+                error: message,
+                data: { sessionId, requestId },
+            });
+            this.streamManager.publish({
+                type: 'status',
+                data: {
+                    sessionId,
+                    status: 'error',
+                    message,
+                    requestId,
+                },
+            });
+        }
+        finally {
+            // Always remove from running executions when done
+            this.runningExecutions.delete(requestId);
+        }
+    }
+    /**
+     * Expose registered engines for UI and diagnostics.
+     */
+    getEngineInfos() {
+        const result = [];
+        for (const engine of this.engines.values()) {
+            result.push({
+                name: engine.name,
+                supportsMcp: engine.supportsMcp,
+            });
+        }
+        return result;
+    }
+}
+exports.AgentChatService = AgentChatService;
+//# sourceMappingURL=chat-service.js.map

package/dist/agent/chat-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-service.js","sourceRoot":"","sources":["../../src/agent/chat-service.ts"],"names":[],"mappings":";;;AAAA,6CAAyC;AAYzC,uDAAmG;AACnG,uDAAyE;AACzE,uDAM2B;AAC3B,6DAA+E;AAQ/E;;;;;GAKG;AACH,MAAa,gBAAgB;IAU3B,YAAY,OAAgC;QAT3B,YAAO,GAAG,IAAI,GAAG,EAA2B,CAAC;QAI9D;;WAEG;QACc,sBAAiB,GAAG,IAAI,GAAG,EAA4B,CAAC;QAGvE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,CAAC;QAE3C,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,OAAO,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC7E,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,0EAA0E;YAC1E,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACvC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;YACnE,CAAC;YACD,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,SAAiB,EAAE,OAAwB;;QACzD,MAAM,OAAO,GAAG,MAAA,OAAO,CAAC,WAAW,0CAAE,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAA,wBAAU,GAAE,CAAC;QACpD,IAAI,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QAClC,sCAAsC;QACtC,MAAM,cAAc,GAClB,OAAO,OAAO,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,MAAM,WAAW,GAAG,cAAc,IAAI,SAAS,CAAC;QAEhD,wDAAwD;QACxD,IAAI,SAAmC,CAAC;QACxC,IAAI,WAAW,EAAE,CAAC;YAChB,SAAS,GAAG,MAAM,IAAA,4BAAU,EAAC,WAAW,CAAC,CAAC;YAC1C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,6BAA6B,WAAW,EAAE,CAAC,CAAC;YAC9D,CAAC;YACD,+BAA+B;YAC/B,IAAI,SAAS,IAAI,SAAS,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACnD,MAAM,IAAI,KAAK,CAAC,WAAW,WAAW,gCAAgC,SAAS,EAAE,CAAC,CAAC;YACrF,CAAC;YACD,mDAAmD;YACnD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YAClC,CAAC;QACH,CAAC;QAED,qEAAqE;QACrE,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,IAAA,4BAAU,EAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,6BAA6B,SAAS,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;QACrC,MAAM,mBAAmB,GAAG,OAAO,CAAC,YAAsC,CAAC;QAC3E,MAAM,oBAAoB,GAAG,OAAO,CAAC,aAAa,CAAC;QACnD,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;QAErC,sFAAsF;QACtF,IAAI,qBAAyC,CAAC;QAC9C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;QACxD,CAAC;QAED,yDAAyD;QACzD,IAAI,UAAsB,CAAC;QAC3B,IAAI,SAAS,EAAE,CAAC;YACd,UAAU,GAAG,SAAS,CAAC,UAAwB,CAAC;YAChD,gDAAgD;YAChD,IAAI,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;gBAClE,MAAM,IAAI,KAAK,CACb,kBAAkB,OAAO,CAAC,aAAa,wCAAwC,UAAU,GAAG,CAC7F,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,IAAI,CAAC,iBAAiB,CACjC,OAAO,CAAC,aAAuC,EAC/C,mBAAmB,CACpB,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,kCAAkC,UAAU,EAAE,CAAC,CAAC;QAClE,CAAC;QAED,8CAA8C;QAC9C,MAAM,cAAc,GAAG,CAAA,MAAA,OAAO,CAAC,KAAK,0CAAE,IAAI,EAAE,MAAI,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,KAAK,CAAA,IAAI,oBAAoB,CAAC;QAEzF,2EAA2E;QAC3E,IAAI,SAAS,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;YACzC,qBAAqB,GAAG,SAAS,CAAC,eAAe,CAAC;QACpD,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,aAAa,GAAG,IAAA,wBAAU,GAAE,CAAC;QAEnC,wCAAwC;QACxC,MAAM,gBAAgB,GAAsB,EAAE,CAAC;QAC/C,IAAI,kBAAoD,CAAC;QACzD,IAAI,kBAAwC,CAAC;QAE7C,IAAI,SAAS,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvE,MAAM,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC;YAE/E,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,IAAI,CAAC;oBACH,OAAO,CAAC,KAAK,CACX,6BAA6B,gBAAgB,CAAC,MAAM,oCAAoC,SAAS,EAAE,CACpG,CAAC;oBAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;wBACjD,MAAM,UAAU,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;wBACvC,MAAM,KAAK,GAAG,MAAM,sCAAiB,CAAC,cAAc,CAAC;4BACnD,SAAS;4BACT,SAAS,EAAE,aAAa;4BACxB,UAAU;4BACV,KAAK,EAAE,CAAC;yBACT,CAAC,CAAC;wBACH,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBAC/B,CAAC;oBAED,+CAA+C;oBAC/C,kBAAkB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBAC7D,2CAA2C;oBAC3C,kBAAkB,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;oBAEjE,OAAO,CAAC,KAAK,CACX,4BAA4B,gBAAgB,CAAC,MAAM,mBAAmB,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtG,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,gDAAgD,EAAE,KAAK,CAAC,CAAC;oBACvE,+DAA+D;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,yCAAyC;QACzC,8DAA8D;QAC9D,IAAI,mBAAwD,CAAC;QAC7D,MAAM,cAAc,GAAG,kBAAkB,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAG,OAAO,CAAC,UAAU,KAAK,SAAS,CAAC;QACvD,MAAM,cAAc,GAAG,OAAO,CAAC,WAAW,KAAK,SAAS,CAAC;QAEzD,IAAI,cAAc,IAAI,aAAa,IAAI,cAAc,EAAE,CAAC;YACtD,mBAAmB,GAAG,EAAE,CAAC;YACzB,IAAI,cAAc,EAAE,CAAC;gBACnB,mBAAmB,CAAC,WAAW,GAAG,kBAAkB,CAAC;YACvD,CAAC;YACD,IAAI,aAAa,EAAE,CAAC;gBAClB,mBAAmB,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;YACtD,CAAC;YACD,IAAI,cAAc,EAAE,CAAC;gBACnB,mBAAmB,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;YACxD,CAAC;QACH,CAAC;QAED,0FAA0F;QAC1F,MAAM,WAAW,GAAiB;YAChC,EAAE,EAAE,aAAa;YACjB,SAAS;YACT,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO;YAChB,WAAW,EAAE,MAAM;YACnB,SAAS,EAAE,UAAU;YACrB,SAAS;YACT,WAAW,EAAE,KAAK;YAClB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,GAAG;YACd,QAAQ,EAAE,mBAAmB;SAC9B,CAAC;QAEF,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAEnE,IAAI,SAAS,EAAE,CAAC;YACd,8DAA8D;YAC9D,IAAI,CAAC;gBACH,MAAM,IAAA,sCAAoB,EAAC,SAAS,CAAC,CAAC;gBACtC,yEAAyE;gBACzE,IAAI,WAAW,EAAE,CAAC;oBAChB,MAAM,IAAA,sCAAoB,EAAC,WAAW,CAAC,CAAC;gBAC1C,CAAC;gBACD,MAAM,IAAA,+BAAmB,EAAC;oBACxB,SAAS;oBACT,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,MAAM;oBACnB,OAAO,EAAE,OAAO;oBAChB,SAAS;oBACT,SAAS,EAAE,UAAU;oBACrB,SAAS;oBACT,EAAE,EAAE,WAAW,CAAC,EAAE;oBAClB,SAAS,EAAE,WAAW,CAAC,SAAS;oBAChC,QAAQ,EAAE,mBAAmB;iBAC9B,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oDAAoD,EAAE,KAAK,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACzB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE;gBACJ,SAAS;gBACT,MAAM,EAAE,UAAU;gBAClB,SAAS;gBACT,OAAO,EAAE,wBAAwB;aAClC;SACF,CAAC,CAAC;QAEH,MAAM,GAAG,GAA2B;YAClC,IAAI,EAAE,CAAC,KAAoB,EAAE,EAAE;;gBAC7B,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAElC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,OAAO;gBACT,CAAC;gBAED,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAC7B,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC;oBACvB,IAAI,CAAC,GAAG;wBAAE,OAAO;oBAEjB,gEAAgE;oBAChE,IAAI,GAAG,CAAC,WAAW,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;wBACpC,OAAO;oBACT,CAAC;oBAED,2CAA2C;oBAC3C,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACxB,OAAO;oBACT,CAAC;oBAED,MAAM,OAAO,GAAG,MAAA,GAAG,CAAC,OAAO,0CAAE,IAAI,EAAE,CAAC;oBACpC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,OAAO;oBACT,CAAC;oBAED,KAAK,IAAA,+BAAmB,EAAC;wBACvB,SAAS;wBACT,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,WAAW,EAAE,GAAG,CAAC,WAAW;wBAC5B,OAAO;wBACP,QAAQ,EAAE,GAAG,CAAC,QAAQ;wBACtB,SAAS,EAAE,GAAG,CAAC,SAAS;wBACxB,cAAc,EAAE,SAAS;wBACzB,SAAS,EAAE,GAAG,CAAC,SAAS;wBACxB,SAAS,EAAE,GAAG,CAAC,SAAS;wBACxB,EAAE,EAAE,GAAG,CAAC,EAAE;wBACV,SAAS,EAAE,GAAG,CAAC,SAAS;qBACzB,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;wBACjB,OAAO,CAAC,KAAK,CAAC,qDAAqD,EAAE,KAAK,CAAC,CAAC;oBAC9E,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,6EAA6E;YAC7E,sDAAsD;YACtD,sBAAsB,EAAE,WAAW;gBACjC,CAAC,CAAC,KAAK,EAAE,eAAuB,EAAE,EAAE;oBAChC,MAAM,IAAA,uCAAqB,EAAC,WAAW,EAAE,eAAe,CAAC,CAAC;gBAC5D,CAAC;gBACH,CAAC,CAAC,SAAS;oBACT,CAAC,CAAC,KAAK,EAAE,eAAuB,EAAE,EAAE;wBAChC,MAAM,IAAA,8CAA4B,EAAC,SAAS,EAAE,eAAe,CAAC,CAAC;oBACjE,CAAC;oBACH,CAAC,CAAC,SAAS;YACf,+DAA+D;YAC/D,sDAAsD;YACtD,qBAAqB,EAAE,WAAW;gBAChC,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;oBACb,MAAM,IAAA,sCAAoB,EAAC,WAAW,EAAE,IAAI,CAAC,CAAC;gBAChD,CAAC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;QAEF,MAAM,aAAa,GAAsB;YACvC,SAAS;YACT,WAAW,EAAE,OAAO;YACpB,KAAK,EAAE,cAAc;YACrB,WAAW;YACX,SAAS;YACT,0EAA0E;YAC1E,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,kBAAkB;YAClB,SAAS;YACT,WAAW;YACX,+CAA+C;YAC/C,cAAc,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,cAAc;YACzC,+BAA+B,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,+BAA+B;YAC3E,kBAAkB,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,kBAAkB;YACjD,aAAa,EAAE,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,aAAa;YACvC,oEAAoE;YACpE,qBAAqB,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,SAAS;YAClF,sEAAsE;YACtE,MAAM,EAAE,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;YAC3D,uDAAuD;YACvD,WAAW,EAAE,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,MAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,aAAa,0CAAE,WAAW,CAAC,CAAC,CAAC,SAAS;SACxF,CAAC;QAEF,mDAAmD;QACnD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,wDAAwD;QACxD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE;YACpC,SAAS;YACT,SAAS;YACT,UAAU;YACV,eAAe;YACf,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC,CAAC;QAEH,uDAAuD;QACvD,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;QAEvF,OAAO,EAAE,SAAS,EAAE,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,SAAiB;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,KAAK,CAAC;QACf,CAAC;QAED,sBAAsB;QACtB,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAElC,wBAAwB;QACxB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACzB,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE;gBACJ,SAAS,EAAE,SAAS,CAAC,SAAS;gBAC9B,MAAM,EAAE,WAAW;gBACnB,SAAS;gBACT,OAAO,EAAE,6BAA6B;aACvC;SACF,CAAC,CAAC;QAEH,uBAAuB;QACvB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAEzC,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;OAGG;IACH,uBAAuB,CAAC,SAAiB;QACvC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC5D,IAAI,SAAS,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACtC,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;gBAClC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;gBACzC,SAAS,EAAE,CAAC;YACd,CAAC;QACH,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;gBACzB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,SAAS;oBACT,MAAM,EAAE,WAAW;oBACnB,OAAO,EAAE,aAAa,SAAS,uBAAuB;iBACvD;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;IACrD,CAAC;IAEO,iBAAiB,CAAC,UAAuB,EAAE,mBAAgC;QACjF,IAAI,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/C,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,IAAI,mBAAmB,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACjE,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAEO,KAAK,CAAC,SAAS,CACrB,MAAmB,EACnB,OAA0B,EAC1B,GAA2B,EAC3B,SAAiB,EACjB,SAAiB,EACjB,eAAgC;QAEhC,IAAI,CAAC;YACH,2CAA2C;YAC3C,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,OAAO;YACT,CAAC;YAED,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;gBACzB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,SAAS;oBACT,MAAM,EAAE,SAAS;oBACjB,SAAS;iBACV;aACF,CAAC,CAAC;YAEH,8BAA8B;YAC9B,MAAM,iBAAiB,GAAsB;gBAC3C,GAAG,OAAO;gBACV,MAAM,EAAE,eAAe,CAAC,MAAM;aAC/B,CAAC;YAEF,MAAM,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YAEtD,qCAAqC;YACrC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;oBACzB,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE;wBACJ,SAAS;wBACT,MAAM,EAAE,WAAW;wBACnB,SAAS;qBACV;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,mCAAmC;YACnC,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnC,kDAAkD;gBAClD,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEvE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;gBACzB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE;aAC/B,CAAC,CAAC;YAEH,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;gBACzB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE;oBACJ,SAAS;oBACT,MAAM,EAAE,OAAO;oBACf,OAAO;oBACP,SAAS;iBACV;aACF,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,kDAAkD;YAClD,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,MAAM,MAAM,GAAuD,EAAE,CAAC;QACtE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,WAAW,EAAE,MAAM,CAAC,WAAW;aAChC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AA3eD,4CA2eC"}

package/dist/agent/db/client.d.ts

@@ -0,0 +1,26 @@
+import { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
+import * as schema from './schema';
+export type DrizzleDB = BetterSQLite3Database<typeof schema>;
+/**
+ * Get the database file path.
+ * Environment: CHROME_MCP_AGENT_DB_FILE overrides the default path.
+ */
+export declare function getDatabasePath(): string;
+/**
+ * Get the Drizzle database instance.
+ * Lazily initializes the connection and schema on first call.
+ */
+export declare function getDb(): DrizzleDB;
+/**
+ * Close the database connection.
+ * Should be called on graceful shutdown.
+ */
+export declare function closeDb(): void;
+/**
+ * Check if database is initialized.
+ */
+export declare function isDbInitialized(): boolean;
+/**
+ * Execute raw SQL (for advanced use cases).
+ */
+export declare function execRawSql(sqlStr: string): void;