File size: 3,225 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | "use server";
import { NextResponse } from "next/server";
const TIMEOUT_MS = 8000;
// Probe MCP server: initialize + tools/list. No auth header — works for authless servers.
// OAuth servers return 401, signal client to skip tool listing.
async function probeMcp(url) {
const headers = {
"Content-Type": "application/json",
"Accept": "application/json, text/event-stream",
"MCP-Protocol-Version": "2025-06-18",
};
const ac = new AbortController();
const timer = setTimeout(() => ac.abort(), TIMEOUT_MS);
try {
// Step 1: initialize
const initRes = await fetch(url, {
method: "POST",
headers,
body: JSON.stringify({
jsonrpc: "2.0", id: 1, method: "initialize",
params: { protocolVersion: "2025-06-18", capabilities: {}, clientInfo: { name: "9router", version: "1" } },
}),
signal: ac.signal,
});
if (initRes.status === 401 || initRes.status === 403) {
return { requiresAuth: true, tools: [] };
}
if (!initRes.ok) {
return { error: `init ${initRes.status}`, tools: [] };
}
const sessionId = initRes.headers.get("mcp-session-id") || "";
await initRes.text().catch(() => {});
const listHeaders = { ...headers };
if (sessionId) listHeaders["mcp-session-id"] = sessionId;
// Step 2: notifications/initialized (required by spec before tools/list)
await fetch(url, {
method: "POST",
headers: listHeaders,
body: JSON.stringify({ jsonrpc: "2.0", method: "notifications/initialized", params: {} }),
signal: ac.signal,
}).catch(() => {});
// Step 3: tools/list
const listRes = await fetch(url, {
method: "POST",
headers: listHeaders,
body: JSON.stringify({ jsonrpc: "2.0", id: 2, method: "tools/list" }),
signal: ac.signal,
});
if (listRes.status === 401 || listRes.status === 403) {
return { requiresAuth: true, tools: [] };
}
const ct = listRes.headers.get("content-type") || "";
let parsed;
if (ct.includes("text/event-stream")) {
// Parse SSE: each "data: {...}" line is a JSON-RPC message
const text = await listRes.text();
const dataLines = text.split("\n").filter((l) => l.startsWith("data:"));
for (const line of dataLines) {
try {
const obj = JSON.parse(line.replace(/^data:\s*/, ""));
if (obj?.id === 2 && obj.result) { parsed = obj; break; }
} catch { /* skip */ }
}
} else {
parsed = await listRes.json().catch(() => null);
}
const tools = parsed?.result?.tools || [];
return {
tools: tools.map((t) => ({ name: t.name, description: t.description || "" })),
};
} catch (e) {
return { error: e.name === "AbortError" ? "timeout" : e.message, tools: [] };
} finally {
clearTimeout(timer);
}
}
export async function POST(request) {
try {
const { url } = await request.json();
if (!url || typeof url !== "string") {
return NextResponse.json({ error: "url required" }, { status: 400 });
}
const result = await probeMcp(url);
return NextResponse.json(result);
} catch (e) {
return NextResponse.json({ error: e.message, tools: [] }, { status: 500 });
}
}
|