mn-docs-mcp 0.1.7 → 0.2.1

package/mcp/cli.mjs

@@ -1,11 +1,16 @@
 #!/usr/bin/env node
 
+import fs from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+
 const args = process.argv.slice(2);
 
 let mode = 'stdio';
 let port;
 let prebuild = false;
-let silent = true;
+let silent = false;
+let root;
 
 for (let i = 0; i < args.length; i += 1) {
 	const arg = args[i];
@@ -25,15 +30,28 @@ for (let i = 0; i < args.length; i += 1) {
 		silent = false;
 		continue;
 	}
+	if (arg === '--silent') {
+		silent = true;
+		continue;
+	}
 	if (arg === '--port' && args[i + 1]) {
 		port = args[i + 1];
 		i += 1;
 		continue;
 	}
+	if (arg === '--root' && args[i + 1]) {
+		root = args[i + 1];
+		i += 1;
+		continue;
+	}
 	if (arg?.startsWith('--port=')) {
 		port = arg.split('=')[1];
 		continue;
 	}
+	if (arg?.startsWith('--root=')) {
+		root = arg.split('=')[1];
+		continue;
+	}
 	if (arg === '--help' || arg === '-h') {
 		process.stderr.write(
 			[
@@ -41,7 +59,9 @@ for (let i = 0; i < args.length; i += 1) {
 				'  --http            启动HTTPStream模式(默认stdio)',
 				'  --port <port>     HTTP端口(默认8788)',
 				'  --prebuild        启动后后台预构建索引',
-				'  --verbose         输出日志(默认静默)',
+				'  --verbose         输出日志(覆盖--silent)',
+				'  --silent          关闭开屏与日志',
+				'  --root <path>     指定文档仓库根目录',
 				'  --stdio           显式使用stdio模式',
 			].join('\n') + '\n'
 		);
@@ -49,13 +69,29 @@ for (let i = 0; i < args.length; i += 1) {
 	}
 }
 
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const packagePath = path.join(__dirname, '../package.json');
+let version = '0.0.0';
+try {
+	const raw = fs.readFileSync(packagePath, 'utf-8');
+	version = JSON.parse(raw).version || version;
+} catch {}
+
 if (mode === 'stdio') {
 	process.env.MCP_STDIO = '1';
 	process.env.MCP_SILENT = silent ? '1' : '0';
+	process.env.MN_DOCS_VERSION = version;
+	process.env.MN_DOCS_MODE = 'stdio';
+	process.env.MN_DOCS_ROOT = root || process.env.MN_DOCS_ROOT || process.cwd();
 	if (prebuild) process.env.MCP_PREBUILD = '1';
 	await import('./server.mjs');
 } else {
 	if (port) process.env.MCP_HTTP_PORT = String(port);
+	process.env.MN_DOCS_VERSION = version;
+	process.env.MN_DOCS_MODE = 'http';
+	if (port) process.env.MN_DOCS_PORT = String(port);
+	process.env.MN_DOCS_ROOT = root || process.env.MN_DOCS_ROOT || process.cwd();
 	if (prebuild) process.env.MCP_PREBUILD = '1';
 	await import('./server-http.mjs');
 }

package/mcp/lib.mjs

@@ -1,4 +1,5 @@
 import fs from 'node:fs/promises';
+import fsSync from 'node:fs';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
 import matter from 'gray-matter';
@@ -9,7 +10,28 @@ import dotenv from 'dotenv';
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
-const ROOT_DIR = path.resolve(__dirname, '../..');
+const DEFAULT_ROOT = path.resolve(__dirname, '..');
+
+function resolveRootDir() {
+	const envRoot = (process.env.MN_DOCS_ROOT || '').trim();
+	if (envRoot) return envRoot;
+	const cwd = process.cwd();
+	const docsFromCwd = path.join(cwd, 'src', 'content', 'docs');
+	try {
+		if (fsSyncExists(docsFromCwd)) return cwd;
+	} catch {}
+	return DEFAULT_ROOT;
+}
+
+function fsSyncExists(p) {
+	try {
+		return fsSync.statSync(p).isDirectory();
+	} catch {
+		return false;
+	}
+}
+
+const ROOT_DIR = resolveRootDir();
 const DOCS_DIR = path.join(ROOT_DIR, 'src', 'content', 'docs');
 const MCP_DIR = path.join(ROOT_DIR, '.mcp');
 const INDEX_PATH = path.join(MCP_DIR, 'index.json');

package/mcp/server-http.mjs

@@ -4,6 +4,42 @@ import { buildIndex, getPaths, isIndexStale, loadIndex, searchDocs } from './lib
 
 const TOOL_NAME = 'search_docs';
 const PORT = Number(process.env.MCP_HTTP_PORT || 8788);
+const IS_SILENT = process.env.MCP_SILENT === '1';
+const NO_COLOR = process.env.MCP_NO_COLOR === '1';
+
+function color(text, code) {
+	if (NO_COLOR) return text;
+	return `\x1b[${code}m${text}\x1b[0m`;
+}
+
+function stripAnsi(text) {
+	return text.replace(/\x1b\[[0-9;]*m/g, '');
+}
+
+function padLine(line, width) {
+	const len = stripAnsi(line).length;
+	const padding = width - len;
+	return line + (padding > 0 ? ' '.repeat(padding) : '');
+}
+
+function renderSplash() {
+	if (IS_SILENT) return;
+	const version = process.env.MN_DOCS_VERSION || '0.0.0';
+	const mode = process.env.MN_DOCS_MODE || 'http';
+	const port = process.env.MN_DOCS_PORT || String(PORT);
+	const contentLines = [
+		color(`mn-docs-mcp v${version}`, '38;5;45'),
+		color(`模式: ${mode}  端口: ${port}`, '38;5;39'),
+	];
+	const maxWidth = Math.max(...contentLines.map((line) => stripAnsi(line).length)) + 4;
+	const top = color('╭' + '─'.repeat(maxWidth) + '╮', '38;5;45');
+	const bottom = color('╰' + '─'.repeat(maxWidth) + '╯', '38;5;45');
+	const body = contentLines.map((line) => {
+		const padded = padLine(line, maxWidth - 4);
+		return color('│', '38;5;45') + '  ' + padded + '  ' + color('│', '38;5;45');
+	});
+	process.stdout.write([top, ...body, bottom].join('\n') + '\n');
+}
 
 async function ensureIndex() {
 	const { INDEX_PATH } = getPaths();
@@ -67,8 +103,7 @@ await server.start({
 	},
 });
 
-// 不阻塞握手，启动后再初始化索引
-// 默认不自动构建，避免阻塞握手；仅在首次查询时构建
-if (process.env.MCP_PREBUILD === '1') {
-	setTimeout(() => initIndexInBackground(), 1000);
-}
+renderSplash();
+
+// 默认自动构建，异步启动避免阻塞握手
+setTimeout(() => initIndexInBackground(), 0);

package/mcp/server.mjs

@@ -4,12 +4,46 @@ import { buildIndex, getPaths, isIndexStale, loadIndex, searchDocs } from './lib
 
 const TOOL_NAME = 'search_docs';
 const IS_SILENT = process.env.MCP_SILENT === '1';
+const NO_COLOR = process.env.MCP_NO_COLOR === '1';
 
 function logError(...args) {
 	if (IS_SILENT) return;
 	console.error(...args);
 }
 
+function color(text, code) {
+	if (NO_COLOR) return text;
+	return `\x1b[${code}m${text}\x1b[0m`;
+}
+
+function stripAnsi(text) {
+	return text.replace(/\x1b\[[0-9;]*m/g, '');
+}
+
+function padLine(line, width) {
+	const len = stripAnsi(line).length;
+	const padding = width - len;
+	return line + (padding > 0 ? ' '.repeat(padding) : '');
+}
+
+function renderSplash() {
+	if (IS_SILENT) return;
+	const version = process.env.MN_DOCS_VERSION || '0.0.0';
+	const mode = process.env.MN_DOCS_MODE || 'stdio';
+	const contentLines = [
+		color(`mn-docs-mcp v${version}`, '38;5;45'),
+		color(`模式: ${mode}`, '38;5;39'),
+	];
+	const maxWidth = Math.max(...contentLines.map((line) => stripAnsi(line).length)) + 4;
+	const top = color('╭' + '─'.repeat(maxWidth) + '╮', '38;5;45');
+	const bottom = color('╰' + '─'.repeat(maxWidth) + '╯', '38;5;45');
+	const body = contentLines.map((line) => {
+		const padded = padLine(line, maxWidth - 4);
+		return color('│', '38;5;45') + '  ' + padded + '  ' + color('│', '38;5;45');
+	});
+	process.stderr.write([top, ...body, bottom].join('\n') + '\n');
+}
+
 async function ensureIndex() {
 	const { INDEX_PATH } = getPaths();
 	try {
@@ -85,7 +119,7 @@ await server.start({
 	transportType: 'stdio',
 });
 
-// 默认不自动构建，避免阻塞握手；仅在首次查询时构建
-if (process.env.MCP_PREBUILD === '1') {
-	setTimeout(() => initIndexInBackground(), 1000);
-}
+renderSplash();
+
+// 默认自动构建，异步启动避免阻塞握手
+setTimeout(() => initIndexInBackground(), 0);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "mn-docs-mcp",
   "type": "module",
-  "version": "0.1.7",
+  "version": "0.2.1",
   "repository": {
     "type": "git",
     "url": "https://github.com/Temsys-Shen/marginnote-addon-docs.git"