@zzzen/pyright-internal 1.1.254

package/dist/languageService/documentSymbolCollector.js

@@ -0,0 +1,344 @@
+"use strict";
+/*
+ * documentSymbolCollector.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ * Author: Eric Traut
+ *
+ * Collects symbols within the given tree that are semantically
+ * equivalent to the requested symbol.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DocumentSymbolCollector = void 0;
+const AnalyzerNodeInfo = __importStar(require("../analyzer/analyzerNodeInfo"));
+const declaration_1 = require("../analyzer/declaration");
+const declarationUtils_1 = require("../analyzer/declarationUtils");
+const parseTreeUtils_1 = require("../analyzer/parseTreeUtils");
+const ParseTreeUtils = __importStar(require("../analyzer/parseTreeUtils"));
+const parseTreeWalker_1 = require("../analyzer/parseTreeWalker");
+const ScopeUtils = __importStar(require("../analyzer/scopeUtils"));
+const sourceMapper_1 = require("../analyzer/sourceMapper");
+const types_1 = require("../analyzer/types");
+const typeUtils_1 = require("../analyzer/typeUtils");
+const cancellationUtils_1 = require("../common/cancellationUtils");
+const collectionUtils_1 = require("../common/collectionUtils");
+const debug_1 = require("../common/debug");
+// This walker looks for symbols that are semantically equivalent
+// to the requested symbol.
+class DocumentSymbolCollector extends parseTreeWalker_1.ParseTreeWalker {
+    constructor(_symbolName, _declarations, _evaluator, _cancellationToken, _startingNode, _treatModuleInImportAndFromImportSame = false, _skipUnreachableCode = true) {
+        super();
+        this._symbolName = _symbolName;
+        this._declarations = _declarations;
+        this._evaluator = _evaluator;
+        this._cancellationToken = _cancellationToken;
+        this._startingNode = _startingNode;
+        this._treatModuleInImportAndFromImportSame = _treatModuleInImportAndFromImportSame;
+        this._skipUnreachableCode = _skipUnreachableCode;
+        this._results = [];
+        this._dunderAllNameNodes = new Set();
+        // Don't report strings in __all__ right away, that will
+        // break the assumption on the result ordering.
+        this._setDunderAllNodes(this._startingNode);
+    }
+    static collectFromNode(node, evaluator, cancellationToken, startingNode, treatModuleInImportAndFromImportSame = false, skipUnreachableCode = true) {
+        const symbolName = node.value;
+        const declarations = this.getDeclarationsForNode(node, evaluator, 
+        /* resolveLocalName */ true, cancellationToken);
+        startingNode = startingNode !== null && startingNode !== void 0 ? startingNode : (0, parseTreeUtils_1.getModuleNode)(node);
+        if (!startingNode) {
+            return [];
+        }
+        const collector = new DocumentSymbolCollector(symbolName, declarations, evaluator, cancellationToken, startingNode, treatModuleInImportAndFromImportSame, skipUnreachableCode);
+        return collector.collect();
+    }
+    static getDeclarationsForNode(node, evaluator, resolveLocalName, token, sourceMapper) {
+        (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+        const declarations = this._getDeclarationsForNode(node, evaluator);
+        const resolvedDeclarations = [];
+        declarations.forEach((decl) => {
+            const resolvedDecl = evaluator.resolveAliasDeclaration(decl, resolveLocalName);
+            if (resolvedDecl) {
+                resolvedDeclarations.push(resolvedDecl);
+                if (sourceMapper && (0, sourceMapper_1.isStubFile)(resolvedDecl.path)) {
+                    const implDecls = sourceMapper.findDeclarations(resolvedDecl);
+                    for (const implDecl of implDecls) {
+                        if (implDecl && implDecl.path) {
+                            this._addIfUnique(resolvedDeclarations, implDecl);
+                        }
+                    }
+                }
+            }
+        });
+        return resolvedDeclarations;
+    }
+    collect() {
+        this.walk(this._startingNode);
+        return this._results;
+    }
+    walk(node) {
+        if (!this._skipUnreachableCode || !AnalyzerNodeInfo.isCodeUnreachable(node)) {
+            super.walk(node);
+        }
+    }
+    visitName(node) {
+        (0, cancellationUtils_1.throwIfCancellationRequested)(this._cancellationToken);
+        // No need to do any more work if the symbol name doesn't match.
+        if (node.value !== this._symbolName) {
+            return false;
+        }
+        if (this._declarations.length > 0) {
+            const declarations = DocumentSymbolCollector._getDeclarationsForNode(node, this._evaluator, this._skipUnreachableCode);
+            if (declarations && declarations.length > 0) {
+                // Does this name share a declaration with the symbol of interest?
+                if (declarations.some((decl) => this._resultsContainsDeclaration(decl))) {
+                    this._addResult(node);
+                }
+            }
+        }
+        else {
+            // There were no declarations
+            this._addResult(node);
+        }
+        return false;
+    }
+    visitString(node) {
+        (0, cancellationUtils_1.throwIfCancellationRequested)(this._cancellationToken);
+        if (this._dunderAllNameNodes.has(node)) {
+            this._addResult(node);
+        }
+        return false;
+    }
+    _addResult(node) {
+        const range = node.nodeType === 38 /* Name */ ? node : (0, parseTreeUtils_1.getStringNodeValueRange)(node);
+        this._results.push({ node, range });
+    }
+    _resultsContainsDeclaration(declaration) {
+        // Resolve the declaration.
+        const resolvedDecl = this._evaluator.resolveAliasDeclaration(declaration, /* resolveLocalNames */ false);
+        if (!resolvedDecl) {
+            return false;
+        }
+        // The reference results declarations are already resolved, so we don't
+        // need to call resolveAliasDeclaration on them.
+        if (this._declarations.some((decl) => (0, declarationUtils_1.areDeclarationsSame)(decl, resolvedDecl, this._treatModuleInImportAndFromImportSame))) {
+            return true;
+        }
+        // We didn't find the declaration using local-only alias resolution. Attempt
+        // it again by fully resolving the alias.
+        const resolvedDeclNonlocal = this._getResolveAliasDeclaration(resolvedDecl);
+        if (!resolvedDeclNonlocal || resolvedDeclNonlocal === resolvedDecl) {
+            return false;
+        }
+        return this._declarations.some((decl) => (0, declarationUtils_1.areDeclarationsSame)(decl, resolvedDeclNonlocal, this._treatModuleInImportAndFromImportSame));
+    }
+    _getResolveAliasDeclaration(declaration) {
+        // TypeEvaluator.resolveAliasDeclaration only resolve alias in AliasDeclaration in the form of
+        // "from x import y as [y]" but don't do thing for alias in "import x as [x]"
+        // Here, alias should have same name as module name.
+        if (isAliasDeclFromImportAsWithAlias(declaration)) {
+            return (0, declarationUtils_1.getDeclarationsWithUsesLocalNameRemoved)([declaration])[0];
+        }
+        const resolvedDecl = this._evaluator.resolveAliasDeclaration(declaration, /* resolveLocalNames */ true);
+        return isAliasDeclFromImportAsWithAlias(resolvedDecl)
+            ? (0, declarationUtils_1.getDeclarationsWithUsesLocalNameRemoved)([resolvedDecl])[0]
+            : resolvedDecl;
+        function isAliasDeclFromImportAsWithAlias(decl) {
+            return (!!decl &&
+                decl.type === 6 /* Alias */ &&
+                decl.node &&
+                decl.usesLocalName &&
+                decl.node.nodeType === 21 /* ImportAs */);
+        }
+    }
+    _setDunderAllNodes(node) {
+        if (node.nodeType !== 36 /* Module */) {
+            return;
+        }
+        const dunderAllInfo = AnalyzerNodeInfo.getDunderAllInfo(node);
+        if (!dunderAllInfo) {
+            return;
+        }
+        const moduleScope = ScopeUtils.getScopeForNode(node);
+        if (!moduleScope) {
+            return;
+        }
+        dunderAllInfo.stringNodes.forEach((stringNode) => {
+            if (stringNode.value !== this._symbolName) {
+                return;
+            }
+            const symbolInScope = moduleScope.lookUpSymbolRecursive(stringNode.value);
+            if (!symbolInScope) {
+                return;
+            }
+            if (!symbolInScope.symbol.getDeclarations().some((d) => this._resultsContainsDeclaration(d))) {
+                return;
+            }
+            this._dunderAllNameNodes.add(stringNode);
+        });
+    }
+    static _addIfUnique(declarations, itemToAdd) {
+        for (const def of declarations) {
+            if ((0, declarationUtils_1.areDeclarationsSame)(def, itemToAdd)) {
+                return;
+            }
+        }
+        declarations.push(itemToAdd);
+    }
+    static _getDeclarationsForNode(node, evaluator, skipUnreachableCode = true) {
+        var _a;
+        // This can handle symbols brought in by wildcard (import *) as long as the declarations that the symbol collector
+        // compares against point to the actual alias declaration, not one that uses local name (ex, import alias)
+        if (((_a = node.parent) === null || _a === void 0 ? void 0 : _a.nodeType) !== 37 /* ModuleName */) {
+            return this._getDeclarationsForNonModuleNameNode(node, evaluator, skipUnreachableCode);
+        }
+        return this._getDeclarationsForModuleNameNode(node, evaluator);
+    }
+    static _getDeclarationsForNonModuleNameNode(node, evaluator, skipUnreachableCode = true) {
+        var _a, _b;
+        (0, debug_1.assert)(((_a = node.parent) === null || _a === void 0 ? void 0 : _a.nodeType) !== 37 /* ModuleName */);
+        let decls = evaluator.getDeclarationsForNameNode(node, skipUnreachableCode) || [];
+        if (((_b = node.parent) === null || _b === void 0 ? void 0 : _b.nodeType) === 23 /* ImportFromAs */) {
+            // Make sure we get the decl for this specific "from import" statement
+            decls = decls.filter((d) => d.node === node.parent);
+        }
+        // If we can't get decl, see whether we can get type from the node.
+        // Some might have synthesized type for the node such as subModule in import X.Y statement.
+        if (decls.length === 0) {
+            const type = evaluator.getType(node);
+            if ((type === null || type === void 0 ? void 0 : type.category) === 8 /* Module */) {
+                // Synthesize decl for the module.
+                return [(0, declarationUtils_1.createSynthesizedAliasDeclaration)(type.filePath)];
+            }
+        }
+        // We would like to make X in import X and import X.Y as Y to match, but path for
+        // X in import X and one in import X.Y as Y might not match since path in X.Y will point
+        // to X.Y rather than X if import statement has an alias.
+        // so, for such case, we put synthesized one so we can treat X in both statement same.
+        for (const aliasDecl of decls.filter((d) => (0, declaration_1.isAliasDeclaration)(d) && !d.loadSymbolsFromPath)) {
+            const node = aliasDecl.node;
+            if (node.nodeType === 23 /* ImportFromAs */) {
+                // from ... import X case, decl in the submodule fallback has the path.
+                continue;
+            }
+            decls.push(...(evaluator.getDeclarationsForNameNode(node.module.nameParts[0], skipUnreachableCode) || []));
+        }
+        // For now, we only support function overriding.
+        for (const decl of decls.filter((d) => (0, declaration_1.isFunctionDeclaration)(d) && d.isMethod && d.node.name.value.length > 0)) {
+            const methodDecl = decl;
+            const enclosingClass = ParseTreeUtils.getEnclosingClass(methodDecl.node);
+            const classResults = enclosingClass ? evaluator.getTypeOfClass(enclosingClass) : undefined;
+            if (!classResults) {
+                continue;
+            }
+            for (const mroClass of classResults.classType.details.mro) {
+                if ((0, types_1.isInstantiableClass)(mroClass)) {
+                    const currentMember = (0, typeUtils_1.lookUpClassMember)(mroClass, methodDecl.node.name.value);
+                    const baseMember = (0, typeUtils_1.lookUpClassMember)(mroClass, methodDecl.node.name.value, 1 /* SkipOriginalClass */);
+                    if (currentMember && !baseMember) {
+                        // Found base decl of the overridden method. Hold onto the decls.
+                        currentMember.symbol
+                            .getDeclarations()
+                            .filter((d) => (0, declaration_1.isFunctionDeclaration)(d) && d.isMethod)
+                            .forEach((d) => this._addIfUnique(decls, d));
+                    }
+                }
+            }
+        }
+        return decls;
+    }
+    static _getDeclarationsForModuleNameNode(node, evaluator) {
+        var _a, _b, _c, _d, _e;
+        (0, debug_1.assert)(((_a = node.parent) === null || _a === void 0 ? void 0 : _a.nodeType) === 37 /* ModuleName */);
+        // We don't have symbols corresponding to ModuleName in our system since those
+        // are not referenceable. but in "find all reference", we want to match those
+        // if it refers to the same module file. Code below handles different kind of
+        // ModuleName cases.
+        const moduleName = node.parent;
+        if (((_b = moduleName.parent) === null || _b === void 0 ? void 0 : _b.nodeType) === 21 /* ImportAs */ ||
+            ((_c = moduleName.parent) === null || _c === void 0 ? void 0 : _c.nodeType) === 22 /* ImportFrom */) {
+            const index = moduleName.nameParts.findIndex((n) => n === node);
+            // Special case, first module name part.
+            if (index === 0) {
+                // 1. import X or from X import ...
+                const decls = [];
+                // First, we need to put decls for module names type evaluator synthesized so that
+                // we can match both "import X" and "from X import ..."
+                decls.push(...(((_d = evaluator
+                    .getDeclarationsForNameNode(moduleName.nameParts[0])) === null || _d === void 0 ? void 0 : _d.filter((d) => (0, declaration_1.isAliasDeclaration)(d))) || []));
+                if (decls.length === 0) {
+                    return decls;
+                }
+                // ex, import X as x
+                const isImportAsWithAlias = moduleName.nameParts.length === 1 &&
+                    moduleName.parent.nodeType === 21 /* ImportAs */ &&
+                    !!moduleName.parent.alias;
+                // if "import" has alias, symbol is assigned to alias, not the module.
+                const importName = isImportAsWithAlias
+                    ? moduleName.parent.alias.value
+                    : moduleName.nameParts[0].value;
+                // And we also need to re-use "decls for X" binder has created
+                // so that it matches with decls type evaluator returns for "references for X".
+                // ex) import X or from .X import ... in init file and etc.
+                const symbolWithScope = (_e = ScopeUtils.getScopeForNode(node)) === null || _e === void 0 ? void 0 : _e.lookUpSymbolRecursive(importName);
+                if (symbolWithScope && moduleName.nameParts.length === 1) {
+                    let declsFromSymbol = [];
+                    (0, collectionUtils_1.appendArray)(declsFromSymbol, symbolWithScope.symbol.getDeclarations().filter((d) => (0, declaration_1.isAliasDeclaration)(d)));
+                    // If symbols are re-used, then find one that belong to this import statement.
+                    if (declsFromSymbol.length > 1) {
+                        declsFromSymbol = declsFromSymbol.filter((d) => {
+                            d = d;
+                            if (d.firstNamePart !== undefined) {
+                                // For multiple import statements with sub modules, decl can be re-used.
+                                // ex) import X.Y and import X.Z or from .X import ... in init file.
+                                // Decls for X will be reused for both import statements, and node will point
+                                // to first import statement. For those case, use firstNamePart instead to check.
+                                return d.firstNamePart === moduleName.nameParts[0].value;
+                            }
+                            return d.node === moduleName.parent;
+                        });
+                    }
+                    // ex, import X as x
+                    // We have decls for the alias "x" not the module name "X". Convert decls for the "X"
+                    if (isImportAsWithAlias) {
+                        declsFromSymbol = (0, declarationUtils_1.getDeclarationsWithUsesLocalNameRemoved)(declsFromSymbol);
+                    }
+                    decls.push(...declsFromSymbol);
+                }
+                return decls;
+            }
+            if (index > 0) {
+                // 2. import X.Y or from X.Y import ....
+                // For submodule "Y", we just use synthesized decls from type evaluator.
+                // Decls for these sub module don't actually exist in the system. Instead, symbol for Y in
+                // "import X.Y" hold onto synthesized module type (without any decl).
+                // And "from X.Y import ..." doesn't have any symbol associated module names.
+                // they can't be referenced in the module.
+                return evaluator.getDeclarationsForNameNode(moduleName.nameParts[index]) || [];
+            }
+            return [];
+        }
+        return [];
+    }
+}
+exports.DocumentSymbolCollector = DocumentSymbolCollector;
+//# sourceMappingURL=documentSymbolCollector.js.map

package/dist/languageService/documentSymbolCollector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"documentSymbolCollector.js","sourceRoot":"","sources":["../../../src/languageService/documentSymbolCollector.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;;;;;;;;;;;;;;;;;;AAIH,+EAAiE;AACjE,yDAOiC;AACjC,mEAIsC;AACtC,+DAAoF;AACpF,2EAA6D;AAC7D,iEAA8D;AAC9D,mEAAqD;AACrD,2DAAoE;AAEpE,6CAAsE;AACtE,qDAAkF;AAClF,mEAA2E;AAC3E,+DAAwD;AACxD,2CAAyC;AASzC,iEAAiE;AACjE,2BAA2B;AAC3B,MAAa,uBAAwB,SAAQ,iCAAe;IAqExD,YACY,WAAmB,EACnB,aAA4B,EAC5B,UAAyB,EACzB,kBAAqC,EACrC,aAAwB,EACxB,wCAAwC,KAAK,EAC7C,uBAAuB,IAAI;QAEnC,KAAK,EAAE,CAAC;QARA,gBAAW,GAAX,WAAW,CAAQ;QACnB,kBAAa,GAAb,aAAa,CAAe;QAC5B,eAAU,GAAV,UAAU,CAAe;QACzB,uBAAkB,GAAlB,kBAAkB,CAAmB;QACrC,kBAAa,GAAb,aAAa,CAAW;QACxB,0CAAqC,GAArC,qCAAqC,CAAQ;QAC7C,yBAAoB,GAApB,oBAAoB,CAAO;QAV/B,aAAQ,GAAuB,EAAE,CAAC;QAClC,wBAAmB,GAAG,IAAI,GAAG,EAAc,CAAC;QAahD,wDAAwD;QACxD,+CAA+C;QAC/C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAChD,CAAC;IAlFD,MAAM,CAAC,eAAe,CAClB,IAAc,EACd,SAAwB,EACxB,iBAAoC,EACpC,YAAwB,EACxB,oCAAoC,GAAG,KAAK,EAC5C,mBAAmB,GAAG,IAAI;QAE1B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC;QAC9B,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAC5C,IAAI,EACJ,SAAS;QACT,sBAAsB,CAAC,IAAI,EAC3B,iBAAiB,CACpB,CAAC;QAEF,YAAY,GAAG,YAAY,aAAZ,YAAY,cAAZ,YAAY,GAAI,IAAA,8BAAa,EAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,YAAY,EAAE;YACf,OAAO,EAAE,CAAC;SACb;QAED,MAAM,SAAS,GAAG,IAAI,uBAAuB,CACzC,UAAU,EACV,YAAY,EACZ,SAAS,EACT,iBAAiB,EACjB,YAAY,EACZ,oCAAoC,EACpC,mBAAmB,CACtB,CAAC;QAEF,OAAO,SAAS,CAAC,OAAO,EAAE,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,sBAAsB,CACzB,IAAc,EACd,SAAwB,EACxB,gBAAyB,EACzB,KAAwB,EACxB,YAA2B;QAE3B,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAEnE,MAAM,oBAAoB,GAAkB,EAAE,CAAC;QAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YAC1B,MAAM,YAAY,GAAG,SAAS,CAAC,uBAAuB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;YAC/E,IAAI,YAAY,EAAE;gBACd,oBAAoB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;gBAExC,IAAI,YAAY,IAAI,IAAA,yBAAU,EAAC,YAAY,CAAC,IAAI,CAAC,EAAE;oBAC/C,MAAM,SAAS,GAAG,YAAY,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;oBAC9D,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE;wBAC9B,IAAI,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE;4BAC3B,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAC;yBACrD;qBACJ;iBACJ;aACJ;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,oBAAoB,CAAC;IAChC,CAAC;IAqBD,OAAO;QACH,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC,QAAQ,CAAC;IACzB,CAAC;IAEQ,IAAI,CAAC,IAAe;QACzB,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE;YACzE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SACpB;IACL,CAAC;IAEQ,SAAS,CAAC,IAAc;QAC7B,IAAA,gDAA4B,EAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAEtD,gEAAgE;QAChE,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,WAAW,EAAE;YACjC,OAAO,KAAK,CAAC;SAChB;QAED,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;YAC/B,MAAM,YAAY,GAAG,uBAAuB,CAAC,uBAAuB,CAChE,IAAI,EACJ,IAAI,CAAC,UAAU,EACf,IAAI,CAAC,oBAAoB,CAC5B,CAAC;YAEF,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBACzC,kEAAkE;gBAClE,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,EAAE;oBACrE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;iBACzB;aACJ;SACJ;aAAM;YACH,6BAA6B;YAC7B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;SACzB;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,WAAW,CAAC,IAAgB;QACjC,IAAA,gDAA4B,EAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAEtD,IAAI,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YACpC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;SACzB;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEO,UAAU,CAAC,IAA2B;QAC1C,MAAM,KAAK,GAAc,IAAI,CAAC,QAAQ,kBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAA,wCAAuB,EAAC,IAAI,CAAC,CAAC;QACrG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACxC,CAAC;IAEO,2BAA2B,CAAC,WAAwB;QACxD,2BAA2B;QAC3B,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC,WAAW,EAAE,uBAAuB,CAAC,KAAK,CAAC,CAAC;QACzG,IAAI,CAAC,YAAY,EAAE;YACf,OAAO,KAAK,CAAC;SAChB;QAED,uEAAuE;QACvE,gDAAgD;QAChD,IACI,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC7B,IAAA,sCAAmB,EAAC,IAAI,EAAE,YAAY,EAAE,IAAI,CAAC,qCAAqC,CAAC,CACtF,EACH;YACE,OAAO,IAAI,CAAC;SACf;QAED,4EAA4E;QAC5E,yCAAyC;QACzC,MAAM,oBAAoB,GAAG,IAAI,CAAC,2BAA2B,CAAC,YAAY,CAAC,CAAC;QAC5E,IAAI,CAAC,oBAAoB,IAAI,oBAAoB,KAAK,YAAY,EAAE;YAChE,OAAO,KAAK,CAAC;SAChB;QAED,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CACpC,IAAA,sCAAmB,EAAC,IAAI,EAAE,oBAAoB,EAAE,IAAI,CAAC,qCAAqC,CAAC,CAC9F,CAAC;IACN,CAAC;IAEO,2BAA2B,CAAC,WAAwB;QACxD,8FAA8F;QAC9F,6EAA6E;QAC7E,oDAAoD;QACpD,IAAI,gCAAgC,CAAC,WAAW,CAAC,EAAE;YAC/C,OAAO,IAAA,0DAAuC,EAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;SACpE;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC,WAAW,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;QACxG,OAAO,gCAAgC,CAAC,YAAY,CAAC;YACjD,CAAC,CAAC,IAAA,0DAAuC,EAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,CAAC,CAAC,YAAY,CAAC;QAEnB,SAAS,gCAAgC,CAAC,IAAkB;YACxD,OAAO,CACH,CAAC,CAAC,IAAI;gBACN,IAAI,CAAC,IAAI,kBAA0B;gBACnC,IAAI,CAAC,IAAI;gBACT,IAAI,CAAC,aAAa;gBAClB,IAAI,CAAC,IAAI,CAAC,QAAQ,sBAA2B,CAChD,CAAC;QACN,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,IAAe;QACtC,IAAI,IAAI,CAAC,QAAQ,oBAAyB,EAAE;YACxC,OAAO;SACV;QAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC9D,IAAI,CAAC,aAAa,EAAE;YAChB,OAAO;SACV;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,WAAW,EAAE;YACd,OAAO;SACV;QAED,aAAa,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YAC7C,IAAI,UAAU,CAAC,KAAK,KAAK,IAAI,CAAC,WAAW,EAAE;gBACvC,OAAO;aACV;YAED,MAAM,aAAa,GAAG,WAAW,CAAC,qBAAqB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC1E,IAAI,CAAC,aAAa,EAAE;gBAChB,OAAO;aACV;YAED,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC1F,OAAO;aACV;YAED,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,YAA2B,EAAE,SAAsB;QAC3E,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE;YAC5B,IAAI,IAAA,sCAAmB,EAAC,GAAG,EAAE,SAAS,CAAC,EAAE;gBACrC,OAAO;aACV;SACJ;QAED,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAClC,IAAc,EACd,SAAwB,EACxB,mBAAmB,GAAG,IAAI;;QAE1B,kHAAkH;QAClH,0GAA0G;QAC1G,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,yBAA6B,EAAE;YACpD,OAAO,IAAI,CAAC,oCAAoC,CAAC,IAAI,EAAE,SAAS,EAAE,mBAAmB,CAAC,CAAC;SAC1F;QAED,OAAO,IAAI,CAAC,iCAAiC,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACnE,CAAC;IAEO,MAAM,CAAC,oCAAoC,CAC/C,IAAc,EACd,SAAwB,EACxB,mBAAmB,GAAG,IAAI;;QAE1B,IAAA,cAAM,EAAC,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,yBAA6B,CAAC,CAAC;QAE3D,IAAI,KAAK,GAAG,SAAS,CAAC,0BAA0B,CAAC,IAAI,EAAE,mBAAmB,CAAC,IAAI,EAAE,CAAC;QAClF,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,2BAA+B,EAAE;YACtD,sEAAsE;YACtE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,CAAC;SACvD;QAED,mEAAmE;QACnE,2FAA2F;QAC3F,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;YACpB,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,QAAQ,oBAAwB,EAAE;gBACxC,kCAAkC;gBAClC,OAAO,CAAC,IAAA,oDAAiC,EAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;aAC7D;SACJ;QAED,iFAAiF;QACjF,wFAAwF;QACxF,yDAAyD;QACzD,sFAAsF;QACtF,KAAK,MAAM,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,gCAAkB,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,mBAAmB,CAAC,EAAE;YAC1F,MAAM,IAAI,GAAI,SAA8B,CAAC,IAAI,CAAC;YAClD,IAAI,IAAI,CAAC,QAAQ,0BAA+B,EAAE;gBAC9C,uEAAuE;gBACvE,SAAS;aACZ;YAED,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,mBAAmB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;SAC9G;QAED,gDAAgD;QAChD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,MAAM,CAC3B,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,mCAAqB,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAChF,EAAE;YACC,MAAM,UAAU,GAAG,IAA2B,CAAC;YAC/C,MAAM,cAAc,GAAG,cAAc,CAAC,iBAAiB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACzE,MAAM,YAAY,GAAG,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC3F,IAAI,CAAC,YAAY,EAAE;gBACf,SAAS;aACZ;YAED,KAAK,MAAM,QAAQ,IAAI,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE;gBACvD,IAAI,IAAA,2BAAmB,EAAC,QAAQ,CAAC,EAAE;oBAC/B,MAAM,aAAa,GAAG,IAAA,6BAAiB,EAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBAC9E,MAAM,UAAU,GAAG,IAAA,6BAAiB,EAChC,QAAQ,EACR,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,4BAE7B,CAAC;oBACF,IAAI,aAAa,IAAI,CAAC,UAAU,EAAE;wBAC9B,iEAAiE;wBACjE,aAAa,CAAC,MAAM;6BACf,eAAe,EAAE;6BACjB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,mCAAqB,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;6BACrD,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;qBACpD;iBACJ;aACJ;SACJ;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEO,MAAM,CAAC,iCAAiC,CAAC,IAAc,EAAE,SAAwB;;QACrF,IAAA,cAAM,EAAC,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,yBAA6B,CAAC,CAAC;QAE3D,8EAA8E;QAC9E,6EAA6E;QAC7E,6EAA6E;QAC7E,oBAAoB;QACpB,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/B,IACI,CAAA,MAAA,UAAU,CAAC,MAAM,0CAAE,QAAQ,uBAA2B;YACtD,CAAA,MAAA,UAAU,CAAC,MAAM,0CAAE,QAAQ,yBAA6B,EAC1D;YACE,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;YAEhE,wCAAwC;YACxC,IAAI,KAAK,KAAK,CAAC,EAAE;gBACb,mCAAmC;gBACnC,MAAM,KAAK,GAAkB,EAAE,CAAC;gBAEhC,kFAAkF;gBAClF,uDAAuD;gBACvD,KAAK,CAAC,IAAI,CACN,GAAG,CAAC,CAAA,MAAA,SAAS;qBACR,0BAA0B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,0CAClD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,gCAAkB,EAAC,CAAC,CAAC,CAAC,KAAI,EAAE,CAAC,CACpD,CAAC;gBAEF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;oBACpB,OAAO,KAAK,CAAC;iBAChB;gBAED,oBAAoB;gBACpB,MAAM,mBAAmB,GACrB,UAAU,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC;oBACjC,UAAU,CAAC,MAAM,CAAC,QAAQ,sBAA2B;oBACrD,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC;gBAE9B,sEAAsE;gBACtE,MAAM,UAAU,GAAG,mBAAmB;oBAClC,CAAC,CAAE,UAAU,CAAC,MAAuB,CAAC,KAAM,CAAC,KAAK;oBAClD,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBAEpC,8DAA8D;gBAC9D,+EAA+E;gBAC/E,2DAA2D;gBAC3D,MAAM,eAAe,GAAG,MAAA,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,0CAAE,qBAAqB,CAAC,UAAU,CAAC,CAAC;gBAC5F,IAAI,eAAe,IAAI,UAAU,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBACtD,IAAI,eAAe,GAAkB,EAAE,CAAC;oBAExC,IAAA,6BAAW,EACP,eAAe,EACf,eAAe,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,gCAAkB,EAAC,CAAC,CAAC,CAAC,CAChF,CAAC;oBAEF,8EAA8E;oBAC9E,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE;wBAC5B,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC3C,CAAC,GAAG,CAAqB,CAAC;4BAE1B,IAAI,CAAC,CAAC,aAAa,KAAK,SAAS,EAAE;gCAC/B,wEAAwE;gCACxE,oEAAoE;gCACpE,6EAA6E;gCAC7E,iFAAiF;gCACjF,OAAO,CAAC,CAAC,aAAa,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;6BAC5D;4BAED,OAAO,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,CAAC;wBACxC,CAAC,CAAC,CAAC;qBACN;oBAED,oBAAoB;oBACpB,qFAAqF;oBACrF,IAAI,mBAAmB,EAAE;wBACrB,eAAe,GAAG,IAAA,0DAAuC,EAAC,eAAe,CAAC,CAAC;qBAC9E;oBAED,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;iBAClC;gBAED,OAAO,KAAK,CAAC;aAChB;YAED,IAAI,KAAK,GAAG,CAAC,EAAE;gBACX,wCAAwC;gBACxC,wEAAwE;gBACxE,0FAA0F;gBAC1F,qEAAqE;gBACrE,6EAA6E;gBAC7E,0CAA0C;gBAC1C,OAAO,SAAS,CAAC,0BAA0B,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;aAClF;YAED,OAAO,EAAE,CAAC;SACb;QAED,OAAO,EAAE,CAAC;IACd,CAAC;CACJ;AAlaD,0DAkaC"}

package/dist/languageService/documentSymbolProvider.d.ts

@@ -0,0 +1,37 @@
+import { CancellationToken, CompletionItemKind, DocumentSymbol, SymbolInformation, SymbolKind } from 'vscode-languageserver';
+import { AnalyzerFileInfo, ImportLookup } from '../analyzer/analyzerFileInfo';
+import { AliasDeclaration } from '../analyzer/declaration';
+import { Range } from '../common/textRange';
+import { ParseResults } from '../parser/parser';
+export interface IndexAliasData {
+    readonly originalName: string;
+    readonly modulePath: string;
+    readonly kind: SymbolKind;
+    readonly itemKind?: CompletionItemKind | undefined;
+}
+export interface IndexSymbolData {
+    readonly name: string;
+    readonly externallyVisible: boolean;
+    readonly kind: SymbolKind;
+    readonly itemKind?: CompletionItemKind | undefined;
+    readonly alias?: IndexAliasData | undefined;
+    readonly range?: Range | undefined;
+    readonly selectionRange?: Range | undefined;
+    readonly children?: IndexSymbolData[] | undefined;
+}
+export interface IndexResults {
+    readonly privateOrProtected: boolean;
+    readonly symbols: IndexSymbolData[];
+}
+export interface IndexOptions {
+    indexingForAutoImportMode: boolean;
+    forceIndexing?: boolean;
+}
+export declare type WorkspaceSymbolCallback = (symbols: SymbolInformation[]) => void;
+export declare function getIndexAliasData(importLookup: ImportLookup, declaration: AliasDeclaration): IndexAliasData | undefined;
+export declare function convertToFlatSymbols(documentUri: string, symbolList: DocumentSymbol[]): SymbolInformation[];
+export declare class DocumentSymbolProvider {
+    static getSymbolsForDocument(fileInfo: AnalyzerFileInfo | undefined, indexResults: IndexResults | undefined, parseResults: ParseResults | undefined, filePath: string, query: string, token: CancellationToken): SymbolInformation[];
+    static addHierarchicalSymbolsForDocument(fileInfo: AnalyzerFileInfo | undefined, indexResults: IndexResults | undefined, parseResults: ParseResults | undefined, symbolList: DocumentSymbol[], token: CancellationToken): void;
+    static indexSymbols(fileInfo: AnalyzerFileInfo, parseResults: ParseResults, options: IndexOptions, token: CancellationToken): IndexSymbolData[];
+}