autosnippet 3.0.0 → 3.0.2

package/lib/core/AstAnalyzer.js

@@ -1,6 +1,6 @@
 /**
  * @module AstAnalyzer
- * @description 基于 Tree-sitter 的多语言 AST 分析器
+ * @description 基于 Tree-sitter 的多语言 AST 分析器（插件注册制）
  *
  * 提供结构化代码分析能力：
  * - 类/协议/扩展 声明与继承关系
@@ -9,22 +9,48 @@
  * - 设计模式检测（Singleton、Delegate、Factory、Observer）
  * - 代码结构指标（圈复杂度、嵌套深度、方法行数）
  *
- * 支持语言：Objective-C、Swift（可扩展 JS/TS）
+ * 支持语言：通过插件注册 — ObjC、Swift、TypeScript、JavaScript、Python、Java、Kotlin
+ * 插件注册入口: lib/core/ast/index.js
  */
 
-import { createRequire } from 'module';
+import { createRequire } from 'node:module';
+
 const require = createRequire(import.meta.url);
 
-let Parser, LangObjC, LangSwift;
+let Parser;
 
 try {
   Parser = require('tree-sitter');
-  LangObjC = require('tree-sitter-objc');
-  LangSwift = require('tree-sitter-swift');
 } catch {
   // 在没有 tree-sitter 的环境中优雅降级
 }
 
+// ──────────────────────────────────────────────────────────────────
+// 插件注册表
+// ──────────────────────────────────────────────────────────────────
+
+/**
+ * @typedef {object} LangPlugin
+ * @property {Function} getGrammar  — () => tree-sitter Language Module (lazy load)
+ * @property {Function} walk        — (rootNode, ctx) => void  (AST 遍历)
+ * @property {Function} [detectPatterns] — (root, lang, methods, properties, classes) => Pattern[]
+ * @property {string[]} [extensions] — 关联的文件扩展名
+ */
+
+/** @type {Map<string, LangPlugin>} */
+const _langPlugins = new Map();
+
+/**
+ * 注册语言 AST 插件
+ * @param {string} langId — 语言标识 (e.g. 'objectivec', 'swift', 'typescript')
+ * @param {LangPlugin} plugin
+ */
+export function registerLanguage(langId, plugin) {
+  _langPlugins.set(langId, plugin);
+  // 清除 parser cache 以便下次使用新语法
+  _parserCache.delete(langId);
+}
+
 // ──────────────────────────────────────────────────────────────────
 // 公共 API
 // ──────────────────────────────────────────────────────────────────
@@ -32,49 +58,58 @@ try {
 /**
  * 分析单个源文件，返回结构化 AST 摘要
  * @param {string} source  源代码文本
- * @param {string} lang    语言标识 'objectivec' | 'swift'
+ * @param {string} lang    语言标识 'objectivec' | 'swift' | 'typescript' | 'javascript' | 'python' | 'java' | 'kotlin' | 'tsx'
  * @returns {AstSummary | null}
  */
 function analyzeFile(source, lang) {
+  const plugin = _langPlugins.get(lang);
+  if (!plugin) {
+    return null; // 无插件 → 优雅降级
+  }
+
   const parser = _getParser(lang);
-  if (!parser) return null;
+  if (!parser) {
+    return null;
+  }
 
   const tree = parser.parse(source);
   const root = tree.rootNode;
 
-  const classes = [];
-  const protocols = [];
-  const categories = [];
-  const methods = [];
-  const properties = [];
-  const patterns = [];
-  const imports = [];
+  const ctx = {
+    classes: [],
+    protocols: [],
+    categories: [],
+    methods: [],
+    properties: [],
+    patterns: [],
+    imports: [],
+    exports: [],
+  };
 
-  if (lang === 'objectivec') {
-    _walkObjC(root, { classes, protocols, categories, methods, properties, patterns, imports });
-  } else if (lang === 'swift') {
-    _walkSwift(root, { classes, protocols, categories, methods, properties, patterns, imports });
-  }
+  plugin.walk(root, ctx);
 
   // 构建继承图谱
-  const inheritanceGraph = _buildInheritanceGraph(classes, protocols, categories);
+  const inheritanceGraph = _buildInheritanceGraph(ctx.classes, ctx.protocols, ctx.categories);
 
-  // 检测设计模式
-  const detectedPatterns = _detectPatterns(root, lang, methods, properties, classes);
-  patterns.push(...detectedPatterns);
+  // 检测设计模式（优先使用插件自带的检测器，否则使用通用检测器）
+  const detectedPatterns = plugin.detectPatterns
+    ? plugin.detectPatterns(root, lang, ctx.methods, ctx.properties, ctx.classes)
+    : _detectPatterns(root, lang, ctx.methods, ctx.properties, ctx.classes);
+  ctx.patterns.push(...detectedPatterns);
 
   // 结构指标
-  const metrics = _computeMetrics(root, lang, methods);
+  const metrics = _computeMetrics(root, lang, ctx.methods);
 
   return {
     lang,
-    classes,
-    protocols,
-    categories,
-    methods,
-    properties,
-    patterns,
-    imports,
+    classes: ctx.classes,
+    protocols: ctx.protocols,
+    categories: ctx.categories,
+    methods: ctx.methods,
+    properties: ctx.properties,
+    patterns: ctx.patterns,
+    imports: ctx.imports,
+    exports: ctx.exports,
     inheritanceGraph,
     metrics,
   };
@@ -84,9 +119,10 @@ function analyzeFile(source, lang) {
  * 批量分析多文件，返回项目级汇总
  * @param {{ name: string, relativePath: string, content: string }[]} files
  * @param {string} lang
+ * @param {{ preprocessFile?: (content: string, ext: string) => { content: string, lang: string } | null }} [options]
  * @returns {ProjectAstSummary}
  */
-function analyzeProject(files, lang) {
+function analyzeProject(files, lang, options) {
   const fileSummaries = [];
   const allClasses = [];
   const allProtocols = [];
@@ -94,18 +130,34 @@ function analyzeProject(files, lang) {
   const allMethods = [];
   const allPatterns = [];
   const allImports = [];
+  const preprocessFile = options?.preprocessFile;
 
   for (const file of files) {
-    const summary = analyzeFile(file.content, lang);
-    if (!summary) continue;
+    let { content } = file;
+    let fileLang = lang;
+
+    // SFC 预处理: .vue / .svelte 等文件 → 提取 <script> 块再交给 AST
+    if (preprocessFile) {
+      const ext = file.name ? ('.' + file.name.split('.').pop()) : '';
+      const result = preprocessFile(content, ext);
+      if (result) {
+        content = result.content;
+        fileLang = result.lang || lang;
+      }
+    }
+
+    const summary = analyzeFile(content, fileLang);
+    if (!summary) {
+      continue;
+    }
 
     fileSummaries.push({ file: file.relativePath, ...summary });
-    allClasses.push(...summary.classes.map(c => ({ ...c, file: file.relativePath })));
-    allProtocols.push(...summary.protocols.map(p => ({ ...p, file: file.relativePath })));
-    allCategories.push(...summary.categories.map(c => ({ ...c, file: file.relativePath })));
-    allMethods.push(...summary.methods.map(m => ({ ...m, file: file.relativePath })));
-    allPatterns.push(...summary.patterns.map(p => ({ ...p, file: file.relativePath })));
-    allImports.push(...summary.imports.map(i => ({ path: i, file: file.relativePath })));
+    allClasses.push(...summary.classes.map((c) => ({ ...c, file: file.relativePath })));
+    allProtocols.push(...summary.protocols.map((p) => ({ ...p, file: file.relativePath })));
+    allCategories.push(...summary.categories.map((c) => ({ ...c, file: file.relativePath })));
+    allMethods.push(...summary.methods.map((m) => ({ ...m, file: file.relativePath })));
+    allPatterns.push(...summary.patterns.map((p) => ({ ...p, file: file.relativePath })));
+    allImports.push(...summary.imports.map((i) => ({ path: i, file: file.relativePath })));
   }
 
   // 项目级继承图（跨文件合并）
@@ -114,7 +166,9 @@ function analyzeProject(files, lang) {
   // 项目级模式统计
   const patternStats = {};
   for (const p of allPatterns) {
-    if (!patternStats[p.type]) patternStats[p.type] = { count: 0, files: [], instances: [] };
+    if (!patternStats[p.type]) {
+      patternStats[p.type] = { count: 0, files: [], instances: [] };
+    }
     patternStats[p.type].count++;
     if (!patternStats[p.type].files.includes(p.file)) {
       patternStats[p.type].files.push(p.file);
@@ -147,7 +201,8 @@ function generateContextForAgent(projectSummary) {
   const lines = ['## 项目代码结构分析（AST）', ''];
 
   // 类型声明概览
-  const { classes, protocols, categories, inheritanceGraph, patternStats, projectMetrics } = projectSummary;
+  const { classes, protocols, categories, inheritanceGraph, patternStats, projectMetrics } =
+    projectSummary;
 
   lines.push(`### 代码规模`);
   lines.push(`- 已分析文件: ${projectSummary.fileCount}`);
@@ -169,13 +224,15 @@ function generateContextForAgent(projectSummary) {
   }
 
   // 协议遵循
-  const conformances = classes.filter(c => c.protocols && c.protocols.length > 0);
+  const conformances = classes.filter((c) => c.protocols && c.protocols.length > 0);
   if (conformances.length > 0) {
     lines.push(`### 协议遵循`);
     for (const c of conformances.slice(0, 20)) {
-      lines.push(`- \`${c.name}\` → ${c.protocols.map(p => '`' + p + '`').join(', ')}`);
+      lines.push(`- \`${c.name}\` → ${c.protocols.map((p) => `\`${p}\``).join(', ')}`);
+    }
+    if (conformances.length > 20) {
+      lines.push(`- ... (共 ${conformances.length} 个)`);
     }
-    if (conformances.length > 20) lines.push(`- ... (共 ${conformances.length} 个)`);
     lines.push('');
   }
 
@@ -183,10 +240,15 @@ function generateContextForAgent(projectSummary) {
   if (categories.length > 0) {
     lines.push(`### Category / Extension`);
     for (const cat of categories.slice(0, 15)) {
-      const methodNames = (cat.methods || []).slice(0, 5).map(m => m.name).join(', ');
+      const methodNames = (cat.methods || [])
+        .slice(0, 5)
+        .map((m) => m.name)
+        .join(', ');
       lines.push(`- \`${cat.className}(${cat.categoryName})\` → ${methodNames || '(无方法)'}`);
     }
-    if (categories.length > 15) lines.push(`- ... (共 ${categories.length} 个)`);
+    if (categories.length > 15) {
+      lines.push(`- ... (共 ${categories.length} 个)`);
+    }
     lines.push('');
   }
 
@@ -194,7 +256,9 @@ function generateContextForAgent(projectSummary) {
   if (Object.keys(patternStats).length > 0) {
     lines.push(`### 检测到的设计模式`);
     for (const [type, stat] of Object.entries(patternStats)) {
-      lines.push(`- **${type}**: ${stat.count} 处 (${stat.files.slice(0, 3).join(', ')}${stat.files.length > 3 ? '...' : ''})`);
+      lines.push(
+        `- **${type}**: ${stat.count} 处 (${stat.files.slice(0, 3).join(', ')}${stat.files.length > 3 ? '...' : ''})`
+      );
     }
     lines.push('');
   }
@@ -204,13 +268,17 @@ function generateContextForAgent(projectSummary) {
   if (projectMetrics.complexMethods.length > 0) {
     lines.push(`- ⚠️ 高复杂度方法 (cyclomatic > 10):`);
     for (const m of projectMetrics.complexMethods.slice(0, 5)) {
-      lines.push(`  - \`${m.className || ''}${m.className ? '.' : ''}${m.name}\` (复杂度: ${m.complexity}, ${m.file}:${m.line})`);
+      lines.push(
+        `  - \`${m.className || ''}${m.className ? '.' : ''}${m.name}\` (复杂度: ${m.complexity}, ${m.file}:${m.line})`
+      );
     }
   }
   if (projectMetrics.longMethods.length > 0) {
     lines.push(`- ⚠️ 过长方法 (> 50 行):`);
     for (const m of projectMetrics.longMethods.slice(0, 5)) {
-      lines.push(`  - \`${m.className || ''}${m.className ? '.' : ''}${m.name}\` (${m.lines} 行, ${m.file}:${m.line})`);
+      lines.push(
+        `  - \`${m.className || ''}${m.className ? '.' : ''}${m.name}\` (${m.lines} 行, ${m.file}:${m.line})`
+      );
     }
   }
   lines.push('');
@@ -219,20 +287,17 @@ function generateContextForAgent(projectSummary) {
 }
 
 /**
- * 检查 Tree-sitter 是否可用
+ * 检查 Tree-sitter 是否可用（至少有一个语言插件注册）
  */
 function isAvailable() {
-  return !!(Parser && (LangObjC || LangSwift));
+  return !!Parser && _langPlugins.size > 0;
 }
 
 /**
  * 获取支持的语言列表
  */
 function supportedLanguages() {
-  const langs = [];
-  if (LangObjC) langs.push('objectivec');
-  if (LangSwift) langs.push('swift');
-  return langs;
+  return [..._langPlugins.keys()];
 }
 
 // ──────────────────────────────────────────────────────────────────
@@ -242,509 +307,39 @@ function supportedLanguages() {
 const _parserCache = new Map();
 
 function _getParser(lang) {
-  if (!Parser) return null;
-  if (_parserCache.has(lang)) return _parserCache.get(lang);
-
-  let langModule;
-  if (lang === 'objectivec' && LangObjC) langModule = LangObjC;
-  else if (lang === 'swift' && LangSwift) langModule = LangSwift;
-  else return null;
-
-  const parser = new Parser();
-  parser.setLanguage(langModule);
-  _parserCache.set(lang, parser);
-  return parser;
-}
-
-// ──────────────────────────────────────────────────────────────────
-// 内部实现 — ObjC AST 遍历
-// ──────────────────────────────────────────────────────────────────
-
-function _walkObjC(root, ctx) {
-  for (let i = 0; i < root.namedChildCount; i++) {
-    const node = root.namedChild(i);
-
-    switch (node.type) {
-      case 'preproc_include': {
-        const pathNode = node.namedChildren.find(c => c.type === 'string_literal' || c.type === 'system_lib_string');
-        if (pathNode) ctx.imports.push(pathNode.text.replace(/^["<]|[">]$/g, ''));
-        break;
-      }
-
-      case 'class_interface': {
-        const classInfo = _parseObjCInterface(node);
-        if (classInfo.isCategory) {
-          ctx.categories.push(classInfo);
-        } else {
-          ctx.classes.push(classInfo);
-        }
-        // 提取类中的方法和属性声明
-        for (const child of node.namedChildren) {
-          if (child.type === 'method_declaration') {
-            ctx.methods.push(_parseObjCMethodDecl(child, classInfo.name));
-          } else if (child.type === 'property_declaration') {
-            ctx.properties.push(_parseObjCProperty(child, classInfo.name));
-          }
-        }
-        break;
-      }
-
-      case 'protocol_declaration': {
-        ctx.protocols.push(_parseObjCProtocol(node));
-        break;
-      }
-
-      case 'class_implementation': {
-        const implName = _findIdentifier(node);
-        // 遍历 implementation_definition 提取方法定义
-        for (const child of node.namedChildren) {
-          if (child.type === 'implementation_definition') {
-            for (const implChild of child.namedChildren) {
-              if (implChild.type === 'method_definition') {
-                const m = _parseObjCMethodDef(implChild, implName);
-                ctx.methods.push(m);
-              }
-            }
-          }
-        }
-        break;
-      }
-
-      case 'category_implementation': {
-        const catImplName = _findIdentifier(node);
-        for (const child of node.namedChildren) {
-          if (child.type === 'implementation_definition') {
-            for (const implChild of child.namedChildren) {
-              if (implChild.type === 'method_definition') {
-                ctx.methods.push(_parseObjCMethodDef(implChild, catImplName));
-              }
-            }
-          }
-        }
-        break;
-      }
-    }
-  }
-}
-
-function _parseObjCInterface(node) {
-  const identifiers = node.namedChildren.filter(c => c.type === 'identifier');
-  const name = identifiers[0]?.text || 'Unknown';
-
-  // 判断 Category: @interface ClassName (CategoryName)
-  // tree-sitter-objc 中 category 的第二个 identifier 是 categoryName
-  const isCategory = node.text.includes('(') && identifiers.length >= 2 &&
-    node.text.indexOf('(') < node.text.indexOf(identifiers[1].text);
-
-  // superclass: 第二个 identifier（非 category 时）
-  let superclass = null;
-  let categoryName = null;
-  if (isCategory) {
-    categoryName = identifiers[1]?.text;
-  } else if (identifiers.length >= 2) {
-    superclass = identifiers[1]?.text;
-  }
-
-  // protocols: parameterized_arguments 中的 type_identifier
-  const protocols = [];
-  const protoList = node.namedChildren.find(c => c.type === 'parameterized_arguments');
-  if (protoList) {
-    for (const child of protoList.namedChildren) {
-      if (child.type === 'type_name') {
-        const ti = child.namedChildren.find(c => c.type === 'type_identifier');
-        if (ti) protocols.push(ti.text);
-      }
-    }
-  }
-
-  const methods = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'method_declaration') {
-      methods.push(_parseObjCMethodDecl(child, name));
-    }
-  }
-
-  const result = {
-    name,
-    superclass,
-    protocols,
-    isCategory,
-    line: node.startPosition.row + 1,
-    endLine: node.endPosition.row + 1,
-  };
-  if (isCategory) {
-    result.className = name;
-    result.categoryName = categoryName;
-    result.methods = methods;
-  }
-  return result;
-}
-
-function _parseObjCProtocol(node) {
-  const name = _findIdentifier(node) || 'Unknown';
-  const inherits = [];
-  const protoRef = node.namedChildren.find(c => c.type === 'protocol_reference_list');
-  if (protoRef) {
-    for (const child of protoRef.namedChildren) {
-      if (child.type === 'identifier') inherits.push(child.text);
-    }
-  }
-
-  const methods = [];
-  let isOptional = false;
-  for (const child of node.namedChildren) {
-    if (child.type === 'qualified_protocol_interface_declaration') {
-      isOptional = true;
-      for (const sub of child.namedChildren) {
-        if (sub.type === 'method_declaration') {
-          const m = _parseObjCMethodDecl(sub, name);
-          m.isOptional = true;
-          methods.push(m);
-        }
-      }
-    } else if (child.type === 'method_declaration') {
-      const m = _parseObjCMethodDecl(child, name);
-      m.isOptional = isOptional;
-      methods.push(m);
-    }
-  }
-
-  return {
-    name,
-    inherits,
-    methods,
-    line: node.startPosition.row + 1,
-  };
-}
-
-function _parseObjCMethodDecl(node, className) {
-  const isClassMethod = node.text.trimStart().startsWith('+');
-  const name = _findIdentifier(node) || 'unknown';
-
-  // 收集参数
-  const params = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'method_parameter') {
-      const paramName = _findIdentifier(child);
-      params.push(paramName || '?');
-    }
+  if (!Parser) {
+    return null;
   }
-
-  // 返回类型
-  let returnType = 'void';
-  const methodType = node.namedChildren.find(c => c.type === 'method_type');
-  if (methodType) {
-    const tn = methodType.namedChildren.find(c => c.type === 'type_name');
-    if (tn) {
-      const ti = tn.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'primitive_type');
-      if (ti) returnType = ti.text;
-    }
+  if (_parserCache.has(lang)) {
+    return _parserCache.get(lang);
   }
 
-  // 构建 ObjC selector
-  const selector = params.length > 0 ? name + ':' + params.slice(1).map(p => p + ':').join('') : name;
-
-  return {
-    name,
-    selector,
-    className,
-    isClassMethod,
-    returnType,
-    paramCount: params.length,
-    line: node.startPosition.row + 1,
-    kind: 'declaration',
-  };
-}
-
-function _parseObjCMethodDef(node, className) {
-  const isClassMethod = node.text.trimStart().startsWith('+');
-  const name = _findIdentifier(node) || 'unknown';
-
-  const params = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'method_parameter') {
-      const paramName = _findIdentifier(child);
-      params.push(paramName || '?');
-    }
-  }
-
-  // 方法体长度
-  const body = node.namedChildren.find(c => c.type === 'compound_statement');
-  const bodyLines = body ? (body.endPosition.row - body.startPosition.row + 1) : 0;
-
-  // 圈复杂度估算
-  const complexity = body ? _estimateComplexity(body) : 1;
-
-  // 嵌套深度
-  const nestingDepth = body ? _maxNesting(body, 0) : 0;
-
-  return {
-    name,
-    className,
-    isClassMethod,
-    paramCount: params.length,
-    bodyLines,
-    complexity,
-    nestingDepth,
-    line: node.startPosition.row + 1,
-    kind: 'definition',
-  };
-}
-
-function _parseObjCProperty(node, className) {
-  // 属性修饰符
-  const attrs = [];
-  const attrDecl = node.namedChildren.find(c => c.type === 'property_attributes_declaration');
-  if (attrDecl) {
-    for (const attr of attrDecl.namedChildren) {
-      if (attr.type === 'property_attribute') {
-        const id = attr.namedChildren.find(c => c.type === 'identifier');
-        if (id) attrs.push(id.text);
-      }
-    }
+  const plugin = _langPlugins.get(lang);
+  if (!plugin) {
+    return null;
   }
 
-  // 属性名 — 从 struct_declaration > struct_declarator 中找 identifier
-  let propName = 'unknown';
-  let propType = 'id';
-  const structDecl = node.namedChildren.find(c => c.type === 'struct_declaration');
-  if (structDecl) {
-    // 类型
-    const ti = structDecl.namedChildren.find(c => c.type === 'type_identifier');
-    if (ti) propType = ti.text;
-
-    // 名字 — 在 struct_declarator 深处
-    const sd = structDecl.namedChildren.find(c => c.type === 'struct_declarator');
-    if (sd) {
-      const findName = (n) => {
-        if (n.type === 'identifier') return n.text;
-        for (let j = 0; j < n.namedChildCount; j++) {
-          const r = findName(n.namedChild(j));
-          if (r) return r;
-        }
-        return null;
-      };
-      propName = findName(sd) || propName;
+  try {
+    const grammar = plugin.getGrammar();
+    if (!grammar) {
+      return null;
     }
+    const parser = new Parser();
+    parser.setLanguage(grammar);
+    _parserCache.set(lang, parser);
+    return parser;
+  } catch {
+    return null;
   }
-
-  return {
-    name: propName,
-    type: propType,
-    attributes: attrs,
-    className,
-    line: node.startPosition.row + 1,
-  };
 }
 
 // ──────────────────────────────────────────────────────────────────
-// 内部实现 — Swift AST 遍历
+// 内部实现 — ObjC/Swift Walker 已迁移到 ast/lang-objc.js 和 ast/lang-swift.js
+// 通过 ast/index.js 自动注册到 _langPlugins
 // ──────────────────────────────────────────────────────────────────
 
-function _walkSwift(root, ctx) {
-  _walkSwiftNode(root, ctx, null);
-}
-
-function _walkSwiftNode(node, ctx, parentClassName) {
-  for (let i = 0; i < node.namedChildCount; i++) {
-    const child = node.namedChild(i);
-
-    switch (child.type) {
-      case 'import_declaration': {
-        const mod = child.namedChildren.find(c => c.type === 'identifier' || c.type === 'simple_identifier');
-        if (mod) ctx.imports.push(mod.text);
-        break;
-      }
-
-      case 'class_declaration':
-      case 'struct_declaration':
-      case 'enum_declaration': {
-        const classInfo = _parseSwiftTypeDecl(child);
-        ctx.classes.push(classInfo);
-
-        // 递归遍历类体
-        const body = child.namedChildren.find(c => c.type === 'class_body' || c.type === 'struct_body' || c.type === 'enum_body');
-        if (body) _walkSwiftNode(body, ctx, classInfo.name);
-        break;
-      }
-
-      case 'protocol_declaration': {
-        const protoInfo = _parseSwiftProtocol(child);
-        ctx.protocols.push(protoInfo);
-        break;
-      }
-
-      case 'extension_declaration': {
-        const extInfo = _parseSwiftExtension(child);
-        ctx.categories.push(extInfo);
-
-        const body = child.namedChildren.find(c => c.type === 'extension_body');
-        if (body) _walkSwiftNode(body, ctx, extInfo.className);
-        break;
-      }
-
-      case 'function_declaration': {
-        const m = _parseSwiftFunction(child, parentClassName);
-        ctx.methods.push(m);
-        break;
-      }
-
-      case 'property_declaration': {
-        const p = _parseSwiftProperty(child, parentClassName);
-        if (p) ctx.properties.push(p);
-        break;
-      }
-
-      default: {
-        // 递归进入未识别节点
-        if (child.namedChildCount > 0 && !['function_body', 'computed_property', 'willSet_didSet_block'].includes(child.type)) {
-          _walkSwiftNode(child, ctx, parentClassName);
-        }
-      }
-    }
-  }
-}
-
-function _parseSwiftTypeDecl(node) {
-  const name = node.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'simple_identifier')?.text || 'Unknown';
-  const kind = node.type.replace('_declaration', ''); // class | struct | enum
-
-  // 继承/遵循
-  const superclass = null;
-  const protocols = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'inheritance_specifier') {
-      const typeNode = child.namedChildren.find(c => c.type === 'user_type');
-      if (typeNode) {
-        const typeName = typeNode.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'simple_identifier')?.text;
-        if (typeName) protocols.push(typeName);
-      }
-    }
-  }
-
-  // Swift 无法区分 superclass 和 protocol（都是 inheritance_specifier），
-  // 约定：第一个继承者如果首字母大写且不含 Protocol/Delegate 后缀可能是 superclass
-  let detectedSuper = null;
-  if (protocols.length > 0 && kind === 'class') {
-    const first = protocols[0];
-    if (!first.endsWith('Protocol') && !first.endsWith('Delegate') && !first.endsWith('DataSource')) {
-      detectedSuper = first;
-    }
-  }
-
-  return {
-    name,
-    kind,
-    superclass: detectedSuper,
-    protocols,
-    line: node.startPosition.row + 1,
-    endLine: node.endPosition.row + 1,
-  };
-}
-
-function _parseSwiftProtocol(node) {
-  const name = node.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'simple_identifier')?.text || 'Unknown';
-  const inherits = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'inheritance_specifier') {
-      const t = child.namedChildren.find(c => c.type === 'user_type');
-      if (t) {
-        const n = t.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'simple_identifier');
-        if (n) inherits.push(n.text);
-      }
-    }
-  }
-  return { name, inherits, line: node.startPosition.row + 1 };
-}
-
-function _parseSwiftExtension(node) {
-  const className = node.namedChildren.find(c => c.type === 'user_type' || c.type === 'type_identifier')?.text || 'Unknown';
-  const protocols = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'inheritance_specifier') {
-      const t = child.namedChildren.find(c => c.type === 'user_type');
-      if (t) {
-        const n = t.namedChildren.find(c => c.type === 'type_identifier' || c.type === 'simple_identifier');
-        if (n) protocols.push(n.text);
-      }
-    }
-  }
-
-  const methods = [];
-  const body = node.namedChildren.find(c => c.type === 'extension_body');
-  if (body) {
-    for (const child of body.namedChildren) {
-      if (child.type === 'function_declaration') {
-        methods.push(_parseSwiftFunction(child, className));
-      }
-    }
-  }
-
-  return {
-    className,
-    categoryName: protocols.length > 0 ? protocols.join('+') : 'ext',
-    protocols,
-    methods,
-    line: node.startPosition.row + 1,
-  };
-}
-
-function _parseSwiftFunction(node, className) {
-  const name = node.namedChildren.find(c => c.type === 'simple_identifier')?.text || 'unknown';
-
-  // static/class
-  const modifiers = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'modifiers' || child.type === 'modifier') {
-      modifiers.push(child.text);
-    }
-  }
-  const isClassMethod = modifiers.some(m => /\b(static|class)\b/.test(m));
-
-  // 方法体
-  const body = node.namedChildren.find(c => c.type === 'function_body');
-  const bodyLines = body ? (body.endPosition.row - body.startPosition.row + 1) : 0;
-  const complexity = body ? _estimateComplexity(body) : 1;
-  const nestingDepth = body ? _maxNesting(body, 0) : 0;
-
-  return {
-    name,
-    className,
-    isClassMethod,
-    bodyLines,
-    complexity,
-    nestingDepth,
-    line: node.startPosition.row + 1,
-    kind: 'definition',
-  };
-}
-
-function _parseSwiftProperty(node, className) {
-  const name = node.namedChildren.find(c => c.type === 'simple_identifier' || c.type === 'pattern')?.text || null;
-  if (!name) return null;
-
-  const modifiers = [];
-  for (const child of node.namedChildren) {
-    if (child.type === 'modifiers' || child.type === 'modifier') {
-      modifiers.push(child.text);
-    }
-  }
-
-  const isStatic = modifiers.some(m => /\b(static|class)\b/.test(m));
-  const isLet = node.text.includes(' let ');
-
-  return {
-    name,
-    className,
-    isStatic,
-    isConstant: isLet,
-    attributes: modifiers,
-    line: node.startPosition.row + 1,
-  };
-}
-
 // ──────────────────────────────────────────────────────────────────
-// 内部实现 — 设计模式检测
+// 内部实现 — 设计模式检测（通用回退，插件可提供自己的 detectPatterns）
 // ──────────────────────────────────────────────────────────────────
 
 function _detectPatterns(root, lang, methods, properties, classes) {
@@ -793,8 +388,7 @@ function _detectPatterns(root, lang, methods, properties, classes) {
 
   // Observer/Notification 检测（通过方法名）
   for (const m of methods) {
-    if (/^observe|^addObserver|^subscribe/.test(m.name) ||
-        /^didChange|^willChange/.test(m.name)) {
+    if (/^observe|^addObserver|^subscribe/.test(m.name) || /^didChange|^willChange/.test(m.name)) {
       patterns.push({
         type: 'observer',
         className: m.className,
@@ -835,7 +429,11 @@ function _buildInheritanceGraph(classes, protocols, categories) {
   }
 
   for (const cat of categories) {
-    edges.push({ from: `${cat.className}(${cat.categoryName})`, to: cat.className, type: 'extends' });
+    edges.push({
+      from: `${cat.className}(${cat.categoryName})`,
+      to: cat.className,
+      type: 'extends',
+    });
     if (cat.protocols) {
       for (const proto of cat.protocols) {
         edges.push({ from: cat.className, to: proto, type: 'conforms' });
@@ -848,24 +446,28 @@ function _buildInheritanceGraph(classes, protocols, categories) {
 
 function _renderInheritanceTree(edges) {
   // 找出根节点（只被继承不继承其他的）
-  const allTargets = new Set(edges.map(e => e.to));
-  const allSources = new Set(edges.map(e => e.from));
-  const roots = [...allTargets].filter(t => !allSources.has(t)).slice(0, 5);
+  const allTargets = new Set(edges.map((e) => e.to));
+  const allSources = new Set(edges.map((e) => e.from));
+  const roots = [...allTargets].filter((t) => !allSources.has(t)).slice(0, 5);
 
   const childMap = {};
   for (const e of edges) {
-    if (!childMap[e.to]) childMap[e.to] = [];
+    if (!childMap[e.to]) {
+      childMap[e.to] = [];
+    }
     const label = e.type === 'conforms' ? `${e.from} ◇` : e.from;
-    if (!childMap[e.to].includes(label)) childMap[e.to].push(label);
+    if (!childMap[e.to].includes(label)) {
+      childMap[e.to].push(label);
+    }
   }
 
   const lines = [];
   function render(name, prefix, isLast) {
-    const connector = prefix.length === 0 ? '' : (isLast ? '└─ ' : '├─ ');
+    const connector = prefix.length === 0 ? '' : isLast ? '└─ ' : '├─ ';
     lines.push(prefix + connector + name);
     const children = childMap[name] || [];
     for (let i = 0; i < children.length && i < 10; i++) {
-      const childPrefix = prefix + (prefix.length === 0 ? '' : (isLast ? '   ' : '│  '));
+      const childPrefix = prefix + (prefix.length === 0 ? '' : isLast ? '   ' : '│  ');
       render(children[i], childPrefix, i === children.length - 1);
     }
   }
@@ -884,19 +486,32 @@ function _renderInheritanceTree(edges) {
 function _estimateComplexity(node) {
   let complexity = 1;
   const BRANCH_TYPES = new Set([
-    'if_statement', 'for_statement', 'for_in_statement', 'while_statement',
-    'switch_statement', 'case_statement', 'catch_clause', 'conditional_expression',
-    'ternary_expression', 'guard_statement',
+    'if_statement',
+    'for_statement',
+    'for_in_statement',
+    'while_statement',
+    'switch_statement',
+    'case_statement',
+    'catch_clause',
+    'conditional_expression',
+    'ternary_expression',
+    'guard_statement',
     // ObjC specific
     'for_in_expression',
   ]);
 
   function walk(n) {
-    if (BRANCH_TYPES.has(n.type)) complexity++;
+    if (BRANCH_TYPES.has(n.type)) {
+      complexity++;
+    }
     // && / || 也增加复杂度
     if (n.type === 'binary_expression') {
-      const op = n.children?.find(c => c.type === '&&' || c.type === '||' || c.text === '&&' || c.text === '||');
-      if (op) complexity++;
+      const op = n.children?.find(
+        (c) => c.type === '&&' || c.type === '||' || c.text === '&&' || c.text === '||'
+      );
+      if (op) {
+        complexity++;
+      }
     }
     for (let i = 0; i < n.namedChildCount; i++) {
       walk(n.namedChild(i));
@@ -909,8 +524,12 @@ function _estimateComplexity(node) {
 
 function _maxNesting(node, depth) {
   const NESTING_TYPES = new Set([
-    'if_statement', 'for_statement', 'for_in_statement', 'while_statement',
-    'switch_statement', 'compound_statement',
+    'if_statement',
+    'for_statement',
+    'for_in_statement',
+    'while_statement',
+    'switch_statement',
+    'compound_statement',
   ]);
 
   let max = depth;
@@ -918,34 +537,38 @@ function _maxNesting(node, depth) {
 
   for (let i = 0; i < node.namedChildCount; i++) {
     const childMax = _maxNesting(node.namedChild(i), nextDepth);
-    if (childMax > max) max = childMax;
+    if (childMax > max) {
+      max = childMax;
+    }
   }
 
   return max;
 }
 
 function _computeMetrics(root, lang, methods) {
-  const defs = methods.filter(m => m.kind === 'definition');
+  const defs = methods.filter((m) => m.kind === 'definition');
   const totalBodyLines = defs.reduce((sum, m) => sum + (m.bodyLines || 0), 0);
 
   return {
     methodCount: defs.length,
     avgBodyLines: defs.length > 0 ? totalBodyLines / defs.length : 0,
-    maxComplexity: defs.length > 0 ? Math.max(...defs.map(m => m.complexity || 1)) : 0,
-    maxNestingDepth: defs.length > 0 ? Math.max(...defs.map(m => m.nestingDepth || 0)) : 0,
-    longMethods: defs.filter(m => (m.bodyLines || 0) > 50),
-    complexMethods: defs.filter(m => (m.complexity || 1) > 10),
+    maxComplexity: defs.length > 0 ? Math.max(...defs.map((m) => m.complexity || 1)) : 0,
+    maxNestingDepth: defs.length > 0 ? Math.max(...defs.map((m) => m.nestingDepth || 0)) : 0,
+    longMethods: defs.filter((m) => (m.bodyLines || 0) > 50),
+    complexMethods: defs.filter((m) => (m.complexity || 1) > 10),
   };
 }
 
 function _aggregateMetrics(fileSummaries) {
-  const allMethods = fileSummaries.flatMap(f => f.methods.filter(m => m.kind === 'definition'));
-  const allClasses = fileSummaries.flatMap(f => f.classes);
+  const allMethods = fileSummaries.flatMap((f) => f.methods.filter((m) => m.kind === 'definition'));
+  const allClasses = fileSummaries.flatMap((f) => f.classes);
 
   const methodsByClass = {};
   for (const m of allMethods) {
     if (m.className) {
-      if (!methodsByClass[m.className]) methodsByClass[m.className] = 0;
+      if (!methodsByClass[m.className]) {
+        methodsByClass[m.className] = 0;
+      }
       methodsByClass[m.className]++;
     }
   }
@@ -954,16 +577,28 @@ function _aggregateMetrics(fileSummaries) {
   return {
     totalMethods: allMethods.length,
     totalClasses: allClasses.length,
-    avgMethodsPerClass: classCounts.length > 0 ? classCounts.reduce((a, b) => a + b, 0) / classCounts.length : 0,
-    maxNestingDepth: allMethods.length > 0 ? Math.max(...allMethods.map(m => m.nestingDepth || 0)) : 0,
-    longMethods: allMethods.filter(m => (m.bodyLines || 0) > 50).map(m => ({
-      name: m.name, className: m.className, lines: m.bodyLines,
-      file: m.file, line: m.line,
-    })),
-    complexMethods: allMethods.filter(m => (m.complexity || 1) > 10).map(m => ({
-      name: m.name, className: m.className, complexity: m.complexity,
-      file: m.file, line: m.line,
-    })),
+    avgMethodsPerClass:
+      classCounts.length > 0 ? classCounts.reduce((a, b) => a + b, 0) / classCounts.length : 0,
+    maxNestingDepth:
+      allMethods.length > 0 ? Math.max(...allMethods.map((m) => m.nestingDepth || 0)) : 0,
+    longMethods: allMethods
+      .filter((m) => (m.bodyLines || 0) > 50)
+      .map((m) => ({
+        name: m.name,
+        className: m.className,
+        lines: m.bodyLines,
+        file: m.file,
+        line: m.line,
+      })),
+    complexMethods: allMethods
+      .filter((m) => (m.complexity || 1) > 10)
+      .map((m) => ({
+        name: m.name,
+        className: m.className,
+        complexity: m.complexity,
+        file: m.file,
+        line: m.line,
+      })),
   };
 }
 
@@ -974,7 +609,11 @@ function _aggregateMetrics(fileSummaries) {
 function _findIdentifier(node) {
   for (let i = 0; i < node.namedChildCount; i++) {
     const child = node.namedChild(i);
-    if (child.type === 'identifier' || child.type === 'simple_identifier' || child.type === 'type_identifier') {
+    if (
+      child.type === 'identifier' ||
+      child.type === 'simple_identifier' ||
+      child.type === 'type_identifier'
+    ) {
       return child.text;
     }
   }
@@ -994,7 +633,9 @@ function _findIdentifier(node) {
  */
 function findCallExpressions(source, lang, targetCallee) {
   const parser = _getParser(lang);
-  if (!parser) return [];
+  if (!parser) {
+    return [];
+  }
 
   const tree = parser.parse(source);
   const results = [];
@@ -1003,12 +644,23 @@ function findCallExpressions(source, lang, targetCallee) {
   function walk(node, enclosingClass) {
     // 更新当前所处的类
     let currentClass = enclosingClass;
-    if (['class_declaration', 'struct_declaration', 'class_interface', 'class_implementation'].includes(node.type)) {
+    if (
+      [
+        'class_declaration',
+        'struct_declaration',
+        'class_interface',
+        'class_implementation',
+      ].includes(node.type)
+    ) {
       currentClass = _findIdentifier(node) || enclosingClass;
     }
 
     // 检查调用表达式
-    const isCallLike = ['call_expression', 'message_expression', 'function_call_expression'].includes(node.type);
+    const isCallLike = [
+      'call_expression',
+      'message_expression',
+      'function_call_expression',
+    ].includes(node.type);
     if (isCallLike) {
       const nodeText = node.text || '';
       if (nodeText.includes(targetCallee)) {
@@ -1059,7 +711,9 @@ function findCallExpressions(source, lang, targetCallee) {
  */
 function findPatternInContext(source, lang, pattern, contextFilter = {}) {
   const parser = _getParser(lang);
-  if (!parser) return [];
+  if (!parser) {
+    return [];
+  }
 
   const tree = parser.parse(source);
   const results = [];
@@ -1068,7 +722,14 @@ function findPatternInContext(source, lang, pattern, contextFilter = {}) {
   function getEnclosingMethodName(node) {
     let current = node.parent;
     while (current) {
-      if (['method_definition', 'method_declaration', 'function_declaration', 'function_definition'].includes(current.type)) {
+      if (
+        [
+          'method_definition',
+          'method_declaration',
+          'function_declaration',
+          'function_definition',
+        ].includes(current.type)
+      ) {
         return _findIdentifier(current) || null;
       }
       current = current.parent;
@@ -1079,7 +740,14 @@ function findPatternInContext(source, lang, pattern, contextFilter = {}) {
   function getEnclosingClassName(node) {
     let current = node.parent;
     while (current) {
-      if (['class_declaration', 'struct_declaration', 'class_interface', 'class_implementation'].includes(current.type)) {
+      if (
+        [
+          'class_declaration',
+          'struct_declaration',
+          'class_interface',
+          'class_implementation',
+        ].includes(current.type)
+      ) {
         return _findIdentifier(current) || null;
       }
       current = current.parent;
@@ -1096,7 +764,10 @@ function findPatternInContext(source, lang, pattern, contextFilter = {}) {
 
       if (contextFilter.forbiddenContext) {
         // 在禁止上下文中出现 → 报告
-        if (methodName === contextFilter.forbiddenContext || className === contextFilter.forbiddenContext) {
+        if (
+          methodName === contextFilter.forbiddenContext ||
+          className === contextFilter.forbiddenContext
+        ) {
           results.push({
             line: node.startPosition.row + 1,
             snippet: lines[node.startPosition.row]?.trim().slice(0, 120) || '',
@@ -1105,7 +776,10 @@ function findPatternInContext(source, lang, pattern, contextFilter = {}) {
         }
       } else if (contextFilter.requiredContext) {
         // 不在要求的上下文中 → 报告
-        if (className !== contextFilter.requiredContext && methodName !== contextFilter.requiredContext) {
+        if (
+          className !== contextFilter.requiredContext &&
+          methodName !== contextFilter.requiredContext
+        ) {
           results.push({
             line: node.startPosition.row + 1,
             snippet: lines[node.startPosition.row]?.trim().slice(0, 120) || '',
@@ -1134,11 +808,15 @@ function findPatternInContext(source, lang, pattern, contextFilter = {}) {
  */
 function checkProtocolConformance(source, lang, className, protocolName) {
   const summary = analyzeFile(source, lang);
-  if (!summary) return { conforms: false, classFound: false, classDeclLine: null };
+  if (!summary) {
+    return { conforms: false, classFound: false, classDeclLine: null };
+  }
 
   // 在 classes 中查找
-  const cls = summary.classes.find(c => c.name === className);
-  if (!cls) return { conforms: false, classFound: false, classDeclLine: null };
+  const cls = summary.classes.find((c) => c.name === className);
+  if (!cls) {
+    return { conforms: false, classFound: false, classDeclLine: null };
+  }
 
   // 直接遵循
   if (cls.protocols?.includes(protocolName)) {
@@ -1147,7 +825,7 @@ function checkProtocolConformance(source, lang, className, protocolName) {
 
   // 通过 extension/category 遵循
   const catConforms = summary.categories.some(
-    cat => cat.className === className && cat.protocols?.includes(protocolName)
+    (cat) => cat.className === className && cat.protocols?.includes(protocolName)
   );
   if (catConforms) {
     return { conforms: true, classFound: true, classDeclLine: cls.line };
@@ -1166,6 +844,7 @@ export {
   generateContextForAgent,
   isAvailable,
   supportedLanguages,
+  // registerLanguage 已在定义处 inline export，此处不再重复
   // Guard AST 查询 API
   findCallExpressions,
   findPatternInContext,