modelcontextprotocol/typescript-sdk
Watch
1
0
Fork
You've already forked typescript-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Build a router to chain middleware better

Browse source
This commit is contained in:
Justin Spahr-Summers 2025-02-17 13:51:37 +00:00
commit a776f5f94a
No known key found for this signature in database
1 changed files with 23 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

39
src/server/auth/handlers/clientRegistration.ts
View file

@ -1,7 +1,7 @@
import { Request, Response } from "express";
import express, { RequestHandler } from "express";
import { OAuthClientInformationFull, OAuthClientMetadataSchema, OAuthClientRegistrationError } from "../../../shared/auth.js";
import crypto from 'node:crypto';
import bodyParser from 'body-parser';
import cors from 'cors';
import { OAuthRegisteredClientsStore } from "../clients.js";
export type ClientRegistrationHandlerOptions = {
@ -20,11 +20,18 @@ export type ClientRegistrationHandlerOptions = {
const DEFAULT_CLIENT_SECRET_EXPIRY_SECONDS = 30 * 24 * 60 * 60; // 30 days
export function clientRegistrationHandler({ store, clientSecretExpirySeconds = DEFAULT_CLIENT_SECRET_EXPIRY_SECONDS }: ClientRegistrationHandlerOptions) {
export function clientRegistrationHandler({ store, clientSecretExpirySeconds = DEFAULT_CLIENT_SECRET_EXPIRY_SECONDS }: ClientRegistrationHandlerOptions): RequestHandler {
if (!store.registerClient) {
throw new Error("Client registration store does not support registering clients");
}
// Nested router so we can configure middleware
const router = express.Router();
router.use(express.json());
// Configure CORS to allow any origin, to make accessible to web-based MCP clients
router.use(cors());
async function register(requestBody: unknown): Promise<OAuthClientInformationFull | OAuthClientRegistrationError> {
let clientMetadata;
try {
@ -53,18 +60,18 @@ export function clientRegistrationHandler({ store, clientSecretExpirySeconds = D
}
// Actual request handler
return (req: Request, res: Response) => bodyParser.json()(req, res, (err) => {
if (err === undefined) {
register(req.body).then((result) => {
if ("error" in result) {
res.status(400).json(result);
} else {
res.status(201).json(result);
}
}, (error) => {
console.error("Uncaught error in client registration handler:", error);
res.status(500).end("Internal Server Error");
});
}
router.post("/", (req, res) => {
register(req.body).then((result) => {
if ("error" in result) {
res.status(400).json(result);
} else {
res.status(201).json(result);
}
}, (error) => {
console.error("Uncaught error in client registration handler:", error);
res.status(500).end("Internal Server Error");
});
});
return router;
}