@blueking/bkui-knowledge 0.0.1-beta.10 → 0.0.1-beta.11

package/bin/bkui-knowledge.js

@@ -26,14 +26,23 @@
  *   }
  * }
  *
+ * Claude Code（无需额外配置，自动从 MCP initialize 请求获取项目路径）：
+ * {
+ *   "bkui-knowledge": {
+ *     "command": "npx",
+ *     "args": ["-y", "@blueking/bkui-knowledge", "--ide=claude-code"]
+ *   }
+ * }
+ *
  * 参数说明：
  * - 第一个非 -- 开头的参数: 项目路径
- * - --ide=cursor|codebuddy: 指定目标 IDE（默认 cursor）
+ * - --ide=cursor|codebuddy|claude-code: 指定目标 IDE（默认 cursor）
  *
  * 项目路径传递方式（优先级从高到低）：
  * 1. 命令行参数: npx @blueking/bkui-knowledge /path/to/project
  * 2. 环境变量: BKUI_PROJECT_ROOT=/path/to/project
- * 3. 当前工作目录 (cwd)
+ * 3. MCP initialize 请求中的 rootUri（Claude Code 等支持此协议的客户端）
+ * 4. 当前工作目录 (cwd)
  *
  * 更新机制：
  * - skills 内置在 npm 包中
@@ -57,9 +66,20 @@ const IDE_CONFIGS = {
     configDir: '.codebuddy',
     skillsDir: '.codebuddy/skills',
     versionFile: '.codebuddy/.bkui-knowledge-version'
+  },
+  'claude-code': {
+    name: 'Claude Code',
+    configDir: '.claude',
+    skillsDir: '.claude/skills',
+    versionFile: '.claude/.bkui-knowledge-version'
   }
 };
 
+// 用于存储从 MCP initialize 请求获取的项目路径
+let mcpRootUri = null;
+// 标记是否已经同步过 skills
+let skillsSynced = false;
+
 // npm 包根目录
 const PKG_ROOT = path.join(__dirname, '..');
 
@@ -95,7 +115,7 @@ function getIDEConfig() {
 
 /**
  * 获取用户项目根目录
- * 优先级：命令行参数 > 环境变量 > cwd
+ * 优先级：命令行参数 > 环境变量 > MCP rootUri > cwd
  */
 function getProjectRoot() {
   // 1. 命令行参数: node bkui-knowledge.js /path/to/project (过滤掉 -- 开头的参数)
@@ -107,10 +127,34 @@ function getProjectRoot() {
   if (process.env.BKUI_PROJECT_ROOT) {
     return process.env.BKUI_PROJECT_ROOT;
   }
-  // 3. 默认使用 cwd
+  // 3. MCP initialize 请求中的 rootUri
+  if (mcpRootUri) {
+    return mcpRootUri;
+  }
+  // 4. 默认使用 cwd
   return process.cwd();
 }
 
+/**
+ * 从 file:// URI 提取路径
+ * @param {string} uri - file:// URI
+ * @returns {string} 文件系统路径
+ */
+function uriToPath(uri) {
+  if (!uri) return null;
+  if (uri.startsWith('file://')) {
+    // file:///path/to/dir -> /path/to/dir
+    let filePath = uri.slice(7);
+    // Windows: file:///C:/path -> C:/path
+    if (filePath.match(/^\/[A-Za-z]:\//)) {
+      filePath = filePath.slice(1);
+    }
+    return decodeURIComponent(filePath);
+  }
+  // 如果不是 file:// URI，直接返回（可能已经是路径）
+  return uri;
+}
+
 /**
  * 获取当前包版本
  */
@@ -268,6 +312,27 @@ async function startMcpServer() {
 
   async function handleMessage(message) {
     if (message.method === "initialize") {
+      // 从 initialize 请求中提取项目路径
+      const params = message.params || {};
+
+      // 尝试从 rootUri 或 workspaceFolders 获取项目路径
+      if (params.rootUri) {
+        mcpRootUri = uriToPath(params.rootUri);
+        log(`从 MCP initialize 获取项目路径: ${mcpRootUri}`);
+      } else if (params.workspaceFolders && params.workspaceFolders.length > 0) {
+        mcpRootUri = uriToPath(params.workspaceFolders[0].uri);
+        log(`从 MCP workspaceFolders 获取项目路径: ${mcpRootUri}`);
+      }
+
+      // 延迟同步 skills（在获取到项目路径后）
+      if (!skillsSynced && mcpRootUri) {
+        const projectRoot = getProjectRoot();
+        const ideConfig = getIDEConfig();
+        log(`[延迟同步] 工作目录: ${projectRoot}, IDE: ${ideConfig.name}`);
+        syncSkills(projectRoot, ideConfig);
+        skillsSynced = true;
+      }
+
       return {
         protocolVersion: "2024-11-05",
         serverInfo: { name: "bkui-knowledge-server", version: "1.0.0" },
@@ -362,21 +427,34 @@ async function startMcpServer() {
   log('MCP stdio 服务已启动');
 }
 
+/**
+ * 检查是否有明确的项目路径（命令行参数或环境变量）
+ */
+function hasExplicitProjectPath() {
+  const pathArg = process.argv.slice(2).find(arg => !arg.startsWith('-'));
+  return !!(pathArg || process.env.BKUI_PROJECT_ROOT);
+}
+
 /**
  * 主入口
  */
 async function main() {
-  const projectRoot = getProjectRoot();
   const ideConfig = getIDEConfig();
 
-  // 调试：输出实际的工作目录和目标 IDE
-  log(`工作目录: ${projectRoot}`);
-  log(`目标 IDE: ${ideConfig.name}`);
-
-  // 1. 同步 skills（同步执行，从 npm 包读取）
-  syncSkills(projectRoot, ideConfig);
+  // 如果有明确的项目路径（命令行参数或环境变量），立即同步
+  // 否则等待 MCP initialize 请求获取项目路径
+  if (hasExplicitProjectPath()) {
+    const projectRoot = getProjectRoot();
+    log(`工作目录: ${projectRoot}`);
+    log(`目标 IDE: ${ideConfig.name}`);
+    syncSkills(projectRoot, ideConfig);
+    skillsSynced = true;
+  } else {
+    log(`等待 MCP initialize 请求获取项目路径...`);
+    log(`目标 IDE: ${ideConfig.name}`);
+  }
 
-  // 2. 启动 MCP 服务
+  // 启动 MCP 服务
   await startMcpServer();
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/bkui-knowledge",
-  "version": "0.0.1-beta.10",
+  "version": "0.0.1-beta.11",
   "description": "蓝鲸前端知识库 MCP 服务 - 自动同步 skills，支持 Cursor/CodeBuddy",
   "main": "server/mcp-core.js",
   "bin": {