@minniexcode/codex-switch 0.1.5 → 0.2.0

package/dist/runtime/copilot-bridge.js

@@ -163,13 +163,13 @@ async function probeCopilotBridgeRuntime(provider, persistedState, runtimeDir) {
 /**
  * Starts or reuses a Copilot bridge worker, then verifies its health before returning.
  */
-async function ensureCopilotBridge(providerName, provider, runtimeDir, runtimesDir) {
-    return startOrReuseCopilotBridge(providerName, provider, runtimeDir, runtimesDir);
+async function ensureCopilotBridge(providerName, provider, runtimeDir, runtimesDir, toolHomeDir) {
+    return startOrReuseCopilotBridge(providerName, provider, runtimeDir, runtimesDir, toolHomeDir);
 }
 /**
  * Starts or reuses a Copilot bridge worker and reports the chosen port.
  */
-async function startOrReuseCopilotBridge(providerName, provider, runtimeDir, runtimesDir) {
+async function startOrReuseCopilotBridge(providerName, provider, runtimeDir, runtimesDir, toolHomeDir) {
     if (!(0, providers_1.isCopilotBridgeProvider)(provider)) {
         throw (0, errors_1.cliError)("RUNTIME_PROVIDER_INVALID", "Provider is not backed by a Copilot bridge runtime.", {
             provider: providerName,
@@ -221,7 +221,7 @@ async function startOrReuseCopilotBridge(providerName, provider, runtimeDir, run
     }
     const selectedPort = await selectBridgePort(runtime.bridgeHost, runtime.bridgePort);
     const selectedBaseUrl = `http://${runtime.bridgeHost}:${selectedPort}${runtime.bridgePath}`;
-    const workerPath = path.join(__dirname, "copilot-bridge-worker.js");
+    const workerPath = path.join(__dirname, "copilot-http-bridge-worker.js");
     ensureBridgeLogFile(logPath);
     appendBridgeLifecycleLog(logPath, `worker start provider=${providerName} host=${runtime.bridgeHost} port=${String(selectedPort)} replaced=${String(replaced)}`);
     let child;
@@ -239,6 +239,7 @@ async function startOrReuseCopilotBridge(providerName, provider, runtimeDir, run
                 CODEX_SWITCH_RUNTIME_DIR: runtimeDir ?? "",
                 CODEX_SWITCH_RUNTIMES_DIR: runtimesDir ?? "",
                 CODEX_SWITCH_BRIDGE_LOG_PATH: logPath,
+                CODEX_SWITCH_TOOL_HOME_DIR: toolHomeDir ?? "",
             },
         });
     }
@@ -349,23 +350,27 @@ function createCopilotBridgeRequestHandler(context) {
                         connection: "keep-alive",
                     });
                     const heartbeat = startSseHeartbeat(response);
-                    const payload = await context.executeChatCompletion(body, {
-                        timeoutMs,
-                        onTextDelta: (delta) => {
-                            response.write(`data: ${JSON.stringify({
-                                choices: [
-                                    {
-                                        index: 0,
-                                        delta: { content: delta },
-                                        finish_reason: null,
-                                    },
-                                ],
-                            })}\n\n`);
-                        },
-                    });
-                    clearInterval(heartbeat);
-                    response.write("data: [DONE]\n\n");
-                    response.end();
+                    try {
+                        const payload = await context.executeChatCompletion(body, {
+                            timeoutMs,
+                            onTextDelta: (delta) => {
+                                response.write(`data: ${JSON.stringify({
+                                    choices: [
+                                        {
+                                            index: 0,
+                                            delta: { content: delta },
+                                            finish_reason: null,
+                                        },
+                                    ],
+                                })}\n\n`);
+                            },
+                        });
+                        response.write("data: [DONE]\n\n");
+                        response.end();
+                    }
+                    finally {
+                        clearInterval(heartbeat);
+                    }
                     return;
                 }
                 const payload = await context.executeChatCompletion(body, { timeoutMs });
@@ -389,31 +394,36 @@ function createCopilotBridgeRequestHandler(context) {
                     const responseId = `resp_${Date.now()}`;
                     const messageId = buildResponsesMessageId(responseId);
                     writeResponsesStreamStart(response, responseId, normalized.model, messageId);
+                    writeResponsesReasoningPartAdded(response, responseId);
                     const heartbeat = startSseHeartbeat(response);
                     let text = "";
-                    const payload = await context.executeChatCompletion(chatPayload, {
-                        timeoutMs: normalized.timeoutMs,
-                        onTextDelta: (delta) => {
-                            text += delta;
-                            writeResponsesTextDelta(response, messageId, delta);
-                        },
-                        onTextDone: (doneText) => {
-                            if (text.length === 0) {
-                                text = doneText;
-                                writeResponsesTextDelta(response, messageId, doneText);
-                            }
-                        },
-                        onRuntimeEvent: (event) => {
-                            writeResponsesRuntimeEvent(response, responseId, event);
-                        },
-                    });
-                    clearInterval(heartbeat);
-                    const outputText = text || getChatCompletionText(payload);
-                    if (text.length === 0 && outputText.length > 0) {
-                        writeResponsesTextDelta(response, messageId, outputText);
+                    try {
+                        const payload = await context.executeChatCompletion(chatPayload, {
+                            timeoutMs: normalized.timeoutMs,
+                            onTextDelta: (delta) => {
+                                text += delta;
+                                writeResponsesTextDelta(response, messageId, delta);
+                            },
+                            onTextDone: (doneText) => {
+                                if (text.length === 0) {
+                                    text = doneText;
+                                    writeResponsesTextDelta(response, messageId, doneText);
+                                }
+                            },
+                            onRuntimeEvent: (event) => {
+                                writeResponsesRuntimeEvent(response, responseId, event);
+                            },
+                        });
+                        const outputText = text || getChatCompletionText(payload);
+                        if (text.length === 0 && outputText.length > 0) {
+                            writeResponsesTextDelta(response, messageId, outputText);
+                        }
+                        writeResponsesStreamDone(response, responseId, normalized.model, messageId, outputText);
+                        response.end();
+                    }
+                    finally {
+                        clearInterval(heartbeat);
                     }
-                    writeResponsesStreamDone(response, responseId, normalized.model, messageId, outputText);
-                    response.end();
                     return;
                 }
                 const payload = await context.executeChatCompletion(chatPayload, {
@@ -432,9 +442,15 @@ function createCopilotBridgeRequestHandler(context) {
             response.end(JSON.stringify({ error: { message: "Not found" } }));
         }
         catch (error) {
-            const statusCode = mapBridgeErrorStatus(error);
-            response.writeHead(statusCode, { "content-type": "application/json" });
-            response.end(JSON.stringify({ error: { message: error instanceof Error ? error.message : String(error), code: isCliError(error) ? error.code : "BRIDGE_RUNTIME_FAILURE" } }));
+            if (!response.headersSent) {
+                const statusCode = mapBridgeErrorStatus(error);
+                response.writeHead(statusCode, { "content-type": "application/json" });
+                response.end(JSON.stringify({ error: { message: error instanceof Error ? error.message : String(error), code: isCliError(error) ? error.code : "BRIDGE_RUNTIME_FAILURE" } }));
+            }
+            else if (!response.writableEnded) {
+                response.write(`data: ${JSON.stringify({ error: { message: error instanceof Error ? error.message : String(error), code: isCliError(error) ? error.code : "BRIDGE_RUNTIME_FAILURE" } })}\n\n`);
+                response.end();
+            }
         }
     };
 }
@@ -631,7 +647,7 @@ function writeResponsesStream(response, payload) {
     });
     writeSseEvent(response, "response.output_item.added", {
         type: "response.output_item.added",
-        output_index: 0,
+        output_index: 1,
         item: {
             id: messageId,
             type: "message",
@@ -643,7 +659,7 @@ function writeResponsesStream(response, payload) {
     writeSseEvent(response, "response.content_part.added", {
         type: "response.content_part.added",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         part: {
             type: "output_text",
@@ -654,21 +670,21 @@ function writeResponsesStream(response, payload) {
     writeSseEvent(response, "response.output_text.delta", {
         type: "response.output_text.delta",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         delta: outputText,
     });
     writeSseEvent(response, "response.output_text.done", {
         type: "response.output_text.done",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         text: outputText,
     });
     writeSseEvent(response, "response.content_part.done", {
         type: "response.content_part.done",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         part: {
             type: "output_text",
@@ -678,7 +694,7 @@ function writeResponsesStream(response, payload) {
     });
     writeSseEvent(response, "response.output_item.done", {
         type: "response.output_item.done",
-        output_index: 0,
+        output_index: 1,
         item: completedMessage,
     });
     writeSseEvent(response, "response.completed", {
@@ -710,7 +726,7 @@ function writeResponsesStreamStart(response, responseId, model, messageId) {
     });
     writeSseEvent(response, "response.output_item.added", {
         type: "response.output_item.added",
-        output_index: 0,
+        output_index: 1,
         item: {
             id: messageId,
             type: "message",
@@ -722,7 +738,7 @@ function writeResponsesStreamStart(response, responseId, model, messageId) {
     writeSseEvent(response, "response.content_part.added", {
         type: "response.content_part.added",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         part: {
             type: "output_text",
@@ -738,7 +754,7 @@ function writeResponsesTextDelta(response, messageId, delta) {
     writeSseEvent(response, "response.output_text.delta", {
         type: "response.output_text.delta",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         delta,
     });
@@ -760,14 +776,14 @@ function writeResponsesStreamDone(response, responseId, model, messageId, output
     writeSseEvent(response, "response.output_text.done", {
         type: "response.output_text.done",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         text: outputText,
     });
     writeSseEvent(response, "response.content_part.done", {
         type: "response.content_part.done",
         item_id: messageId,
-        output_index: 0,
+        output_index: 1,
         content_index: 0,
         part: {
             type: "output_text",
@@ -777,7 +793,7 @@ function writeResponsesStreamDone(response, responseId, model, messageId, output
     });
     writeSseEvent(response, "response.output_item.done", {
         type: "response.output_item.done",
-        output_index: 0,
+        output_index: 1,
         item: completedMessage,
     });
     writeSseEvent(response, "response.completed", {
@@ -797,23 +813,40 @@ function writeResponsesRuntimeEvent(response, responseId, event) {
     if (event.type === "assistant.message_delta") {
         return;
     }
-    if (event.type === "assistant.reasoning_delta") {
-        writeResponsesReasoningDelta(response, responseId, formatRuntimeEventText(event));
-        return;
-    }
     if (event.type === "session.unknown") {
-        process.stderr.write(`[${new Date().toISOString()}] bridge runtime event ignored type=${event.sdkType} summary=${truncateBridgeText(event.summary, 240)}\n`);
+        process.stderr.write(`[${new Date().toISOString()}] bridge runtime event ignored type=${event.sdkType ?? "unknown"} summary=${truncateBridgeText(event.summary ?? "", 240)}\n`);
         return;
     }
     const text = formatRuntimeEventText(event);
     if (text.length === 0) {
         return;
     }
+    writeResponsesReasoningDelta(response, responseId, text);
     writeResponsesCommentaryItem(response, responseId, text);
     process.stderr.write(`[${new Date().toISOString()}] bridge runtime event type=${event.type} summary=${truncateBridgeText(text, 240)}\n`);
 }
 function writeResponsesReasoningDelta(response, responseId, text) {
     const reasoningId = `${responseId}_rs_0`;
+    writeSseEvent(response, "response.reasoning_summary_text.delta", {
+        type: "response.reasoning_summary_text.delta",
+        item_id: reasoningId,
+        output_index: 0,
+        summary_index: 0,
+        delta: text,
+    });
+}
+function writeResponsesReasoningPartAdded(response, responseId) {
+    const reasoningId = `${responseId}_rs_0`;
+    writeSseEvent(response, "response.output_item.added", {
+        type: "response.output_item.added",
+        output_index: 0,
+        item: {
+            id: reasoningId,
+            type: "reasoning",
+            status: "in_progress",
+            summary: [],
+        },
+    });
     writeSseEvent(response, "response.reasoning_summary_part.added", {
         type: "response.reasoning_summary_part.added",
         item_id: reasoningId,
@@ -824,19 +857,12 @@ function writeResponsesReasoningDelta(response, responseId, text) {
             text: "",
         },
     });
-    writeSseEvent(response, "response.reasoning_summary_text.delta", {
-        type: "response.reasoning_summary_text.delta",
-        item_id: reasoningId,
-        output_index: 0,
-        summary_index: 0,
-        delta: text,
-    });
 }
 function writeResponsesCommentaryItem(response, responseId, text) {
     const itemId = `${responseId}_commentary_${Date.now()}_${Math.floor(Math.random() * 100000)}`;
     writeSseEvent(response, "response.output_item.done", {
         type: "response.output_item.done",
-        output_index: 0,
+        output_index: 1,
         item: {
             id: itemId,
             type: "message",
@@ -856,9 +882,9 @@ function writeResponsesCommentaryItem(response, responseId, text) {
 function formatRuntimeEventText(event) {
     switch (event.type) {
         case "assistant.intent":
-            return truncateBridgeText(event.text, 600);
+            return truncateBridgeText(event.text ?? "", 600);
         case "assistant.reasoning_delta":
-            return truncateBridgeText(event.text, 600);
+            return truncateBridgeText(event.text ?? "", 600);
         case "tool.execution_start":
             return truncateBridgeText(`Copilot started ${formatToolName(event.name)}${formatRequestId(event.requestId)}${formatSummarySuffix(event.summary)}`, 600);
         case "tool.execution_progress":
@@ -878,10 +904,12 @@ function formatRuntimeEventText(event) {
         case "session.error":
             return truncateBridgeText(`Copilot session error${formatSummarySuffix(event.summary)}`, 600);
         case "session.idle":
-            return truncateBridgeText(event.summary, 600);
+            return truncateBridgeText(event.summary ?? "", 600);
         case "assistant.message_delta":
         case "session.unknown":
             return "";
+        default:
+            return "";
     }
 }
 function formatToolName(name) {
@@ -891,7 +919,10 @@ function formatRequestId(requestId) {
     return requestId ? ` (${requestId})` : "";
 }
 function formatSummarySuffix(summary) {
-    return summary.length > 0 ? `: ${summary}` : "";
+    if (!summary || summary.length === 0) {
+        return "";
+    }
+    return `: ${summary}`;
 }
 function truncateBridgeText(value, maxLength) {
     if (value.length <= maxLength) {
@@ -901,7 +932,9 @@ function truncateBridgeText(value, maxLength) {
 }
 function startSseHeartbeat(response) {
     return setInterval(() => {
-        response.write(": keep-alive\n\n");
+        if (!response.writableEnded) {
+            response.write(": keep-alive\n\n");
+        }
     }, 15000);
 }
 function getChatCompletionText(payload) {

package/dist/runtime/copilot-http-bridge-worker.js

@@ -0,0 +1,228 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const http = __importStar(require("node:http"));
+const https = __importStar(require("node:https"));
+const crypto = __importStar(require("node:crypto"));
+const copilot_token_1 = require("./copilot-token");
+let tokenManager = null;
+function logWorkerEvent(message) {
+    process.stderr.write(`[${new Date().toISOString()}] ${message}\n`);
+}
+async function main() {
+    const provider = process.env.CODEX_SWITCH_BRIDGE_PROVIDER ?? "copilot";
+    const host = process.env.CODEX_SWITCH_BRIDGE_HOST ?? "127.0.0.1";
+    const port = Number(process.env.CODEX_SWITCH_BRIDGE_PORT ?? "41415");
+    const localApiKey = process.env.CODEX_SWITCH_BRIDGE_API_KEY ?? "";
+    const toolHomeDir = process.env.CODEX_SWITCH_TOOL_HOME_DIR || undefined;
+    const staticCopilotToken = process.env.CODEX_SWITCH_BRIDGE_COPILOT_TOKEN || undefined;
+    logWorkerEvent(`worker startup provider=${provider} host=${host} port=${String(port)}`);
+    if (staticCopilotToken) {
+        tokenManager = (0, copilot_token_1.createStaticTokenManager)(staticCopilotToken);
+        logWorkerEvent("copilot token acquired (static), api base: https://api.githubcopilot.com");
+    }
+    else {
+        const githubPat = process.env.CODEX_SWITCH_GITHUB_TOKEN || (0, copilot_token_1.readGithubToken)(toolHomeDir);
+        if (!githubPat) {
+            throw new Error("No GitHub token found. Run `codexs login copilot` first.");
+        }
+        tokenManager = (0, copilot_token_1.createTokenManager)(githubPat);
+        await tokenManager.getToken();
+        logWorkerEvent(`copilot token acquired, api base: ${tokenManager.getApiBaseUrl()}`);
+    }
+    const server = http.createServer(async (req, res) => {
+        try {
+            await handleRequest(req, res, localApiKey);
+        }
+        catch (error) {
+            if (!res.headersSent) {
+                res.writeHead(500, { "content-type": "application/json" });
+                res.end(JSON.stringify({ error: { message: error instanceof Error ? error.message : String(error) } }));
+            }
+        }
+    });
+    const stopWorker = () => {
+        logWorkerEvent(`worker shutdown provider=${provider}`);
+        tokenManager?.stop();
+        server.close();
+        process.exit(0);
+    };
+    process.once("SIGINT", stopWorker);
+    process.once("SIGTERM", stopWorker);
+    await new Promise((resolve, reject) => {
+        server.once("error", reject);
+        server.listen(port, host, () => {
+            server.off("error", reject);
+            resolve();
+        });
+    });
+    logWorkerEvent(`worker ready provider=${provider} host=${host} port=${String(port)}`);
+}
+async function handleRequest(req, res, localApiKey) {
+    const method = req.method ?? "GET";
+    const url = req.url ?? "/";
+    if (method === "GET" && url === "/healthz") {
+        res.writeHead(200, { "content-type": "application/json" });
+        res.end(JSON.stringify({ ok: true }));
+        return;
+    }
+    if (!isAuthorized(req, localApiKey)) {
+        res.writeHead(401, { "content-type": "application/json" });
+        res.end(JSON.stringify({ error: { message: "Unauthorized" } }));
+        return;
+    }
+    if (method === "GET" && url === "/v1/models") {
+        await proxyGet("/models", res);
+        return;
+    }
+    if (method === "POST" && url === "/v1/chat/completions") {
+        await proxyPost("/chat/completions", req, res);
+        return;
+    }
+    if (method === "POST" && url === "/v1/responses") {
+        await proxyPost("/responses", req, res);
+        return;
+    }
+    res.writeHead(404, { "content-type": "application/json" });
+    res.end(JSON.stringify({ error: { message: "Not found" } }));
+}
+async function proxyGet(upstreamPath, res) {
+    const copilotToken = await tokenManager.getToken();
+    const apiBase = tokenManager.getApiBaseUrl();
+    const targetUrl = new URL(upstreamPath, apiBase);
+    const headers = (0, copilot_token_1.getCopilotRequestHeaders)(copilotToken);
+    const upstreamRes = await httpsRequest({
+        method: "GET",
+        url: targetUrl,
+        headers,
+    });
+    res.writeHead(upstreamRes.statusCode, filterResponseHeaders(upstreamRes.headers));
+    upstreamRes.pipe(res);
+}
+async function proxyPost(upstreamPath, req, res) {
+    const body = await readRequestBody(req);
+    const copilotToken = await tokenManager.getToken();
+    const apiBase = tokenManager.getApiBaseUrl();
+    const targetUrl = new URL(upstreamPath, apiBase);
+    const requestId = crypto.randomUUID();
+    const headers = (0, copilot_token_1.getCopilotRequestHeaders)(copilotToken, requestId);
+    // Preserve content-length for the body
+    headers["content-length"] = Buffer.byteLength(body).toString();
+    const upstreamRes = await httpsRequest({
+        method: "POST",
+        url: targetUrl,
+        headers,
+        body,
+    });
+    // If upstream returned 401, try refreshing token once and retry
+    if (upstreamRes.statusCode === 401) {
+        upstreamRes.resume();
+        logWorkerEvent("upstream 401, invalidating token and retrying");
+        tokenManager.invalidate();
+        const freshToken = await tokenManager.getToken();
+        const retryHeaders = (0, copilot_token_1.getCopilotRequestHeaders)(freshToken, crypto.randomUUID());
+        retryHeaders["content-length"] = Buffer.byteLength(body).toString();
+        const retryRes = await httpsRequest({
+            method: "POST",
+            url: targetUrl,
+            headers: retryHeaders,
+            body,
+        });
+        res.writeHead(retryRes.statusCode, filterResponseHeaders(retryRes.headers));
+        retryRes.pipe(res);
+        return;
+    }
+    res.writeHead(upstreamRes.statusCode, filterResponseHeaders(upstreamRes.headers));
+    upstreamRes.pipe(res);
+}
+function httpsRequest(args) {
+    return new Promise((resolve, reject) => {
+        const req = https.request({
+            hostname: args.url.hostname,
+            port: args.url.port || 443,
+            path: args.url.pathname + args.url.search,
+            method: args.method,
+            headers: args.headers,
+        }, (res) => {
+            resolve(res);
+        });
+        req.on("error", reject);
+        if (args.body) {
+            req.write(args.body);
+        }
+        req.end();
+    });
+}
+function readRequestBody(req) {
+    return new Promise((resolve, reject) => {
+        const chunks = [];
+        req.on("data", (chunk) => chunks.push(chunk));
+        req.on("end", () => resolve(Buffer.concat(chunks).toString("utf8")));
+        req.on("error", reject);
+    });
+}
+function filterResponseHeaders(headers) {
+    const filtered = {};
+    const passthrough = ["content-type", "transfer-encoding", "x-request-id"];
+    for (const key of passthrough) {
+        if (headers[key]) {
+            filtered[key] = headers[key];
+        }
+    }
+    // Always allow cache-control for SSE
+    if (headers["cache-control"]) {
+        filtered["cache-control"] = headers["cache-control"];
+    }
+    return filtered;
+}
+function isAuthorized(req, expectedApiKey) {
+    const authorization = req.headers.authorization;
+    if (!authorization || !authorization.startsWith("Bearer ")) {
+        return false;
+    }
+    return authorization.slice("Bearer ".length) === expectedApiKey;
+}
+if (require.main === module) {
+    process.on("uncaughtException", (error) => {
+        logWorkerEvent(`worker uncaught exception: ${error.message}`);
+    });
+    process.on("unhandledRejection", (reason) => {
+        logWorkerEvent(`worker unhandled rejection: ${reason instanceof Error ? reason.message : String(reason)}`);
+    });
+    void main().catch((error) => {
+        logWorkerEvent(`worker startup failure: ${error instanceof Error ? error.message : String(error)}`);
+        process.exit(1);
+    });
+}