autosnippet 2.8.3 → 2.10.0

package/lib/injection/ServiceContainer.js

@@ -3,11 +3,15 @@ import Logger from '../infrastructure/logging/Logger.js';
 import AuditStore from '../infrastructure/audit/AuditStore.js';
 import AuditLogger from '../infrastructure/audit/AuditLogger.js';
 import Gateway from '../core/gateway/Gateway.js';
-
-import { CandidateRepositoryImpl } from '../repository/candidate/CandidateRepository.impl.js';
-import { RecipeRepositoryImpl } from '../repository/recipe/RecipeRepository.impl.js';
+import { readdirSync, statSync } from 'node:fs';
+import { join as pathJoin, relative as pathRelative, extname as pathExtname } from 'node:path';
+
+import { KnowledgeRepositoryImpl } from '../repository/knowledge/KnowledgeRepository.impl.js';
+import { KnowledgeFileWriter } from '../service/knowledge/KnowledgeFileWriter.js';
+import { KnowledgeSyncService } from '../cli/KnowledgeSyncService.js';
+import { KnowledgeService } from '../service/knowledge/KnowledgeService.js';
+import { ConfidenceRouter } from '../service/knowledge/ConfidenceRouter.js';
 import { SnippetFactory } from '../service/snippet/SnippetFactory.js';
-import { RecipeService } from '../service/recipe/RecipeService.js';
 import { GuardService } from '../service/guard/GuardService.js';
 import { KnowledgeGraphService } from '../service/knowledge/KnowledgeGraphService.js';
 import { SearchEngine } from '../service/search/SearchEngine.js';
@@ -20,13 +24,9 @@ import { RetrievalFunnel } from '../service/search/RetrievalFunnel.js';
 import { JsonVectorAdapter } from '../infrastructure/vector/JsonVectorAdapter.js';
 import { IndexingPipeline } from '../infrastructure/vector/IndexingPipeline.js';
 
-// ─── P1: Injection / Snippet / Recipe ─────────────────
+// ─── P1: Injection / Snippet ─────────────────
 import { RecipeParser } from '../service/recipe/RecipeParser.js';
-import { RecipeStatsTracker } from '../service/recipe/RecipeStatsTracker.js';
 import { RecipeCandidateValidator } from '../service/recipe/RecipeCandidateValidator.js';
-import { RecipeFileWriter } from '../service/recipe/RecipeFileWriter.js';
-import { CandidateFileWriter } from '../service/candidate/CandidateFileWriter.js';
-import { CandidateService } from '../service/candidate/CandidateService.js';
 import { SnippetInstaller } from '../service/snippet/SnippetInstaller.js';
 
 // ─── P1: Guard Advanced ──────────────────────────────
@@ -34,6 +34,9 @@ import { ExclusionManager } from '../service/guard/ExclusionManager.js';
 import { RuleLearner } from '../service/guard/RuleLearner.js';
 import { ViolationsStore } from '../service/guard/ViolationsStore.js';
 
+// ─── P1: Token Usage Tracking ─────────────────────────
+import { TokenUsageStore } from '../repository/token/TokenUsageStore.js';
+
 // ─── P2: Quality ──────────────────────────────────────
 import { QualityScorer } from '../service/quality/QualityScorer.js';
 import { FeedbackCollector } from '../service/quality/FeedbackCollector.js';
@@ -53,6 +56,7 @@ import { SkillHooks } from '../service/skills/SkillHooks.js';
 
 // ─── v3.0: AST ProjectGraph ──────────────────────────
 import ProjectGraph from '../core/ast/ProjectGraph.js';
+import { GraphCache } from '../infrastructure/cache/GraphCache.js';
 
 // ─── P3: Infrastructure ──────────────────────────────
 import { EventBus } from '../infrastructure/event/EventBus.js';
@@ -232,22 +236,31 @@ export class ServiceContainer {
    * 注册仓储
    */
   _registerRepositories() {
-    // CandidateRepository
-    this.register('candidateRepository', () => {
-      if (!this.singletons.candidateRepository) {
+    // KnowledgeRepository (统一知识实体)
+    this.register('knowledgeRepository', () => {
+      if (!this.singletons.knowledgeRepository) {
         const database = this.get('database');
-        this.singletons.candidateRepository = new CandidateRepositoryImpl(database);
+        this.singletons.knowledgeRepository = new KnowledgeRepositoryImpl(database);
       }
-      return this.singletons.candidateRepository;
+      return this.singletons.knowledgeRepository;
     });
 
-    // RecipeRepository
-    this.register('recipeRepository', () => {
-      if (!this.singletons.recipeRepository) {
-        const database = this.get('database');
-        this.singletons.recipeRepository = new RecipeRepositoryImpl(database);
+    // KnowledgeFileWriter (统一 .md 序列化/落盘)
+    this.register('knowledgeFileWriter', () => {
+      if (!this.singletons.knowledgeFileWriter) {
+        const projectRoot = this.singletons._projectRoot || process.cwd();
+        this.singletons.knowledgeFileWriter = new KnowledgeFileWriter(projectRoot);
+      }
+      return this.singletons.knowledgeFileWriter;
+    });
+
+    // KnowledgeSyncService (统一 .md ↔ DB 同步)
+    this.register('knowledgeSyncService', () => {
+      if (!this.singletons.knowledgeSyncService) {
+        const projectRoot = this.singletons._projectRoot || process.cwd();
+        this.singletons.knowledgeSyncService = new KnowledgeSyncService(projectRoot);
       }
-      return this.singletons.recipeRepository;
+      return this.singletons.knowledgeSyncService;
     });
   }
 
@@ -255,54 +268,45 @@ export class ServiceContainer {
    * 注册服务
    */
   _registerServices() {
-    // CandidateService
-    this.register('candidateService', () => {
-      if (!this.singletons.candidateService) {
-        const candidateRepository = this.get('candidateRepository');
-        const auditLogger = this.get('auditLogger');
-        const gateway = this.get('gateway');
-        const projectRoot = this.singletons._projectRoot || process.cwd();
-        const candidateFileWriter = new CandidateFileWriter(projectRoot);
-        const skillHooks = this.get('skillHooks');
-        this.singletons.candidateService = new CandidateService(
-          candidateRepository,
-          auditLogger,
-          gateway,
-          { fileWriter: candidateFileWriter, skillHooks }
-        );
+    // ConfidenceRouter (V3: 知识条目自动审核路由)
+    this.register('confidenceRouter', () => {
+      if (!this.singletons.confidenceRouter) {
+        const qualityScorer = this.get('qualityScorer');
+        this.singletons.confidenceRouter = new ConfidenceRouter({}, qualityScorer);
       }
-      return this.singletons.candidateService;
+      return this.singletons.confidenceRouter;
     });
 
-    // RecipeService
-    this.register('recipeService', () => {
-      if (!this.singletons.recipeService) {
-        const recipeRepository = this.get('recipeRepository');
+    // KnowledgeService (V3: 统一知识服务)
+    this.register('knowledgeService', () => {
+      if (!this.singletons.knowledgeService) {
+        const knowledgeRepository = this.get('knowledgeRepository');
         const auditLogger = this.get('auditLogger');
         const gateway = this.get('gateway');
         const knowledgeGraphService = this.get('knowledgeGraphService');
-        const projectRoot = this.singletons._projectRoot || process.cwd();
-        const fileWriter = new RecipeFileWriter(projectRoot);
+        const fileWriter = this.get('knowledgeFileWriter');
         const skillHooks = this.get('skillHooks');
-        this.singletons.recipeService = new RecipeService(
-          recipeRepository,
+        const confidenceRouter = this.get('confidenceRouter');
+        const qualityScorer = this.get('qualityScorer');
+        this.singletons.knowledgeService = new KnowledgeService(
+          knowledgeRepository,
           auditLogger,
           gateway,
           knowledgeGraphService,
-          { fileWriter, skillHooks }
+          { fileWriter, skillHooks, confidenceRouter, qualityScorer }
         );
       }
-      return this.singletons.recipeService;
+      return this.singletons.knowledgeService;
     });
 
-    // GuardService (now uses recipeRepository)
+    // GuardService (V3: uses knowledgeRepository)
     this.register('guardService', () => {
       if (!this.singletons.guardService) {
-        const recipeRepository = this.get('recipeRepository');
+        const knowledgeRepository = this.get('knowledgeRepository');
         const auditLogger = this.get('auditLogger');
         const gateway = this.get('gateway');
         this.singletons.guardService = new GuardService(
-          recipeRepository,
+          knowledgeRepository,
           auditLogger,
           gateway
         );
@@ -388,15 +392,6 @@ export class ServiceContainer {
       return this.singletons.recipeParser;
     });
 
-    // RecipeStatsTracker
-    this.register('recipeStatsTracker', () => {
-      if (!this.singletons.recipeStatsTracker) {
-        const projectRoot = this.singletons._projectRoot || process.cwd();
-        this.singletons.recipeStatsTracker = new RecipeStatsTracker(projectRoot);
-      }
-      return this.singletons.recipeStatsTracker;
-    });
-
     // RecipeCandidateValidator
     this.register('recipeCandidateValidator', () => {
       if (!this.singletons.recipeCandidateValidator) {
@@ -405,11 +400,11 @@ export class ServiceContainer {
       return this.singletons.recipeCandidateValidator;
     });
 
-    // SnippetFactory (no DB — generates from recipes on-the-fly)
+    // SnippetFactory (V3: uses knowledgeRepository)
     this.register('snippetFactory', () => {
       if (!this.singletons.snippetFactory) {
-        const recipeRepo = this.get('recipeRepository');
-        this.singletons.snippetFactory = new SnippetFactory(recipeRepo);
+        const knowledgeRepo = this.get('knowledgeRepository');
+        this.singletons.snippetFactory = new SnippetFactory(knowledgeRepo);
       }
       return this.singletons.snippetFactory;
     });
@@ -450,6 +445,15 @@ export class ServiceContainer {
       return this.singletons.violationsStore;
     });
 
+    // Token Usage: 持久化 AI token 消耗
+    this.register('tokenUsageStore', () => {
+      if (!this.singletons.tokenUsageStore) {
+        const db = this.get('database').getDb();
+        this.singletons.tokenUsageStore = new TokenUsageStore(db);
+      }
+      return this.singletons.tokenUsageStore;
+    });
+
     // QualityScorer
     this.register('qualityScorer', () => {
       if (!this.singletons.qualityScorer) {
@@ -566,7 +570,7 @@ export class ServiceContainer {
 
   /**
    * 构建 ProjectGraph (v3.0 AST 结构图)
-   * 应在 bootstrap 流程开始前调用一次
+   * 优先从磁盘缓存加载，支持 per-file hash 增量更新
    * @param {string} projectRoot 项目根目录
    * @param {object} [options] 传递给 ProjectGraph.build() 的选项
    * @returns {Promise<import('../core/ast/ProjectGraph.js').default|null>}
@@ -575,14 +579,70 @@ export class ServiceContainer {
     if (this.singletons.projectGraph) {
       return this.singletons.projectGraph;
     }
+
+    const cache = new GraphCache(projectRoot);
+    const startTime = Date.now();
+
     try {
+      // ── 尝试从缓存恢复 + 增量更新 ──
+      const cached = cache.load('project-graph');
+      if (cached?.data && cached.fileHashes) {
+        const graph = ProjectGraph.fromJSON(cached.data);
+        const currentFiles = this.#collectSourceFilePaths(projectRoot, options);
+        const oldHashes = cached.fileHashes || {};
+
+        // 计算差异：新增 / 变更 / 删除
+        const changedPaths = [];
+        const newHashes = {};
+        for (const fp of currentFiles) {
+          const rel = pathRelative(projectRoot, fp);
+          const h = cache.computeFileHash(fp);
+          newHashes[rel] = h;
+          if (!oldHashes[rel] || oldHashes[rel] !== h) {
+            changedPaths.push(fp);
+          }
+        }
+        const deletedPaths = Object.keys(oldHashes).filter(rel => !newHashes[rel]);
+
+        if (changedPaths.length === 0 && deletedPaths.length === 0) {
+          // 完全命中
+          this.singletons.projectGraph = graph;
+          this.logger.info(
+            `[ServiceContainer] ProjectGraph ⚡ 缓存命中 (${graph.getOverview().totalClasses} classes, ` +
+            `${Date.now() - startTime}ms)`
+          );
+          return graph;
+        }
+
+        // 增量更新
+        const diff = await graph.incrementalUpdate(changedPaths, deletedPaths, options);
+        this.singletons.projectGraph = graph;
+
+        // 写回缓存
+        cache.save('project-graph', graph.toJSON(), { fileHashes: newHashes });
+
+        const overview = graph.getOverview();
+        this.logger.info(
+          `[ServiceContainer] ProjectGraph 增量更新: +${diff.added} ~${diff.updated} -${diff.deleted} ` +
+          `(${overview.totalClasses} classes, ${Date.now() - startTime}ms)`
+        );
+        return graph;
+      }
+
+      // ── 无缓存，全量构建 ──
       const graph = await ProjectGraph.build(projectRoot, options);
       this.singletons.projectGraph = graph;
       const overview = graph.getOverview();
+
+      // 计算文件 hash 并写入缓存
+      const currentFiles = this.#collectSourceFilePaths(projectRoot, options);
+      const fileHashes = cache.computeFileHashes(currentFiles, projectRoot);
+      cache.save('project-graph', graph.toJSON(), { fileHashes });
+
       this.logger.info(
         `[ServiceContainer] ProjectGraph built: ${overview.totalClasses} classes, ` +
         `${overview.totalProtocols} protocols, ${overview.totalCategories} categories ` +
-        `(${overview.buildTimeMs}ms)`
+        `(${overview.buildTimeMs}ms) — 缓存已写入`
       );
       return graph;
     } catch (err) {
@@ -590,6 +650,47 @@ export class ServiceContainer {
       return null;
     }
   }
+
+  /**
+   * 收集项目源码文件路径（用于 hash 计算）
+   * @param {string} projectRoot
+   * @param {object} options
+   * @returns {string[]}
+   */
+  #collectSourceFilePaths(projectRoot, options = {}) {
+    const DEFAULTS_EXT = { '.m': true, '.h': true, '.swift': true };
+    const extSet = new Set(options.extensions || Object.keys(DEFAULTS_EXT));
+    const excludePatterns = options.excludePatterns || [
+      'Pods/', 'Carthage/', 'node_modules/', '.build/', 'build/',
+      'DerivedData/', 'vendor/', '.git/', '__tests__/', 'Tests/',
+    ];
+    const maxFiles = options.maxFiles || 500;
+    const maxFileSizeBytes = options.maxFileSizeBytes || 500_000;
+    const results = [];
+
+    function walk(dir) {
+      if (results.length >= maxFiles) return;
+      let entries;
+      try { entries = readdirSync(dir, { withFileTypes: true }); } catch { return; }
+      for (const entry of entries) {
+        if (results.length >= maxFiles) return;
+        const fullPath = pathJoin(dir, entry.name);
+        const relativePath = pathRelative(projectRoot, fullPath);
+        if (excludePatterns.some(p => relativePath.includes(p))) continue;
+        if (entry.isDirectory()) {
+          walk(fullPath);
+        } else if (entry.isFile() && extSet.has(pathExtname(entry.name))) {
+          try {
+            const stat = statSync(fullPath);
+            if (stat.size <= maxFileSizeBytes) results.push(fullPath);
+          } catch { /* skip */ }
+        }
+      }
+    }
+
+    walk(projectRoot);
+    return results;
+  }
 }
 
 let containerInstance = null;