mirror of
https://github.com/Nezumi-2711/9router.git
synced 2026-09-22 20:00:47 +00:00
# v0.4.30 (2026-05-11)
## Features - MCP stdio→SSE bridge: expose local stdio MCP plugins over SSE (api/mcp/[plugin]/sse, /message) - Dynamic Linux cert resolution + NSS DB injection (Debian/Arch/Fedora/openSUSE, Chrome/Chromium/Firefox incl. snap) (#1010) - Cowork tool: expanded settings UI & API - GitBook docs (DocsContent, DocsLayout) ## Fixes - OAuth callback postMessage scoped to expected origins (CWE-1385) (#998) - Re-enable TLS verification on DNS-bypass fetch (CWE-295) (#998) - Normalize `developer` role → `system` for OpenAI-format providers (Deepseek, Groq, …) (#1011, closes #773) - Respect `PORT` env in internal model-test fetch (#1014) - Dropdown text readability in dark theme on usage page (#997) ## Improvements - Refactor Claude CLI spoof headers into shared constant - Tool deduper utility in open-sse handlers
This commit is contained in:
decolua
@@ -0,0 +1,190 @@
|
||||
// Inline stdio<->SSE bridge for MCP. Spawns one child per plugin on demand,
|
||||
// broadcasts JSON-RPC frames over SSE, accepts client messages via HTTP POST.
|
||||
|
||||
const { spawn } = require("child_process");
|
||||
const fs = require("fs");
|
||||
const path = require("path");
|
||||
const crypto = require("crypto");
|
||||
const { LOCAL_STDIO_PLUGINS } = require("@/shared/constants/coworkPlugins");
|
||||
const { DATA_DIR } = require("@/lib/dataDir");
|
||||
|
||||
const CUSTOM_FILE = path.join(DATA_DIR, "mcp", "customPlugins.json");
|
||||
|
||||
const G_KEY = "__9routerMcpBridges";
|
||||
const MAX_TEXT_CHARS = 50000;
|
||||
const COLLAPSE_THRESHOLD = 30;
|
||||
const COLLAPSE_KEEP_HEAD = 10;
|
||||
const COLLAPSE_KEEP_TAIL = 5;
|
||||
|
||||
// Drop noise nodes, collapse repeated siblings, hard-truncate. Preserve [ref=eXX].
|
||||
function smartFilterText(text) {
|
||||
if (typeof text !== "string" || text.length < 2000) return text;
|
||||
let out = text;
|
||||
out = out.replace(/^\s*-\s*generic:?\s*$/gm, "");
|
||||
out = out.replace(/^\s*-\s*text:\s*""\s*$/gm, "");
|
||||
out = collapseRepeated(out);
|
||||
if (out.length > MAX_TEXT_CHARS) {
|
||||
const head = out.slice(0, MAX_TEXT_CHARS - 300);
|
||||
out = `${head}\n\n... [truncated ${text.length - head.length} chars by 9router bridge. Page is large; ask user to scroll/navigate to a specific section, or click an element with the refs shown above]`;
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
// Group consecutive lines sharing the same leading indent + role prefix; collapse if >= COLLAPSE_THRESHOLD.
|
||||
function collapseRepeated(text) {
|
||||
const lines = text.split("\n");
|
||||
const out = [];
|
||||
let i = 0;
|
||||
while (i < lines.length) {
|
||||
const line = lines[i];
|
||||
const m = line.match(/^(\s*)-\s*([a-zA-Z]+)\b/);
|
||||
if (!m) { out.push(line); i++; continue; }
|
||||
const indent = m[1];
|
||||
const role = m[2];
|
||||
let j = i;
|
||||
while (j < lines.length) {
|
||||
const ln = lines[j];
|
||||
const mm = ln.match(/^(\s*)-\s*([a-zA-Z]+)\b/);
|
||||
if (mm && mm[1] === indent && mm[2] === role) { j++; continue; }
|
||||
if (ln.startsWith(`${indent} `) || ln.startsWith(`${indent}\t`)) { j++; continue; }
|
||||
break;
|
||||
}
|
||||
const groupLen = j - i;
|
||||
if (groupLen >= COLLAPSE_THRESHOLD) {
|
||||
const headEnd = findNthSiblingEnd(lines, i, indent, role, COLLAPSE_KEEP_HEAD);
|
||||
const tailStart = findLastNSiblingStart(lines, j, indent, role, COLLAPSE_KEEP_TAIL);
|
||||
for (let k = i; k < headEnd; k++) out.push(lines[k]);
|
||||
out.push(`${indent}... [${groupLen - COLLAPSE_KEEP_HEAD - COLLAPSE_KEEP_TAIL} similar "${role}" items omitted by 9router bridge]`);
|
||||
for (let k = tailStart; k < j; k++) out.push(lines[k]);
|
||||
} else {
|
||||
for (let k = i; k < j; k++) out.push(lines[k]);
|
||||
}
|
||||
i = j;
|
||||
}
|
||||
return out.join("\n");
|
||||
}
|
||||
|
||||
function findNthSiblingEnd(lines, start, indent, role, n) {
|
||||
let count = 0;
|
||||
for (let k = start; k < lines.length; k++) {
|
||||
const mm = lines[k].match(/^(\s*)-\s*([a-zA-Z]+)\b/);
|
||||
if (mm && mm[1] === indent && mm[2] === role) {
|
||||
count++;
|
||||
if (count > n) return k;
|
||||
}
|
||||
}
|
||||
return lines.length;
|
||||
}
|
||||
|
||||
function findLastNSiblingStart(lines, end, indent, role, n) {
|
||||
const positions = [];
|
||||
for (let k = 0; k < end; k++) {
|
||||
const mm = lines[k].match(/^(\s*)-\s*([a-zA-Z]+)\b/);
|
||||
if (mm && mm[1] === indent && mm[2] === role) positions.push(k);
|
||||
}
|
||||
return positions.length > n ? positions[positions.length - n] : end;
|
||||
}
|
||||
|
||||
// Apply filter to JSON-RPC tool/result content text blocks only.
|
||||
function filterFrame(line) {
|
||||
try {
|
||||
const msg = JSON.parse(line);
|
||||
const content = msg?.result?.content;
|
||||
if (!Array.isArray(content)) return line;
|
||||
let mutated = false;
|
||||
for (const item of content) {
|
||||
if (item?.type === "text" && typeof item.text === "string") {
|
||||
const filtered = smartFilterText(item.text);
|
||||
if (filtered !== item.text) { item.text = filtered; mutated = true; }
|
||||
}
|
||||
}
|
||||
return mutated ? JSON.stringify(msg) : line;
|
||||
} catch { return line; }
|
||||
}
|
||||
const getStore = () => {
|
||||
if (!globalThis[G_KEY]) globalThis[G_KEY] = new Map();
|
||||
return globalThis[G_KEY];
|
||||
};
|
||||
|
||||
const getCustomStore = () => {
|
||||
if (!globalThis.__9routerCustomPlugins) globalThis.__9routerCustomPlugins = new Map();
|
||||
return globalThis.__9routerCustomPlugins;
|
||||
};
|
||||
|
||||
function registerCustomPlugin(def) {
|
||||
getCustomStore().set(def.name, def);
|
||||
}
|
||||
|
||||
function findPlugin(name) {
|
||||
const fromMem = getCustomStore().get(name) || LOCAL_STDIO_PLUGINS.find((p) => p.name === name);
|
||||
if (fromMem) return fromMem;
|
||||
// Lazy-load custom plugins from disk (survives app restart).
|
||||
try {
|
||||
const list = JSON.parse(fs.readFileSync(CUSTOM_FILE, "utf-8"));
|
||||
const def = Array.isArray(list) ? list.find((p) => p.name === name && p.command) : null;
|
||||
if (def) { getCustomStore().set(def.name, def); return def; }
|
||||
} catch { /* file missing or invalid */ }
|
||||
return null;
|
||||
}
|
||||
|
||||
function getOrSpawn(name) {
|
||||
const store = getStore();
|
||||
let entry = store.get(name);
|
||||
if (entry?.proc && !entry.proc.killed && entry.proc.exitCode === null) return entry;
|
||||
|
||||
const plugin = findPlugin(name);
|
||||
if (!plugin) throw new Error(`Unknown local plugin: ${name}`);
|
||||
|
||||
const proc = spawn(plugin.command, plugin.args, { stdio: ["pipe", "pipe", "pipe"], env: process.env });
|
||||
entry = { proc, sessions: new Map(), buffer: "" };
|
||||
store.set(name, entry);
|
||||
|
||||
// Parse newline-delimited JSON-RPC from child stdout, broadcast to all sessions.
|
||||
proc.stdout.on("data", (chunk) => {
|
||||
entry.buffer += chunk.toString("utf8");
|
||||
let idx;
|
||||
while ((idx = entry.buffer.indexOf("\n")) >= 0) {
|
||||
const raw = entry.buffer.slice(0, idx).trim();
|
||||
entry.buffer = entry.buffer.slice(idx + 1);
|
||||
if (!raw) continue;
|
||||
const line = filterFrame(raw);
|
||||
for (const send of entry.sessions.values()) {
|
||||
try { send(`event: message\ndata: ${line}\n\n`); } catch { /* ignore broken pipe */ }
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
proc.stderr.on("data", (d) => console.log(`[mcp:${name}]`, d.toString().trim()));
|
||||
proc.on("exit", (code) => {
|
||||
console.log(`[mcp:${name}] exited`, code);
|
||||
store.delete(name);
|
||||
});
|
||||
|
||||
return entry;
|
||||
}
|
||||
|
||||
function registerSession(name, sendFn) {
|
||||
const entry = getOrSpawn(name);
|
||||
const sid = crypto.randomUUID();
|
||||
entry.sessions.set(sid, sendFn);
|
||||
return sid;
|
||||
}
|
||||
|
||||
function unregisterSession(name, sid) {
|
||||
const entry = getStore().get(name);
|
||||
if (!entry) return;
|
||||
entry.sessions.delete(sid);
|
||||
}
|
||||
|
||||
function sendToChild(name, jsonRpc) {
|
||||
const entry = getStore().get(name);
|
||||
if (!entry?.proc?.stdin?.writable) throw new Error(`Bridge not running: ${name}`);
|
||||
entry.proc.stdin.write(`${JSON.stringify(jsonRpc)}\n`);
|
||||
}
|
||||
|
||||
function isRunning(name) {
|
||||
const entry = getStore().get(name);
|
||||
return !!(entry?.proc && !entry.proc.killed && entry.proc.exitCode === null);
|
||||
}
|
||||
|
||||
module.exports = { getOrSpawn, registerSession, unregisterSession, sendToChild, isRunning, findPlugin, registerCustomPlugin };
|
||||
Reference in New Issue
Block a user