ace-tool 0.1.5 → 0.2.0

package/dist/enhancer/PromptEnhancer.js

@@ -0,0 +1,243 @@
+/**
+ * Prompt Enhancer - 核心增强逻辑
+ * 基于 Augment VSCode 插件的实现
+ */
+import axios from 'axios';
+import fs from 'fs';
+import { exec } from 'child_process';
+import { sendMcpLog } from '../mcpLogger.js';
+/**
+ * Prompt Enhancer 类
+ */
+export class PromptEnhancer {
+    indexManager;
+    baseUrl;
+    token;
+    httpClient;
+    server;
+    constructor(indexManager, baseUrl, token, server) {
+        this.indexManager = indexManager;
+        this.baseUrl = baseUrl;
+        this.token = token;
+        this.baseUrl = baseUrl.replace(/\/$/, '');
+        this.server = server;
+        this.httpClient = axios.create({
+            timeout: 60000,
+            headers: {
+                Authorization: `Bearer ${this.token}`,
+                'Content-Type': 'application/json',
+            },
+        });
+    }
+    /**
+     * 增强 prompt
+     * @param originalPrompt 原始用户输入
+     * @param conversationHistory 对话历史（5-10轮）
+     * @returns 增强后的 prompt
+     */
+    async enhance(originalPrompt, conversationHistory) {
+        sendMcpLog('info', '🎨 开始增强 prompt...');
+        try {
+            // 1. 加载已索引的 blob 列表
+            sendMcpLog('info', '📂 加载索引数据...');
+            const blobNames = this.loadBlobNames();
+            if (blobNames.length === 0) {
+                sendMcpLog('warning', '⚠️ 未找到索引数据，将在没有代码上下文的情况下增强');
+            }
+            else {
+                sendMcpLog('info', `📊 已加载 ${blobNames.length} 个文件块`);
+            }
+            // 2. 设置增强回调函数（用于"继续增强"功能）
+            this.server.setEnhanceCallback(async (prompt, history, blobs) => {
+                return await this.callPromptEnhancerApi(prompt, history, blobs);
+            });
+            // 3. 调用 prompt-enhancer API 生成增强内容
+            sendMcpLog('info', '🤖 调用 prompt-enhancer API 生成增强内容...');
+            const enhancedPrompt = await this.callPromptEnhancerApi(originalPrompt, conversationHistory, blobNames);
+            sendMcpLog('info', '✅ 增强内容生成成功');
+            // 4. 启动 Web UI 交互
+            sendMcpLog('info', '🌐 启动 Web UI 等待用户确认...');
+            const finalPrompt = await this.interactWithUser(enhancedPrompt, originalPrompt, conversationHistory, blobNames);
+            sendMcpLog('info', '✅ Prompt 增强完成');
+            return finalPrompt;
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            sendMcpLog('error', `❌ Prompt 增强失败: ${errorMessage}`);
+            throw error;
+        }
+    }
+    /**
+     * 加载已索引的 blob 名称列表
+     */
+    loadBlobNames() {
+        try {
+            // 使用 IndexManager 的私有��法需要通过反射或者添加公共方法
+            // 这里我们直接读取索引文件
+            const indexFilePath = this.indexManager.indexFilePath;
+            if (!fs.existsSync(indexFilePath)) {
+                return [];
+            }
+            const content = fs.readFileSync(indexFilePath, 'utf-8');
+            return JSON.parse(content);
+        }
+        catch (error) {
+            sendMcpLog('warning', `⚠️ 加载索引失败: ${error}`);
+            return [];
+        }
+    }
+    /**
+     * 调用 prompt-enhancer API 生成增强内容
+     */
+    async callPromptEnhancerApi(originalPrompt, conversationHistory, blobNames) {
+        // 解析对话历史
+        const chatHistory = this.parseChatHistory(conversationHistory);
+        // 检测原始 prompt 的语言
+        const isChinese = /[\u4e00-\u9fa5]/.test(originalPrompt);
+        const languageGuideline = isChinese
+            ? 'Please respond in Chinese (Simplified Chinese). 请用中文回复。'
+            : '';
+        // 构造符合 Augment 格式的 payload
+        // nodes 包含用户的原始 prompt
+        const payload = {
+            nodes: [
+                {
+                    id: 1,
+                    type: 0, // text node type
+                    text_node: {
+                        content: originalPrompt
+                    }
+                }
+            ],
+            chat_history: chatHistory,
+            blobs: {
+                checkpoint_id: null,
+                added_blobs: blobNames,
+                deleted_blobs: [],
+            },
+            conversation_id: null,
+            model: 'claude-sonnet-4-5',
+            mode: 'CHAT',
+            user_guided_blobs: [],
+            external_source_ids: [],
+            user_guidelines: languageGuideline,
+            workspace_guidelines: '',
+            rules: []
+        };
+        try {
+            const response = await this.httpClient.post(`${this.baseUrl}/prompt-enhancer`, payload);
+            let enhancedText = response.data.text || '';
+            if (!enhancedText) {
+                throw new Error('Prompt enhancer API 返回空结果');
+            }
+            // 替换 Augment 特定的工具名称为 ace-tool 的工具名称
+            enhancedText = this.replaceToolNames(enhancedText);
+            return enhancedText;
+        }
+        catch (error) {
+            if (error.response?.status === 401) {
+                throw new Error('Token 已失效或无效，请检查配置');
+            }
+            else if (error.response?.status === 403) {
+                throw new Error('访问被拒绝，Token 可能已被禁用');
+            }
+            else if (error.code === 'ECONNREFUSED') {
+                throw new Error('无法连接到服务器，请检查 base-url 配置');
+            }
+            throw new Error(`Prompt enhancer API 调用失败: ${error.message}`);
+        }
+    }
+    /**
+     * 替换 Augment 特定的工具名称为 ace-tool 的工具名称
+     */
+    replaceToolNames(text) {
+        // 替换 codebase-retrieval 为 search_context
+        // 支持多种可能的格式：
+        // 1. 工具调用格式：codebase-retrieval 或 codebase_retrieval
+        // 2. 文本引用格式：`codebase-retrieval` 或 "codebase-retrieval"
+        let result = text;
+        // 替换各种格式的 codebase-retrieval
+        result = result.replace(/codebase-retrieval/g, 'search_context');
+        result = result.replace(/codebase_retrieval/g, 'search_context');
+        // 如果有其他 Augment 特定的工具需要替换，可以在这里添加
+        return result;
+    }
+    /**
+     * 解析对话历史为 chat_history 格式
+     */
+    parseChatHistory(conversationHistory) {
+        // 简单解析对话历史
+        // 格式示例: "User: xxx\nAI: yyy\nUser: zzz"
+        const lines = conversationHistory.split('\n');
+        const chatHistory = [];
+        for (const line of lines) {
+            const trimmed = line.trim();
+            if (trimmed.startsWith('User:') || trimmed.startsWith('用户:')) {
+                chatHistory.push({
+                    role: 'user',
+                    content: trimmed.replace(/^(User:|用户:)\s*/, '')
+                });
+            }
+            else if (trimmed.startsWith('AI:') || trimmed.startsWith('Assistant:') || trimmed.startsWith('助手:')) {
+                chatHistory.push({
+                    role: 'assistant',
+                    content: trimmed.replace(/^(AI:|Assistant:|助手:)\s*/, '')
+                });
+            }
+        }
+        return chatHistory;
+    }
+    /**
+     * 通过 Web UI 与用户交互
+     */
+    async interactWithUser(enhancedPrompt, originalPrompt, conversationHistory, blobNames) {
+        // 确保服务器已启动
+        await this.server.start();
+        // 创建 session（传入所有必要的参数以支持"继续增强"）
+        const sessionId = this.server.createSession(enhancedPrompt, originalPrompt, conversationHistory, blobNames);
+        // 打开浏览器
+        const url = `http://localhost:${this.server.getPort()}/enhance?session=${sessionId}`;
+        sendMcpLog('info', `🌐 请在浏览器中查看: ${url}`);
+        // 尝试自动打开浏览器
+        this.openBrowser(url);
+        // 等待用户操作
+        try {
+            const finalPrompt = await this.server.waitForSession(sessionId);
+            if (!finalPrompt || finalPrompt.trim() === '') {
+                sendMcpLog('info', '❌ 用户取消了增强');
+                throw new Error('User cancelled the enhancement');
+            }
+            return finalPrompt;
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('timeout')) {
+                sendMcpLog('error', '⏱️ 用户交互超时（8分钟）');
+                throw new Error('User interaction timeout (8 minutes)');
+            }
+            throw error;
+        }
+    }
+    /**
+     * 打开浏览器
+     */
+    openBrowser(url) {
+        const platform = process.platform;
+        let command;
+        if (platform === 'darwin') {
+            command = `open "${url}"`;
+        }
+        else if (platform === 'win32') {
+            command = `start "${url}"`;
+        }
+        else {
+            command = `xdg-open "${url}"`;
+        }
+        exec(command, (error) => {
+            if (error) {
+                sendMcpLog('warning', `⚠️ 无法自动打开浏览器: ${error.message}`);
+                sendMcpLog('info', `请手动打开: ${url}`);
+            }
+        });
+    }
+}
+//# sourceMappingURL=PromptEnhancer.js.map

package/dist/enhancer/PromptEnhancer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PromptEnhancer.js","sourceRoot":"","sources":["../../src/enhancer/PromptEnhancer.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAwB,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAGrC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAE7C;;GAEG;AACH,MAAM,OAAO,cAAc;IAKf;IACA;IACA;IANF,UAAU,CAAgB;IAC1B,MAAM,CAAiB;IAE/B,YACU,YAA0B,EAC1B,OAAe,EACf,KAAa,EACrB,MAAsB;QAHd,iBAAY,GAAZ,YAAY,CAAc;QAC1B,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAQ;QAGrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC;YAC7B,OAAO,EAAE,KAAK;YACd,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE;gBACrC,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAC;IACL,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO,CAAC,cAAsB,EAAE,mBAA2B;QAC/D,UAAU,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;QAExC,IAAI,CAAC;YACH,oBAAoB;YACpB,UAAU,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;YACnC,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;YAEvC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3B,UAAU,CAAC,SAAS,EAAE,4BAA4B,CAAC,CAAC;YACtD,CAAC;iBAAM,CAAC;gBACN,UAAU,CAAC,MAAM,EAAE,UAAU,SAAS,CAAC,MAAM,OAAO,CAAC,CAAC;YACxD,CAAC;YAED,0BAA0B;YAC1B,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;gBAC9D,OAAO,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;YAEH,mCAAmC;YACnC,UAAU,CAAC,MAAM,EAAE,qCAAqC,CAAC,CAAC;YAC1D,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,qBAAqB,CACrD,cAAc,EACd,mBAAmB,EACnB,SAAS,CACV,CAAC;YACF,UAAU,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAEjC,kBAAkB;YAClB,UAAU,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;YAC7C,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC7C,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,SAAS,CACV,CAAC;YAEF,UAAU,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YACpC,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,UAAU,CAAC,OAAO,EAAE,kBAAkB,YAAY,EAAE,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,CAAC;YACH,uCAAuC;YACvC,eAAe;YACf,MAAM,aAAa,GAAI,IAAI,CAAC,YAAoB,CAAC,aAAa,CAAC;YAE/D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClC,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACxD,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,UAAU,CAAC,SAAS,EAAE,cAAc,KAAK,EAAE,CAAC,CAAC;YAC7C,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB,CACjC,cAAsB,EACtB,mBAA2B,EAC3B,SAAmB;QAEnB,SAAS;QACT,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,CAAC;QAE/D,kBAAkB;QAClB,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzD,MAAM,iBAAiB,GAAG,SAAS;YACjC,CAAC,CAAC,yDAAyD;YAC3D,CAAC,CAAC,EAAE,CAAC;QAEP,2BAA2B;QAC3B,uBAAuB;QACvB,MAAM,OAAO,GAAG;YACd,KAAK,EAAE;gBACL;oBACE,EAAE,EAAE,CAAC;oBACL,IAAI,EAAE,CAAC,EAAE,iBAAiB;oBAC1B,SAAS,EAAE;wBACT,OAAO,EAAE,cAAc;qBACxB;iBACF;aACF;YACD,YAAY,EAAE,WAAW;YACzB,KAAK,EAAE;gBACL,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,SAAS;gBACtB,aAAa,EAAE,EAAE;aAClB;YACD,eAAe,EAAE,IAAI;YACrB,KAAK,EAAE,mBAAmB;YAC1B,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE,EAAE;YACrB,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,iBAAiB;YAClC,oBAAoB,EAAE,EAAE;YACxB,KAAK,EAAE,EAAE;SACV,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACzC,GAAG,IAAI,CAAC,OAAO,kBAAkB,EACjC,OAAO,CACR,CAAC;YAEF,IAAI,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;YAE5C,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YAED,qCAAqC;YACrC,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;YAEnD,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACxC,CAAC;iBAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC1C,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACxC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9C,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,IAAY;QACnC,yCAAyC;QACzC,aAAa;QACb,oDAAoD;QACpD,wDAAwD;QAExD,IAAI,MAAM,GAAG,IAAI,CAAC;QAElB,6BAA6B;QAC7B,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,gBAAgB,CAAC,CAAC;QAEjE,kCAAkC;QAElC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,mBAA2B;QAClD,WAAW;QACX,wCAAwC;QACxC,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,WAAW,GAA2C,EAAE,CAAC;QAE/D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7D,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC;iBAChD,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtG,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC;iBACzD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAC5B,cAAsB,EACtB,cAAsB,EACtB,mBAA2B,EAC3B,SAAmB;QAEnB,WAAW;QACX,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAE1B,iCAAiC;QACjC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CACzC,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,SAAS,CACV,CAAC;QAEF,QAAQ;QACR,MAAM,GAAG,GAAG,oBAAoB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,oBAAoB,SAAS,EAAE,CAAC;QACrF,UAAU,CAAC,MAAM,EAAE,gBAAgB,GAAG,EAAE,CAAC,CAAC;QAE1C,YAAY;QACZ,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAEtB,SAAS;QACT,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YAEhE,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC9C,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YACpD,CAAC;YAED,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAChE,UAAU,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,GAAW;QAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QAElC,IAAI,OAAe,CAAC;QACpB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,GAAG,SAAS,GAAG,GAAG,CAAC;QAC5B,CAAC;aAAM,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YAChC,OAAO,GAAG,UAAU,GAAG,GAAG,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,aAAa,GAAG,GAAG,CAAC;QAChC,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;YAC3B,IAAI,KAAK,EAAE,CAAC;gBACV,UAAU,CAAC,SAAS,EAAE,iBAAiB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxD,UAAU,CAAC,MAAM,EAAE,UAAU,GAAG,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/enhancer/templates.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Prompt Enhancer 模板
+ * 基于 Augment VSCode 插件的官方模板
+ */
+/**
+ * 增强 Prompt 的模板
+ * 包含对话历史和原始 prompt 的占位符
+ */
+export declare const ENHANCE_PROMPT_TEMPLATE = "\u26A0\uFE0F NO TOOLS ALLOWED \u26A0\uFE0F\n\nHere is an instruction that I'd like to give you, but it needs to be improved.\nRewrite and enhance this instruction to make it clearer, more specific, less ambiguous,\nand correct any mistakes. Do not use any tools: reply immediately with your answer,\neven if you're not sure. Consider the context of our conversation history when enhancing\nthe prompt.\n\nConversation history:\n{conversation_history}\n\nReply with the following format:\n\n### BEGIN RESPONSE ###\nHere is an enhanced version of the original instruction that is more specific and clear:\n<augment-enhanced-prompt>enhanced prompt goes here</augment-enhanced-prompt>\n\n### END RESPONSE ###\n\nHere is my original instruction:\n\n{original_prompt}";
+/**
+ * Web UI 的 HTML 模板
+ */
+export declare const ENHANCER_UI_HTML = "<!DOCTYPE html>\n<html lang=\"zh-CN\">\n<head>\n  <meta charset=\"UTF-8\">\n  <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">\n  <title>Prompt Enhancer - ACE Tool</title>\n  <style>\n    * {\n      margin: 0;\n      padding: 0;\n      box-sizing: border-box;\n    }\n\n    body {\n      font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', 'Roboto', 'Helvetica Neue', sans-serif;\n      background: #f5f5f5;\n      min-height: 100vh;\n      padding: 20px;\n      display: flex;\n      align-items: center;\n      justify-content: center;\n    }\n\n    .container {\n      background: white;\n      border-radius: 8px;\n      box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);\n      border: 1px solid #e0e0e0;\n      max-width: 1000px;\n      width: 100%;\n      overflow: hidden;\n    }\n\n    .header {\n      background: white;\n      color: #333;\n      padding: 30px;\n      text-align: center;\n      border-bottom: 1px solid #e0e0e0;\n    }\n\n    .header h1 {\n      font-size: 24px;\n      font-weight: 600;\n      margin-bottom: 8px;\n      display: flex;\n      align-items: center;\n      justify-content: center;\n      gap: 10px;\n      color: #333;\n    }\n\n    .header p {\n      font-size: 14px;\n      color: #666;\n    }\n\n    .countdown {\n      margin-top: 12px;\n      padding: 8px 16px;\n      background: #f0f0f0;\n      border-radius: 6px;\n      display: inline-block;\n      font-size: 13px;\n      font-weight: 500;\n      color: #555;\n    }\n\n    .countdown.warning {\n      background: #fff3cd;\n      color: #856404;\n    }\n\n    .countdown.danger {\n      background: #f8d7da;\n      color: #721c24;\n      animation: pulse 1s ease-in-out infinite;\n    }\n\n    @keyframes pulse {\n      0%, 100% { opacity: 1; }\n      50% { opacity: 0.7; }\n    }\n\n    .content {\n      padding: 30px;\n    }\n\n    .section {\n      margin-bottom: 25px;\n    }\n\n    .section-title {\n      font-size: 14px;\n      font-weight: 600;\n      color: #333;\n      margin-bottom: 10px;\n      text-transform: uppercase;\n      letter-spacing: 0.5px;\n    }\n\n    .editor-wrapper {\n      position: relative;\n    }\n\n    textarea {\n      width: 100%;\n      min-height: 350px;\n      padding: 16px;\n      border: 2px solid #e0e0e0;\n      border-radius: 8px;\n      font-family: 'SF Mono', 'Monaco', 'Menlo', 'Consolas', monospace;\n      font-size: 14px;\n      line-height: 1.6;\n      resize: vertical;\n      transition: border-color 0.3s;\n      background: #fafafa;\n    }\n\n    textarea:focus {\n      outline: none;\n      border-color: #333;\n      background: white;\n    }\n\n    .char-count {\n      position: absolute;\n      bottom: 12px;\n      right: 12px;\n      background: rgba(255, 255, 255, 0.9);\n      padding: 4px 10px;\n      border-radius: 12px;\n      font-size: 12px;\n      color: #666;\n      pointer-events: none;\n      box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);\n    }\n\n    .info-box {\n      background: #f9f9f9;\n      border-left: 4px solid #333;\n      padding: 15px;\n      border-radius: 4px;\n      margin-bottom: 20px;\n    }\n\n    .info-box p {\n      font-size: 14px;\n      color: #555;\n      line-height: 1.6;\n    }\n\n    .buttons {\n      display: flex;\n      gap: 12px;\n      justify-content: flex-end;\n      margin-top: 25px;\n    }\n\n    button {\n      padding: 12px 28px;\n      border: none;\n      border-radius: 8px;\n      font-size: 15px;\n      font-weight: 600;\n      cursor: pointer;\n      transition: all 0.3s;\n      display: flex;\n      align-items: center;\n      gap: 8px;\n    }\n\n    .send-btn {\n      background: #333;\n      color: white;\n      box-shadow: none;\n    }\n\n    .send-btn:hover:not(:disabled) {\n      background: #000;\n    }\n\n    .send-btn:active:not(:disabled) {\n      background: #000;\n    }\n\n    .send-btn:disabled {\n      background: #ccc;\n      cursor: not-allowed;\n      box-shadow: none;\n    }\n\n    .cancel-btn {\n      background: white;\n      color: #666;\n      border: 2px solid #e0e0e0;\n    }\n\n    .cancel-btn:hover {\n      background: #f5f5f5;\n      border-color: #ccc;\n    }\n\n    .re-enhance-btn {\n      background: white;\n      color: #333;\n      border: 2px solid #333;\n    }\n\n    .re-enhance-btn:hover:not(:disabled) {\n      background: #f5f5f5;\n      border-color: #000;\n    }\n\n    .re-enhance-btn:disabled {\n      background: #f5f5f5;\n      color: #ccc;\n      border-color: #e0e0e0;\n      cursor: not-allowed;\n    }\n\n    .status {\n      margin-top: 20px;\n      padding: 15px;\n      border-radius: 8px;\n      display: none;\n      animation: slideIn 0.3s ease;\n    }\n\n    @keyframes slideIn {\n      from {\n        opacity: 0;\n        transform: translateY(-10px);\n      }\n      to {\n        opacity: 1;\n        transform: translateY(0);\n      }\n    }\n\n    .status.success {\n      background: #d4edda;\n      color: #155724;\n      border-left: 4px solid #28a745;\n      display: block;\n    }\n\n    .status.error {\n      background: #f8d7da;\n      color: #721c24;\n      border-left: 4px solid #dc3545;\n      display: block;\n    }\n\n    .loading {\n      display: none;\n      text-align: center;\n      padding: 40px;\n    }\n\n    .loading.active {\n      display: block;\n    }\n\n    .spinner {\n      border: 3px solid #f3f3f3;\n      border-top: 3px solid #333;\n      border-radius: 50%;\n      width: 40px;\n      height: 40px;\n      animation: spin 1s linear infinite;\n      margin: 0 auto 15px;\n    }\n\n    @keyframes spin {\n      0% { transform: rotate(0deg); }\n      100% { transform: rotate(360deg); }\n    }\n\n    .keyboard-hint {\n      font-size: 12px;\n      color: #999;\n      text-align: center;\n      margin-top: 15px;\n    }\n\n    .keyboard-hint kbd {\n      background: #f5f5f5;\n      border: 1px solid #ddd;\n      border-radius: 4px;\n      padding: 2px 6px;\n      font-family: monospace;\n      font-size: 11px;\n    }\n\n    @media (max-width: 768px) {\n      body {\n        padding: 10px;\n      }\n\n      .header {\n        padding: 20px;\n      }\n\n      .header h1 {\n        font-size: 22px;\n      }\n\n      .content {\n        padding: 20px;\n      }\n\n      textarea {\n        min-height: 250px;\n        font-size: 13px;\n      }\n\n      .buttons {\n        flex-direction: column-reverse;\n      }\n\n      button {\n        width: 100%;\n        justify-content: center;\n      }\n    }\n  </style>\n</head>\n<body>\n  <div class=\"container\">\n    <div class=\"header\">\n      <h1>\n        <svg width=\"24\" height=\"24\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\">\n          <path d=\"M12 2L2 7l10 5 10-5-10-5z\"/>\n          <path d=\"M2 17l10 5 10-5\"/>\n          <path d=\"M2 12l10 5 10-5\"/>\n        </svg>\n        Prompt Enhancer\n      </h1>\n      <p>Review and refine your enhanced prompt</p>\n      <div class=\"countdown\" id=\"countdown\">\u23F1\uFE0F \u52A0\u8F7D\u4E2D...</div>\n    </div>\n\n    <div class=\"content\">\n      <div class=\"loading\" id=\"loading\">\n        <div class=\"spinner\"></div>\n        <p>Loading your enhanced prompt...</p>\n      </div>\n\n      <div id=\"mainContent\" style=\"display: none;\">\n        <div class=\"info-box\">\n          <p>\n            <strong>\uD83D\uDCA1 \u63D0\u793A\uFF1A</strong>AI \u5DF2\u7ECF\u6839\u636E\u5BF9\u8BDD\u5386\u53F2\u548C\u4EE3\u7801\u4E0A\u4E0B\u6587\u589E\u5F3A\u4E86\u4F60\u7684 prompt\u3002\n            \u4F60\u53EF\u4EE5\u5728\u4E0B\u65B9\u7F16\u8F91\u5668\u4E2D\u8FDB\u4E00\u6B65\u4FEE\u6539\uFF0C\u7136\u540E\u70B9\u51FB\"\u53D1\u9001\"\u7EE7\u7EED\u3002\n          </p>\n        </div>\n\n        <div class=\"section\">\n          <div class=\"section-title\">Enhanced Prompt</div>\n          <div class=\"editor-wrapper\">\n            <textarea\n              id=\"promptText\"\n              placeholder=\"Your enhanced prompt will appear here...\"\n              spellcheck=\"false\"\n            ></textarea>\n            <div class=\"char-count\" id=\"charCount\">0 \u5B57\u7B26</div>\n          </div>\n        </div>\n\n        <div class=\"buttons\">\n          <button class=\"cancel-btn\" onclick=\"endConversation()\">\n            <svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\">\n              <line x1=\"18\" y1=\"6\" x2=\"6\" y2=\"18\"/>\n              <line x1=\"6\" y1=\"6\" x2=\"18\" y2=\"18\"/>\n            </svg>\n            \u7ED3\u675F\u5BF9\u8BDD\n          </button>\n          <button class=\"re-enhance-btn\" id=\"reEnhanceBtn\" onclick=\"reEnhance()\">\n            <svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\">\n              <polyline points=\"23 4 23 10 17 10\"/>\n              <path d=\"M20.49 15a9 9 0 1 1-2.12-9.36L23 10\"/>\n            </svg>\n            \u7EE7\u7EED\u589E\u5F3A\n          </button>\n          <button class=\"cancel-btn\" onclick=\"useOriginal()\">\n            <svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\">\n              <path d=\"M3 12h18M3 6h18M3 18h18\"/>\n            </svg>\n            \u4F7F\u7528\u539F\u59CB\n          </button>\n          <button class=\"send-btn\" id=\"sendBtn\" onclick=\"sendPrompt()\">\n            <svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\">\n              <line x1=\"22\" y1=\"2\" x2=\"11\" y2=\"13\"/>\n              <polygon points=\"22 2 15 22 11 13 2 9 22 2\"/>\n            </svg>\n            \u53D1\u9001\u589E\u5F3A\n          </button>\n        </div>\n\n        <div class=\"keyboard-hint\">\n          \u5FEB\u6377\u952E: <kbd>Ctrl</kbd> + <kbd>Enter</kbd> \u53D1\u9001\u589E\u5F3A | <kbd>Esc</kbd> \u7ED3\u675F\u5BF9\u8BDD\n        </div>\n\n        <div id=\"status\" class=\"status\"></div>\n      </div>\n    </div>\n  </div>\n\n  <script>\n    const urlParams = new URLSearchParams(window.location.search);\n    const sessionId = urlParams.get('session');\n    const promptText = document.getElementById('promptText');\n    const charCount = document.getElementById('charCount');\n    const loading = document.getElementById('loading');\n    const mainContent = document.getElementById('mainContent');\n    const countdownEl = document.getElementById('countdown');\n\n    let countdownInterval = null;\n    let sessionCreatedAt = null;\n    let sessionTimeoutMs = null;\n\n    // \u66F4\u65B0\u5B57\u7B26\u8BA1\u6570\n    function updateCharCount() {\n      const count = promptText.value.length;\n      charCount.textContent = count + ' \u5B57\u7B26';\n    }\n\n    promptText.addEventListener('input', updateCharCount);\n\n    // \u683C\u5F0F\u5316\u65F6\u95F4\u663E\u793A\n    function formatTime(ms) {\n      const totalSeconds = Math.floor(ms / 1000);\n      const minutes = Math.floor(totalSeconds / 60);\n      const seconds = totalSeconds % 60;\n      return minutes + ':' + seconds.toString().padStart(2, '0');\n    }\n\n    // \u66F4\u65B0\u5012\u8BA1\u65F6\u663E\u793A\n    function updateCountdown() {\n      if (!sessionCreatedAt || !sessionTimeoutMs) return;\n\n      const now = Date.now();\n      const elapsed = now - sessionCreatedAt;\n      const remaining = sessionTimeoutMs - elapsed;\n\n      if (remaining <= 0) {\n        countdownEl.textContent = '\u23F1\uFE0F \u5DF2\u8D85\u65F6';\n        countdownEl.className = 'countdown danger';\n        if (countdownInterval) {\n          clearInterval(countdownInterval);\n          countdownInterval = null;\n        }\n        return;\n      }\n\n      const remainingMinutes = remaining / 60000;\n\n      // \u66F4\u65B0\u6837\u5F0F\n      if (remainingMinutes <= 1) {\n        countdownEl.className = 'countdown danger';\n      } else if (remainingMinutes <= 3) {\n        countdownEl.className = 'countdown warning';\n      } else {\n        countdownEl.className = 'countdown';\n      }\n\n      countdownEl.textContent = '\u23F1\uFE0F \u5269\u4F59\u65F6\u95F4: ' + formatTime(remaining);\n    }\n\n    // \u542F\u52A8\u5012\u8BA1\u65F6\n    function startCountdown(createdAt, timeoutMs) {\n      sessionCreatedAt = createdAt;\n      sessionTimeoutMs = timeoutMs;\n\n      updateCountdown();\n\n      if (countdownInterval) {\n        clearInterval(countdownInterval);\n      }\n\n      countdownInterval = setInterval(updateCountdown, 1000);\n    }\n\n    // \u952E\u76D8\u5FEB\u6377\u952E\n    document.addEventListener('keydown', (e) => {\n      if ((e.ctrlKey || e.metaKey) && e.key === 'Enter') {\n        e.preventDefault();\n        sendPrompt();\n      } else if (e.key === 'Escape') {\n        e.preventDefault();\n        endConversation();\n      }\n    });\n\n    // \u52A0\u8F7D session \u6570\u636E\n    if (!sessionId) {\n      loading.style.display = 'none';\n      mainContent.style.display = 'block';\n      showStatus('\u9519\u8BEF: \u672A\u63D0\u4F9B session ID', 'error');\n    } else {\n      loading.classList.add('active');\n\n      fetch('/api/session?session=' + encodeURIComponent(sessionId))\n        .then(r => r.json())\n        .then(data => {\n          promptText.value = data.enhancedPrompt;\n          updateCharCount();\n          loading.classList.remove('active');\n          mainContent.style.display = 'block';\n          promptText.focus();\n\n          // \u542F\u52A8\u5012\u8BA1\u65F6\n          if (data.createdAt && data.timeoutMs) {\n            startCountdown(data.createdAt, data.timeoutMs);\n          }\n        })\n        .catch(err => {\n          loading.classList.remove('active');\n          mainContent.style.display = 'block';\n          showStatus('\u52A0\u8F7D\u5931\u8D25: ' + err.message, 'error');\n        });\n    }\n\n    function reEnhance() {\n      const currentContent = promptText.value.trim();\n\n      if (!currentContent) {\n        showStatus('\u8BF7\u8F93\u5165\u5185\u5BB9\u540E\u518D\u589E\u5F3A', 'error');\n        return;\n      }\n\n      const reEnhanceBtn = document.getElementById('reEnhanceBtn');\n      const sendBtn = document.getElementById('sendBtn');\n\n      reEnhanceBtn.disabled = true;\n      sendBtn.disabled = true;\n      reEnhanceBtn.innerHTML = '<div class=\"spinner\" style=\"width: 16px; height: 16px; border-width: 2px; margin: 0;\"></div> \u589E\u5F3A\u4E2D...';\n\n      fetch('/api/re-enhance', {\n        method: 'POST',\n        headers: { 'Content-Type': 'application/json' },\n        body: JSON.stringify({\n          sessionId: sessionId,\n          currentPrompt: currentContent\n        })\n      })\n      .then(r => r.json())\n      .then(data => {\n        if (data.error) {\n          throw new Error(data.error);\n        }\n\n        promptText.value = data.enhancedPrompt;\n        updateCharCount();\n        showStatus('\u2705 \u589E\u5F3A\u6210\u529F\uFF01\u4F60\u53EF\u4EE5\u7EE7\u7EED\u7F16\u8F91\u6216\u53D1\u9001', 'success');\n\n        reEnhanceBtn.disabled = false;\n        sendBtn.disabled = false;\n        reEnhanceBtn.innerHTML = '<svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\"><polyline points=\"23 4 23 10 17 10\"/><path d=\"M20.49 15a9 9 0 1 1-2.12-9.36L23 10\"/></svg> \u7EE7\u7EED\u589E\u5F3A';\n      })\n      .catch(err => {\n        showStatus('\u589E\u5F3A\u5931\u8D25: ' + err.message, 'error');\n        reEnhanceBtn.disabled = false;\n        sendBtn.disabled = false;\n        reEnhanceBtn.innerHTML = '<svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\"><polyline points=\"23 4 23 10 17 10\"/><path d=\"M20.49 15a9 9 0 1 1-2.12-9.36L23 10\"/></svg> \u7EE7\u7EED\u589E\u5F3A';\n      });\n    }\n\n    function sendPrompt() {\n      const content = promptText.value.trim();\n\n      if (!content) {\n        showStatus('\u8BF7\u8F93\u5165\u5185\u5BB9\u540E\u518D\u53D1\u9001', 'error');\n        return;\n      }\n\n      const sendBtn = document.getElementById('sendBtn');\n      const reEnhanceBtn = document.getElementById('reEnhanceBtn');\n\n      sendBtn.disabled = true;\n      reEnhanceBtn.disabled = true;\n      sendBtn.innerHTML = '<div class=\"spinner\" style=\"width: 16px; height: 16px; border-width: 2px; margin: 0;\"></div> \u53D1\u9001\u4E2D...';\n\n      fetch('/api/submit', {\n        method: 'POST',\n        headers: { 'Content-Type': 'application/json' },\n        body: JSON.stringify({ sessionId: sessionId, content: content })\n      })\n      .then(r => r.json())\n      .then(data => {\n        showStatus('\u2705 \u53D1\u9001\u6210\u529F\uFF01\u7A97\u53E3\u5C06\u5728 2 \u79D2\u540E\u81EA\u52A8\u5173\u95ED...', 'success');\n        setTimeout(() => window.close(), 2000);\n      })\n      .catch(err => {\n        showStatus('\u53D1\u9001\u5931\u8D25: ' + err.message, 'error');\n        sendBtn.disabled = false;\n        reEnhanceBtn.disabled = false;\n        sendBtn.innerHTML = '<svg width=\"16\" height=\"16\" viewBox=\"0 0 24 24\" fill=\"none\" stroke=\"currentColor\" stroke-width=\"2\"><line x1=\"22\" y1=\"2\" x2=\"11\" y2=\"13\"/><polygon points=\"22 2 15 22 11 13 2 9 22 2\"/></svg> \u53D1\u9001';\n      });\n    }\n\n    function useOriginal() {\n      if (confirm('\u786E\u5B9A\u4F7F\u7528\u539F\u59CB prompt \u7EE7\u7EED\u5BF9\u8BDD\u5417\uFF1F')) {\n        fetch('/api/submit', {\n          method: 'POST',\n          headers: { 'Content-Type': 'application/json' },\n          body: JSON.stringify({ sessionId: sessionId, content: '__USE_ORIGINAL__' })\n        })\n        .then(() => {\n          showStatus('\u2705 \u5C06\u4F7F\u7528\u539F\u59CB prompt \u7EE7\u7EED...', 'success');\n          setTimeout(() => window.close(), 1000);\n        })\n        .catch(() => window.close());\n      }\n    }\n\n    function endConversation() {\n      if (confirm('\u786E\u5B9A\u8981\u7ED3\u675F\u672C\u6B21\u5BF9\u8BDD\u5417\uFF1F')) {\n        fetch('/api/submit', {\n          method: 'POST',\n          headers: { 'Content-Type': 'application/json' },\n          body: JSON.stringify({ sessionId: sessionId, content: '__END_CONVERSATION__' })\n        })\n        .then(() => {\n          showStatus('\u2705 \u5BF9\u8BDD\u5DF2\u7ED3\u675F', 'success');\n          setTimeout(() => window.close(), 1000);\n        })\n        .catch(() => window.close());\n      }\n    }\n\n    function showStatus(message, type) {\n      const status = document.getElementById('status');\n      status.textContent = message;\n      status.className = 'status ' + type;\n    }\n  </script>\n</body>\n</html>";
+//# sourceMappingURL=templates.d.ts.map

package/dist/enhancer/templates.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"templates.d.ts","sourceRoot":"","sources":["../../src/enhancer/templates.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;GAGG;AACH,eAAO,MAAM,uBAAuB,8vBAqBlB,CAAC;AAEnB;;GAEG;AACH,eAAO,MAAM,gBAAgB,qwkBA8nBrB,CAAC"}