autosnippet 3.0.6 → 3.0.8

package/lib/core/AstAnalyzer.js

@@ -13,17 +13,7 @@
  * 插件注册入口: lib/core/ast/index.js
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
-let Parser;
-
-try {
-  Parser = require('tree-sitter');
-} catch {
-  // 在没有 tree-sitter 的环境中优雅降级
-}
+import { getParserClass, isParserReady } from './ast/parser-init.js';
 
 // ──────────────────────────────────────────────────────────────────
 // 插件注册表
@@ -290,7 +280,7 @@ function generateContextForAgent(projectSummary) {
  * 检查 Tree-sitter 是否可用（至少有一个语言插件注册）
  */
 function isAvailable() {
-  return !!Parser && _langPlugins.size > 0;
+  return isParserReady() && _langPlugins.size > 0;
 }
 
 /**
@@ -307,7 +297,8 @@ function supportedLanguages() {
 const _parserCache = new Map();
 
 function _getParser(lang) {
-  if (!Parser) {
+  const ParserClass = getParserClass();
+  if (!ParserClass) {
     return null;
   }
   if (_parserCache.has(lang)) {
@@ -324,7 +315,7 @@ function _getParser(lang) {
     if (!grammar) {
       return null;
     }
-    const parser = new Parser();
+    const parser = new ParserClass();
     parser.setLanguage(grammar);
     _parserCache.set(lang, parser);
     return parser;

package/lib/core/ast/ensure-grammars.js

@@ -1,200 +1,97 @@
 /**
  * @module ast/ensure-grammars
- * @description 按需安装缺失的 tree-sitter 语法包
+ * @description 检查 .wasm 语法文件可用性
  *
- * 在冷启动检测到项目语言后，检查对应 tree-sitter 包是否已安装，
- * 未安装时自动通过 npm install 补装，然后重新加载 AST 插件。
+ * 迁移至 web-tree-sitter (WASM) 后，不再需要运行时 npm install。
+ * 所有 .wasm 文件随包一起发布在 resources/grammars/。
+ * 此模块保留旧接口以兼容调用方，但内部逻辑改为检查 .wasm 文件。
  *
  * 使用方式:
  *   import { ensureGrammars } from '../core/ast/ensure-grammars.js';
  *   const result = await ensureGrammars(['typescript', 'javascript'], { logger });
  */
 
-import { execFile } from 'node:child_process';
-import { createRequire } from 'node:module';
+import fs from 'node:fs';
 import path from 'node:path';
 import { fileURLToPath } from 'node:url';
-import { promisify } from 'node:util';
 import { LanguageService } from '../../shared/LanguageService.js';
 
-const execFileAsync = promisify(execFile);
-const require = createRequire(import.meta.url);
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
-/**
- * 语言 ID → npm 包名映射
- */
-const LANG_TO_PACKAGE = {
-  objectivec: 'tree-sitter-objc',
-  swift: 'tree-sitter-swift',
-  typescript: 'tree-sitter-typescript',
-  tsx: 'tree-sitter-typescript', // tsx 与 typescript 共用同一个包
-  javascript: 'tree-sitter-javascript',
-  python: 'tree-sitter-python',
-  java: 'tree-sitter-java',
-  kotlin: 'tree-sitter-kotlin',
-  go: 'tree-sitter-go',
-  dart: 'tree-sitter-dart',
-  rust: 'tree-sitter-rust',
-};
+/** .wasm 文件存放目录 */
+const GRAMMARS_DIR = path.resolve(__dirname, '..', '..', '..', 'resources', 'grammars');
 
 /**
- * package.json 中声明的版本范围（保持和 optionalDependencies 一致）
+ * 语言 ID → .wasm 文件名映射
  */
-const PACKAGE_VERSIONS = {
-  'tree-sitter-objc': '^3.0.2',
-  'tree-sitter-swift': '^0.7.1',
-  'tree-sitter-typescript': '^0.23.2',
-  'tree-sitter-javascript': '^0.23.1',
-  'tree-sitter-python': '^0.23.5',
-  'tree-sitter-java': '^0.23.4',
-  'tree-sitter-kotlin': '^0.3.8',
-  'tree-sitter-go': '^0.25.0',
-  'tree-sitter-dart': '^1.0.0',
-  'tree-sitter-rust': '^0.23.2',
+const LANG_TO_WASM = {
+  objectivec: 'tree-sitter-objc.wasm',
+  swift: 'tree-sitter-swift.wasm',
+  typescript: 'tree-sitter-typescript.wasm',
+  tsx: 'tree-sitter-tsx.wasm',
+  javascript: 'tree-sitter-javascript.wasm',
+  python: 'tree-sitter-python.wasm',
+  java: 'tree-sitter-java.wasm',
+  kotlin: 'tree-sitter-kotlin.wasm',
+  go: 'tree-sitter-go.wasm',
+  dart: 'tree-sitter-dart.wasm',
+  rust: 'tree-sitter-rust.wasm',
 };
 
 /**
- * 检测某个 npm 包是否已安装可用
+ * 检查 .wasm 文件是否存在
  */
-function isPackageInstalled(pkgName) {
-  try {
-    require.resolve(pkgName);
-    return true;
-  } catch {
-    return false;
-  }
+function isWasmAvailable(wasmFileName) {
+  return fs.existsSync(path.join(GRAMMARS_DIR, wasmFileName));
 }
 
 /**
- * 获取 AutoSnippet 包的安装根目录（npm install 的 cwd）
- */
-function getPackageRoot() {
-  const thisFile = fileURLToPath(import.meta.url);
-  // lib/core/ast/ensure-grammars.js → 向上 3 级到包根
-  return path.resolve(path.dirname(thisFile), '..', '..', '..');
-}
-
-/**
- * 按需安装缺失的 tree-sitter 语法包
+ * 检查所需语言的 .wasm 文件是否就绪
+ *
+ * 保持旧接口签名以兼容 bootstrap 等调用方。
+ * WASM 模式下不会执行 npm install —— 文件随包分发。
  *
- * @param {string[]} detectedLanguages - 检测到的语言列表 (如 ['typescript', 'javascript', 'python'])
+ * @param {string[]} detectedLanguages - 检测到的语言列表
  * @param {object} [options]
  * @param {object} [options.logger] - Logger 实例（可选）
- * @param {number} [options.timeout=60000] - npm install 超时 (ms)
  * @returns {Promise<{installed: string[], skipped: string[], failed: string[], alreadyAvailable: string[]}>}
  */
 export async function ensureGrammars(detectedLanguages, options = {}) {
-  const { logger, timeout = 60_000 } = options;
+  const { logger } = options;
 
   const result = {
-    installed: [], // 本次新安装的包
-    skipped: [], // 不需要安装（已有）
-    failed: [], // 安装失败的包
-    alreadyAvailable: [], // 已经可用的语言
+    installed: [], // WASM 模式下始终为空（不再运行时安装）
+    skipped: [],
+    failed: [],
+    alreadyAvailable: [],
   };
 
   if (!detectedLanguages || detectedLanguages.length === 0) {
     return result;
   }
 
-  // 1) 去重: 多个语言可能映射到同一个包 (如 typescript + tsx → tree-sitter-typescript)
-  const neededPackages = new Map(); // pkgName → [langIds]
   for (const lang of detectedLanguages) {
-    const pkg = LANG_TO_PACKAGE[lang];
-    if (!pkg) {
+    const wasmFile = LANG_TO_WASM[lang];
+    if (!wasmFile) {
+      result.skipped.push(lang);
       continue;
     }
-    if (!neededPackages.has(pkg)) {
-      neededPackages.set(pkg, []);
-    }
-    neededPackages.get(pkg).push(lang);
-  }
 
-  // 2) 检查哪些已安装
-  const toInstall = [];
-  for (const [pkg, langs] of neededPackages) {
-    if (isPackageInstalled(pkg)) {
-      result.skipped.push(pkg);
-      result.alreadyAvailable.push(...langs);
+    if (isWasmAvailable(wasmFile)) {
+      result.alreadyAvailable.push(lang);
     } else {
-      toInstall.push(pkg);
+      result.failed.push(lang);
+      logger?.warn?.(`[ensure-grammars] Missing .wasm file: ${wasmFile} for language "${lang}"`);
     }
   }
 
-  if (toInstall.length === 0) {
-    logger?.info?.('[ensure-grammars] All required grammars already installed');
-    return result;
-  }
-
-  // 3) 批量安装缺失的包
-  const pkgRoot = getPackageRoot();
-  const installArgs = toInstall.map((pkg) => {
-    const ver = PACKAGE_VERSIONS[pkg];
-    return ver ? `${pkg}@${ver}` : pkg;
-  });
-
-  logger?.info?.(`[ensure-grammars] Installing missing grammars: ${toInstall.join(', ')}`);
-
-  try {
-    const { stderr } = await execFileAsync(
-      'npm',
-      ['install', '--no-save', '--no-audit', '--no-fund', ...installArgs],
-      {
-        cwd: pkgRoot,
-        timeout,
-        env: { ...process.env, NODE_ENV: '' }, // 避免 NODE_ENV=production 跳过 optional
-      }
+  if (result.failed.length > 0) {
+    logger?.warn?.(
+      `[ensure-grammars] ${result.failed.length} grammar(s) missing. ` +
+      `Expected in: ${GRAMMARS_DIR}`
     );
-
-    if (stderr && !stderr.includes('npm warn')) {
-      logger?.warn?.(`[ensure-grammars] npm stderr: ${stderr.slice(0, 200)}`);
-    }
-
-    // 4) 逐个验证安装结果
-    for (const pkg of toInstall) {
-      // 清除 require cache 以便重新检测
-      try {
-        delete require.cache[require.resolve(pkg)];
-      } catch {
-        /* not cached */
-      }
-
-      if (isPackageInstalled(pkg)) {
-        result.installed.push(pkg);
-        const langs = neededPackages.get(pkg) || [];
-        result.alreadyAvailable.push(...langs);
-        logger?.info?.(`[ensure-grammars] ✓ ${pkg} installed successfully`);
-      } else {
-        result.failed.push(pkg);
-        logger?.warn?.(
-          `[ensure-grammars] ✗ ${pkg} install reported success but package not resolvable`
-        );
-      }
-    }
-  } catch (err) {
-    logger?.warn?.(`[ensure-grammars] npm install failed: ${err.message}`);
-    // 批量失败 → 逐个重试
-    for (const pkg of toInstall) {
-      try {
-        const ver = PACKAGE_VERSIONS[pkg];
-        const spec = ver ? `${pkg}@${ver}` : pkg;
-        await execFileAsync('npm', ['install', '--no-save', '--no-audit', '--no-fund', spec], {
-          cwd: pkgRoot,
-          timeout: timeout / 2,
-        });
-        if (isPackageInstalled(pkg)) {
-          result.installed.push(pkg);
-          const langs = neededPackages.get(pkg) || [];
-          result.alreadyAvailable.push(...langs);
-          logger?.info?.(`[ensure-grammars] ✓ ${pkg} installed (retry)`);
-        } else {
-          result.failed.push(pkg);
-        }
-      } catch {
-        result.failed.push(pkg);
-        logger?.warn?.(`[ensure-grammars] ✗ ${pkg} install failed permanently`);
-      }
-    }
+  } else {
+    logger?.info?.('[ensure-grammars] All required grammar .wasm files available');
   }
 
   return result;
@@ -205,7 +102,6 @@ export async function ensureGrammars(detectedLanguages, options = {}) {
  * 由于 loadPlugins() 是幂等的（_loaded 标志），需要重置标志后重新加载
  */
 export async function reloadPlugins() {
-  // 动态 import 获取模块并重置 _loaded 状态
   const astIndex = await import('./index.js');
   if (typeof astIndex._resetForReload === 'function') {
     astIndex._resetForReload();
@@ -220,7 +116,6 @@ export async function reloadPlugins() {
  * @returns {string[]} 需要的语言 ID 列表
  */
 export function inferLanguagesFromStats(langStats) {
-  // 从 LanguageService 派生，仅覆盖 tsx（tree-sitter 需要独立解析器）
   const bareMap = LanguageService.bareExtToLangMap;
 
   const langs = new Set();

package/lib/core/ast/index.js

@@ -1,9 +1,15 @@
 /**
  * @module ast/index
- * @description 语言 AST 插件自动加载器
+ * @description 语言 AST 插件自动加载器（web-tree-sitter WASM 版）
  *
- * 按 try/catch 逐个加载每个语言插件并注册到 AstAnalyzer。
- * 缺少对应 tree-sitter 包时静默跳过（优雅降级）。
+ * 初始化流程:
+ *   1. 调用 initParser() — 初始化 web-tree-sitter WASM 运行时
+ *   2. 并行加载所有 .wasm 语法文件
+ *   3. 将 Language 对象注入每个 lang-*.js 插件
+ *   4. 注册到 AstAnalyzer
+ *
+ * .wasm 文件位于 resources/grammars/，随 npm 包一起发布。
+ * 不再依赖原生 tree-sitter 编译，任何平台即装即用。
  *
  * 使用方式:
  *   import '../core/ast/index.js';  // 副作用: 注册所有可用语言插件
@@ -14,6 +20,7 @@
  */
 
 import { registerLanguage } from '../AstAnalyzer.js';
+import { initParser, loadLanguageWasm, isParserReady } from './parser-init.js';
 
 let _loaded = false;
 
@@ -25,6 +32,23 @@ export function _resetForReload() {
   _loaded = false;
 }
 
+/**
+ * 语言注册表 — langId → { wasmFile, module, setGrammarFn, langId, tsxWasmFile?, setTsxGrammarFn? }
+ */
+const LANG_REGISTRY = [
+  { langId: 'objectivec', wasmFile: 'tree-sitter-objc.wasm', module: './lang-objc.js', setFn: 'setGrammar' },
+  { langId: 'swift', wasmFile: 'tree-sitter-swift.wasm', module: './lang-swift.js', setFn: 'setGrammar' },
+  { langId: 'typescript', wasmFile: 'tree-sitter-typescript.wasm', module: './lang-typescript.js', setFn: 'setGrammar' },
+  { langId: 'tsx', wasmFile: 'tree-sitter-tsx.wasm', module: './lang-typescript.js', setFn: 'setTsxGrammar', pluginKey: 'tsxPlugin' },
+  { langId: 'javascript', wasmFile: 'tree-sitter-javascript.wasm', module: './lang-javascript.js', setFn: 'setGrammar' },
+  { langId: 'python', wasmFile: 'tree-sitter-python.wasm', module: './lang-python.js', setFn: 'setGrammar' },
+  { langId: 'java', wasmFile: 'tree-sitter-java.wasm', module: './lang-java.js', setFn: 'setGrammar' },
+  { langId: 'kotlin', wasmFile: 'tree-sitter-kotlin.wasm', module: './lang-kotlin.js', setFn: 'setGrammar' },
+  { langId: 'go', wasmFile: 'tree-sitter-go.wasm', module: './lang-go.js', setFn: 'setGrammar' },
+  { langId: 'dart', wasmFile: 'tree-sitter-dart.wasm', module: './lang-dart.js', setFn: 'setGrammar' },
+  { langId: 'rust', wasmFile: 'tree-sitter-rust.wasm', module: './lang-rust.js', setFn: 'setGrammar' },
+];
+
 /**
  * 加载并注册所有可用的语言 AST 插件
  * 幂等 — 多次调用只执行一次
@@ -35,87 +59,56 @@ export async function loadPlugins() {
   }
   _loaded = true;
 
-  // ObjC
-  try {
-    const { plugin } = await import('./lang-objc.js');
-    registerLanguage('objectivec', plugin);
-  } catch {
-    /* tree-sitter-objc not installed */
+  // 1. 初始化 web-tree-sitter WASM 运行时
+  await initParser();
+  if (!isParserReady()) {
+    return; // web-tree-sitter 不可用，优雅降级（和以前缺少 tree-sitter 一样）
   }
 
-  // Swift
-  try {
-    const { plugin } = await import('./lang-swift.js');
-    registerLanguage('swift', plugin);
-  } catch {
-    /* tree-sitter-swift not installed */
-  }
-
-  // TypeScript
-  try {
-    const { plugin: tsPlugin, tsxPlugin } = await import('./lang-typescript.js');
-    registerLanguage('typescript', tsPlugin);
-    if (tsxPlugin) {
-      registerLanguage('tsx', tsxPlugin);
+  // 2. 按顺序加载所有 .wasm 语法文件（并行加载偶发竞态导致失败）
+  const wasmResults = [];
+  for (const entry of LANG_REGISTRY) {
+    try {
+      const lang = await loadLanguageWasm(entry.wasmFile);
+      wasmResults.push({ status: 'fulfilled', value: lang });
+    } catch (err) {
+      wasmResults.push({ status: 'rejected', reason: err });
     }
-  } catch {
-    /* tree-sitter-typescript not installed */
   }
 
-  // JavaScript
-  try {
-    const { plugin } = await import('./lang-javascript.js');
-    registerLanguage('javascript', plugin);
-  } catch {
-    /* tree-sitter-javascript not installed */
-  }
+  // 3. 逐个加载插件模块并注入 Grammar
+  const moduleCache = new Map();
 
-  // Python
-  try {
-    const { plugin } = await import('./lang-python.js');
-    registerLanguage('python', plugin);
-  } catch {
-    /* tree-sitter-python not installed */
-  }
+  for (let i = 0; i < LANG_REGISTRY.length; i++) {
+    const entry = LANG_REGISTRY[i];
+    const wasmResult = wasmResults[i];
 
-  // Java
-  try {
-    const { plugin } = await import('./lang-java.js');
-    registerLanguage('java', plugin);
-  } catch {
-    /* tree-sitter-java not installed */
-  }
+    if (wasmResult.status !== 'fulfilled' || !wasmResult.value) {
+      continue; // wasm 加载失败，跳过此语言
+    }
 
-  // Kotlin
-  try {
-    const { plugin } = await import('./lang-kotlin.js');
-    registerLanguage('kotlin', plugin);
-  } catch {
-    /* tree-sitter-kotlin not installed */
-  }
+    const language = wasmResult.value;
 
-  // Go
-  try {
-    const { plugin } = await import('./lang-go.js');
-    registerLanguage('go', plugin);
-  } catch {
-    /* tree-sitter-go not installed */
-  }
+    try {
+      // 模块缓存（TypeScript 模块被 typescript + tsx 共用）
+      let mod = moduleCache.get(entry.module);
+      if (!mod) {
+        mod = await import(entry.module);
+        moduleCache.set(entry.module, mod);
+      }
 
-  // Dart
-  try {
-    const { plugin } = await import('./lang-dart.js');
-    registerLanguage('dart', plugin);
-  } catch {
-    /* tree-sitter-dart not installed */
-  }
+      // 注入 Grammar
+      mod[entry.setFn](language);
 
-  // Rust
-  try {
-    const { plugin } = await import('./lang-rust.js');
-    registerLanguage('rust', plugin);
-  } catch {
-    /* tree-sitter-rust not installed */
+      // 注册到 AstAnalyzer
+      const pluginKey = entry.pluginKey || 'plugin';
+      const plugin = mod[pluginKey];
+      if (plugin) {
+        registerLanguage(entry.langId, plugin);
+      }
+    } catch {
+      /* 插件加载失败，静默跳过 */
+    }
   }
 }
 

package/lib/core/ast/lang-dart.js

@@ -7,13 +7,9 @@
  *        Builder, BLoC/Cubit, Provider/Riverpod, Freezed
  *
  * 注意: tree-sitter-dart 目前尚无兼容 tree-sitter ≥0.25 的稳定版。
- *       本插件已实现完整的 walk/detectPatterns 逻辑，等待上游发布兼容版本后即可启用。
+ *       已迁移至 web-tree-sitter (WASM)，无原生编译依赖。
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkDart(root, ctx) {
   _walkNode(root, ctx, null);
 }
@@ -647,11 +643,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-dart');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-go.js

@@ -7,10 +7,6 @@
  *        Goroutine, Channel, Middleware (http.Handler chain)
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkGo(root, ctx) {
   for (let i = 0; i < root.namedChildCount; i++) {
     const child = root.namedChild(i);
@@ -516,11 +512,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-go');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-java.js

@@ -6,10 +6,6 @@
  * 模式: Singleton, Builder, Factory, DI, Stream Pipeline
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkJava(root, ctx) {
   _walkJavaNode(root, ctx, null);
 }
@@ -421,11 +417,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-java');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-javascript.js

@@ -2,13 +2,9 @@
  * @module lang-javascript
  * @description JavaScript AST Walker 插件
  *
- * 与 TypeScript walker 共享大部分逻辑，grammar 使用 tree-sitter-javascript
+ * 与 TypeScript walker 共享大部分逻辑，grammar 使用 web-tree-sitter (WASM)
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 // JavaScript walker 与 TypeScript walker 结构相同
 // 复用 lang-typescript 的 walker 逻辑
 
@@ -258,11 +254,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-javascript');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-kotlin.js

@@ -6,10 +6,6 @@
  * 模式: Singleton (object), Factory (companion), DSL, Flow, Sealed
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkKotlin(root, ctx) {
   _walkKtNode(root, ctx, null);
 }
@@ -409,11 +405,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-kotlin');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-objc.js

@@ -3,10 +3,6 @@
  * @description ObjC AST Walker 插件 — 从 AstAnalyzer.js 迁移
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 // ── ObjC AST 遍历 ──
 
 function walkObjC(root, ctx) {
@@ -374,11 +370,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-objc');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-python.js

@@ -6,10 +6,6 @@
  * 模式: Singleton, Factory, Context Manager, Decorator pattern, Data Class
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkPython(root, ctx) {
   _walkPyNode(root, ctx, null);
 }
@@ -357,11 +353,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-python');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-rust.js

@@ -7,10 +7,6 @@
  *        Async (tokio/async-std), Unsafe block, Derive macro
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkRust(root, ctx) {
   for (let i = 0; i < root.namedChildCount; i++) {
     const child = root.namedChild(i);
@@ -681,11 +677,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-rust');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-swift.js

@@ -3,10 +3,6 @@
  * @description Swift AST Walker 插件 — 从 AstAnalyzer.js 迁移
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 // ── Swift AST 遍历 ──
 
 function walkSwift(root, ctx) {
@@ -323,11 +319,11 @@ function _maxNesting(node, depth) {
 
 let _grammar = null;
 function getGrammar() {
-  if (!_grammar) {
-    _grammar = require('tree-sitter-swift');
-  }
   return _grammar;
 }
+export function setGrammar(grammar) {
+  _grammar = grammar;
+}
 
 export const plugin = {
   getGrammar,

package/lib/core/ast/lang-typescript.js

@@ -6,10 +6,6 @@
  * 模式检测: Singleton, Factory, Observer, React Hook/Component, Middleware, Decorator
  */
 
-import { createRequire } from 'node:module';
-
-const require = createRequire(import.meta.url);
-
 function walkTypeScript(root, ctx) {
   _walkTSNode(root, ctx, null);
 }
@@ -471,12 +467,11 @@ function _maxNesting(node, depth) {
 
 let _tsGrammar = null;
 function getGrammar() {
-  if (!_tsGrammar) {
-    const ts = require('tree-sitter-typescript');
-    _tsGrammar = ts.typescript;
-  }
   return _tsGrammar;
 }
+export function setGrammar(grammar) {
+  _tsGrammar = grammar;
+}
 
 export const plugin = {
   getGrammar,
@@ -488,12 +483,11 @@ export const plugin = {
 // TSX 插件 — 共享 walker，不同 grammar
 let _tsxGrammar = null;
 function getTsxGrammar() {
-  if (!_tsxGrammar) {
-    const ts = require('tree-sitter-typescript');
-    _tsxGrammar = ts.tsx;
-  }
   return _tsxGrammar;
 }
+export function setTsxGrammar(grammar) {
+  _tsxGrammar = grammar;
+}
 
 export const tsxPlugin = {
   getGrammar: getTsxGrammar,

package/lib/core/ast/parser-init.js

@@ -0,0 +1,82 @@
+/**
+ * @module ast/parser-init
+ * @description web-tree-sitter 初始化器
+ *
+ * 统一管理 WASM 版 Parser 的生命周期：
+ *   1. 调用 Parser.init() 初始化 WASM 运行时（仅一次）
+ *   2. 加载 .wasm 语法文件为 Language 对象
+ *   3. 提供同步的 Parser 构造与语言设置 API
+ *
+ * 所有 async 操作（init + wasm 加载）集中在 loadPlugins() 阶段完成，
+ * 下游 analyzeFile / findCallExpressions 等保持同步调用。
+ */
+
+import { fileURLToPath } from 'node:url';
+import path from 'node:path';
+import { readFile } from 'node:fs/promises';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+
+/** 预编译 .wasm 文件存放目录 */
+const GRAMMARS_DIR = path.resolve(__dirname, '..', '..', '..', 'resources', 'grammars');
+
+let Parser = null;
+/** web-tree-sitter 模块命名空间 — Language.load 在这里 */
+let _namespace = null;
+let _initialized = false;
+
+/**
+ * 初始化 web-tree-sitter WASM 运行时
+ * 幂等 — 多次调用只执行一次
+ */
+export async function initParser() {
+  if (_initialized) return;
+
+  try {
+    // web-tree-sitter ESM: 导出 { Parser, Language, ... } 命名空间
+    const mod = await import('web-tree-sitter');
+    _namespace = mod.default || mod;
+    // v0.25 导出 { Parser, Language, ... }，需要提取 Parser 类
+    Parser = typeof _namespace === 'function' ? _namespace : _namespace.Parser;
+    await Parser.init();
+    _initialized = true;
+  } catch (err) {
+    // web-tree-sitter 不可用时优雅降级
+    Parser = null;
+    _initialized = false;
+  }
+}
+
+/**
+ * 获取 Parser 构造函数
+ * @returns {typeof import('web-tree-sitter') | null}
+ */
+export function getParserClass() {
+  return Parser;
+}
+
+/**
+ * 检查 parser 是否已初始化
+ */
+export function isParserReady() {
+  return _initialized && Parser !== null;
+}
+
+/**
+ * 从 resources/grammars/ 加载指定语言的 .wasm 文件
+ * @param {string} wasmFileName — 如 'tree-sitter-javascript.wasm'
+ * @returns {Promise<object|null>} Language 对象，失败返回 null
+ */
+export async function loadLanguageWasm(wasmFileName) {
+  if (!_initialized || !_namespace) return null;
+
+  const wasmPath = path.join(GRAMMARS_DIR, wasmFileName);
+  try {
+    // 自行读取 wasm 文件为 Uint8Array，绕过 ESM 下 __require("fs/promises") 的兼容问题
+    const buffer = await readFile(wasmPath);
+    const Language = _namespace.Language || Parser.Language;
+    return await Language.load(new Uint8Array(buffer));
+  } catch {
+    return null;
+  }
+}

package/lib/service/guard/GuardCheckEngine.js

@@ -122,6 +122,7 @@ const BUILT_IN_RULES = {
     languages: ['javascript', 'typescript'],
     dimension: 'file',
     category: 'style',
+    excludePaths: /(?:^|[\/\\])(?:test|tests|__tests__|spec|__mocks__|mock|mocks|fixtures?)[\/\\]|[\/\\](?:test_|spec_)[^\/\\]*\.(?:js|ts)$|\.(?:test|spec)\.(?:js|ts)$/,
   },
   'js-no-console-log': {
     message: '生产代码应移除 console.log，使用专用日志库',
@@ -130,6 +131,7 @@ const BUILT_IN_RULES = {
     languages: ['javascript', 'typescript'],
     dimension: 'file',
     category: 'style',
+    excludePaths: /(?:^|[\/\\])(?:test|tests|__tests__|spec|mock|mocks|__mocks__|scripts|tools|debug)[\/\\]|[\/\\](?:test_|spec_|mock)[^\/\\]*\.(?:js|ts)$|\.(?:test|spec)\.(?:js|ts)$/,
   },
   'js-no-debugger': {
     message: '生产代码中不应包含 debugger 语句',
@@ -267,6 +269,7 @@ const BUILT_IN_RULES = {
     languages: ['go'],
     dimension: 'file',
     category: 'correctness',
+    excludePaths: /(?:^|[\/\\])(?:tests?|testdata|_test)[\/\\]|_test\.go$/,
   },
   'go-no-init-abuse': {
     message: 'init() 函数副作用难以追踪，避免在 init 中执行复杂逻辑',
@@ -279,10 +282,11 @@ const BUILT_IN_RULES = {
   'go-no-global-var': {
     message: '全局可变变量导致并发安全问题，考虑使用依赖注入',
     severity: 'info',
-    pattern: '^var\\s+\\w+\\s+',
+    pattern: '^var\\s+(?!_\\s)[a-zA-Z]\\w*\\s+(?!=[^=])',
     languages: ['go'],
     dimension: 'file',
     category: 'style',
+    excludePaths: /(?:^|[\/\\])(?:tests?|testdata)[\/\\]|_test\.go$/,
   },
 
   // ══════════════════════════════════════════════════════════
@@ -298,17 +302,18 @@ const BUILT_IN_RULES = {
     category: 'style',
   },
   'dart-avoid-dynamic': {
-    message: '避免使用 dynamic 类型，使用具体类型或泛型提升类型安全',
+    message: '避免直接使用 dynamic 作为变量/参数类型，使用具体类型或泛型提升类型安全',
     severity: 'warning',
-    pattern: '\\bdynamic\\b',
+    pattern: '(?<!<\\w*,\\s*)(?<!<)\\bdynamic\\b(?!\\s*>)',
     languages: ['dart'],
     dimension: 'file',
     category: 'style',
+    fixSuggestion: '使用 Object? 或具体类型替代 dynamic；Map<String, dynamic> 用于 JSON 序列化时可保留',
   },
   'dart-no-set-state-after-dispose': {
     message: 'setState 调用前应检查 mounted 状态，避免 disposed 后调用',
-    severity: 'warning',
-    pattern: 'setState\\s*\\(',
+    severity: 'info',
+    pattern: '(?<!mounted\\)\\s*)setState\\s*\\(',
     languages: ['dart'],
     dimension: 'file',
     category: 'correctness',
@@ -372,7 +377,9 @@ const BUILT_IN_RULES = {
     dimension: 'file',
     category: 'correctness',
     fixSuggestion: '使用 ? 操作符传播错误，或 .unwrap_or_default() / .expect("原因")',
-    excludePaths: /(?:^|[\/\\])tests?[\/\\]|[\/\\]test_|_test\.rs$|[\/\\]benches[\/\\]/,
+    excludePaths: /(?:^|[\/\\])(?:tests?|test_helpers|benches|examples)[\/\\]|[\/\\]test_[^\/\\]*\.rs$|_test\.rs$/,
+    skipComments: true,
+    skipTestBlocks: true,
   },
   'rust-no-expect-without-msg': {
     message: 'expect() 应提供有意义的错误消息，帮助定位 panic 原因',
@@ -399,7 +406,9 @@ const BUILT_IN_RULES = {
     languages: ['rust'],
     dimension: 'file',
     category: 'correctness',
-    excludePaths: /(?:^|[\/\\])tests?[\/\\]|[\/\\]examples[\/\\]/,
+    excludePaths: /(?:^|[\/\\])(?:tests?|test_helpers|benches|examples)[\/\\]|_test\.rs$/,
+    skipComments: true,
+    skipTestBlocks: true,
   },
   'rust-clone-overuse': {
     message: '频繁 .clone() 可能暗示所有权设计问题，考虑使用借用或 Cow',
@@ -409,6 +418,9 @@ const BUILT_IN_RULES = {
     dimension: 'file',
     category: 'performance',
     fixSuggestion: '分析是否可用 &T 借用替代，或使用 Cow<T> 延迟克隆',
+    excludePaths: /(?:^|[\/\\])(?:tests?|test_helpers|benches|examples)[\/\\]|_test\.rs$/,
+    skipComments: true,
+    skipTestBlocks: true,
   },
   'rust-no-panic-in-lib': {
     message: 'panic!() 在库代码中应避免使用，返回 Result 让调用方决定如何处理',
@@ -417,7 +429,9 @@ const BUILT_IN_RULES = {
     languages: ['rust'],
     dimension: 'file',
     category: 'correctness',
-    excludePaths: /(?:^|[\/\\])tests?[\/\\]|[\/\\]benches[\/\\]|main\.rs$/,
+    excludePaths: /(?:^|[\/\\])(?:tests?|test_helpers|benches|examples)[\/\\]|main\.rs$/,
+    skipComments: true,
+    skipTestBlocks: true,
   },
   'rust-std-mutex-in-async': {
     message: 'async 代码中不应使用 std::sync::Mutex，MutexGuard 不是 Send',
@@ -589,6 +603,8 @@ export class GuardCheckEngine {
           type: 'regex',
           fixSuggestion: rule.fixSuggestion || null,
           ...(rule.excludePaths ? { excludePaths: rule.excludePaths } : {}),
+          ...(rule.skipComments ? { skipComments: true } : {}),
+          ...(rule.skipTestBlocks ? { skipTestBlocks: true } : {}),
         });
       }
     }
@@ -655,6 +671,14 @@ export class GuardCheckEngine {
 
     const lines = (code || '').split(/\r?\n/);
 
+    // 预计算注释行掩码 — 供 skipComments 规则使用
+    // 识别: // 行注释, /// doc, //! inner doc, /* block */, # Python/Shell 行注释
+    const commentLines = this._buildCommentMask(lines, language);
+
+    // 预计算测试块掩码 — 供 skipTestBlocks 规则使用
+    // Rust: #[cfg(test)] mod tests { ... } 内联测试模块
+    const testBlockLines = this._buildTestBlockMask(lines, language);
+
     for (const rule of rules) {
       // 跳过空模式或特殊标记 (?!) — 由 code-level 检查接管
       if (!rule.pattern || rule.pattern === '(?!)') {
@@ -669,7 +693,19 @@ export class GuardCheckEngine {
         continue;
       }
 
+      const shouldSkipComments = !!rule.skipComments;
+      const shouldSkipTestBlocks = !!rule.skipTestBlocks;
+
       for (let i = 0; i < lines.length; i++) {
+        // skipComments: 跳过注释行（doc comments / 行注释 / 块注释内）
+        if (shouldSkipComments && commentLines[i]) {
+          continue;
+        }
+        // skipTestBlocks: 跳过内联测试模块（Rust #[cfg(test)] 块等）
+        if (shouldSkipTestBlocks && testBlockLines[i]) {
+          continue;
+        }
+
         if (re.test(lines[i])) {
           violations.push({
             ruleId: rule.id || rule.name,
@@ -1179,6 +1215,134 @@ export class GuardCheckEngine {
     return cached;
   }
 
+  /**
+   * 构建内联测试块掩码
+   * 目前支持 Rust #[cfg(test)] mod xxx { ... } 块
+   * @param {string[]} lines
+   * @param {string} language
+   * @returns {boolean[]} 每行是否在测试块内
+   */
+  _buildTestBlockMask(lines, language) {
+    const mask = new Array(lines.length).fill(false);
+
+    // 目前仅 Rust 需要 — #[cfg(test)] 内联测试模块
+    if (language !== 'rust') return mask;
+
+    let inTestBlock = false;
+    let braceDepth = 0;
+
+    for (let i = 0; i < lines.length; i++) {
+      const trimmed = lines[i].trimStart();
+
+      if (!inTestBlock) {
+        // 检测 #[cfg(test)] 属性行
+        if (/^#\[cfg\(test\)\]/.test(trimmed)) {
+          // 向后找 mod xxx { — 标记为测试块起始
+          // 可能在同一行: #[cfg(test)] mod tests {
+          // 也可能在下一行: mod tests {
+          const restOfLine = trimmed.slice('#[cfg(test)]'.length).trim();
+          if (/^mod\s+\w+/.test(restOfLine)) {
+            // 同一行有 mod 声明
+            inTestBlock = true;
+            braceDepth = 0;
+            // 计算本行的花括号
+            for (const ch of lines[i]) {
+              if (ch === '{') braceDepth++;
+              else if (ch === '}') braceDepth--;
+            }
+            mask[i] = true;
+            if (braceDepth <= 0) inTestBlock = false; // 单行 mod 声明 (mod tests;)
+            continue;
+          }
+          // 检查下一行是否是 mod xxx {
+          if (i + 1 < lines.length && /^\s*mod\s+\w+/.test(lines[i + 1])) {
+            mask[i] = true; // #[cfg(test)] 行本身也标记
+            inTestBlock = true;
+            braceDepth = 0;
+            // 下一行会在循环中处理
+            continue;
+          }
+          // 单行 #[cfg(test)] 但后面不是 mod — 不处理
+        }
+      } else {
+        // 正在测试块内 — 追踪花括号深度
+        mask[i] = true;
+        for (const ch of lines[i]) {
+          if (ch === '{') braceDepth++;
+          else if (ch === '}') braceDepth--;
+        }
+        if (braceDepth <= 0) {
+          inTestBlock = false; // 测试块结束
+        }
+      }
+    }
+
+    return mask;
+  }
+
+  /**
+   * 构建注释行掩码 — 识别行注释和块注释内部行
+   *
+   * 支持的注释形式:
+   *   // 行注释,  /// 文档注释,  //! 内部文档注释  (C/Java/JS/TS/Go/Rust/Swift/Kotlin/Dart)
+   *   # 行注释  (Python)
+   *   /* ... * / 块注释  (C/Java/JS/TS/Go/Rust/Swift/Kotlin)
+   *   \"\"\" ... \"\"\"  (Python doc-string — 简化: 整行以 \"\"\" 开头的行)
+   *
+   * @param {string[]} lines
+   * @param {string} language
+   * @returns {boolean[]} 每行是否为注释行
+   */
+  _buildCommentMask(lines, language) {
+    const mask = new Array(lines.length).fill(false);
+    let inBlock = false; // 是否在 /* ... */ 块内
+
+    const usesHash = language === 'python'; // Python 用 # 注释
+    const usesSlash = !usesHash;            // 其他语言用 //
+
+    for (let i = 0; i < lines.length; i++) {
+      const trimmed = lines[i].trimStart();
+
+      // 块注释延续
+      if (inBlock) {
+        mask[i] = true;
+        if (trimmed.includes('*/')) {
+          inBlock = false;
+        }
+        continue;
+      }
+
+      // 块注释开始（同行不闭合）
+      if (usesSlash && /^\s*\/\*/.test(lines[i])) {
+        mask[i] = true;
+        if (!trimmed.includes('*/')) {
+          inBlock = true;
+        }
+        continue;
+      }
+
+      // 行注释: // 或 /// 或 //!
+      if (usesSlash && /^\s*\/\//.test(lines[i])) {
+        mask[i] = true;
+        continue;
+      }
+
+      // Python 行注释: #
+      if (usesHash && /^\s*#/.test(lines[i])) {
+        mask[i] = true;
+        continue;
+      }
+
+      // Python docstring 行 (简化: 整行以 """ 或 ''' 开头)
+      if (usesHash && /^\s*("""|''')/.test(lines[i])) {
+        mask[i] = true;
+        continue;
+      }
+    }
+
+    return mask;
+  }
+
   /**
    * 获取内置规则列表
    */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "autosnippet",
-  "version": "3.0.6",
+  "version": "3.0.8",
   "description": "Extract code patterns into a knowledge base for AI coding assistants",
   "type": "module",
   "main": "lib/bootstrap.js",
@@ -88,9 +88,9 @@
     "open": "^8.0.4",
     "ora": "^8.0.1",
     "socket.io": "^4.8.1",
-    "tree-sitter": "^0.25.0",
     "undici": "^6.23.0",
     "uuid": "^9.0.1",
+    "web-tree-sitter": "^0.25.0",
     "winston": "^3.11.0",
     "zod": "^4.3.6"
   },
@@ -110,7 +110,8 @@
     "resources/native-ui/combined-window.swift",
     "resources/native-ui/native-ui",
     "resources/native-ui/README.md",
-    "resources/openChrome.applescript"
+    "resources/openChrome.applescript",
+    "resources/grammars"
   ],
   "directories": {
     "doc": "docs",
@@ -126,45 +127,7 @@
     "prettier": "^3.1.1",
     "supertest": "^6.3.3"
   },
-  "optionalDependencies": {
-    "tree-sitter-go": "^0.25.0",
-    "tree-sitter-java": "^0.23.5",
-    "tree-sitter-javascript": "^0.25.0",
-    "tree-sitter-kotlin": "^0.3.8",
-    "tree-sitter-objc": "^3.0.2",
-    "tree-sitter-python": "^0.25.0",
-    "tree-sitter-swift": "^0.7.1",
-    "tree-sitter-typescript": "^0.23.2",
-    "tree-sitter-rust": "^0.23.2"
-  },
   "overrides": {
-    "tree-sitter-kotlin": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-java": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-python": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-javascript": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-typescript": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-swift": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-objc": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-rust": {
-      "tree-sitter": "$tree-sitter"
-    },
-    "tree-sitter-c": {
-      "tree-sitter": "$tree-sitter"
-    },
     "minimatch": "^10.2.2",
     "diff": "^8.0.3",
     "glob": "^11.0.0"