autosnippet 3.0.1 → 3.0.2

package/lib/external/mcp/handlers/wiki.js

@@ -9,16 +9,16 @@
 
 import fs from 'node:fs';
 import path from 'node:path';
-import { envelope } from '../envelope.js';
-import { WikiGenerator } from '../../../service/wiki/WikiGenerator.js';
 import Logger from '../../../infrastructure/logging/Logger.js';
+import { WikiGenerator } from '../../../service/wiki/WikiGenerator.js';
+import { envelope } from '../envelope.js';
 
 const logger = Logger.getInstance();
 
 // ─── 进程内任务状态（与 HTTP routes/wiki.js 独立） ─────────
 
 let mcpWikiTask = {
-  status: 'idle',   // idle | running | done | error
+  status: 'idle', // idle | running | done | error
   phase: null,
   progress: 0,
   message: null,
@@ -29,14 +29,20 @@ let mcpWikiTask = {
 };
 
 /** @type {WikiGenerator|null} */
-let currentGenerator = null;
+let _currentGenerator = null;
 
-function resetTask() {
+function _resetTask() {
   mcpWikiTask = {
-    status: 'idle', phase: null, progress: 0, message: null,
-    startedAt: null, finishedAt: null, result: null, error: null,
+    status: 'idle',
+    phase: null,
+    progress: 0,
+    message: null,
+    startedAt: null,
+    finishedAt: null,
+    result: null,
+    error: null,
   };
-  currentGenerator = null;
+  _currentGenerator = null;
 }
 
 // ─── generateWiki ───────────────────────────────────────────
@@ -62,20 +68,38 @@ export async function generateWiki(ctx, args) {
   }
 
   const container = ctx.container;
-  const projectRoot = container.singletons?._projectRoot
-    || process.env.ASD_PROJECT_DIR
-    || process.cwd();
+  const projectRoot =
+    container.singletons?._projectRoot || process.env.ASD_PROJECT_DIR || process.cwd();
 
   // 收集可用服务（全部可选，WikiGenerator 会优雅降级）
-  let spmService = null, knowledgeService = null, projectGraph = null, codeEntityGraph = null;
-  try { spmService = container.get('spmService'); } catch { /* optional */ }
-  try { knowledgeService = container.get('knowledgeService'); } catch { /* optional */ }
-  try { projectGraph = container.get('projectGraph'); } catch { /* optional */ }
-  try { codeEntityGraph = container.get('codeEntityGraph'); } catch { /* optional */ }
+  let moduleService = null,
+    knowledgeService = null,
+    projectGraph = null,
+    codeEntityGraph = null;
+  try {
+    moduleService = container.get('moduleService');
+  } catch {
+    /* optional */
+  }
+  try {
+    knowledgeService = container.get('knowledgeService');
+  } catch {
+    /* optional */
+  }
+  try {
+    projectGraph = container.get('projectGraph');
+  } catch {
+    /* optional */
+  }
+  try {
+    codeEntityGraph = container.get('codeEntityGraph');
+  } catch {
+    /* optional */
+  }
 
   const generator = new WikiGenerator({
     projectRoot,
-    spmService,
+    moduleService,
     knowledgeService,
     projectGraph,
     codeEntityGraph,
@@ -93,7 +117,7 @@ export async function generateWiki(ctx, args) {
     },
   });
 
-  currentGenerator = generator;
+  _currentGenerator = generator;
   mcpWikiTask.status = 'running';
   mcpWikiTask.startedAt = Date.now();
   mcpWikiTask.progress = 0;
@@ -103,12 +127,10 @@ export async function generateWiki(ctx, args) {
   mcpWikiTask.result = null;
 
   // 异步执行生成
-  const genPromise = mode === 'incremental'
-    ? generator.update()
-    : generator.generate();
+  const genPromise = mode === 'incremental' ? generator.update() : generator.generate();
 
   genPromise
-    .then(result => {
+    .then((result) => {
       mcpWikiTask.status = 'done';
       mcpWikiTask.progress = 100;
       mcpWikiTask.finishedAt = Date.now();
@@ -116,7 +138,7 @@ export async function generateWiki(ctx, args) {
       mcpWikiTask.message = `Wiki 生成完成，共 ${result.filesGenerated || 0} 个文件。`;
       logger.info('[MCP Wiki] Generation completed', { filesGenerated: result.filesGenerated });
     })
-    .catch(err => {
+    .catch((err) => {
       mcpWikiTask.status = 'error';
       mcpWikiTask.finishedAt = Date.now();
       mcpWikiTask.error = err.message;
@@ -127,9 +149,10 @@ export async function generateWiki(ctx, args) {
   return envelope({
     success: true,
     data: {
-      message: mode === 'incremental'
-        ? 'Wiki 增量更新已启动，调用 autosnippet_wiki_status 查看进度。'
-        : 'Wiki 全量生成已启动，调用 autosnippet_wiki_status 查看进度。',
+      message:
+        mode === 'incremental'
+          ? 'Wiki 增量更新已启动，调用 autosnippet_wiki_status 查看进度。'
+          : 'Wiki 全量生成已启动，调用 autosnippet_wiki_status 查看进度。',
       mode,
       status: 'running',
     },
@@ -146,9 +169,8 @@ export async function generateWiki(ctx, args) {
  */
 export async function wikiStatus(ctx, _args) {
   const container = ctx.container;
-  const projectRoot = container.singletons?._projectRoot
-    || process.env.ASD_PROJECT_DIR
-    || process.cwd();
+  const projectRoot =
+    container.singletons?._projectRoot || process.env.ASD_PROJECT_DIR || process.cwd();
 
   const wikiDir = path.join(projectRoot, 'AutoSnippet', 'wiki');
   const files = [];
@@ -173,7 +195,11 @@ export async function wikiStatus(ctx, _args) {
   let meta = null;
   const metaPath = path.join(wikiDir, 'meta.json');
   if (fs.existsSync(metaPath)) {
-    try { meta = JSON.parse(fs.readFileSync(metaPath, 'utf-8')); } catch { /* ignore */ }
+    try {
+      meta = JSON.parse(fs.readFileSync(metaPath, 'utf-8'));
+    } catch {
+      /* ignore */
+    }
   }
 
   return envelope({
@@ -213,9 +239,8 @@ export async function readWikiFile(ctx, args) {
   }
 
   const container = ctx.container;
-  const projectRoot = container.singletons?._projectRoot
-    || process.env.ASD_PROJECT_DIR
-    || process.cwd();
+  const projectRoot =
+    container.singletons?._projectRoot || process.env.ASD_PROJECT_DIR || process.cwd();
 
   const wikiDir = path.join(projectRoot, 'AutoSnippet', 'wiki');
   const fullPath = path.join(wikiDir, filePath);