@jonathanleelx/meddle 0.0.1

package/dist/core/plugin-generator.js

@@ -0,0 +1,697 @@
+"use strict";
+/**
+ * AI插件生成器
+ * 根据用户需求，使用AI生成符合插件系统规范的插件代码
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cleanAIResponse = cleanAIResponse;
+exports.generatePluginStream = generatePluginStream;
+exports.fixPluginWithAI = fixPluginWithAI;
+exports.fixPluginWithAIStream = fixPluginWithAIStream;
+exports.revisePluginWithAIStream = revisePluginWithAIStream;
+exports.generatePlugin = generatePlugin;
+const AI_CODE_MAX_TOKENS = 4000;
+function extractTextContent(content) {
+    if (typeof content === 'string')
+        return content;
+    if (!Array.isArray(content))
+        return '';
+    return content
+        .map((part) => {
+        if (typeof part === 'string')
+            return part;
+        if (part && typeof part === 'object' && 'text' in part && typeof part.text === 'string') {
+            return part.text;
+        }
+        return '';
+    })
+        .join('');
+}
+function readOpenAIMessageContent(data) {
+    return extractTextContent(data?.choices?.[0]?.message?.content).trim();
+}
+function readOpenAIStreamChoice(data) {
+    const choice = data?.choices?.[0];
+    return {
+        content: extractTextContent(choice?.delta?.content),
+        finishReason: typeof choice?.finish_reason === 'string' ? choice.finish_reason : undefined,
+    };
+}
+async function readOpenAIStream(response, onChunk) {
+    const reader = response.body?.getReader();
+    const decoder = new TextDecoder();
+    if (!reader) {
+        throw new Error('无法读取响应流');
+    }
+    let fullCode = '';
+    let buffer = '';
+    let finishReason = '';
+    const consumeLine = (line) => {
+        if (!line.startsWith('data:'))
+            return;
+        const data = line.slice(5).trim();
+        if (!data || data === '[DONE]')
+            return;
+        try {
+            const parsed = JSON.parse(data);
+            const { content, finishReason: nextFinishReason } = readOpenAIStreamChoice(parsed);
+            if (nextFinishReason) {
+                finishReason = nextFinishReason;
+            }
+            if (content) {
+                fullCode += content;
+                onChunk(content);
+            }
+        }
+        catch (_) {
+            // Ignore malformed SSE lines from compatible providers.
+        }
+    };
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split('\n');
+            buffer = lines.pop() || '';
+            for (const line of lines) {
+                consumeLine(line);
+            }
+        }
+        if (buffer.trim()) {
+            consumeLine(buffer);
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+    if (finishReason === 'length') {
+        throw new Error('AI 输出被截断，请缩短补充需求后重试');
+    }
+    if (!fullCode.trim()) {
+        throw new Error('OpenAI 返回了空结果');
+    }
+    return cleanAIResponse(fullCode);
+}
+/**
+ * 生成插件系统设计文档prompt
+ */
+function getPluginSystemDesignPrompt() {
+    return `# Meddle 插件系统设计
+
+## 插件接口规范
+
+插件必须实现以下接口（JavaScript对象）：
+
+\`\`\`javascript
+const plugin = {
+  manifest: {
+    id: 'local.plugin-name',      // 唯一标识
+    name: '插件名称',
+    version: '1.0.0',
+    apiVersion: '1.x',
+    permissions: ['proxy:read'],  // 权限列表
+    hooks: ['onRequestStart'],    // Hook列表
+    priority: 100,
+    type: 'local'
+  },
+  
+  // 必需：初始化方法
+  setup(context) {
+    context.log.info('插件初始化');
+  },
+  
+  // 可选：启动方法
+  start() {
+    // 插件启动时执行
+  },
+  
+  // 可选：Hook方法
+  async onRequestStart(context) {
+    context.log.info('请求开始');
+  }
+}
+
+export interface PluginManifest {
+  id: string;                    // 唯一标识，格式：org.plugin-name
+  name?: string;                 // 显示名称
+  version: string;               // 版本号
+  apiVersion: string;            // API版本（当前为 '1.x'）
+  permissions: string[];         // 权限列表
+  hooks: string[];               // 声明的Hook列表
+  priority?: number;             // 优先级（默认100，数字越小优先级越高）
+  type?: string;                 // 'builtin' | 'local'
+}
+
+export interface PluginContext {
+  manifest: PluginManifest;
+  log: Logger;                   // 日志接口
+  [key: string]: any;
+}
+
+export interface HookContext {
+  request: Request;              // 请求信息
+  target: string;                // 目标地址
+  meta: Record<string, any>;     // 元数据存储
+  setTarget(target: string): void;  // 设置目标地址
+  respond(response: Response): void; // 短路响应（仅 onBeforeProxy）
+}
+
+export interface ResponseContext {
+  request: Request;              // 请求信息
+  target: string;                // 目标地址
+  meta: Record<string, any>;     // 元数据存储
+  response: Response;            // 响应信息
+}
+\`\`\`
+
+## 权限模型
+
+可用权限：
+- \`proxy:read\` - 读取请求/响应信息
+- \`proxy:write\` - 修改请求/响应
+- \`response:shortcircuit\` - 短路响应（Mock）
+- \`config:read\` - 读取配置
+- \`config:write\` - 写入配置
+- \`storage:read\` - 读取存储
+- \`storage:write\` - 写入存储
+- \`network:outbound\` - 外部网络访问
+
+## Hook 生命周期
+
+1. \`onRequestStart\` - 请求刚进入代理时调用
+2. \`onBeforeProxy\` - 即将发起上游请求之前调用（可短路）
+3. \`onBeforeResponse\` - 收到上游响应，返回客户端之前调用
+4. \`onAfterResponse\` - 响应完成后异步调用
+5. \`onError\` - 任意阶段出现错误后调用
+
+## 插件示例
+
+\`\`\`javascript
+// example-plugin.js
+exports.plugin = {
+    manifest: {
+        id: 'local.example',
+        name: 'Example Plugin',
+        version: '1.0.0',
+        apiVersion: '1.x',
+        permissions: ['proxy:read'],
+        hooks: ['onRequestStart', 'onAfterResponse'],
+        priority: 100,
+        type: 'local'
+    },
+    
+    setup(context) {
+        context.log.info(this.manifest.name + ' setup complete');
+    },
+    
+    async onRequestStart(context) {
+        context.log.info('Request: ' + context.request.method + ' ' + context.request.url);
+    },
+    
+    async onAfterResponse(context) {
+        context.log.info('Response: ' + context.response.statusCode);
+    }
+};
+\`\`\`
+
+**重要**: 必须使用 CommonJS 格式（exports.plugin），不要使用 ES6 模块（export const）。`;
+}
+/**
+ * 使用 OpenAI 生成插件代码（流式）
+ */
+async function generateWithOpenAIStream(requirement, config, onChunk) {
+    const systemPrompt = `你是一个专业的插件开发助手，精通 JavaScript 和 Meddle 插件系统。你的任务是根据用户需求生成高质量的插件代码。
+
+${getPluginSystemDesignPrompt()}
+
+## 代码生成要求
+
+1. 必须遵循上述插件接口规范
+2. 代码必须是**纯 JavaScript**（不要使用TypeScript类型注解）
+3. 使用 CommonJS 格式：exports.plugin = { ... }
+4. 插件 ID 格式：local.{plugin-name}（小写，用连字符分隔）
+5. 只返回代码，不要包含任何解释性文字
+6. 不要使用 markdown 代码块标记
+7. 代码要有适当的注释说明功能
+8. 权限申请要最小化，只申请必需的权限
+9. 代码必须可以直接在Node.js中执行
+10. 使用 async/await 处理异步操作`;
+    const userPrompt = `请生成一个符合 Meddle 插件系统规范的插件代码。
+
+插件需求：
+名称：${requirement.name}
+描述：${requirement.description}
+${requirement.hooks ? `需要的 Hooks：${requirement.hooks.join(', ')}` : ''}
+${requirement.permissions ? `需要的权限：${requirement.permissions.join(', ')}` : ''}
+
+请直接返回完整的 JavaScript 插件代码，使用 exports.plugin = {...} 格式导出。
+代码必须可以直接在Node.js中执行，不要使用任何TypeScript语法。`;
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${config.apiKey}`
+        },
+        body: JSON.stringify({
+            model: config.model,
+            messages: [
+                { role: 'system', content: systemPrompt },
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3,
+            max_tokens: AI_CODE_MAX_TOKENS,
+            stream: true
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`OpenAI API 错误 (${response.status}): ${errorText}`);
+    }
+    return readOpenAIStream(response, onChunk);
+}
+/**
+ * 使用 Anthropic 生成插件代码（流式）
+ */
+async function generateWithAnthropicStream(requirement, config, onChunk) {
+    const systemPrompt = `你是一个专业的插件开发助手，精通 JavaScript 和 Meddle 插件系统。你的任务是根据用户需求生成高质量的插件代码。
+
+${getPluginSystemDesignPrompt()}
+
+## 代码生成要求
+
+1. 必须遵循上述插件接口规范
+2. 代码必须是**纯 JavaScript**（不要使用TypeScript类型注解）
+3. 使用 CommonJS 格式：exports.plugin = { ... }
+4. 插件 ID 格式：local.{plugin-name}（小写，用连字符分隔）
+5. 只返回代码，不要包含任何解释性文字
+6. 不要使用 markdown 代码块标记
+7. 代码要有适当的注释说明功能
+8. 权限申请要最小化，只申请必需的权限
+9. 代码必须可以直接在Node.js中执行
+10. 使用 async/await 处理异步操作`;
+    const userPrompt = `请生成一个符合 Meddle 插件系统规范的插件代码。
+
+插件需求：
+名称：${requirement.name}
+描述：${requirement.description}
+${requirement.hooks ? `需要的 Hooks：${requirement.hooks.join(', ')}` : ''}
+${requirement.permissions ? `需要的权限：${requirement.permissions.join(', ')}` : ''}
+
+请直接返回完整的 JavaScript 插件代码，使用 exports.plugin = {...} 格式导出。
+代码必须可以直接在Node.js中执行，不要使用任何TypeScript语法。`;
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'x-api-key': config.apiKey,
+            'anthropic-version': '2023-06-01'
+        },
+        body: JSON.stringify({
+            model: config.model,
+            max_tokens: AI_CODE_MAX_TOKENS,
+            system: systemPrompt,
+            messages: [
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3,
+            stream: true
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Anthropic API 错误 (${response.status}): ${errorText}`);
+    }
+    let fullCode = '';
+    const reader = response.body?.getReader();
+    const decoder = new TextDecoder();
+    if (!reader) {
+        throw new Error('无法读取响应流');
+    }
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            const chunk = decoder.decode(value, { stream: true });
+            const lines = chunk.split('\n').filter(line => line.trim() !== '');
+            for (const line of lines) {
+                if (line.startsWith('data: ')) {
+                    const data = line.slice(6);
+                    try {
+                        const parsed = JSON.parse(data);
+                        if (parsed.type === 'content_block_delta') {
+                            const content = parsed.delta?.text;
+                            if (content) {
+                                fullCode += content;
+                                onChunk(content);
+                            }
+                        }
+                    }
+                    catch (e) {
+                        // 忽略解析错误
+                    }
+                }
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+    if (!fullCode.trim()) {
+        throw new Error('Anthropic 返回了空结果');
+    }
+    return cleanAIResponse(fullCode);
+}
+/**
+ * 清理 AI 响应，移除 markdown 代码块标记
+ */
+function cleanAIResponse(text) {
+    // 移除可能的 markdown 代码块标记
+    let cleaned = text
+        .replace(/^```[a-z]*\n?/i, '')
+        .replace(/\n?```$/i, '')
+        .trim();
+    return cleaned;
+}
+/**
+ * 从生成的代码中提取 manifest 信息
+ */
+function extractManifest(code) {
+    try {
+        // 尝试通过正则提取 manifest 对象
+        const manifestMatch = code.match(/manifest\s*:\s*{([^}]+({[^}]+}[^}]+)*[^}]+)}/s);
+        if (!manifestMatch) {
+            throw new Error('无法提取 manifest 信息');
+        }
+        const manifestText = manifestMatch[0];
+        // 提取各个字段
+        const idMatch = manifestText.match(/id\s*:\s*['"]([^'"]+)['"]/);
+        const nameMatch = manifestText.match(/name\s*:\s*['"]([^'"]+)['"]/);
+        const versionMatch = manifestText.match(/version\s*:\s*['"]([^'"]+)['"]/);
+        const apiVersionMatch = manifestText.match(/apiVersion\s*:\s*['"]([^'"]+)['"]/);
+        const permissionsMatch = manifestText.match(/permissions\s*:\s*\[([^\]]*)\]/);
+        const hooksMatch = manifestText.match(/hooks\s*:\s*\[([^\]]*)\]/);
+        const permissions = permissionsMatch
+            ? permissionsMatch[1].split(',').map(p => p.trim().replace(/['"]/g, '')).filter(Boolean)
+            : [];
+        const hooks = hooksMatch
+            ? hooksMatch[1].split(',').map(h => h.trim().replace(/['"]/g, '')).filter(Boolean)
+            : [];
+        return {
+            id: idMatch ? idMatch[1] : 'local.unknown',
+            name: nameMatch ? nameMatch[1] : 'Unknown Plugin',
+            version: versionMatch ? versionMatch[1] : '1.0.0',
+            apiVersion: apiVersionMatch ? apiVersionMatch[1] : '1.x',
+            permissions,
+            hooks
+        };
+    }
+    catch (error) {
+        // 返回默认 manifest
+        return {
+            id: 'local.unknown',
+            name: 'Unknown Plugin',
+            version: '1.0.0',
+            apiVersion: '1.x',
+            permissions: [],
+            hooks: []
+        };
+    }
+}
+/**
+ * 生成插件代码（流式）
+ */
+async function generatePluginStream(requirement, config, onChunk) {
+    let code;
+    if (config.provider === 'anthropic') {
+        code = await generateWithAnthropicStream(requirement, config, onChunk);
+    }
+    else {
+        code = await generateWithOpenAIStream(requirement, config, onChunk);
+    }
+    // 提取 manifest 信息
+    const manifest = extractManifest(code);
+    // 生成文件名（直接是.js）
+    const pluginId = manifest.id.replace('local.', '');
+    const filename = `${pluginId}.js`;
+    return {
+        code,
+        filename,
+        manifest
+    };
+}
+/**
+ * 根据测试错误自动修复插件代码
+ */
+async function fixPluginWithAI(originalCode, testError, requirement, config) {
+    const systemPrompt = `你是一个专业的插件调试助手。用户的插件代码在测试时出现了错误，你需要分析错误并修复代码。
+
+${getPluginSystemDesignPrompt()}
+
+## 修复要求
+
+1. 仔细分析测试错误信息
+2. 找出代码中的问题
+3. 修复bug，确保代码正确
+4. 返回修复后的完整代码
+5. 只返回代码，不要解释
+6. 代码必须是纯JavaScript（CommonJS格式）`;
+    const userPrompt = `以下是插件代码和测试错误，请修复代码中的问题。
+
+原始需求：
+${requirement.description}
+
+当前代码：
+\`\`\`javascript
+${originalCode}
+\`\`\`
+
+测试错误：
+${testError}
+
+请分析错误原因并返回修复后的完整JavaScript代码。`;
+    if (config.provider === 'anthropic') {
+        return fixWithAnthropicNonStream(userPrompt, systemPrompt, config);
+    }
+    else {
+        return fixWithOpenAINonStream(userPrompt, systemPrompt, config);
+    }
+}
+async function fixWithOpenAINonStream(userPrompt, systemPrompt, config) {
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${config.apiKey}`
+        },
+        body: JSON.stringify({
+            model: config.model,
+            messages: [
+                { role: 'system', content: systemPrompt },
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3,
+            max_tokens: AI_CODE_MAX_TOKENS
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`OpenAI API 错误 (${response.status}): ${errorText}`);
+    }
+    const data = await response.json();
+    const code = readOpenAIMessageContent(data);
+    if (!code) {
+        throw new Error('OpenAI 返回了空结果');
+    }
+    return cleanAIResponse(code);
+}
+async function fixWithOpenAIStream(userPrompt, systemPrompt, config, onChunk) {
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${config.apiKey}`
+        },
+        body: JSON.stringify({
+            model: config.model,
+            messages: [
+                { role: 'system', content: systemPrompt },
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3,
+            max_tokens: AI_CODE_MAX_TOKENS,
+            stream: true
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`OpenAI API 错误 (${response.status}): ${errorText}`);
+    }
+    return readOpenAIStream(response, onChunk);
+}
+async function fixWithAnthropicNonStream(userPrompt, systemPrompt, config) {
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'x-api-key': config.apiKey,
+            'anthropic-version': '2023-06-01'
+        },
+        body: JSON.stringify({
+            model: config.model,
+            max_tokens: AI_CODE_MAX_TOKENS,
+            system: systemPrompt,
+            messages: [
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Anthropic API 错误 (${response.status}): ${errorText}`);
+    }
+    const data = await response.json();
+    const code = data.content?.[0]?.text?.trim();
+    if (!code) {
+        throw new Error('Anthropic 返回了空结果');
+    }
+    return cleanAIResponse(code);
+}
+async function fixWithAnthropicStream(userPrompt, systemPrompt, config, onChunk) {
+    const response = await fetch(config.baseUrl, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            'x-api-key': config.apiKey,
+            'anthropic-version': '2023-06-01'
+        },
+        body: JSON.stringify({
+            model: config.model,
+            max_tokens: AI_CODE_MAX_TOKENS,
+            system: systemPrompt,
+            messages: [
+                { role: 'user', content: userPrompt }
+            ],
+            temperature: 0.3,
+            stream: true
+        })
+    });
+    if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(`Anthropic API 错误 (${response.status}): ${errorText}`);
+    }
+    const reader = response.body?.getReader();
+    const decoder = new TextDecoder();
+    if (!reader) {
+        throw new Error('无法读取响应流');
+    }
+    let fullCode = '';
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            const chunk = decoder.decode(value, { stream: true });
+            const lines = chunk.split('\n').filter(line => line.trim() !== '');
+            for (const line of lines) {
+                if (!line.startsWith('data: '))
+                    continue;
+                const data = line.slice(6);
+                try {
+                    const parsed = JSON.parse(data);
+                    if (parsed.type === 'content_block_delta') {
+                        const content = parsed.delta?.text;
+                        if (content) {
+                            fullCode += content;
+                            onChunk(content);
+                        }
+                    }
+                }
+                catch (_) {
+                    // ignore parse errors
+                }
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+    if (!fullCode.trim()) {
+        throw new Error('Anthropic 返回了空结果');
+    }
+    return cleanAIResponse(fullCode);
+}
+async function fixPluginWithAIStream(originalCode, testError, requirement, config, onChunk) {
+    const systemPrompt = `你是一个专业的插件调试助手。用户的插件代码在测试时出现了错误，你需要分析错误并修复代码。
+
+${getPluginSystemDesignPrompt()}
+
+## 修复要求
+
+1. 仔细分析测试错误信息
+2. 找出代码中的问题
+3. 修复bug，确保代码正确
+4. 返回修复后的完整代码
+5. 只返回代码，不要解释
+6. 代码必须是纯JavaScript（CommonJS格式）`;
+    const userPrompt = `以下是插件代码和测试错误，请修复代码中的问题。
+
+原始需求：
+${requirement.description}
+
+当前代码：
+\`\`\`javascript
+${originalCode}
+\`\`\`
+
+测试错误：
+${testError}
+
+请分析错误原因并返回修复后的完整JavaScript代码。`;
+    if (config.provider === 'anthropic') {
+        return fixWithAnthropicStream(userPrompt, systemPrompt, config, onChunk);
+    }
+    return fixWithOpenAIStream(userPrompt, systemPrompt, config, onChunk);
+}
+async function revisePluginWithAIStream(originalCode, instruction, requirement, config, onChunk) {
+    const systemPrompt = `你是一个专业的 Meddle 插件开发助手。用户会给你现有插件代码和额外修改要求，你需要在保持插件可运行的前提下更新代码。
+
+${getPluginSystemDesignPrompt()}
+
+## 更新要求
+
+1. 基于现有代码做增量修改，不要无关重写
+2. 严格保留插件 manifest 的核心语义，除非用户明确要求修改
+3. 保持已有 hooks 与权限声明正确
+4. 返回更新后的完整代码
+5. 只返回代码，不要解释
+6. 代码必须是纯JavaScript（CommonJS格式）`;
+    const userPrompt = `以下是当前插件代码和新的修改要求，请直接更新插件代码。
+
+原始需求：
+${requirement.description}
+
+当前代码：
+\`\`\`javascript
+${originalCode}
+\`\`\`
+
+额外修改要求：
+${instruction}
+
+请返回更新后的完整 JavaScript 插件代码。`;
+    if (config.provider === 'anthropic') {
+        return fixWithAnthropicStream(userPrompt, systemPrompt, config, onChunk);
+    }
+    return fixWithOpenAIStream(userPrompt, systemPrompt, config, onChunk);
+}
+/**
+ * 生成插件代码（非流式，向后兼容）
+ */
+async function generatePlugin(requirement, config) {
+    return generatePluginStream(requirement, config, () => { });
+}
+//# sourceMappingURL=plugin-generator.js.map