opencode-codebuddy-external-auth 1.0.7 → 1.0.9

package/dist/plugin.js

@@ -6,12 +6,8 @@ exports.CodeBuddyExternalAuthPlugin = void 0;
 // ============================================================================
 const PROVIDER_ID = "codebuddy-external";
 const CONFIG = {
-    // IOA 版本使用 copilot.tencent.com
+    // IOA 版本使用 copilot.tencent.com 进行认证
     serverUrl: "https://copilot.tencent.com",
-    // API 端点 - CodeBuddy IOA 直接使用 endpoint 作为 baseURL
-    // OpenAI SDK 会自动追加 /chat/completions
-    // 注意：不需要 /v1 前缀，CodeBuddy 服务端直接接收 /chat/completions
-    apiBaseUrl: "https://copilot.tencent.com",
     // 平台标识
     platform: "CLI",
     appVersion: "2.37.20",
@@ -23,22 +19,216 @@ function sleep(ms) {
     return new Promise((resolve) => setTimeout(resolve, ms));
 }
 /**
- * Creates an authenticated fetch function with CodeBuddy headers
+ * Convert OpenAI chat messages to a single prompt string
  */
-function createAuthenticatedFetch(accessToken, userId) {
+function convertMessagesToPrompt(messages) {
+    let systemPrompt;
+    const conversationParts = [];
+    for (const msg of messages) {
+        if (msg.role === "system") {
+            systemPrompt = msg.content;
+        }
+        else if (msg.role === "user") {
+            conversationParts.push(`User: ${msg.content}`);
+        }
+        else if (msg.role === "assistant") {
+            conversationParts.push(`Assistant: ${msg.content}`);
+        }
+    }
+    // If there's only one user message, use it directly
+    const userMessages = messages.filter((m) => m.role === "user");
+    if (userMessages.length === 1 && conversationParts.length === 1) {
+        return {
+            prompt: userMessages[0].content,
+            systemPrompt,
+        };
+    }
+    return {
+        prompt: conversationParts.join("\n\n"),
+        systemPrompt,
+    };
+}
+/**
+ * Create a streaming response in OpenAI format from CodeBuddy response
+ */
+function createOpenAIStreamResponse(text, model) {
+    const encoder = new TextEncoder();
+    const id = `chatcmpl-${Date.now()}`;
+    const created = Math.floor(Date.now() / 1000);
+    return new ReadableStream({
+        start(controller) {
+            // First chunk with role
+            const roleChunk = {
+                id,
+                object: "chat.completion.chunk",
+                created,
+                model,
+                choices: [
+                    {
+                        index: 0,
+                        delta: { role: "assistant" },
+                        finish_reason: null,
+                    },
+                ],
+            };
+            controller.enqueue(encoder.encode(`data: ${JSON.stringify(roleChunk)}\n\n`));
+            // Content chunk
+            const contentChunk = {
+                id,
+                object: "chat.completion.chunk",
+                created,
+                model,
+                choices: [
+                    {
+                        index: 0,
+                        delta: { content: text },
+                        finish_reason: null,
+                    },
+                ],
+            };
+            controller.enqueue(encoder.encode(`data: ${JSON.stringify(contentChunk)}\n\n`));
+            // Done chunk
+            const doneChunk = {
+                id,
+                object: "chat.completion.chunk",
+                created,
+                model,
+                choices: [
+                    {
+                        index: 0,
+                        delta: {},
+                        finish_reason: "stop",
+                    },
+                ],
+            };
+            controller.enqueue(encoder.encode(`data: ${JSON.stringify(doneChunk)}\n\n`));
+            controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+            controller.close();
+        },
+    });
+}
+/**
+ * Create a non-streaming response in OpenAI format
+ */
+function createOpenAIResponse(text, model) {
+    return JSON.stringify({
+        id: `chatcmpl-${Date.now()}`,
+        object: "chat.completion",
+        created: Math.floor(Date.now() / 1000),
+        model: model,
+        choices: [
+            {
+                index: 0,
+                message: {
+                    role: "assistant",
+                    content: text,
+                },
+                finish_reason: "stop",
+            },
+        ],
+        usage: {
+            prompt_tokens: 0,
+            completion_tokens: 0,
+            total_tokens: 0,
+        },
+    });
+}
+/**
+ * Execute codebuddy CLI command and return result
+ */
+async function executeCodeBuddyCLI(prompt, model, systemPrompt) {
+    const { spawn } = await import("child_process");
+    return new Promise((resolve, reject) => {
+        const args = ["-p", "--output-format", "text"];
+        if (model) {
+            args.push("--model", model);
+        }
+        if (systemPrompt) {
+            args.push("--system-prompt", systemPrompt);
+        }
+        args.push(prompt);
+        console.log(`[codebuddy-external] Executing: codebuddy ${args.join(" ").substring(0, 100)}...`);
+        const child = spawn("codebuddy", args, {
+            env: { ...process.env },
+            stdio: ["pipe", "pipe", "pipe"],
+        });
+        let stdout = "";
+        let stderr = "";
+        child.stdout.on("data", (data) => {
+            stdout += data.toString();
+        });
+        child.stderr.on("data", (data) => {
+            stderr += data.toString();
+        });
+        child.on("close", (code) => {
+            if (code === 0) {
+                resolve(stdout.trim());
+            }
+            else {
+                reject(new Error(`CodeBuddy CLI exited with code ${code}: ${stderr}`));
+            }
+        });
+        child.on("error", (err) => {
+            reject(new Error(`Failed to spawn codebuddy: ${err.message}`));
+        });
+        // Timeout after 5 minutes
+        setTimeout(() => {
+            child.kill();
+            reject(new Error("CodeBuddy CLI timeout after 5 minutes"));
+        }, 5 * 60 * 1000);
+    });
+}
+/**
+ * Creates a fetch function that invokes codebuddy CLI
+ * and converts between OpenAI and CodeBuddy formats
+ */
+function createProxyFetch() {
     return async (url, init) => {
-        const headers = new Headers(init?.headers);
-        // Bearer token authentication
-        headers.set("Authorization", `Bearer ${accessToken}`);
-        // User identification (URL encoded)
-        if (userId) {
-            headers.set("X-User-Id", encodeURIComponent(userId));
+        const urlStr = url.toString();
+        // Check if this is a chat completions request
+        if (urlStr.includes("/chat/completions") || urlStr.includes("/v1/chat/completions")) {
+            try {
+                // Parse the OpenAI request
+                const body = init?.body;
+                if (!body) {
+                    return new Response(JSON.stringify({ error: "Missing request body" }), {
+                        status: 400,
+                        headers: { "Content-Type": "application/json" },
+                    });
+                }
+                const openaiRequest = JSON.parse(typeof body === "string" ? body : await new Response(body).text());
+                // Convert to CodeBuddy format
+                const { prompt, systemPrompt } = convertMessagesToPrompt(openaiRequest.messages);
+                console.log(`[codebuddy-external] Invoking CodeBuddy CLI`);
+                console.log(`[codebuddy-external] Model: ${openaiRequest.model}, Prompt length: ${prompt.length}`);
+                // Execute codebuddy CLI
+                const resultText = await executeCodeBuddyCLI(prompt, openaiRequest.model, systemPrompt);
+                // Convert response to OpenAI format
+                if (openaiRequest.stream) {
+                    return new Response(createOpenAIStreamResponse(resultText, openaiRequest.model), {
+                        headers: {
+                            "Content-Type": "text/event-stream",
+                            "Cache-Control": "no-cache",
+                            "Connection": "keep-alive",
+                        },
+                    });
+                }
+                else {
+                    return new Response(createOpenAIResponse(resultText, openaiRequest.model), {
+                        headers: { "Content-Type": "application/json" },
+                    });
+                }
+            }
+            catch (error) {
+                console.error(`[codebuddy-external] CLI error:`, error);
+                return new Response(JSON.stringify({ error: `CodeBuddy CLI error: ${error}` }), {
+                    status: 500,
+                    headers: { "Content-Type": "application/json" },
+                });
+            }
         }
-        // Request tracing
-        headers.set("X-B3-TraceId", crypto.randomUUID().replace(/-/g, ""));
-        headers.set("X-B3-SpanId", crypto.randomUUID().substring(0, 16));
-        headers.set("X-B3-Sampled", "1");
-        return fetch(url, { ...init, headers });
+        // For non-chat-completions requests, pass through normally
+        return fetch(url, init);
     };
 }
 // ============================================================================
@@ -165,10 +355,12 @@ const CodeBuddyExternalAuthPlugin = async (_input) => {
             async loader(getAuth) {
                 const auth = await getAuth();
                 if (auth.type === "oauth" && auth.access) {
+                    // 返回代理 fetch 函数，通过 codebuddy CLI 处理请求
+                    // baseURL 可以是任意值，因为 fetch 会拦截并调用 CLI
                     return {
-                        apiKey: auth.access,
-                        baseURL: CONFIG.apiBaseUrl,
-                        fetch: createAuthenticatedFetch(auth.access, auth.userId),
+                        apiKey: "cli-proxy", // 占位符，实际认证由 codebuddy CLI 处理
+                        baseURL: "http://localhost", // 占位符，fetch 会拦截请求
+                        fetch: createProxyFetch(),
                     };
                 }
                 return {};
@@ -225,7 +417,8 @@ const CodeBuddyExternalAuthPlugin = async (_input) => {
             if (input.model.providerID !== PROVIDER_ID) {
                 return;
             }
-            output.options.baseURL = CONFIG.apiBaseUrl;
+            // 占位符，实际请求由 fetch 函数拦截并通过 CLI 处理
+            output.options.baseURL = "http://localhost";
         },
     };
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-codebuddy-external-auth",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "OpenCode plugin for CodeBuddy External (IOA) authentication",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -24,6 +24,7 @@
   "devDependencies": {
     "@opencode-ai/plugin": "^1.0.168",
     "@opencode-ai/sdk": "^1.0.168",
+    "@types/node": "^25.0.10",
     "typescript": "^5.7.2"
   },
   "files": [