yujunjun/context7
1
0
Fork 0
mirror of https://github.com/upstash/context7.git synced 2025-07-03 07:04:25 +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 { createServer } from "http";
import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js"; import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js"; import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
import { parse } from "url";
import { Command } from "commander"; import { Command } from "commander";
const DEFAULT_MINIMUM_TOKENS = 10000; const DEFAULT_MINIMUM_TOKENS = 10000;
@ -26,6 +25,15 @@ const cliOptions = program.opts<{
port: string; 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 // Transport configuration
const TRANSPORT_TYPE = (cliOptions.transport || "stdio") as "stdio" | "http" | "sse"; const TRANSPORT_TYPE = (cliOptions.transport || "stdio") as "stdio" | "http" | "sse";
@ -217,8 +225,9 @@ async function main() {
await requestServer.connect(sseTransport); await requestServer.connect(sseTransport);
} else if (url === "/messages" && req.method === "POST") { } else if (url === "/messages" && req.method === "POST") {
// Get session ID from query parameters // Get session ID from query parameters
const parsedUrl = parse(req.url || "", true); const sessionId =
const sessionId = parsedUrl.query.sessionId as string; new URL(req.url || "", `http://${req.headers.host}`).searchParams.get("sessionId") ??
"";
if (!sessionId) { if (!sessionId) {
res.writeHead(400); res.writeHead(400);