vite-plus 0.1.2 → 0.1.3

package/dist/global/mcp.js

@@ -1,7 +1,6 @@
 import { dirname, join } from "node:path";
 import { existsSync, readFileSync, readdirSync } from "node:fs";
 import { fileURLToPath } from "node:url";
-
 //#region src/mcp/bin.ts
 const PROTOCOL_VERSION = "2024-11-05";
 const TOOLS = [
@@ -41,69 +40,50 @@ const TOOLS = [
 function findPackageRoot(from) {
 	let dir = from;
 	while (true) {
-		if (existsSync(join(dir, "package.json"))) {
-			return dir;
-		}
+		if (existsSync(join(dir, "package.json"))) return dir;
 		const parent = dirname(dir);
-		if (parent === dir) {
-			break;
-		}
+		if (parent === dir) break;
 		dir = parent;
 	}
 	throw new Error("Could not find package.json from: " + from);
 }
 function readPackageVersion(pkgRoot) {
 	const raw = readFileSync(join(pkgRoot, "package.json"), "utf8");
-	const pkg = JSON.parse(raw);
-	return pkg.version ?? "0.0.0";
+	return JSON.parse(raw).version ?? "0.0.0";
 }
 function resolveDocsDir(pkgRoot) {
 	const bundledDocsDir = join(pkgRoot, "skills", "vite-plus", "docs");
-	if (existsSync(bundledDocsDir)) {
-		return bundledDocsDir;
-	}
+	if (existsSync(bundledDocsDir)) return bundledDocsDir;
 	const workspaceDocsDir = join(pkgRoot, "..", "..", "docs");
-	if (existsSync(workspaceDocsDir)) {
-		return workspaceDocsDir;
-	}
+	if (existsSync(workspaceDocsDir)) return workspaceDocsDir;
 	throw new Error(`Vite+ docs directory not found. Expected bundled docs at: ${bundledDocsDir}`);
 }
 function collectMarkdownFiles(rootDir, relativeDir = "") {
-	const currentDir = join(rootDir, relativeDir);
-	const entries = readdirSync(currentDir, { withFileTypes: true });
+	const entries = readdirSync(join(rootDir, relativeDir), { withFileTypes: true });
 	const files = [];
 	for (const entry of entries) {
 		const relPath = relativeDir ? `${relativeDir}/${entry.name}` : entry.name;
 		if (entry.isDirectory()) {
-			if (entry.name === "node_modules") {
-				continue;
-			}
+			if (entry.name === "node_modules") continue;
 			files.push(...collectMarkdownFiles(rootDir, relPath));
 			continue;
 		}
-		if (entry.isFile() && entry.name.endsWith(".md")) {
-			files.push(relPath);
-		}
+		if (entry.isFile() && entry.name.endsWith(".md")) files.push(relPath);
 	}
 	files.sort();
 	return files;
 }
 function normalizeDocId(value) {
-	const normalized = value.trim().replace(/\\/g, "/").replace(/^docs\//, "").replace(/\.md$/, "").replace(/\/index$/, "").replace(/^\/+|\/+$/g, "");
-	return normalized || "index";
+	return value.trim().replace(/\\/g, "/").replace(/^docs\//, "").replace(/\.md$/, "").replace(/\/index$/, "").replace(/^\/+|\/+$/g, "") || "index";
 }
 function createSlug(relativePath) {
 	const withoutExt = relativePath.replace(/\.md$/, "");
-	if (withoutExt === "index") {
-		return "index";
-	}
-	if (withoutExt.endsWith("/index")) {
-		return withoutExt.slice(0, -"/index".length);
-	}
+	if (withoutExt === "index") return "index";
+	if (withoutExt.endsWith("/index")) return withoutExt.slice(0, -6);
 	return withoutExt;
 }
 function buildAliases(page, basenameCounts) {
-	const aliases = new Set();
+	const aliases = /* @__PURE__ */ new Set();
 	const relativeNoExt = page.relativePath.replace(/\.md$/, "");
 	const flatSlug = page.slug.replaceAll("/", "-");
 	const baseName = page.slug.split("/").at(-1);
@@ -123,45 +103,34 @@ function buildAliases(page, basenameCounts) {
 }
 function loadDocs(pkgRoot) {
 	const docsDir = resolveDocsDir(pkgRoot);
-	const files = collectMarkdownFiles(docsDir);
-	const pages = files.map((relativePath) => {
-		const raw = readFileSync(join(docsDir, relativePath), "utf8");
-		const content = raw.replace(/^---\n[\s\S]*?\n---\n/, "");
+	const pages = collectMarkdownFiles(docsDir).map((relativePath) => {
+		const content = readFileSync(join(docsDir, relativePath), "utf8").replace(/^---\n[\s\S]*?\n---\n/, "");
 		const titleMatch = content.match(/^#\s+(.+)/m);
 		const slug = createSlug(relativePath);
-		const title = titleMatch ? titleMatch[1].trim() : slug;
 		return {
 			slug,
 			relativePath,
-			title,
+			title: titleMatch ? titleMatch[1].trim() : slug,
 			content
 		};
 	});
-	const basenameCounts = new Map();
+	const basenameCounts = /* @__PURE__ */ new Map();
 	for (const page of pages) {
 		const baseName = page.slug.split("/").at(-1);
-		if (!baseName) {
-			continue;
-		}
+		if (!baseName) continue;
 		basenameCounts.set(baseName, (basenameCounts.get(baseName) ?? 0) + 1);
 	}
-	const aliasCounts = new Map();
-	const aliasSources = new Map();
+	const aliasCounts = /* @__PURE__ */ new Map();
+	const aliasSources = /* @__PURE__ */ new Map();
 	for (const page of pages) {
 		const aliases = [...new Set(buildAliases(page, basenameCounts).map(normalizeDocId))];
 		aliasSources.set(page, aliases);
-		for (const alias of aliases) {
-			aliasCounts.set(alias, (aliasCounts.get(alias) ?? 0) + 1);
-		}
+		for (const alias of aliases) aliasCounts.set(alias, (aliasCounts.get(alias) ?? 0) + 1);
 	}
-	const byAlias = new Map();
-	for (const page of pages) {
-		for (const alias of aliasSources.get(page) ?? []) {
-			if ((aliasCounts.get(alias) ?? 0) !== 1) {
-				continue;
-			}
-			byAlias.set(alias, page);
-		}
+	const byAlias = /* @__PURE__ */ new Map();
+	for (const page of pages) for (const alias of aliasSources.get(page) ?? []) {
+		if ((aliasCounts.get(alias) ?? 0) !== 1) continue;
+		byAlias.set(alias, page);
 	}
 	return {
 		pages,
@@ -170,9 +139,7 @@ function loadDocs(pkgRoot) {
 }
 function searchDocs(pages, query) {
 	const terms = query.toLowerCase().split(/\s+/).filter((term) => term.length > 0);
-	if (terms.length === 0) {
-		return [];
-	}
+	if (terms.length === 0) return [];
 	const scored = [];
 	for (const page of pages) {
 		const titleLower = page.title.toLowerCase();
@@ -188,23 +155,17 @@ function searchDocs(pages, query) {
 			idx = 0;
 			while ((idx = contentLower.indexOf(term, idx)) !== -1) {
 				score += 1;
-				if (firstMatchIndex === -1) {
-					firstMatchIndex = idx;
-				}
+				if (firstMatchIndex === -1) firstMatchIndex = idx;
 				idx += term.length;
 			}
 		}
-		if (score === 0) {
-			continue;
-		}
+		if (score === 0) continue;
 		let snippet;
 		if (firstMatchIndex !== -1) {
 			const start = Math.max(0, firstMatchIndex - 80);
 			const end = Math.min(page.content.length, firstMatchIndex + 120);
 			snippet = (start > 0 ? "..." : "") + page.content.slice(start, end).trim() + (end < page.content.length ? "..." : "");
-		} else {
-			snippet = page.content.slice(0, 200).trim() + "...";
-		}
+		} else snippet = page.content.slice(0, 200).trim() + "...";
 		scored.push({
 			slug: page.slug,
 			title: page.title,
@@ -230,37 +191,29 @@ function makeErrorResponse(id, code, message) {
 }
 function handleRequest(index, serverVersion, req) {
 	const { method, id, params } = req;
-	if (method === "initialize") {
-		return {
-			jsonrpc: "2.0",
-			id: id ?? null,
-			result: {
-				protocolVersion: PROTOCOL_VERSION,
-				capabilities: { tools: {} },
-				serverInfo: {
-					name: "vite-plus",
-					version: serverVersion
-				}
+	if (method === "initialize") return {
+		jsonrpc: "2.0",
+		id: id ?? null,
+		result: {
+			protocolVersion: PROTOCOL_VERSION,
+			capabilities: { tools: {} },
+			serverInfo: {
+				name: "vite-plus",
+				version: serverVersion
 			}
-		};
-	}
-	if (method === "notifications/initialized" || method === "$/cancelRequest") {
-		return null;
-	}
-	if (method === "ping") {
-		return {
-			jsonrpc: "2.0",
-			id: id ?? null,
-			result: {}
-		};
-	}
-	if (method === "tools/list") {
-		return {
-			jsonrpc: "2.0",
-			id: id ?? null,
-			result: { tools: TOOLS }
-		};
-	}
+		}
+	};
+	if (method === "notifications/initialized" || method === "$/cancelRequest") return null;
+	if (method === "ping") return {
+		jsonrpc: "2.0",
+		id: id ?? null,
+		result: {}
+	};
+	if (method === "tools/list") return {
+		jsonrpc: "2.0",
+		id: id ?? null,
+		result: { tools: TOOLS }
+	};
 	if (method === "tools/call") {
 		const toolName = params?.name ?? "";
 		const toolArgs = params?.arguments ?? {};
@@ -281,33 +234,29 @@ function handleRequest(index, serverVersion, req) {
 		}
 		if (toolName === "get_page") {
 			const slug = toolArgs.slug;
-			if (typeof slug !== "string" || slug.trim().length === 0) {
-				return {
-					jsonrpc: "2.0",
-					id: id ?? null,
-					result: {
-						content: [{
-							type: "text",
-							text: "Missing required string argument: slug"
-						}],
-						isError: true
-					}
-				};
-			}
+			if (typeof slug !== "string" || slug.trim().length === 0) return {
+				jsonrpc: "2.0",
+				id: id ?? null,
+				result: {
+					content: [{
+						type: "text",
+						text: "Missing required string argument: slug"
+					}],
+					isError: true
+				}
+			};
 			const page = resolvePageBySlug(index, slug);
-			if (!page) {
-				return {
-					jsonrpc: "2.0",
-					id: id ?? null,
-					result: {
-						content: [{
-							type: "text",
-							text: `Page not found: ${slug}`
-						}],
-						isError: true
-					}
-				};
-			}
+			if (!page) return {
+				jsonrpc: "2.0",
+				id: id ?? null,
+				result: {
+					content: [{
+						type: "text",
+						text: `Page not found: ${slug}`
+					}],
+					isError: true
+				}
+			};
 			return {
 				jsonrpc: "2.0",
 				id: id ?? null,
@@ -319,19 +268,17 @@ function handleRequest(index, serverVersion, req) {
 		}
 		if (toolName === "search_docs") {
 			const query = toolArgs.query;
-			if (typeof query !== "string" || query.trim().length === 0) {
-				return {
-					jsonrpc: "2.0",
-					id: id ?? null,
-					result: {
-						content: [{
-							type: "text",
-							text: "Missing required string argument: query"
-						}],
-						isError: true
-					}
-				};
-			}
+			if (typeof query !== "string" || query.trim().length === 0) return {
+				jsonrpc: "2.0",
+				id: id ?? null,
+				result: {
+					content: [{
+						type: "text",
+						text: "Missing required string argument: query"
+					}],
+					isError: true
+				}
+			};
 			const results = searchDocs(index.pages, query);
 			return {
 				jsonrpc: "2.0",
@@ -342,14 +289,10 @@ function handleRequest(index, serverVersion, req) {
 				}] }
 			};
 		}
-		if (id === undefined) {
-			return null;
-		}
+		if (id === void 0) return null;
 		return makeErrorResponse(id, -32601, `Unknown tool: ${toolName}`);
 	}
-	if (id === undefined) {
-		return null;
-	}
+	if (id === void 0) return null;
 	return makeErrorResponse(id, -32601, `Unknown method: ${method}`);
 }
 function writeMessage(payload) {
@@ -360,15 +303,11 @@ function writeMessage(payload) {
 function findHeadersBoundary(buffer) {
 	const crlf = buffer.indexOf("\r\n\r\n");
 	const lf = buffer.indexOf("\n\n");
-	if (crlf === -1 && lf === -1) {
-		return null;
-	}
-	if (crlf !== -1 && (lf === -1 || crlf < lf)) {
-		return {
-			end: crlf,
-			separatorLength: 4
-		};
-	}
+	if (crlf === -1 && lf === -1) return null;
+	if (crlf !== -1 && (lf === -1 || crlf < lf)) return {
+		end: crlf,
+		separatorLength: 4
+	};
 	return {
 		end: lf,
 		separatorLength: 2
@@ -380,9 +319,7 @@ function parseContentLength(rawHeaders) {
 		const match = line.match(/^content-length:\s*(\d+)\s*$/i);
 		if (match) {
 			const value = Number.parseInt(match[1], 10);
-			if (Number.isSafeInteger(value) && value >= 0) {
-				return value;
-			}
+			if (Number.isSafeInteger(value) && value >= 0) return value;
 			return null;
 		}
 	}
@@ -400,11 +337,8 @@ function startStdioServer(index, serverVersion) {
 		buffer = Buffer.concat([buffer, chunk]);
 		while (true) {
 			const boundary = findHeadersBoundary(buffer);
-			if (!boundary) {
-				break;
-			}
-			const headerText = buffer.subarray(0, boundary.end).toString("utf8");
-			const contentLength = parseContentLength(headerText);
+			if (!boundary) break;
+			const contentLength = parseContentLength(buffer.subarray(0, boundary.end).toString("utf8"));
 			const bodyStart = boundary.end + boundary.separatorLength;
 			if (contentLength === null) {
 				writeMessage(makeErrorResponse(null, -32600, "Missing or invalid Content-Length header"));
@@ -412,9 +346,7 @@ function startStdioServer(index, serverVersion) {
 				continue;
 			}
 			const bodyEnd = bodyStart + contentLength;
-			if (buffer.length < bodyEnd) {
-				break;
-			}
+			if (buffer.length < bodyEnd) break;
 			const body = buffer.subarray(bodyStart, bodyEnd).toString("utf8");
 			buffer = buffer.subarray(bodyEnd);
 			let request;
@@ -429,20 +361,16 @@ function startStdioServer(index, serverVersion) {
 				continue;
 			}
 			const response = handleRequest(index, serverVersion, request);
-			if (response !== null) {
-				writeMessage(response);
-			}
+			if (response !== null) writeMessage(response);
 		}
 	});
 }
 try {
 	const packageRoot = findPackageRoot(dirname(fileURLToPath(import.meta.url)));
 	const serverVersion = readPackageVersion(packageRoot);
-	const docs = loadDocs(packageRoot);
-	startStdioServer(docs, serverVersion);
+	startStdioServer(loadDocs(packageRoot), serverVersion);
 } catch (err) {
 	process.stderr.write(`[vite-plus mcp] Failed to start: ${err instanceof Error ? err.message : String(err)}\n`);
 	process.exit(1);
 }
-
-//#endregion
+//#endregion