agent-ide 0.2.0 → 0.4.0

package/dist/plugins/python/symbol-extractor.js

@@ -0,0 +1,389 @@
+/**
+ * Python 符號提取器
+ * 從 AST 中提取 class、function、variable 等符號
+ */
+import { SymbolType, createSymbol, createScope } from '../../shared/types/index.js';
+import { PythonNodeKind } from './types.js';
+import { getNodeText, getFieldNode } from './tree-sitter-bridge.js';
+/**
+ * Python 符號提取器類別
+ */
+export class PythonSymbolExtractor {
+    /**
+     * 從 AST 提取所有符號
+     */
+    async extractSymbols(ast) {
+        const symbols = [];
+        const globalScope = createScope('module', ast.sourceFile);
+        this.visitNode(ast.root, symbols, globalScope, ast.sourceFile);
+        return symbols;
+    }
+    /**
+     * 遞歸訪問節點並提取符號
+     */
+    visitNode(node, symbols, scope, filePath) {
+        switch (node.pythonKind) {
+            case PythonNodeKind.ClassDefinition:
+                this.extractClassSymbol(node, symbols, scope, filePath);
+                break;
+            case PythonNodeKind.FunctionDefinition:
+            case PythonNodeKind.AsyncFunctionDefinition:
+                this.extractFunctionSymbol(node, symbols, scope, filePath);
+                break;
+            case PythonNodeKind.DecoratedDefinition:
+                this.handleDecoratedDefinition(node, symbols, scope, filePath);
+                break;
+            case PythonNodeKind.Assignment:
+            case PythonNodeKind.AnnotatedAssignment:
+                this.extractVariableSymbols(node, symbols, scope, filePath);
+                break;
+            case PythonNodeKind.GlobalStatement:
+            case PythonNodeKind.NonlocalStatement:
+                // 這些語句不創建新符號，只是引用現有符號
+                break;
+            default:
+                // 遞歸處理子節點
+                this.visitChildren(node, symbols, scope, filePath);
+        }
+    }
+    /**
+     * 訪問子節點
+     */
+    visitChildren(node, symbols, scope, filePath) {
+        for (const child of node.children) {
+            this.visitNode(child, symbols, scope, filePath);
+        }
+    }
+    /**
+     * 提取類別符號
+     */
+    extractClassSymbol(node, symbols, scope, filePath) {
+        const name = this.getNodeName(node);
+        if (!name) {
+            return;
+        }
+        const modifiers = this.extractClassModifiers(node);
+        const superclass = this.extractSuperclass(node);
+        const implementsProtocols = this.extractImplementedProtocols(node);
+        const decorators = node.decorators;
+        const location = this.createLocation(node, filePath);
+        const symbol = createSymbol(name, SymbolType.Class, location, scope, modifiers, decorators, superclass, implementsProtocols);
+        symbols.push(symbol);
+        // 創建類別作用域，遞歸處理類別內部
+        const classScope = createScope('class', name, scope);
+        this.visitClassBody(node, symbols, classScope, filePath);
+    }
+    /**
+     * 提取函式符號
+     */
+    extractFunctionSymbol(node, symbols, scope, filePath) {
+        const name = this.getNodeName(node);
+        if (!name) {
+            return;
+        }
+        const modifiers = this.extractFunctionModifiers(node);
+        const decorators = node.decorators;
+        const location = this.createLocation(node, filePath);
+        const symbol = createSymbol(name, SymbolType.Function, location, scope, modifiers, decorators);
+        symbols.push(symbol);
+        // 提取函式參數
+        this.extractParameters(node, symbols, scope, filePath);
+        // 創建函式作用域，遞歸處理函式內部
+        const funcScope = createScope('function', name, scope);
+        this.visitFunctionBody(node, symbols, funcScope, filePath);
+    }
+    /**
+     * 處理帶裝飾器的定義
+     */
+    handleDecoratedDefinition(node, symbols, scope, filePath) {
+        // 收集裝飾器
+        const decorators = node.decorators || [];
+        // 找到被裝飾的定義（class 或 function）
+        for (const child of node.children) {
+            const childNode = child;
+            if (childNode.pythonKind === PythonNodeKind.ClassDefinition
+                || childNode.pythonKind === PythonNodeKind.FunctionDefinition
+                || childNode.pythonKind === PythonNodeKind.AsyncFunctionDefinition) {
+                // 將裝飾器傳遞給子節點
+                const decoratedNode = {
+                    ...childNode,
+                    decorators: decorators
+                };
+                if (childNode.pythonKind === PythonNodeKind.ClassDefinition) {
+                    this.extractClassSymbol(decoratedNode, symbols, scope, filePath);
+                }
+                else {
+                    this.extractFunctionSymbol(decoratedNode, symbols, scope, filePath);
+                }
+                return;
+            }
+        }
+    }
+    /**
+     * 提取變數符號
+     */
+    extractVariableSymbols(node, symbols, scope, filePath) {
+        // 獲取賦值左側的目標
+        const targets = this.getAssignmentTargets(node);
+        for (const target of targets) {
+            const name = target.name;
+            if (!name || this.isPrivateOrDunder(name)) {
+                continue;
+            }
+            const modifiers = this.extractVariableModifiers(node, name);
+            const location = this.createLocation(target.node || node, filePath);
+            // 判斷是常量還是變數
+            const symbolType = this.isConstant(name) ? SymbolType.Constant : SymbolType.Variable;
+            const symbol = createSymbol(name, symbolType, location, scope, modifiers);
+            symbols.push(symbol);
+        }
+    }
+    /**
+     * 提取函式參數
+     */
+    extractParameters(funcNode, symbols, scope, filePath) {
+        const parametersNode = this.findChildByKind(funcNode, PythonNodeKind.Parameters);
+        if (!parametersNode) {
+            return;
+        }
+        for (const child of parametersNode.children) {
+            const childNode = child;
+            const paramName = this.extractParameterName(childNode);
+            if (paramName && !this.isSpecialParameter(paramName)) {
+                const location = this.createLocation(childNode, filePath);
+                const symbol = createSymbol(paramName, SymbolType.Variable, location, scope, ['parameter']);
+                symbols.push(symbol);
+            }
+        }
+    }
+    /**
+     * 訪問類別主體
+     */
+    visitClassBody(classNode, symbols, scope, filePath) {
+        const bodyNode = this.findChildByKind(classNode, PythonNodeKind.Block);
+        if (!bodyNode) {
+            return;
+        }
+        for (const child of bodyNode.children) {
+            this.visitNode(child, symbols, scope, filePath);
+        }
+    }
+    /**
+     * 訪問函式主體
+     */
+    visitFunctionBody(funcNode, symbols, scope, filePath) {
+        const bodyNode = this.findChildByKind(funcNode, PythonNodeKind.Block);
+        if (!bodyNode) {
+            return;
+        }
+        for (const child of bodyNode.children) {
+            this.visitNode(child, symbols, scope, filePath);
+        }
+    }
+    /**
+     * 獲取節點名稱
+     */
+    getNodeName(node) {
+        const nameNode = getFieldNode(node, 'name');
+        if (nameNode) {
+            return getNodeText(nameNode);
+        }
+        // 嘗試從 properties 獲取
+        if (node.properties.name) {
+            return node.properties.name;
+        }
+        return undefined;
+    }
+    /**
+     * 提取類別修飾符
+     */
+    extractClassModifiers(node) {
+        const modifiers = [];
+        const decorators = node.decorators || [];
+        // 檢查常見的類別裝飾器
+        if (decorators.includes('abstractmethod') || decorators.includes('abc.abstractmethod')) {
+            modifiers.push('abstract');
+        }
+        if (decorators.includes('dataclass') || decorators.includes('dataclasses.dataclass')) {
+            modifiers.push('dataclass');
+        }
+        if (decorators.includes('final') || decorators.includes('typing.final')) {
+            modifiers.push('final');
+        }
+        return modifiers;
+    }
+    /**
+     * 提取函式修飾符
+     */
+    extractFunctionModifiers(node) {
+        const modifiers = [];
+        // async 函式
+        if (node.pythonKind === PythonNodeKind.AsyncFunctionDefinition) {
+            modifiers.push('async');
+        }
+        const decorators = node.decorators || [];
+        // 檢查常見的函式裝飾器
+        if (decorators.includes('staticmethod')) {
+            modifiers.push('static');
+        }
+        if (decorators.includes('classmethod')) {
+            modifiers.push('classmethod');
+        }
+        if (decorators.includes('property')) {
+            modifiers.push('property');
+        }
+        if (decorators.includes('abstractmethod') || decorators.includes('abc.abstractmethod')) {
+            modifiers.push('abstract');
+        }
+        return modifiers;
+    }
+    /**
+     * 提取變數修飾符
+     */
+    extractVariableModifiers(node, name) {
+        const modifiers = [];
+        // 檢查是否為常量（全大寫）
+        if (this.isConstant(name)) {
+            modifiers.push('const');
+        }
+        // 檢查是否有型別註解
+        if (node.pythonKind === PythonNodeKind.AnnotatedAssignment) {
+            modifiers.push('typed');
+        }
+        return modifiers;
+    }
+    /**
+     * 提取父類別
+     */
+    extractSuperclass(node) {
+        const superclassesNode = node.treeSitterNode.childForFieldName('superclasses');
+        if (!superclassesNode) {
+            return undefined;
+        }
+        // 取第一個父類別
+        const firstArg = superclassesNode.namedChild(0);
+        if (firstArg) {
+            return firstArg.text;
+        }
+        return undefined;
+    }
+    /**
+     * 提取實現的協議/介面
+     */
+    extractImplementedProtocols(node) {
+        const superclassesNode = node.treeSitterNode.childForFieldName('superclasses');
+        if (!superclassesNode) {
+            return undefined;
+        }
+        const protocols = [];
+        for (let i = 1; i < superclassesNode.namedChildCount; i++) {
+            const child = superclassesNode.namedChild(i);
+            if (child) {
+                protocols.push(child.text);
+            }
+        }
+        return protocols.length > 0 ? protocols : undefined;
+    }
+    /**
+     * 獲取賦值目標
+     */
+    getAssignmentTargets(node) {
+        const targets = [];
+        // 尋找賦值左側
+        const leftNode = node.treeSitterNode.childForFieldName('left');
+        if (leftNode) {
+            // 簡單變數賦值
+            if (leftNode.type === 'identifier') {
+                targets.push({ name: leftNode.text });
+            }
+            // 元組解包
+            else if (leftNode.type === 'pattern_list' || leftNode.type === 'tuple_pattern') {
+                for (let i = 0; i < leftNode.namedChildCount; i++) {
+                    const child = leftNode.namedChild(i);
+                    if (child?.type === 'identifier') {
+                        targets.push({ name: child.text });
+                    }
+                }
+            }
+        }
+        // annotated_assignment 的情況
+        const nameNode = node.treeSitterNode.childForFieldName('name');
+        if (nameNode?.type === 'identifier') {
+            targets.push({ name: nameNode.text });
+        }
+        return targets;
+    }
+    /**
+     * 提取參數名稱
+     */
+    extractParameterName(node) {
+        const nodeType = node.pythonKind;
+        switch (nodeType) {
+            case PythonNodeKind.Identifier:
+                return getNodeText(node);
+            case PythonNodeKind.Parameter:
+            case PythonNodeKind.TypedParameter:
+            case PythonNodeKind.DefaultParameter:
+            case PythonNodeKind.TypedDefaultParameter: {
+                const nameNode = node.treeSitterNode.childForFieldName('name');
+                return nameNode?.text;
+            }
+            case PythonNodeKind.ListSplatPattern:
+            case PythonNodeKind.DictionarySplatPattern: {
+                // *args, **kwargs
+                const child = node.treeSitterNode.namedChild(0);
+                return child?.text;
+            }
+            default:
+                return undefined;
+        }
+    }
+    /**
+     * 查找指定類型的子節點
+     */
+    findChildByKind(node, kind) {
+        for (const child of node.children) {
+            const childNode = child;
+            if (childNode.pythonKind === kind) {
+                return childNode;
+            }
+        }
+        return undefined;
+    }
+    /**
+     * 創建位置資訊
+     */
+    createLocation(node, filePath) {
+        return {
+            filePath,
+            range: node.range
+        };
+    }
+    /**
+     * 檢查是否為常量（全大寫命名）
+     */
+    isConstant(name) {
+        return /^[A-Z][A-Z0-9_]*$/.test(name);
+    }
+    /**
+     * 檢查是否為私有或 dunder 名稱
+     */
+    isPrivateOrDunder(name) {
+        // 跳過 _single_leading_underscore 和 __double_leading_underscore
+        // 但保留 __dunder__ 方法
+        return name.startsWith('_') && !name.endsWith('_');
+    }
+    /**
+     * 檢查是否為特殊參數
+     */
+    isSpecialParameter(name) {
+        return name === 'self' || name === 'cls';
+    }
+}
+/**
+ * 創建符號提取器實例
+ */
+export function createSymbolExtractor() {
+    return new PythonSymbolExtractor();
+}
+//# sourceMappingURL=symbol-extractor.js.map

package/dist/plugins/python/symbol-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"symbol-extractor.js","sourceRoot":"","sources":["../../../src/plugins/python/symbol-extractor.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC/E,OAAO,EAGL,cAAc,EAEf,MAAM,YAAY,CAAC;AACpB,OAAO,EAAe,WAAW,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAEjF;;GAEG;AACH,MAAM,OAAO,qBAAqB;IAChC;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,GAAc;QACjC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,WAAW,GAAG,WAAW,CAAC,QAAQ,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;QAE1D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;QAE/D,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,SAAS,CACf,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,QAAQ,IAAI,CAAC,UAAU,EAAE,CAAC;YACxB,KAAK,cAAc,CAAC,eAAe;gBACjC,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBACxD,MAAM;YAER,KAAK,cAAc,CAAC,kBAAkB,CAAC;YACvC,KAAK,cAAc,CAAC,uBAAuB;gBACzC,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBAC3D,MAAM;YAER,KAAK,cAAc,CAAC,mBAAmB;gBACrC,IAAI,CAAC,yBAAyB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBAC/D,MAAM;YAER,KAAK,cAAc,CAAC,UAAU,CAAC;YAC/B,KAAK,cAAc,CAAC,mBAAmB;gBACrC,IAAI,CAAC,sBAAsB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBAC5D,MAAM;YAER,KAAK,cAAc,CAAC,eAAe,CAAC;YACpC,KAAK,cAAc,CAAC,iBAAiB;gBACnC,sBAAsB;gBACtB,MAAM;YAER;gBACE,UAAU;gBACV,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CACnB,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,CAAC,SAAS,CAAC,KAAsB,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,mBAAmB,GAAG,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,UAAkC,CAAC;QAE3D,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,YAAY,CACzB,IAAI,EACJ,UAAU,CAAC,KAAK,EAChB,QAAQ,EACR,KAAK,EACL,SAAS,EACT,UAAU,EACV,UAAU,EACV,mBAAmB,CACpB,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAErB,mBAAmB;QACnB,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACK,qBAAqB,CAC3B,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,IAAI,CAAC,UAAkC,CAAC;QAE3D,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,YAAY,CACzB,IAAI,EACJ,UAAU,CAAC,QAAQ,EACnB,QAAQ,EACR,KAAK,EACL,SAAS,EACT,UAAU,CACX,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAErB,SAAS;QACT,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEvD,mBAAmB;QACnB,MAAM,SAAS,GAAG,WAAW,CAAC,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACvD,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACK,yBAAyB,CAC/B,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,QAAQ;QACR,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QAEzC,6BAA6B;QAC7B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,KAAsB,CAAC;YACzC,IACE,SAAS,CAAC,UAAU,KAAK,cAAc,CAAC,eAAe;mBACpD,SAAS,CAAC,UAAU,KAAK,cAAc,CAAC,kBAAkB;mBAC1D,SAAS,CAAC,UAAU,KAAK,cAAc,CAAC,uBAAuB,EAClE,CAAC;gBACD,aAAa;gBACb,MAAM,aAAa,GAAkB;oBACnC,GAAG,SAAS;oBACZ,UAAU,EAAE,UAA+B;iBAC5C,CAAC;gBAEF,IAAI,SAAS,CAAC,UAAU,KAAK,cAAc,CAAC,eAAe,EAAE,CAAC;oBAC5D,IAAI,CAAC,kBAAkB,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBACnE,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,qBAAqB,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;gBACtE,CAAC;gBACD,OAAO;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,IAAmB,EACnB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,YAAY;QACZ,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAEhD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACzB,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAAA,SAAS;YAAA,CAAC;YAEtD,MAAM,SAAS,GAAG,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI,EAAE,QAAQ,CAAC,CAAC;YAEpE,YAAY;YACZ,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC;YAErF,MAAM,MAAM,GAAG,YAAY,CACzB,IAAI,EACJ,UAAU,EACV,QAAQ,EACR,KAAK,EACL,SAAS,CACV,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAuB,EACvB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;QACjF,IAAI,CAAC,cAAc,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAE9B,KAAK,MAAM,KAAK,IAAI,cAAc,CAAC,QAAQ,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,KAAsB,CAAC;YACzC,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;YAEvD,IAAI,SAAS,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC;gBACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;gBAC1D,MAAM,MAAM,GAAG,YAAY,CACzB,SAAS,EACT,UAAU,CAAC,QAAQ,EACnB,QAAQ,EACR,KAAK,EACL,CAAC,WAAW,CAAC,CACd,CAAC;gBACF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CACpB,SAAwB,EACxB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC;QACvE,IAAI,CAAC,QAAQ,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtC,IAAI,CAAC,SAAS,CAAC,KAAsB,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAuB,EACvB,OAAiB,EACjB,KAAY,EACZ,QAAgB;QAEhB,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC;QACtE,IAAI,CAAC,QAAQ,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAExB,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACtC,IAAI,CAAC,SAAS,CAAC,KAAsB,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,IAAmB;QACrC,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC5C,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAED,oBAAoB;QACpB,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAc,CAAC;QACxC,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,IAAmB;QAC/C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QAEzC,aAAa;QACb,IAAI,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACvF,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE,CAAC;YACrF,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YACxE,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,IAAmB;QAClD,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,WAAW;QACX,IAAI,IAAI,CAAC,UAAU,KAAK,cAAc,CAAC,uBAAuB,EAAE,CAAC;YAC/D,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;QAEzC,aAAa;QACb,IAAI,UAAU,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YACxC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACvC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,UAAU,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YACvF,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,IAAmB,EAAE,IAAY;QAChE,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,eAAe;QACf,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1B,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QAED,YAAY;QACZ,IAAI,IAAI,CAAC,UAAU,KAAK,cAAc,CAAC,mBAAmB,EAAE,CAAC;YAC3D,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,IAAmB;QAC3C,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAC/E,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAAA,OAAO,SAAS,CAAC;QAAA,CAAC;QAE1C,UAAU;QACV,MAAM,QAAQ,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,2BAA2B,CAAC,IAAmB;QACrD,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAC/E,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAAA,OAAO,SAAS,CAAC;QAAA,CAAC;QAE1C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1D,MAAM,KAAK,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC7C,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACtD,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,IAAmB;QAC9C,MAAM,OAAO,GAAkD,EAAE,CAAC;QAElE,SAAS;QACT,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC/D,IAAI,QAAQ,EAAE,CAAC;YACb,SAAS;YACT,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACnC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;YACxC,CAAC;YACD,OAAO;iBACF,IAAI,QAAQ,CAAC,IAAI,KAAK,cAAc,IAAI,QAAQ,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBAC/E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;oBAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;oBACrC,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;wBACjC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC/D,IAAI,QAAQ,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;YACpC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,IAAmB;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC;QAEjC,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,cAAc,CAAC,UAAU;gBAC5B,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC;YAE3B,KAAK,cAAc,CAAC,SAAS,CAAC;YAC9B,KAAK,cAAc,CAAC,cAAc,CAAC;YACnC,KAAK,cAAc,CAAC,gBAAgB,CAAC;YACrC,KAAK,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC/D,OAAO,QAAQ,EAAE,IAAI,CAAC;YACxB,CAAC;YAED,KAAK,cAAc,CAAC,gBAAgB,CAAC;YACrC,KAAK,cAAc,CAAC,sBAAsB,CAAC,CAAC,CAAC;gBAC3C,kBAAkB;gBAClB,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBAChD,OAAO,KAAK,EAAE,IAAI,CAAC;YACrB,CAAC;YAED;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,IAAmB,EAAE,IAAoB;QAC/D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,KAAsB,CAAC;YACzC,IAAI,SAAS,CAAC,UAAU,KAAK,IAAI,EAAE,CAAC;gBAClC,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,IAAmB,EAAE,QAAgB;QAC1D,OAAO;YACL,QAAQ;YACR,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,IAAY;QAC7B,OAAO,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,IAAY;QACpC,8DAA8D;QAC9D,oBAAoB;QACpB,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACrD,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,IAAY;QACrC,OAAO,IAAI,KAAK,MAAM,IAAI,IAAI,KAAK,KAAK,CAAC;IAC3C,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB;IACnC,OAAO,IAAI,qBAAqB,EAAE,CAAC;AACrC,CAAC"}

package/dist/plugins/python/tree-sitter-bridge.d.ts

@@ -0,0 +1,57 @@
+/**
+ * tree-sitter 橋接層
+ * 負責 Parser 初始化和節點轉換
+ */
+import { Parser as TreeSitterParser, type Node as TreeSitterNode, type Tree as TreeSitterTree } from 'web-tree-sitter';
+import { type PythonASTNode, type PythonAST, PythonNodeKind } from './types.js';
+/**
+ * 初始化 tree-sitter Parser
+ * 使用單例模式，確保只初始化一次
+ */
+export declare function initializeParser(): Promise<TreeSitterParser>;
+/**
+ * 獲取 Parser 實例
+ * @throws 如果 Parser 未初始化
+ */
+export declare function getParser(): TreeSitterParser;
+/**
+ * 檢查 Parser 是否已初始化
+ */
+export declare function isParserInitialized(): boolean;
+/**
+ * 解析 Python 程式碼
+ */
+export declare function parseCode(code: string): Promise<TreeSitterTree>;
+/**
+ * 轉換 tree-sitter Node 為 PythonASTNode
+ */
+export declare function convertNode(node: TreeSitterNode, sourceFile: string): PythonASTNode;
+/**
+ * 創建 PythonAST
+ */
+export declare function createPythonAST(tree: TreeSitterTree, sourceFile: string, parseTime: number): PythonAST;
+/**
+ * 遍歷 AST 節點
+ */
+export declare function traverseAST(node: PythonASTNode, callback: (node: PythonASTNode) => boolean | void): void;
+/**
+ * 查找特定類型的節點
+ */
+export declare function findNodesByKind(root: PythonASTNode, kind: PythonNodeKind): PythonASTNode[];
+/**
+ * 查找包含指定位置的節點
+ */
+export declare function findNodeAtPosition(root: PythonASTNode, line: number, column: number): PythonASTNode | null;
+/**
+ * 獲取節點的文字內容
+ */
+export declare function getNodeText(node: PythonASTNode): string;
+/**
+ * 獲取節點的指定欄位
+ */
+export declare function getFieldNode(node: PythonASTNode, fieldName: string): PythonASTNode | null;
+/**
+ * 釋放 Parser 資源
+ */
+export declare function disposeParser(): Promise<void>;
+//# sourceMappingURL=tree-sitter-bridge.d.ts.map

package/dist/plugins/python/tree-sitter-bridge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tree-sitter-bridge.d.ts","sourceRoot":"","sources":["../../../src/plugins/python/tree-sitter-bridge.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,MAAM,IAAI,gBAAgB,EAAkC,KAAK,IAAI,IAAI,cAAc,EAAE,KAAK,IAAI,IAAI,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAIvJ,OAAO,EACL,KAAK,aAAa,EAClB,KAAK,SAAS,EACd,cAAc,EAGf,MAAM,YAAY,CAAC;AAcpB;;;GAGG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,CAAC,CAyBlE;AAsBD;;;GAGG;AACH,wBAAgB,SAAS,IAAI,gBAAgB,CAK5C;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,OAAO,CAE7C;AAED;;GAEG;AACH,wBAAsB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAOrE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,GAAG,aAAa,CA+DnF;AAwBD;;GAEG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,SAAS,CAkBtG;AAaD;;GAEG;AACH,wBAAgB,WAAW,CACzB,IAAI,EAAE,aAAa,EACnB,QAAQ,EAAE,CAAC,IAAI,EAAE,aAAa,KAAK,OAAO,GAAG,IAAI,GAChD,IAAI,CASN;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,GAAG,aAAa,EAAE,CAU1F;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CAgB1G;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,aAAa,GAAG,MAAM,CAEvD;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CAgBzF;AAED;;GAEG;AACH,wBAAsB,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC,CAQnD"}