@congzhen/changewayguard 6.8.12

package/dist/gateway/gateway/handlers/models.js

@@ -0,0 +1,36 @@
+import { findDefaultBackend } from "../config.js";
+export async function handleModelsRequest(res, config) {
+    try {
+        // Find an OpenAI-compatible backend for models listing
+        const resolved = findDefaultBackend("openai", config);
+        if (!resolved) {
+            res.writeHead(500, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: "No OpenAI-compatible backend configured" }));
+            return;
+        }
+        const { backend } = resolved;
+        const modelsUrl = `${backend.baseUrl}/v1/models`;
+        const headers = {
+            "Authorization": `Bearer ${backend.apiKey}`,
+        };
+        if (backend.referer) {
+            headers["HTTP-Referer"] = backend.referer;
+        }
+        if (backend.title) {
+            headers["X-Title"] = backend.title;
+        }
+        const response = await fetch(modelsUrl, { headers });
+        const body = await response.text();
+        res.writeHead(response.status, { "Content-Type": "application/json" });
+        res.end(body);
+    }
+    catch (error) {
+        console.error("[ai-security-gateway] Models request error:", error);
+        res.writeHead(500, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({
+            error: "Internal gateway error",
+            message: error instanceof Error ? error.message : String(error),
+        }));
+    }
+}
+//# sourceMappingURL=models.js.map

package/dist/gateway/gateway/handlers/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/handlers/models.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAqB,MAAM,cAAc,CAAC;AAErE,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,GAAmB,EACnB,MAAqB;IAErB,IAAI,CAAC;QACH,uDAAuD;QACvD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,yCAAyC,EAAE,CAAC,CAAC,CAAC;YAC9E,OAAO;QACT,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC;QAC7B,MAAM,SAAS,GAAG,GAAG,OAAO,CAAC,OAAO,YAAY,CAAC;QACjD,MAAM,OAAO,GAA2B;YACtC,eAAe,EAAE,UAAU,OAAO,CAAC,MAAM,EAAE;SAC5C,CAAC;QAEF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QACrC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACvE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;QACpE,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC3D,GAAG,CAAC,GAAG,CACL,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,wBAAwB;YAC/B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAChE,CAAC,CACH,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/gateway/gateway/handlers/openai.d.ts

@@ -0,0 +1,16 @@
+/**
+ * AI Security Gateway - OpenAI Chat Completions API handler
+ *
+ * Handles POST /v1/chat/completions requests in OpenAI's format.
+ * Also compatible with OpenAI-compatible APIs (Kimi, DeepSeek, etc.)
+ */
+import type { IncomingMessage, ServerResponse } from "node:http";
+import type { BackendConfig } from "../types.js";
+/**
+ * Handle OpenAI API request
+ *
+ * @param backend - Config for OpenAI-compatible backend
+ * @param extraHeaders - Optional additional headers (e.g., OpenRouter attribution)
+ */
+export declare function handleOpenAIRequest(req: IncomingMessage, res: ServerResponse, backend: BackendConfig, extraHeaders?: Record<string, string>): Promise<void>;
+//# sourceMappingURL=openai.d.ts.map

package/dist/gateway/gateway/handlers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/handlers/openai.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AACjE,OAAO,KAAK,EAAE,aAAa,EAAgB,MAAM,aAAa,CAAC;AAK/D;;;;;GAKG;AACH,wBAAsB,mBAAmB,CACvC,GAAG,EAAE,eAAe,EACpB,GAAG,EAAE,cAAc,EACnB,OAAO,EAAE,aAAa,EACtB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACpC,OAAO,CAAC,IAAI,CAAC,CAiGf"}

package/dist/gateway/gateway/handlers/openai.js

@@ -0,0 +1,254 @@
+/**
+ * AI Security Gateway - OpenAI Chat Completions API handler
+ *
+ * Handles POST /v1/chat/completions requests in OpenAI's format.
+ * Also compatible with OpenAI-compatible APIs (Kimi, DeepSeek, etc.)
+ */
+import { sanitize } from "../sanitizer.js";
+import { restore, createStreamRestorer } from "../restorer.js";
+import { generateRequestId, logSanitizeEvent, logRestoreEvent } from "../activity.js";
+/**
+ * Handle OpenAI API request
+ *
+ * @param backend - Config for OpenAI-compatible backend
+ * @param extraHeaders - Optional additional headers (e.g., OpenRouter attribution)
+ */
+export async function handleOpenAIRequest(req, res, backend, extraHeaders) {
+    try {
+        const requestId = generateRequestId();
+        const sanitizeStart = Date.now();
+        // 1. Parse request body
+        const body = await readBody(req);
+        const requestData = JSON.parse(body);
+        const { model, messages, tools, tool_choice, temperature, max_tokens, stream = false, ...rest } = requestData;
+        // 2. Sanitize messages
+        const { sanitized: sanitizedMessages, mappingTable, redactionCount } = sanitize(messages);
+        // Debug: log what was sanitized
+        console.log(`[ai-security-gateway] Sanitized ${redactionCount} items`);
+        if (mappingTable.size > 0) {
+            for (const [placeholder, original] of mappingTable.entries()) {
+                console.log(`[ai-security-gateway]   ${placeholder} <- (${original.length} chars)`);
+            }
+        }
+        // Log sanitization event
+        if (redactionCount > 0) {
+            logSanitizeEvent({
+                requestId,
+                backend: "openai",
+                endpoint: "/v1/chat/completions",
+                model,
+                mappingTable,
+                redactionCount,
+                durationMs: Date.now() - sanitizeStart,
+            });
+        }
+        // 3. Build sanitized request
+        const sanitizedRequest = {
+            model,
+            messages: sanitizedMessages,
+            ...(tools && { tools }),
+            ...(tool_choice && { tool_choice }),
+            ...(temperature !== undefined && { temperature }),
+            ...(max_tokens && { max_tokens }),
+            stream,
+            ...rest,
+        };
+        // 4. Use provided backend config
+        // Note: baseUrl already includes the full path prefix (e.g., /v1 or /v1/coding)
+        const apiUrl = `${backend.baseUrl}/chat/completions`;
+        const headers = {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${backend.apiKey}`,
+        };
+        // Merge extra headers (e.g., OpenRouter attribution headers)
+        if (extraHeaders) {
+            Object.assign(headers, extraHeaders);
+        }
+        const response = await fetch(apiUrl, {
+            method: "POST",
+            headers,
+            body: JSON.stringify(sanitizedRequest),
+        });
+        if (!response.ok) {
+            // Forward error response
+            res.writeHead(response.status, { "Content-Type": "application/json" });
+            const errorBody = await response.text();
+            res.end(errorBody);
+            return;
+        }
+        // 6. Handle streaming or non-streaming response
+        if (stream) {
+            await handleOpenAIStream(response, res, mappingTable, requestId, model);
+        }
+        else {
+            await handleOpenAINonStream(response, res, mappingTable, requestId, model);
+        }
+    }
+    catch (error) {
+        console.error("[ai-security-gateway] OpenAI handler error:", error);
+        res.writeHead(500, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({
+            error: "Internal gateway error",
+            message: error instanceof Error ? error.message : String(error),
+        }));
+    }
+}
+/**
+ * Handle streaming response (SSE) with smart placeholder restoration
+ *
+ * Uses StreamRestorer to detect `__` and buffer potential placeholders.
+ * Only buffers when necessary, maintaining streaming UX.
+ */
+async function handleOpenAIStream(response, res, mappingTable, requestId, model) {
+    const restoreStart = Date.now();
+    // Debug: log mapping table
+    if (mappingTable.size > 0) {
+        console.log(`[ai-security-gateway] Streaming with ${mappingTable.size} placeholders to restore`);
+    }
+    // Set SSE headers
+    res.writeHead(200, {
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        "Connection": "keep-alive",
+    });
+    const reader = response.body?.getReader();
+    if (!reader) {
+        res.end();
+        return;
+    }
+    const decoder = new TextDecoder();
+    let lineBuffer = "";
+    // Create stream restorer for text content
+    const streamRestorer = createStreamRestorer(mappingTable);
+    // Buffer for SSE chunks waiting for restoration
+    const pendingChunks = [];
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            // Decode chunk
+            lineBuffer += decoder.decode(value, { stream: true });
+            // Process complete lines
+            const lines = lineBuffer.split("\n");
+            lineBuffer = lines.pop() || ""; // Keep incomplete line in buffer
+            for (const line of lines) {
+                if (!line.trim()) {
+                    // Flush pending chunks before empty line
+                    flushPendingChunks(pendingChunks, streamRestorer, res);
+                    res.write("\n");
+                    continue;
+                }
+                if (!line.startsWith("data: ")) {
+                    res.write(line + "\n");
+                    continue;
+                }
+                const dataContent = line.slice(6);
+                if (dataContent === "[DONE]") {
+                    // Finalize any pending content
+                    flushPendingChunks(pendingChunks, streamRestorer, res);
+                    res.write(line + "\n");
+                    continue;
+                }
+                try {
+                    const parsed = JSON.parse(dataContent);
+                    const textContent = parsed.choices?.[0]?.delta?.content;
+                    if (textContent !== undefined && mappingTable.size > 0) {
+                        // Process text through stream restorer
+                        const restored = streamRestorer.process(textContent);
+                        if (restored.length > 0) {
+                            // We have restorable content - flush it
+                            const restoredChunk = { ...parsed };
+                            restoredChunk.choices = parsed.choices.map((c, i) => i === 0 ? { ...c, delta: { ...c.delta, content: restored } } : c);
+                            res.write(`data: ${JSON.stringify(restoredChunk)}\n`);
+                        }
+                        // If restorer is buffering, we don't output anything yet
+                        // Content will be output when buffer is flushed
+                    }
+                    else {
+                        // No text content or no mappings - pass through
+                        res.write(line + "\n");
+                    }
+                }
+                catch {
+                    // Not valid JSON, pass through
+                    res.write(line + "\n");
+                }
+            }
+        }
+        // Write any remaining line buffer
+        if (lineBuffer.trim()) {
+            res.write(lineBuffer + "\n");
+        }
+        // Finalize stream restorer - flush any remaining buffered content
+        const finalContent = streamRestorer.finalize();
+        if (finalContent.length > 0) {
+            // Create a final chunk with remaining content
+            const finalChunk = {
+                choices: [{ delta: { content: finalContent }, index: 0, finish_reason: null }],
+            };
+            res.write(`data: ${JSON.stringify(finalChunk)}\n`);
+        }
+        // Log restoration event
+        if (mappingTable.size > 0) {
+            logRestoreEvent({
+                requestId,
+                backend: "openai",
+                endpoint: "/v1/chat/completions",
+                model,
+                mappingTable,
+                restorationCount: mappingTable.size,
+                durationMs: Date.now() - restoreStart,
+            });
+        }
+        res.end();
+    }
+    catch (error) {
+        console.error("[ai-security-gateway] Stream error:", error);
+        res.end();
+    }
+}
+/**
+ * Flush pending chunks with restored content
+ */
+function flushPendingChunks(_pendingChunks, _streamRestorer, _res) {
+    // Currently unused - StreamRestorer handles buffering internally
+}
+/**
+ * Handle non-streaming response
+ */
+async function handleOpenAINonStream(response, res, mappingTable, requestId, model) {
+    const restoreStart = Date.now();
+    const responseBody = await response.text();
+    const responseData = JSON.parse(responseBody);
+    // Restore placeholders in response
+    const restoredData = restore(responseData, mappingTable);
+    // Log restoration event
+    if (mappingTable.size > 0) {
+        logRestoreEvent({
+            requestId,
+            backend: "openai",
+            endpoint: "/v1/chat/completions",
+            model,
+            mappingTable,
+            restorationCount: mappingTable.size,
+            durationMs: Date.now() - restoreStart,
+        });
+    }
+    res.writeHead(200, { "Content-Type": "application/json" });
+    res.end(JSON.stringify(restoredData));
+}
+/**
+ * Read request body as string
+ */
+function readBody(req) {
+    return new Promise((resolve, reject) => {
+        let body = "";
+        req.on("data", (chunk) => {
+            body += chunk.toString();
+        });
+        req.on("end", () => resolve(body));
+        req.on("error", reject);
+    });
+}
+//# sourceMappingURL=openai.js.map

package/dist/gateway/gateway/handlers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/handlers/openai.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtF;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,GAAoB,EACpB,GAAmB,EACnB,OAAsB,EACtB,YAAqC;IAErC,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,iBAAiB,EAAE,CAAC;QACtC,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEjC,wBAAwB;QACxB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAErC,MAAM,EACJ,KAAK,EACL,QAAQ,EACR,KAAK,EACL,WAAW,EACX,WAAW,EACX,UAAU,EACV,MAAM,GAAG,KAAK,EACd,GAAG,IAAI,EACR,GAAG,WAAW,CAAC;QAEhB,uBAAuB;QACvB,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,YAAY,EAAE,cAAc,EAAE,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAE1F,gCAAgC;QAChC,OAAO,CAAC,GAAG,CAAC,mCAAmC,cAAc,QAAQ,CAAC,CAAC;QACvE,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,KAAK,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC7D,OAAO,CAAC,GAAG,CAAC,2BAA2B,WAAW,QAAQ,QAAQ,CAAC,MAAM,SAAS,CAAC,CAAC;YACtF,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;YACvB,gBAAgB,CAAC;gBACf,SAAS;gBACT,OAAO,EAAE,QAAQ;gBACjB,QAAQ,EAAE,sBAAsB;gBAChC,KAAK;gBACL,YAAY;gBACZ,cAAc;gBACd,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa;aACvC,CAAC,CAAC;QACL,CAAC;QAED,6BAA6B;QAC7B,MAAM,gBAAgB,GAAG;YACvB,KAAK;YACL,QAAQ,EAAE,iBAAiB;YAC3B,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;YACvB,GAAG,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,CAAC;YACnC,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,CAAC;YACjD,GAAG,CAAC,UAAU,IAAI,EAAE,UAAU,EAAE,CAAC;YACjC,MAAM;YACN,GAAG,IAAI;SACR,CAAC;QAEF,iCAAiC;QACjC,gFAAgF;QAChF,MAAM,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,mBAAmB,CAAC;QACrD,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,OAAO,CAAC,MAAM,EAAE;SAC5C,CAAC;QACF,6DAA6D;QAC7D,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QACvC,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,EAAE;YACnC,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;SACvC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,yBAAyB;YACzB,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvE,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnB,OAAO;QACT,CAAC;QAED,gDAAgD;QAChD,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,kBAAkB,CAAC,QAAQ,EAAE,GAAG,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC1E,CAAC;aAAM,CAAC;YACN,MAAM,qBAAqB,CAAC,QAAQ,EAAE,GAAG,EAAE,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;QACpE,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC3D,GAAG,CAAC,GAAG,CACL,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,wBAAwB;YAC/B,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAChE,CAAC,CACH,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,KAAK,UAAU,kBAAkB,CAC/B,QAAkB,EAClB,GAAmB,EACnB,YAA0B,EAC1B,SAAiB,EACjB,KAAc;IAEd,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEhC,2BAA2B;IAC3B,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,wCAAwC,YAAY,CAAC,IAAI,0BAA0B,CAAC,CAAC;IACnG,CAAC;IAED,kBAAkB;IAClB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;QACjB,cAAc,EAAE,mBAAmB;QACnC,eAAe,EAAE,UAAU;QAC3B,YAAY,EAAE,YAAY;KAC3B,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,GAAG,CAAC,GAAG,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,UAAU,GAAG,EAAE,CAAC;IAEpB,0CAA0C;IAC1C,MAAM,cAAc,GAAG,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAE1D,gDAAgD;IAChD,MAAM,aAAa,GAA4D,EAAE,CAAC;IAElF,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,eAAe;YACf,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAEtD,yBAAyB;YACzB,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACrC,UAAU,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,iCAAiC;YAEjE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBACjB,yCAAyC;oBACzC,kBAAkB,CAAC,aAAa,EAAE,cAAc,EAAE,GAAG,CAAC,CAAC;oBACvD,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAChB,SAAS;gBACX,CAAC;gBAED,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/B,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;oBACvB,SAAS;gBACX,CAAC;gBAED,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAClC,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;oBAC7B,+BAA+B;oBAC/B,kBAAkB,CAAC,aAAa,EAAE,cAAc,EAAE,GAAG,CAAC,CAAC;oBACvD,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;oBACvB,SAAS;gBACX,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAmB,CAAC;oBACzD,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;oBAExD,IAAI,WAAW,KAAK,SAAS,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;wBACvD,uCAAuC;wBACvC,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;wBAErD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BACxB,wCAAwC;4BACxC,MAAM,aAAa,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;4BACpC,aAAa,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAClD,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,KAAK,EAAE,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CACjE,CAAC;4BACF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;wBACxD,CAAC;wBAED,yDAAyD;wBACzD,gDAAgD;oBAClD,CAAC;yBAAM,CAAC;wBACN,gDAAgD;wBAChD,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;oBACzB,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,+BAA+B;oBAC/B,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,IAAI,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;YACtB,GAAG,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC;QAC/B,CAAC;QAED,kEAAkE;QAClE,MAAM,YAAY,GAAG,cAAc,CAAC,QAAQ,EAAE,CAAC;QAC/C,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,8CAA8C;YAC9C,MAAM,UAAU,GAAmB;gBACjC,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;aAC/E,CAAC;YACF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC;QAED,wBAAwB;QACxB,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,eAAe,CAAC;gBACd,SAAS;gBACT,OAAO,EAAE,QAAQ;gBACjB,QAAQ,EAAE,sBAAsB;gBAChC,KAAK;gBACL,YAAY;gBACZ,gBAAgB,EAAE,YAAY,CAAC,IAAI;gBACnC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY;aACtC,CAAC,CAAC;QACL,CAAC;QAED,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;QAC5D,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAcD;;GAEG;AACH,SAAS,kBAAkB,CACzB,cAAuE,EACvE,eAAwD,EACxD,IAAoB;IAEpB,iEAAiE;AACnE,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,qBAAqB,CAClC,QAAkB,EAClB,GAAmB,EACnB,YAA0B,EAC1B,SAAiB,EACjB,KAAc;IAEd,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAChC,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAE9C,mCAAmC;IACnC,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAEzD,wBAAwB;IACxB,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,eAAe,CAAC;YACd,SAAS;YACT,OAAO,EAAE,QAAQ;YACjB,QAAQ,EAAE,sBAAsB;YAChC,KAAK;YACL,YAAY;YACZ,gBAAgB,EAAE,YAAY,CAAC,IAAI;YACnC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY;SACtC,CAAC,CAAC;IACL,CAAC;IAED,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;AACxC,CAAC;AAED;;GAEG;AACH,SAAS,QAAQ,CAAC,GAAoB;IACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YACvB,IAAI,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QACH,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QACnC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/gateway/gateway/index.d.ts

@@ -0,0 +1,27 @@
+#!/usr/bin/env node
+/**
+ * OpenGuardrails AI Security Gateway
+ *
+ * Local HTTP proxy that intercepts LLM API calls, sanitizes sensitive data
+ * before sending to providers, and restores it in responses.
+ * Supports Anthropic, OpenAI, and Gemini protocols.
+ */
+/**
+ * Stop the gateway server
+ */
+export declare function stopGateway(): Promise<void>;
+/**
+ * Check if gateway is running
+ */
+export declare function isGatewayServerRunning(): boolean;
+/**
+ * Start gateway server
+ * @param configPath - Path to config file
+ * @param embedded - If true, don't call process.exit on errors (for in-process use)
+ */
+export declare function startGateway(configPath?: string, embedded?: boolean): void;
+export { sanitize, sanitizeMessages } from "./sanitizer.js";
+export { restore, restoreJSON, restoreSSELine } from "./restorer.js";
+export { addActivityListener, removeActivityListener, clearActivityListeners, } from "./activity.js";
+export type { GatewayConfig, MappingTable, SanitizeResult, EntityMatch, GatewayActivityEvent, ActivityListener, } from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/gateway/gateway/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;GAMG;AAwMH;;GAEG;AACH,wBAAgB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAY3C;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,OAAO,CAEhD;AAED;;;;GAIG;AACH,wBAAgB,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,QAAQ,UAAQ,GAAG,IAAI,CAmFxE;AAGD,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAC5D,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACrE,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,eAAe,CAAC;AAEvB,YAAY,EACV,aAAa,EACb,YAAY,EACZ,cAAc,EACd,WAAW,EACX,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,YAAY,CAAC"}

package/dist/gateway/gateway/index.js

@@ -0,0 +1,293 @@
+#!/usr/bin/env node
+/**
+ * OpenGuardrails AI Security Gateway
+ *
+ * Local HTTP proxy that intercepts LLM API calls, sanitizes sensitive data
+ * before sending to providers, and restores it in responses.
+ * Supports Anthropic, OpenAI, and Gemini protocols.
+ */
+import { createServer } from "node:http";
+import { loadConfig, validateConfig, findBackendByApiKey, findDefaultBackend, findBackendByPathPrefix } from "./config.js";
+import { handleAnthropicRequest } from "./handlers/anthropic.js";
+import { handleOpenAIRequest } from "./handlers/openai.js";
+import { handleGeminiRequest } from "./handlers/gemini.js";
+import { handleModelsRequest } from "./handlers/models.js";
+let config;
+let currentServer = null;
+/**
+ * Extract API key from request headers
+ */
+function extractApiKey(req) {
+    // Try x-api-key header (Anthropic style)
+    const xApiKey = req.headers["x-api-key"];
+    if (xApiKey && typeof xApiKey === "string") {
+        return xApiKey;
+    }
+    // Try Authorization: Bearer (OpenAI style)
+    const auth = req.headers["authorization"];
+    if (auth && typeof auth === "string" && auth.startsWith("Bearer ")) {
+        return auth.slice(7);
+    }
+    // Try x-goog-api-key (Gemini style)
+    const googKey = req.headers["x-goog-api-key"];
+    if (googKey && typeof googKey === "string") {
+        return googKey;
+    }
+    return null;
+}
+/**
+ * Resolve backend for a request based on path prefix, API key, or defaults
+ * Priority: pathPrefix > apiKey > defaultBackend
+ */
+function resolveBackend(req, apiType) {
+    const url = req.url || "";
+    // 1. Try to find backend by path prefix (most specific)
+    const byPath = findBackendByPathPrefix(url, config);
+    if (byPath) {
+        return byPath;
+    }
+    // 2. Try to find backend by API key
+    const apiKey = extractApiKey(req);
+    if (apiKey) {
+        const byKey = findBackendByApiKey(apiKey, config);
+        if (byKey) {
+            return byKey;
+        }
+    }
+    // 3. Fall back to default backend for the API type
+    return findDefaultBackend(apiType, config);
+}
+/**
+ * Main request handler
+ */
+async function handleRequest(req, res) {
+    const { method, url } = req;
+    // Log request (skip health checks to reduce noise)
+    if (url !== "/health") {
+        console.log(`[ai-security-gateway] ${method} ${url}`);
+    }
+    // CORS headers (for browser-based clients)
+    res.setHeader("Access-Control-Allow-Origin", "*");
+    res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
+    res.setHeader("Access-Control-Allow-Headers", "Content-Type, Authorization, x-api-key, anthropic-version");
+    // Handle OPTIONS for CORS preflight
+    if (method === "OPTIONS") {
+        res.writeHead(204);
+        res.end();
+        return;
+    }
+    // Health check (allow GET)
+    if (url === "/health") {
+        res.writeHead(200, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({ status: "ok", version: "1.0.0" }));
+        return;
+    }
+    // Handle GET /v1/models — proxy to configured backend's models endpoint
+    if (method === "GET" && url === "/v1/models") {
+        await handleModelsRequest(res, config);
+        return;
+    }
+    // Only allow POST for API endpoints
+    if (method !== "POST") {
+        res.writeHead(405, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({ error: "Method not allowed" }));
+        return;
+    }
+    // Route to appropriate handler based on path suffix
+    // This allows flexible path prefixes (e.g., /v1/coding/chat/completions)
+    try {
+        if (url?.endsWith("/messages")) {
+            // Anthropic Messages API (matches /v1/messages, /v1/xxx/messages, etc.)
+            const resolved = resolveBackend(req, "anthropic");
+            if (!resolved) {
+                res.writeHead(500, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ error: "No Anthropic-compatible backend configured" }));
+                return;
+            }
+            await handleAnthropicRequest(req, res, resolved.backend);
+        }
+        else if (url?.endsWith("/chat/completions")) {
+            // OpenAI/OpenRouter Chat Completions API
+            // Try to extract backend name from URL: /backend/{name}/chat/completions
+            const backendMatch = url.match(/^\/backend\/([^/]+)\//);
+            let resolved = null;
+            if (backendMatch) {
+                const backendName = backendMatch[1];
+                const backend = config.backends[backendName];
+                if (backend) {
+                    resolved = { name: backendName, backend };
+                    console.log(`[ai-security-gateway] Backend from URL: ${backendName}`);
+                }
+            }
+            // Fallback to path prefix or default
+            if (!resolved) {
+                resolved = resolveBackend(req, "openai");
+                console.log(`[ai-security-gateway] Resolved backend: ${resolved?.name}`);
+            }
+            // Check explicit routing config
+            const explicitBackendName = config.routing?.["/v1/chat/completions"];
+            const backend = explicitBackendName
+                ? config.backends[explicitBackendName]
+                : resolved?.backend;
+            if (!backend) {
+                res.writeHead(500, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ error: "No OpenAI-compatible backend configured" }));
+                return;
+            }
+            const extraHeaders = {};
+            if (backend.referer) {
+                extraHeaders["HTTP-Referer"] = backend.referer;
+            }
+            if (backend.title) {
+                extraHeaders["X-Title"] = backend.title;
+            }
+            await handleOpenAIRequest(req, res, backend, extraHeaders);
+        }
+        else if (url?.match(/\/models\/(.+):generateContent$/)) {
+            // Gemini API (matches any path ending with /models/{model}:generateContent)
+            const match = url.match(/\/models\/(.+):generateContent$/);
+            const modelName = match?.[1];
+            if (modelName) {
+                const resolved = resolveBackend(req, "gemini");
+                if (!resolved) {
+                    res.writeHead(500, { "Content-Type": "application/json" });
+                    res.end(JSON.stringify({ error: "No Gemini backend configured" }));
+                    return;
+                }
+                await handleGeminiRequest(req, res, resolved.backend, modelName);
+            }
+            else {
+                res.writeHead(404, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ error: "Model name required" }));
+            }
+        }
+        else {
+            // Unknown endpoint
+            res.writeHead(404, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: "Not found", url }));
+        }
+    }
+    catch (error) {
+        console.error("[ai-security-gateway] Request handler error:", error);
+        res.writeHead(500, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({
+            error: "Internal server error",
+            message: error instanceof Error ? error.message : String(error),
+        }));
+    }
+}
+/**
+ * Stop the gateway server
+ */
+export function stopGateway() {
+    return new Promise((resolve) => {
+        if (currentServer) {
+            currentServer.close(() => {
+                currentServer = null;
+                console.log("[ai-security-gateway] Server stopped");
+                resolve();
+            });
+        }
+        else {
+            resolve();
+        }
+    });
+}
+/**
+ * Check if gateway is running
+ */
+export function isGatewayServerRunning() {
+    return currentServer !== null;
+}
+/**
+ * Start gateway server
+ * @param configPath - Path to config file
+ * @param embedded - If true, don't call process.exit on errors (for in-process use)
+ */
+export function startGateway(configPath, embedded = false) {
+    // Stop existing server if running (same process)
+    if (currentServer) {
+        if (!embedded)
+            console.log("[ai-security-gateway] Stopping existing server for restart...");
+        currentServer.close();
+        currentServer = null;
+    }
+    try {
+        // Load and validate configuration
+        config = loadConfig(configPath);
+        validateConfig(config);
+        if (!embedded) {
+            console.log("[ai-security-gateway] Configuration loaded:");
+            console.log(`  Port: ${config.port}`);
+            console.log(`  Backends: ${Object.keys(config.backends).join(", ") || "(none)"}`);
+        }
+        // Create HTTP server
+        const server = createServer(handleRequest);
+        currentServer = server;
+        // Handle server errors
+        server.on("error", (err) => {
+            console.error("[ai-security-gateway] Server error:", err);
+            currentServer = null;
+            if (!embedded) {
+                process.exit(1);
+                return;
+            }
+            // Embedded mode: never crash host process on gateway bind/runtime errors.
+            // The plugin can continue running and other features remain available.
+            console.warn("[ai-security-gateway] Embedded mode: gateway error ignored");
+        });
+        // Start listening
+        server.listen(config.port, "127.0.0.1", () => {
+            if (!embedded) {
+                console.log(`[ai-security-gateway] Server listening on http://127.0.0.1:${config.port}`);
+                console.log("[ai-security-gateway] Ready to proxy requests");
+                console.log("");
+                console.log("Endpoints:");
+                console.log(`  POST http://127.0.0.1:${config.port}/v1/messages - Anthropic`);
+                console.log(`  POST http://127.0.0.1:${config.port}/v1/chat/completions - OpenAI / OpenRouter`);
+                console.log(`  POST http://127.0.0.1:${config.port}/v1/models/:model:generateContent - Gemini`);
+                console.log(`  GET  http://127.0.0.1:${config.port}/v1/models - List models (OpenAI / OpenRouter)`);
+                console.log(`  GET  http://127.0.0.1:${config.port}/health - Health check`);
+            }
+        });
+        // In embedded mode, don't let the server prevent process exit
+        if (embedded) {
+            server.unref();
+        }
+        // Only register shutdown handlers if not embedded
+        if (!embedded) {
+            process.on("SIGINT", () => {
+                console.log("\n[ai-security-gateway] Shutting down...");
+                server.close(() => {
+                    console.log("[ai-security-gateway] Server stopped");
+                    process.exit(0);
+                });
+            });
+            process.on("SIGTERM", () => {
+                console.log("\n[ai-security-gateway] Shutting down...");
+                server.close(() => {
+                    console.log("[ai-security-gateway] Server stopped");
+                    process.exit(0);
+                });
+            });
+        }
+    }
+    catch (error) {
+        console.error("[ai-security-gateway] Failed to start:", error);
+        currentServer = null;
+        if (!embedded) {
+            process.exit(1);
+        }
+        throw error;
+    }
+}
+// Re-export for programmatic use
+export { sanitize, sanitizeMessages } from "./sanitizer.js";
+export { restore, restoreJSON, restoreSSELine } from "./restorer.js";
+export { addActivityListener, removeActivityListener, clearActivityListeners, } from "./activity.js";
+// Start if run directly
+if (import.meta.url === `file://${process.argv[1]}`) {
+    const configPath = process.argv[2];
+    startGateway(configPath);
+}
+//# sourceMappingURL=index.js.map