mn-docs-mcp 0.2.0 → 0.2.1

package/mcp/cli.mjs

@@ -10,6 +10,7 @@ let mode = 'stdio';
 let port;
 let prebuild = false;
 let silent = false;
+let root;
 
 for (let i = 0; i < args.length; i += 1) {
 	const arg = args[i];
@@ -38,10 +39,19 @@ for (let i = 0; i < args.length; 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(
 			[
@@ -51,6 +61,7 @@ for (let i = 0; i < args.length; i += 1) {
 				'  --prebuild        启动后后台预构建索引',
 				'  --verbose         输出日志(覆盖--silent)',
 				'  --silent          关闭开屏与日志',
+				'  --root <path>     指定文档仓库根目录',
 				'  --stdio           显式使用stdio模式',
 			].join('\n') + '\n'
 		);
@@ -72,6 +83,7 @@ if (mode === 'stdio') {
 	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 {
@@ -79,6 +91,7 @@ if (mode === 'stdio') {
 	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

@@ -12,20 +12,33 @@ function color(text, code) {
 	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 lines = [
-		color('╭──────────────────────────────────────────────────────────────╮', '38;5;45'),
-		color('│  mn-docs-mcp', '38;5;45') + color(` v${version}`, '38;5;214') + color('  (Charm风格启动)         │', '38;5;45'),
-		color('│                                                              │', '38;5;45'),
-		color(`│  模式: ${mode.padEnd(12)}  端口: ${port.padEnd(6)}  状态: 已启动          │`, '38;5;39'),
-		color('│                                                              │', '38;5;45'),
-		color('╰──────────────────────────────────────────────────────────────╯', '38;5;45'),
+	const contentLines = [
+		color(`mn-docs-mcp v${version}`, '38;5;45'),
+		color(`模式: ${mode}  端口: ${port}`, '38;5;39'),
 	];
-	process.stdout.write(lines.join('\n') + '\n');
+	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() {

package/mcp/server.mjs

@@ -16,19 +16,32 @@ function color(text, code) {
 	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 lines = [
-		color('╭──────────────────────────────────────────────────────────────╮', '38;5;45'),
-		color('│  mn-docs-mcp', '38;5;45') + color(` v${version}`, '38;5;214') + color('  (Charm风格启动)         │', '38;5;45'),
-		color('│                                                              │', '38;5;45'),
-		color(`│  模式: ${mode.padEnd(12)}  状态: 已启动                         │`, '38;5;39'),
-		color('│                                                              │', '38;5;45'),
-		color('╰──────────────────────────────────────────────────────────────╯', '38;5;45'),
+	const contentLines = [
+		color(`mn-docs-mcp v${version}`, '38;5;45'),
+		color(`模式: ${mode}`, '38;5;39'),
 	];
-	process.stderr.write(lines.join('\n') + '\n');
+	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() {

package/package.json

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