autosnippet/dist/lib/core/ast/index.js

Extract code patterns into a knowledge base for AI coding assistants

/**
 * @module ast/index
 * @description 语言 AST 插件自动加载器（web-tree-sitter WASM 版）
 *
 * 初始化流程:
 *   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';  // 副作用: 注册所有可用语言插件
 *
 * 或按需:
 *   import { loadPlugins } from '../core/ast/index.js';
 *   await loadPlugins();
 */
import { registerLanguage } from '../AstAnalyzer.js';
import { initParser, isParserReady, loadLanguageWasm } from './parser-init.js';
let _loaded = false;
/**
 * 重置加载标志，允许 loadPlugins() 再次执行
 * 仅由 ensure-grammars.js 在安装新包后调用
 */
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 插件
 * 幂等 — 多次调用只执行一次
 */
export async function loadPlugins() {
    if (_loaded) {
        return;
    }
    _loaded = true;
    // 1. 初始化 web-tree-sitter WASM 运行时
    await initParser();
    if (!isParserReady()) {
        return; // web-tree-sitter 不可用，优雅降级（和以前缺少 tree-sitter 一样）
    }
    // 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 });
        }
    }
    // 3. 逐个加载插件模块并注入 Grammar
    const moduleCache = new Map();
    for (let i = 0; i < LANG_REGISTRY.length; i++) {
        const entry = LANG_REGISTRY[i];
        const wasmResult = wasmResults[i];
        if (wasmResult.status !== 'fulfilled' || !wasmResult.value) {
            continue; // wasm 加载失败，跳过此语言
        }
        const language = wasmResult.value;
        try {
            // 模块缓存（TypeScript 模块被 typescript + tsx 共用）
            let mod = moduleCache.get(entry.module);
            if (!mod) {
                mod = await import(entry.module);
                moduleCache.set(entry.module, mod);
            }
            // 注入 Grammar
            mod[entry.setFn](language);
            // 注册到 AstAnalyzer
            const pluginKey = entry.pluginKey || 'plugin';
            const plugin = mod[pluginKey];
            if (plugin) {
                registerLanguage(entry.langId, plugin);
            }
        }
        catch {
            /* 插件加载失败，静默跳过 */
        }
    }
}
// 自动加载（ESM 模块顶层 await）
await loadPlugins();