81 lines
2.4 KiB
TypeScript
81 lines
2.4 KiB
TypeScript
 
|
import { NextResponse } from "next/server";
|
||
|
|
import { authenticateBearer, UnauthorizedError } from "@/lib/auth/jwt";
|
||
|
|
import { userContextFromClaims } from "@/lib/mcp/context";
|
||
|
|
import { dispatchMcpMessage } from "@/lib/mcp/server";
|
||
|
|
|
||
|
|
export const runtime = "nodejs";
|
||
|
|
export const dynamic = "force-dynamic";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* MCP streamable-HTTP endpoint.
|
||
|
|
*
|
||
|
|
* Auth: Bearer token (Authentik-issued JWT). Unauthed requests get 401 with
|
||
|
|
* a WWW-Authenticate header pointing at our RFC 9728 resource metadata
|
||
|
|
* so MCP clients can discover the authorization server.
|
||
|
|
*
|
||
|
|
* Body: JSON-RPC 2.0 message (request or notification).
|
||
|
|
*
|
||
|
|
* Reply: For requests, the JSON-RPC response in the body with
|
||
|
|
* `Content-Type: application/json`.
|
||
|
|
* For notifications, HTTP 202 with empty body.
|
||
|
|
*/
|
||
|
|
|
||
|
|
export async function POST(req: Request) {
|
||
|
|
// ---- auth ----
|
||
|
|
let claims;
|
||
|
|
try {
|
||
|
|
claims = await authenticateBearer(req.headers.get("authorization"));
|
||
|
|
} catch (e) {
|
||
|
|
if (e instanceof UnauthorizedError) {
|
||
|
|
return new NextResponse(JSON.stringify({ error: e.reason }), {
|
||
|
|
status: 401,
|
||
|
|
headers: {
|
||
|
|
"WWW-Authenticate": e.wwwAuthenticate,
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
},
|
||
|
|
});
|
||
|
|
}
|
||
|
|
throw e;
|
||
|
|
}
|
||
|
|
|
||
|
|
// ---- parse body ----
|
||
|
|
let body: unknown;
|
||
|
|
try {
|
||
|
|
body = await req.json();
|
||
|
|
} catch {
|
||
|
|
return NextResponse.json(
|
||
|
|
{ jsonrpc: "2.0", id: null, error: { code: -32700, message: "parse error" } },
|
||
|
|
{ status: 400 },
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// ---- resolve user, dispatch ----
|
||
|
|
const ctx = await userContextFromClaims(claims);
|
||
|
|
|
||
|
|
// MCP supports batched requests (array) and single. Handle both.
|
||
|
|
if (Array.isArray(body)) {
|
||
|
|
const responses = await Promise.all(body.map((m) => dispatchMcpMessage(m, ctx)));
|
||
|
|
const filtered = responses.filter((r) => r !== null);
|
||
|
|
if (filtered.length === 0) {
|
||
|
|
return new NextResponse(null, { status: 202 });
|
||
|
|
}
|
||
|
|
return NextResponse.json(filtered, { status: 200 });
|
||
|
|
}
|
||
|
|
|
||
|
|
const response = await dispatchMcpMessage(body, ctx);
|
||
|
|
if (response === null) {
|
||
|
|
// Notification — no body expected.
|
||
|
|
return new NextResponse(null, { status: 202 });
|
||
|
|
}
|
||
|
|
return NextResponse.json(response, { status: 200 });
|
||
|
|
}
|
||
|
|
|
||
|
|
// MCP clients sometimes probe with GET (for SSE). We don't support
|
||
|
|
// server-initiated events in Phase 1 — return 405 with a discoverable header.
|
||
|
|
export function GET() {
|
||
|
|
return new NextResponse(null, {
|
||
|
|
status: 405,
|
||
|
|
headers: { Allow: "POST" },
|
||
|
|
});
|
||
|
|
}
|