@zzzen/pyright-internal 1.1.254

package/dist/languageService/hoverProvider.js

@@ -0,0 +1,366 @@
+"use strict";
+/*
+ * hoverProvider.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ * Author: Eric Traut
+ *
+ * Logic that maps a position within a Python program file into
+ * markdown text that is displayed when the user hovers over that
+ * position within a smart editor.
+ */
+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.convertHoverResults = exports.HoverProvider = void 0;
+const vscode_languageserver_1 = require("vscode-languageserver");
+const docStringConversion_1 = require("../analyzer/docStringConversion");
+const ParseTreeUtils = __importStar(require("../analyzer/parseTreeUtils"));
+const types_1 = require("../analyzer/types");
+const typeUtils_1 = require("../analyzer/typeUtils");
+const cancellationUtils_1 = require("../common/cancellationUtils");
+const debug_1 = require("../common/debug");
+const positionUtils_1 = require("../common/positionUtils");
+const textRange_1 = require("../common/textRange");
+const tooltipUtils_1 = require("./tooltipUtils");
+class HoverProvider {
+    static getHoverForPosition(sourceMapper, parseResults, position, format, evaluator, token) {
+        var _a;
+        (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+        const offset = (0, positionUtils_1.convertPositionToOffset)(position, parseResults.tokenizerOutput.lines);
+        if (offset === undefined) {
+            return undefined;
+        }
+        const node = ParseTreeUtils.findNodeByOffset(parseResults.parseTree, offset);
+        if (node === undefined) {
+            return undefined;
+        }
+        const results = {
+            parts: [],
+            range: {
+                start: (0, positionUtils_1.convertOffsetToPosition)(node.start, parseResults.tokenizerOutput.lines),
+                end: (0, positionUtils_1.convertOffsetToPosition)(textRange_1.TextRange.getEnd(node), parseResults.tokenizerOutput.lines),
+            },
+        };
+        if (node.nodeType === 38 /* Name */) {
+            const declarations = evaluator.getDeclarationsForNameNode(node);
+            if (declarations && declarations.length > 0) {
+                // In most cases, it's best to treat the first declaration as the
+                // "primary". This works well for properties that have setters
+                // which often have doc strings on the getter but not the setter.
+                // The one case where using the first declaration doesn't work as
+                // well is the case where an import statement within an __init__.py
+                // file uses the form "from .A import A". In this case, if we use
+                // the first declaration, it will show up as a module rather than
+                // the imported symbol type.
+                let primaryDeclaration = declarations[0];
+                if (primaryDeclaration.type === 6 /* Alias */ && declarations.length > 1) {
+                    primaryDeclaration = declarations[1];
+                }
+                this._addResultsForDeclaration(format, sourceMapper, results.parts, primaryDeclaration, node, evaluator);
+            }
+            else if (!node.parent || node.parent.nodeType !== 37 /* ModuleName */) {
+                // If we had no declaration, see if we can provide a minimal tooltip. We'll skip
+                // this if it's part of a module name, since a module name part with no declaration
+                // is a directory (a namespace package), and we don't want to provide any hover
+                // information in that case.
+                if (results.parts.length === 0) {
+                    const type = evaluator.getType(node) || types_1.UnknownType.create();
+                    let typeText = '';
+                    if ((0, types_1.isModule)(type)) {
+                        // Handle modules specially because submodules aren't associated with
+                        // declarations, but we want them to be presented in the same way as
+                        // the top-level module, which does have a declaration.
+                        typeText = '(module) ' + node.value;
+                    }
+                    else {
+                        typeText = node.value + ': ' + evaluator.printType(type, /* expandTypeAlias */ false);
+                    }
+                    this._addResultsPart(results.parts, typeText, /* python */ true);
+                    this._addDocumentationPart(format, sourceMapper, results.parts, node, evaluator, 
+                    /* resolvedDecl */ undefined);
+                }
+            }
+        }
+        else if (node.nodeType === 49 /* String */) {
+            const type = (_a = evaluator.getExpectedType(node)) === null || _a === void 0 ? void 0 : _a.type;
+            if (type !== undefined) {
+                this._tryAddPartsForTypedDictKey(format, sourceMapper, evaluator, node, type, results.parts);
+            }
+        }
+        return results.parts.length > 0 ? results : undefined;
+    }
+    static _addResultsForDeclaration(format, sourceMapper, parts, declaration, node, evaluator) {
+        var _a;
+        const resolvedDecl = evaluator.resolveAliasDeclaration(declaration, /* resolveLocalNames */ true);
+        if (!resolvedDecl) {
+            this._addResultsPart(parts, `(import) ` + node.value + this._getTypeText(node, evaluator), 
+            /* python */ true);
+            return;
+        }
+        switch (resolvedDecl.type) {
+            case 0 /* Intrinsic */: {
+                this._addResultsPart(parts, node.value + this._getTypeText(node, evaluator), /* python */ true);
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+            case 1 /* Variable */: {
+                let label = resolvedDecl.isConstant || resolvedDecl.isFinal ? 'constant' : 'variable';
+                // If the named node is an aliased import symbol, we can't call
+                // getType on the original name because it's not in the symbol
+                // table. Instead, use the node from the resolved alias.
+                let typeNode = node;
+                if (declaration.node.nodeType === 21 /* ImportAs */ ||
+                    declaration.node.nodeType === 23 /* ImportFromAs */) {
+                    if (declaration.node.alias && node !== declaration.node.alias) {
+                        if (resolvedDecl.node.nodeType === 38 /* Name */) {
+                            typeNode = resolvedDecl.node;
+                        }
+                    }
+                }
+                else if (((_a = node.parent) === null || _a === void 0 ? void 0 : _a.nodeType) === 1 /* Argument */ && node.parent.name === node) {
+                    // If this is a named argument, we would normally have received a Parameter declaration
+                    // rather than a variable declaration, but we can get here in the case of a dataclass.
+                    // Replace the typeNode with the node of the variable declaration.
+                    if (declaration.node.nodeType === 38 /* Name */) {
+                        typeNode = declaration.node;
+                    }
+                }
+                // Determine if this identifier is a type alias. If so, expand
+                // the type alias when printing the type information.
+                const type = evaluator.getType(typeNode);
+                let expandTypeAlias = false;
+                let typeVarName;
+                if (type === null || type === void 0 ? void 0 : type.typeAliasInfo) {
+                    const typeAliasInfo = (0, types_1.getTypeAliasInfo)(type);
+                    if ((typeAliasInfo === null || typeAliasInfo === void 0 ? void 0 : typeAliasInfo.name) === typeNode.value) {
+                        if ((0, types_1.isTypeVar)(type)) {
+                            label = type.details.isParamSpec ? 'param spec' : 'type variable';
+                            typeVarName = type.details.name;
+                        }
+                        else {
+                            expandTypeAlias = true;
+                            label = 'type alias';
+                        }
+                    }
+                }
+                const typeText = typeVarName || node.value + this._getTypeText(typeNode, evaluator, expandTypeAlias);
+                this._addResultsPart(parts, `(${label}) ${typeText}`, /* python */ true);
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+            case 2 /* Parameter */: {
+                this._addResultsPart(parts, '(parameter) ' + node.value + this._getTypeText(node, evaluator), 
+                /* python */ true);
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+            case 4 /* Class */:
+            case 5 /* SpecialBuiltInClass */: {
+                if (this._addInitMethodInsteadIfCallNode(format, node, evaluator, parts, sourceMapper, resolvedDecl)) {
+                    return;
+                }
+                this._addResultsPart(parts, '(class) ' + node.value, /* python */ true);
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+            case 3 /* Function */: {
+                let label = 'function';
+                if (resolvedDecl.isMethod) {
+                    const declaredType = evaluator.getTypeForDeclaration(resolvedDecl);
+                    label =
+                        declaredType && (0, typeUtils_1.isMaybeDescriptorInstance)(declaredType, /* requireSetter */ false)
+                            ? 'property'
+                            : 'method';
+                }
+                const type = evaluator.getType(node);
+                if (type && (0, types_1.isOverloadedFunction)(type)) {
+                    this._addResultsPart(parts, `(${label})\n${(0, tooltipUtils_1.getOverloadedFunctionTooltip)(type, evaluator)}`, 
+                    /* python */ true);
+                }
+                else {
+                    this._addResultsPart(parts, `(${label}) ` + node.value + this._getTypeText(node, evaluator), 
+                    /* python */ true);
+                }
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+            case 6 /* Alias */: {
+                this._addResultsPart(parts, '(module) ' + node.value, /* python */ true);
+                this._addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl);
+                break;
+            }
+        }
+    }
+    static _tryAddPartsForTypedDictKey(format, sourceMapper, evaluator, node, type, parts) {
+        // If the expected type is a TypedDict and the current node is a key entry then we can provide a tooltip
+        // with the type of the TypedDict key and its docstring, if available.
+        (0, typeUtils_1.doForEachSubtype)(type, (subtype) => {
+            var _a, _b;
+            if ((0, types_1.isClassInstance)(subtype) && types_1.ClassType.isTypedDictClass(subtype)) {
+                const entry = (_a = subtype.details.typedDictEntries) === null || _a === void 0 ? void 0 : _a.get(node.value);
+                if (entry) {
+                    // If we have already added parts for another declaration (e.g. for a union of TypedDicts that share the same key)
+                    // then we need to add a separator to prevent a visual bug.
+                    if (parts.length > 0) {
+                        parts.push({ text: '\n\n---\n' });
+                    }
+                    // e.g. (key) name: str
+                    const text = '(key) ' +
+                        node.value +
+                        ': ' +
+                        evaluator.printType(entry.valueType, /* expandTypeAlias */ false);
+                    this._addResultsPart(parts, text, /* python */ true);
+                    const declarations = (_b = subtype.details.fields.get(node.value)) === null || _b === void 0 ? void 0 : _b.getDeclarations();
+                    if (declarations !== undefined && (declarations === null || declarations === void 0 ? void 0 : declarations.length) !== 0) {
+                        // As we are just interested in the docString we don't have to worry about
+                        // anything other than the first declaration. There also shouldn't be more
+                        // than one declaration for a TypedDict key variable.
+                        const declaration = declarations[0];
+                        if (declaration.type === 1 /* Variable */ && declaration.docString !== undefined) {
+                            this._addDocumentationPartForType(format, sourceMapper, parts, subtype, declaration, evaluator);
+                        }
+                    }
+                }
+            }
+        });
+    }
+    static _addInitMethodInsteadIfCallNode(format, node, evaluator, parts, sourceMapper, declaration) {
+        // If the class is used as part of a call (i.e. it is being
+        // instantiated), include the constructor arguments within the
+        // hover text.
+        let callLeftNode = node;
+        // Allow the left to be a member access chain (e.g. a.b.c) if the
+        // node in question is the last item in the chain.
+        if (callLeftNode.parent &&
+            callLeftNode.parent.nodeType === 35 /* MemberAccess */ &&
+            node === callLeftNode.parent.memberName) {
+            callLeftNode = node.parent;
+        }
+        if (!callLeftNode ||
+            !callLeftNode.parent ||
+            callLeftNode.parent.nodeType !== 9 /* Call */ ||
+            callLeftNode.parent.leftExpression !== callLeftNode) {
+            return false;
+        }
+        // Get the init method for this class.
+        const classType = evaluator.getType(node);
+        if (!classType || !(0, types_1.isInstantiableClass)(classType)) {
+            return false;
+        }
+        const initMethodMember = (0, typeUtils_1.lookUpClassMember)(classType, '__init__', 8 /* SkipInstanceVariables */);
+        if (!initMethodMember) {
+            return false;
+        }
+        const instanceType = evaluator.getType(callLeftNode.parent);
+        const functionType = evaluator.getTypeOfMember(initMethodMember);
+        if (!instanceType || !functionType || !(0, types_1.isClassInstance)(instanceType) || !(0, types_1.isFunction)(functionType)) {
+            return false;
+        }
+        const initMethodType = evaluator.bindFunctionToClassOrObject(instanceType, functionType);
+        if (!initMethodType || !(0, types_1.isFunction)(initMethodType)) {
+            return false;
+        }
+        const functionParts = evaluator.printFunctionParts(initMethodType);
+        const classText = `${node.value}(${functionParts[0].join(', ')})`;
+        this._addResultsPart(parts, '(class) ' + classText, /* python */ true);
+        const addedDoc = this._addDocumentationPartForType(format, sourceMapper, parts, initMethodType, declaration, evaluator);
+        if (!addedDoc) {
+            this._addDocumentationPartForType(format, sourceMapper, parts, classType, declaration, evaluator);
+        }
+        return true;
+    }
+    static _getTypeText(node, evaluator, expandTypeAlias = false) {
+        const type = evaluator.getType(node) || types_1.UnknownType.create();
+        return ': ' + evaluator.printType(type, expandTypeAlias);
+    }
+    static _addDocumentationPart(format, sourceMapper, parts, node, evaluator, resolvedDecl) {
+        const type = evaluator.getType(node);
+        if (type) {
+            this._addDocumentationPartForType(format, sourceMapper, parts, type, resolvedDecl, evaluator);
+        }
+    }
+    static _addDocumentationPartForType(format, sourceMapper, parts, type, resolvedDecl, evaluator) {
+        const docStrings = (0, tooltipUtils_1.getDocumentationPartsForTypeAndDecl)(sourceMapper, type, resolvedDecl, evaluator);
+        let addedDoc = false;
+        for (const docString of docStrings) {
+            if (docString) {
+                addedDoc = true;
+                this._addDocumentationResultsPart(format, parts, docString);
+            }
+        }
+        return addedDoc;
+    }
+    static _addDocumentationResultsPart(format, parts, docString) {
+        if (docString) {
+            if (format === vscode_languageserver_1.MarkupKind.Markdown) {
+                const markDown = (0, docStringConversion_1.convertDocStringToMarkdown)(docString);
+                if (parts.length > 0 && markDown.length > 0) {
+                    parts.push({ text: '---\n' });
+                }
+                this._addResultsPart(parts, markDown);
+            }
+            else if (format === vscode_languageserver_1.MarkupKind.PlainText) {
+                this._addResultsPart(parts, (0, docStringConversion_1.convertDocStringToPlainText)(docString));
+            }
+            else {
+                (0, debug_1.fail)(`Unsupported markup type: ${format}`);
+            }
+        }
+    }
+    static _addResultsPart(parts, text, python = false) {
+        parts.push({
+            python,
+            text,
+        });
+    }
+}
+exports.HoverProvider = HoverProvider;
+function convertHoverResults(format, hoverResults) {
+    if (!hoverResults) {
+        return undefined;
+    }
+    const markupString = hoverResults.parts
+        .map((part) => {
+        if (part.python) {
+            if (format === vscode_languageserver_1.MarkupKind.Markdown) {
+                return '```python\n' + part.text + '\n```\n';
+            }
+            else if (format === vscode_languageserver_1.MarkupKind.PlainText) {
+                return part.text + '\n\n';
+            }
+            else {
+                (0, debug_1.fail)(`Unsupported markup type: ${format}`);
+            }
+        }
+        return part.text;
+    })
+        .join('')
+        .trimEnd();
+    return {
+        contents: {
+            kind: format,
+            value: markupString,
+        },
+        range: hoverResults.range,
+    };
+}
+exports.convertHoverResults = convertHoverResults;
+//# sourceMappingURL=hoverProvider.js.map

package/dist/languageService/hoverProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hoverProvider.js","sourceRoot":"","sources":["../../../src/languageService/hoverProvider.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;;;;;;;;;;;;;;;;;AAEH,iEAA6E;AAG7E,yEAA0G;AAC1G,2EAA6D;AAG7D,6CAW2B;AAC3B,qDAK+B;AAC/B,mEAA2E;AAC3E,2CAAuC;AACvC,2DAA2F;AAE3F,mDAAgD;AAGhD,iDAAmG;AAYnG,MAAa,aAAa;IACtB,MAAM,CAAC,mBAAmB,CACtB,YAA0B,EAC1B,YAA0B,EAC1B,QAAkB,EAClB,MAAkB,EAClB,SAAwB,EACxB,KAAwB;;QAExB,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,MAAM,GAAG,IAAA,uCAAuB,EAAC,QAAQ,EAAE,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACrF,IAAI,MAAM,KAAK,SAAS,EAAE;YACtB,OAAO,SAAS,CAAC;SACpB;QAED,MAAM,IAAI,GAAG,cAAc,CAAC,gBAAgB,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC7E,IAAI,IAAI,KAAK,SAAS,EAAE;YACpB,OAAO,SAAS,CAAC;SACpB;QAED,MAAM,OAAO,GAAiB;YAC1B,KAAK,EAAE,EAAE;YACT,KAAK,EAAE;gBACH,KAAK,EAAE,IAAA,uCAAuB,EAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC;gBAC9E,GAAG,EAAE,IAAA,uCAAuB,EAAC,qBAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC;aAC3F;SACJ,CAAC;QAEF,IAAI,IAAI,CAAC,QAAQ,kBAAuB,EAAE;YACtC,MAAM,YAAY,GAAG,SAAS,CAAC,0BAA0B,CAAC,IAAI,CAAC,CAAC;YAChE,IAAI,YAAY,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBACzC,iEAAiE;gBACjE,8DAA8D;gBAC9D,iEAAiE;gBACjE,iEAAiE;gBACjE,mEAAmE;gBACnE,iEAAiE;gBACjE,iEAAiE;gBACjE,4BAA4B;gBAC5B,IAAI,kBAAkB,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;gBACzC,IAAI,kBAAkB,CAAC,IAAI,kBAA0B,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;oBAC9E,kBAAkB,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;iBACxC;gBAED,IAAI,CAAC,yBAAyB,CAC1B,MAAM,EACN,YAAY,EACZ,OAAO,CAAC,KAAK,EACb,kBAAkB,EAClB,IAAI,EACJ,SAAS,CACZ,CAAC;aACL;iBAAM,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,wBAA6B,EAAE;gBAC1E,gFAAgF;gBAChF,mFAAmF;gBACnF,+EAA+E;gBAC/E,4BAA4B;gBAC5B,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC5B,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,mBAAW,CAAC,MAAM,EAAE,CAAC;oBAE7D,IAAI,QAAQ,GAAG,EAAE,CAAC;oBAClB,IAAI,IAAA,gBAAQ,EAAC,IAAI,CAAC,EAAE;wBAChB,qEAAqE;wBACrE,oEAAoE;wBACpE,uDAAuD;wBACvD,QAAQ,GAAG,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC;qBACvC;yBAAM;wBACH,QAAQ,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,qBAAqB,CAAC,KAAK,CAAC,CAAC;qBACzF;oBAED,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;oBACjE,IAAI,CAAC,qBAAqB,CACtB,MAAM,EACN,YAAY,EACZ,OAAO,CAAC,KAAK,EACb,IAAI,EACJ,SAAS;oBACT,kBAAkB,CAAC,SAAS,CAC/B,CAAC;iBACL;aACJ;SACJ;aAAM,IAAI,IAAI,CAAC,QAAQ,oBAAyB,EAAE;YAC/C,MAAM,IAAI,GAAG,MAAA,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC,0CAAE,IAAI,CAAC;YACnD,IAAI,IAAI,KAAK,SAAS,EAAE;gBACpB,IAAI,CAAC,2BAA2B,CAAC,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;aAChG;SACJ;QAED,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1D,CAAC;IAEO,MAAM,CAAC,yBAAyB,CACpC,MAAkB,EAClB,YAA0B,EAC1B,KAAsB,EACtB,WAAwB,EACxB,IAAc,EACd,SAAwB;;QAExB,MAAM,YAAY,GAAG,SAAS,CAAC,uBAAuB,CAAC,WAAW,EAAE,uBAAuB,CAAC,IAAI,CAAC,CAAC;QAClG,IAAI,CAAC,YAAY,EAAE;YACf,IAAI,CAAC,eAAe,CAChB,KAAK,EACL,WAAW,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC;YAC7D,YAAY,CAAC,IAAI,CACpB,CAAC;YACF,OAAO;SACV;QAED,QAAQ,YAAY,CAAC,IAAI,EAAE;YACvB,sBAA8B,CAAC,CAAC;gBAC5B,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;gBAChG,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;YAED,qBAA6B,CAAC,CAAC;gBAC3B,IAAI,KAAK,GAAG,YAAY,CAAC,UAAU,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;gBAEtF,+DAA+D;gBAC/D,8DAA8D;gBAC9D,wDAAwD;gBACxD,IAAI,QAAQ,GAAG,IAAI,CAAC;gBACpB,IACI,WAAW,CAAC,IAAI,CAAC,QAAQ,sBAA2B;oBACpD,WAAW,CAAC,IAAI,CAAC,QAAQ,0BAA+B,EAC1D;oBACE,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,KAAK,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE;wBAC3D,IAAI,YAAY,CAAC,IAAI,CAAC,QAAQ,kBAAuB,EAAE;4BACnD,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC;yBAChC;qBACJ;iBACJ;qBAAM,IAAI,CAAA,MAAA,IAAI,CAAC,MAAM,0CAAE,QAAQ,sBAA2B,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,IAAI,EAAE;oBACtF,uFAAuF;oBACvF,sFAAsF;oBACtF,kEAAkE;oBAClE,IAAI,WAAW,CAAC,IAAI,CAAC,QAAQ,kBAAuB,EAAE;wBAClD,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC;qBAC/B;iBACJ;gBAED,8DAA8D;gBAC9D,qDAAqD;gBACrD,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACzC,IAAI,eAAe,GAAG,KAAK,CAAC;gBAC5B,IAAI,WAA+B,CAAC;gBAEpC,IAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,aAAa,EAAE;oBACrB,MAAM,aAAa,GAAG,IAAA,wBAAgB,EAAC,IAAI,CAAC,CAAC;oBAC7C,IAAI,CAAA,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,IAAI,MAAK,QAAQ,CAAC,KAAK,EAAE;wBACxC,IAAI,IAAA,iBAAS,EAAC,IAAI,CAAC,EAAE;4BACjB,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,eAAe,CAAC;4BAClE,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;yBACnC;6BAAM;4BACH,eAAe,GAAG,IAAI,CAAC;4BACvB,KAAK,GAAG,YAAY,CAAC;yBACxB;qBACJ;iBACJ;gBAED,MAAM,QAAQ,GAAG,WAAW,IAAI,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;gBACrG,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,KAAK,KAAK,QAAQ,EAAE,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;gBACzE,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;YAED,sBAA8B,CAAC,CAAC;gBAC5B,IAAI,CAAC,eAAe,CAChB,KAAK,EACL,cAAc,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC;gBAChE,YAAY,CAAC,IAAI,CACpB,CAAC;gBACF,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;YAED,mBAA2B;YAC3B,gCAAwC,CAAC,CAAC;gBACtC,IAAI,IAAI,CAAC,+BAA+B,CAAC,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,YAAY,CAAC,EAAE;oBAClG,OAAO;iBACV;gBAED,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;gBACxE,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;YAED,qBAA6B,CAAC,CAAC;gBAC3B,IAAI,KAAK,GAAG,UAAU,CAAC;gBACvB,IAAI,YAAY,CAAC,QAAQ,EAAE;oBACvB,MAAM,YAAY,GAAG,SAAS,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC;oBACnE,KAAK;wBACD,YAAY,IAAI,IAAA,qCAAyB,EAAC,YAAY,EAAE,mBAAmB,CAAC,KAAK,CAAC;4BAC9E,CAAC,CAAC,UAAU;4BACZ,CAAC,CAAC,QAAQ,CAAC;iBACtB;gBAED,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACrC,IAAI,IAAI,IAAI,IAAA,4BAAoB,EAAC,IAAI,CAAC,EAAE;oBACpC,IAAI,CAAC,eAAe,CAChB,KAAK,EACL,IAAI,KAAK,MAAM,IAAA,2CAA4B,EAAC,IAAI,EAAE,SAAS,CAAC,EAAE;oBAC9D,YAAY,CAAC,IAAI,CACpB,CAAC;iBACL;qBAAM;oBACH,IAAI,CAAC,eAAe,CAChB,KAAK,EACL,IAAI,KAAK,IAAI,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,SAAS,CAAC;oBAC/D,YAAY,CAAC,IAAI,CACpB,CAAC;iBACL;gBAED,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;YAED,kBAA0B,CAAC,CAAC;gBACxB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;gBACzE,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;gBACvF,MAAM;aACT;SACJ;IACL,CAAC;IAEO,MAAM,CAAC,2BAA2B,CACtC,MAAkB,EAClB,YAA0B,EAC1B,SAAwB,EACxB,IAAgB,EAChB,IAAU,EACV,KAAsB;QAEtB,wGAAwG;QACxG,sEAAsE;QACtE,IAAA,4BAAgB,EAAC,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE;;YAC/B,IAAI,IAAA,uBAAe,EAAC,OAAO,CAAC,IAAI,iBAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE;gBACjE,MAAM,KAAK,GAAG,MAAA,OAAO,CAAC,OAAO,CAAC,gBAAgB,0CAAE,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChE,IAAI,KAAK,EAAE;oBACP,kHAAkH;oBAClH,2DAA2D;oBAC3D,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE;wBAClB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;qBACrC;oBAED,uBAAuB;oBACvB,MAAM,IAAI,GACN,QAAQ;wBACR,IAAI,CAAC,KAAK;wBACV,IAAI;wBACJ,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,qBAAqB,CAAC,KAAK,CAAC,CAAC;oBACtE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;oBAErD,MAAM,YAAY,GAAG,MAAA,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,0CAAE,eAAe,EAAE,CAAC;oBAC/E,IAAI,YAAY,KAAK,SAAS,IAAI,CAAA,YAAY,aAAZ,YAAY,uBAAZ,YAAY,CAAE,MAAM,MAAK,CAAC,EAAE;wBAC1D,0EAA0E;wBAC1E,0EAA0E;wBAC1E,qDAAqD;wBACrD,MAAM,WAAW,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;wBACpC,IAAI,WAAW,CAAC,IAAI,qBAA6B,IAAI,WAAW,CAAC,SAAS,KAAK,SAAS,EAAE;4BACtF,IAAI,CAAC,4BAA4B,CAC7B,MAAM,EACN,YAAY,EACZ,KAAK,EACL,OAAO,EACP,WAAW,EACX,SAAS,CACZ,CAAC;yBACL;qBACJ;iBACJ;aACJ;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,MAAM,CAAC,+BAA+B,CAC1C,MAAkB,EAClB,IAAc,EACd,SAAwB,EACxB,KAAsB,EACtB,YAA0B,EAC1B,WAAwB;QAExB,2DAA2D;QAC3D,8DAA8D;QAC9D,cAAc;QACd,IAAI,YAAY,GAA0B,IAAI,CAAC;QAE/C,iEAAiE;QACjE,kDAAkD;QAClD,IACI,YAAY,CAAC,MAAM;YACnB,YAAY,CAAC,MAAM,CAAC,QAAQ,0BAA+B;YAC3D,IAAI,KAAK,YAAY,CAAC,MAAM,CAAC,UAAU,EACzC;YACE,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC;SAC9B;QAED,IACI,CAAC,YAAY;YACb,CAAC,YAAY,CAAC,MAAM;YACpB,YAAY,CAAC,MAAM,CAAC,QAAQ,iBAAuB;YACnD,YAAY,CAAC,MAAM,CAAC,cAAc,KAAK,YAAY,EACrD;YACE,OAAO,KAAK,CAAC;SAChB;QAED,sCAAsC;QACtC,MAAM,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS,IAAI,CAAC,IAAA,2BAAmB,EAAC,SAAS,CAAC,EAAE;YAC/C,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,gBAAgB,GAAG,IAAA,6BAAiB,EAAC,SAAS,EAAE,UAAU,gCAA+C,CAAC;QAEhH,IAAI,CAAC,gBAAgB,EAAE;YACnB,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5D,MAAM,YAAY,GAAG,SAAS,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;QAEjE,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY,IAAI,CAAC,IAAA,uBAAe,EAAC,YAAY,CAAC,IAAI,CAAC,IAAA,kBAAU,EAAC,YAAY,CAAC,EAAE;YAC/F,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,cAAc,GAAG,SAAS,CAAC,2BAA2B,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAEzF,IAAI,CAAC,cAAc,IAAI,CAAC,IAAA,kBAAU,EAAC,cAAc,CAAC,EAAE;YAChD,OAAO,KAAK,CAAC;SAChB;QAED,MAAM,aAAa,GAAG,SAAS,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,GAAG,IAAI,CAAC,KAAK,IAAI,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QAElE,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,UAAU,GAAG,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,4BAA4B,CAC9C,MAAM,EACN,YAAY,EACZ,KAAK,EACL,cAAc,EACd,WAAW,EACX,SAAS,CACZ,CAAC;QACF,IAAI,CAAC,QAAQ,EAAE;YACX,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;SACrG;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,IAAc,EAAE,SAAwB,EAAE,eAAe,GAAG,KAAK;QACzF,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,mBAAW,CAAC,MAAM,EAAE,CAAC;QAC7D,OAAO,IAAI,GAAG,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IAC7D,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAChC,MAAkB,EAClB,YAA0B,EAC1B,KAAsB,EACtB,IAAc,EACd,SAAwB,EACxB,YAAqC;QAErC,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACrC,IAAI,IAAI,EAAE;YACN,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;SACjG;IACL,CAAC;IAEO,MAAM,CAAC,4BAA4B,CACvC,MAAkB,EAClB,YAA0B,EAC1B,KAAsB,EACtB,IAAU,EACV,YAAqC,EACrC,SAAwB;QAExB,MAAM,UAAU,GAAG,IAAA,kDAAmC,EAAC,YAAY,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QACpG,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE;YAChC,IAAI,SAAS,EAAE;gBACX,QAAQ,GAAG,IAAI,CAAC;gBAChB,IAAI,CAAC,4BAA4B,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;aAC/D;SACJ;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,4BAA4B,CAAC,MAAkB,EAAE,KAAsB,EAAE,SAAkB;QACtG,IAAI,SAAS,EAAE;YACX,IAAI,MAAM,KAAK,kCAAU,CAAC,QAAQ,EAAE;gBAChC,MAAM,QAAQ,GAAG,IAAA,gDAA0B,EAAC,SAAS,CAAC,CAAC;gBAEvD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;oBACzC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;iBACjC;gBAED,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;aACzC;iBAAM,IAAI,MAAM,KAAK,kCAAU,CAAC,SAAS,EAAE;gBACxC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,IAAA,iDAA2B,EAAC,SAAS,CAAC,CAAC,CAAC;aACvE;iBAAM;gBACH,IAAA,YAAI,EAAC,4BAA4B,MAAM,EAAE,CAAC,CAAC;aAC9C;SACJ;IACL,CAAC;IAEO,MAAM,CAAC,eAAe,CAAC,KAAsB,EAAE,IAAY,EAAE,MAAM,GAAG,KAAK;QAC/E,KAAK,CAAC,IAAI,CAAC;YACP,MAAM;YACN,IAAI;SACP,CAAC,CAAC;IACP,CAAC;CACJ;AA7ZD,sCA6ZC;AAED,SAAgB,mBAAmB,CAAC,MAAkB,EAAE,YAAsC;IAC1F,IAAI,CAAC,YAAY,EAAE;QACf,OAAO,SAAS,CAAC;KACpB;IAED,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK;SAClC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACV,IAAI,IAAI,CAAC,MAAM,EAAE;YACb,IAAI,MAAM,KAAK,kCAAU,CAAC,QAAQ,EAAE;gBAChC,OAAO,aAAa,GAAG,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;aAChD;iBAAM,IAAI,MAAM,KAAK,kCAAU,CAAC,SAAS,EAAE;gBACxC,OAAO,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC;aAC7B;iBAAM;gBACH,IAAA,YAAI,EAAC,4BAA4B,MAAM,EAAE,CAAC,CAAC;aAC9C;SACJ;QACD,OAAO,IAAI,CAAC,IAAI,CAAC;IACrB,CAAC,CAAC;SACD,IAAI,CAAC,EAAE,CAAC;SACR,OAAO,EAAE,CAAC;IAEf,OAAO;QACH,QAAQ,EAAE;YACN,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,YAAY;SACtB;QACD,KAAK,EAAE,YAAY,CAAC,KAAK;KAC5B,CAAC;AACN,CAAC;AA5BD,kDA4BC"}

package/dist/languageService/importAdder.d.ts

@@ -0,0 +1,24 @@
+import { CancellationToken } from 'vscode-languageserver';
+import { Declaration } from '../analyzer/declaration';
+import { ImportResolver } from '../analyzer/importResolver';
+import { TypeEvaluator } from '../analyzer/typeEvaluatorTypes';
+import { ConfigOptions } from '../common/configOptions';
+import { TextEditAction } from '../common/editAction';
+import { TextRange } from '../common/textRange';
+import { NameNode } from '../parser/parseNodes';
+import { ParseResults } from '../parser/parser';
+export interface ImportData {
+    containsUnreferenceableSymbols: boolean;
+    declarations: Map<Declaration, NameNode[]>;
+}
+export declare class ImportAdder {
+    private _configOptions;
+    private _importResolver;
+    private _evaluator;
+    constructor(_configOptions: ConfigOptions, _importResolver: ImportResolver, _evaluator: TypeEvaluator);
+    collectImportsForSymbolsUsed(parseResults: ParseResults, range: TextRange, token: CancellationToken): ImportData;
+    applyImports(result: ImportData, parseResults: ParseResults, insertionPosition: number, token: CancellationToken): TextEditAction[];
+    private _tryProcessExistingImports;
+    private _getImportInfo;
+    private _areSame;
+}