yujunjun/context7
1
0
Fork 0
mirror of https://github.com/upstash/context7.git synced 2025-06-26 23:50:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace url.parse with URL API for session ID extraction and validate transports

Browse Source
This commit is contained in:
Abdusshh 2025-06-11 14:58:26 +03:00
parent 2e6c9a6c6b
commit 4f04e02851
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/index.ts
View File

@ -9,7 +9,6 @@ import { SearchResponse } from "./lib/types.js";
import { createServer } from "http";
import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
import { parse } from "url";
import { Command } from "commander";
const DEFAULT_MINIMUM_TOKENS = 10000;
@ -26,6 +25,15 @@ const cliOptions = program.opts<{
port: string;
}>();
// Validate transport option
const allowedTransports = ["stdio", "http", "sse"];
if (!allowedTransports.includes(cliOptions.transport)) {
console.error(
`Invalid --transport value: '${cliOptions.transport}'. Must be one of: stdio, http, sse.`
);
process.exit(1);
}
// Transport configuration
const TRANSPORT_TYPE = (cliOptions.transport || "stdio") as "stdio" | "http" | "sse";
@ -217,8 +225,9 @@ async function main() {
await requestServer.connect(sseTransport);
} else if (url === "/messages" && req.method === "POST") {
// Get session ID from query parameters
const parsedUrl = parse(req.url || "", true);
const sessionId = parsedUrl.query.sessionId as string;
const sessionId =
new URL(req.url || "", `http://${req.headers.host}`).searchParams.get("sessionId") ??
"";
if (!sessionId) {
res.writeHead(400);