@zzzen/pyright-internal 1.2.0-dev.20230430 → 1.2.0-dev.20230507

package/dist/analyzer/program.js

@@ -31,6 +31,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.Program = exports.MaxWorkspaceIndexFileCount = void 0;
 const vscode_languageserver_1 = require("vscode-languageserver");
 const vscode_languageserver_types_1 = require("vscode-languageserver-types");
+const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
 const cancellationUtils_1 = require("../common/cancellationUtils");
 const collectionUtils_1 = require("../common/collectionUtils");
 const configOptions_1 = require("../common/configOptions");
@@ -42,23 +43,13 @@ const logTracker_1 = require("../common/logTracker");
 const pathUtils_1 = require("../common/pathUtils");
 const positionUtils_1 = require("../common/positionUtils");
 const stringUtils_1 = require("../common/stringUtils");
-const textEditTracker_1 = require("../common/textEditTracker");
 const textRange_1 = require("../common/textRange");
-const textRangeCollection_1 = require("../common/textRangeCollection");
 const timing_1 = require("../common/timing");
-const workspaceEditUtils_1 = require("../common/workspaceEditUtils");
 const autoImporter_1 = require("../languageService/autoImporter");
 const completionProvider_1 = require("../languageService/completionProvider");
-const documentSymbolCollector_1 = require("../languageService/documentSymbolCollector");
-const importAdder_1 = require("../languageService/importAdder");
-const indentationUtils_1 = require("../languageService/indentationUtils");
-const insertionPointUtils_1 = require("../languageService/insertionPointUtils");
-const referencesProvider_1 = require("../languageService/referencesProvider");
-const renameModuleProvider_1 = require("../languageService/renameModuleProvider");
 const AnalyzerNodeInfo = __importStar(require("./analyzerNodeInfo"));
 const cacheManager_1 = require("./cacheManager");
 const circularDependency_1 = require("./circularDependency");
-const declaration_1 = require("./declaration");
 const parseTreeUtils_1 = require("./parseTreeUtils");
 const scopeUtils_1 = require("./scopeUtils");
 const sourceFile_1 = require("./sourceFile");
@@ -82,6 +73,7 @@ class Program {
         this._sourceFileList = [];
         this._sourceFileMap = new Map();
         this._parsedFileCount = 0;
+        this._isEditMode = false;
         this._lookUpImport = (filePathOrModule, options) => {
             var _a;
             let sourceFileInfo;
@@ -171,6 +163,41 @@ class Program {
     dispose() {
         this._cacheManager.unregisterCacheOwner(this);
     }
+    enterEditMode() {
+        // Keep track of edit mode so we can apply it to new source files.
+        this._isEditMode = true;
+        // Tell all source files we're in edit mode.
+        this._sourceFileList.forEach((sourceFile) => {
+            sourceFile.sourceFile.enterEditMode();
+        });
+    }
+    exitEditMode() {
+        // Tell all source files we're no longer in edit mode. Gather
+        // up all of their edits.
+        const edits = [];
+        this._sourceFileList.forEach((sourceFile) => {
+            const newContents = sourceFile.sourceFile.exitEditMode();
+            if (newContents) {
+                // This means this source file was modified. We need to recompute its imports after
+                // we put it back to how it was.
+                this._updateSourceFileImports(sourceFile, this.configOptions);
+                // Create a text document so we can compute the edits.
+                const textDocument = vscode_languageserver_textdocument_1.TextDocument.create(sourceFile.sourceFile.getFilePath(), 'python', 1, sourceFile.sourceFile.getFileContent() || '');
+                // Add an edit action to the list.
+                edits.push({
+                    filePath: sourceFile.sourceFile.getFilePath(),
+                    range: {
+                        start: { line: 0, character: 0 },
+                        end: { line: textDocument.lineCount, character: 0 },
+                    },
+                    replacementText: newContents,
+                });
+            }
+        });
+        // Stop applying edit mode to new source files.
+        this._isEditMode = false;
+        return edits;
+    }
     setConfigOptions(configOptions) {
         this._configOptions = configOptions;
         this._importResolver.setConfigOptions(configOptions);
@@ -242,7 +269,7 @@ class Program {
             sourceFileInfo.isTracked = true;
             return sourceFileInfo.sourceFile;
         }
-        const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, filePath, importName, isThirdPartyImport, isInPyTypedPackage, this._console, this._logTracker);
+        const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, filePath, importName, isThirdPartyImport, isInPyTypedPackage, this._isEditMode, this._console, this._logTracker);
         sourceFileInfo = {
             sourceFile,
             isTracked: true,
@@ -266,7 +293,7 @@ class Program {
             const importName = this._getImportNameForFile(filePath);
             const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, filePath, importName, 
             /* isThirdPartyImport */ false, 
-            /* isInPyTypedPackage */ false, this._console, this._logTracker, options === null || options === void 0 ? void 0 : options.realFilePath, (_a = options === null || options === void 0 ? void 0 : options.ipythonMode) !== null && _a !== void 0 ? _a : sourceFile_1.IPythonMode.None);
+            /* isInPyTypedPackage */ false, this._isEditMode, this._console, this._logTracker, options === null || options === void 0 ? void 0 : options.realFilePath, (_a = options === null || options === void 0 ? void 0 : options.ipythonMode) !== null && _a !== void 0 ? _a : sourceFile_1.IPythonMode.None);
             const chainedFilePath = options === null || options === void 0 ? void 0 : options.chainedFilePath;
             sourceFileInfo = {
                 sourceFile,
@@ -862,14 +889,11 @@ class Program {
         if (!sourceFileInfo) {
             return undefined;
         }
-        let sourceMapper;
         const completionResult = this._logTracker.log(`completion at ${filePath}:${position.line}:${position.character}`, (ls) => {
             var _a;
             const result = this._runEvaluatorWithCancellationToken(token, () => {
                 this._bindFile(sourceFileInfo);
-                const execEnv = this._configOptions.findExecEnvironment(filePath);
-                sourceMapper = this._createSourceMapper(execEnv, token, sourceFileInfo, /* mapCompiled */ true);
-                return sourceFileInfo.sourceFile.getCompletionsForPosition(position, workspacePath, this._configOptions, this._importResolver, this._lookUpImport, this._evaluator, options, sourceMapper, nameMap, libraryMap, () => this._buildModuleSymbolsMap(sourceFileInfo, libraryMap, options.includeUserSymbolsInAutoImport, token), token);
+                return sourceFileInfo.sourceFile.getCompletionsForPosition(this, position, workspacePath, this._lookUpImport, options, nameMap, libraryMap, () => this._buildModuleSymbolsMap(sourceFileInfo, libraryMap, options.includeUserSymbolsInAutoImport, token), token);
             });
             ls.add(`found ${(_a = result === null || result === void 0 ? void 0 : result.completionMap.size) !== null && _a !== void 0 ? _a : 'null'} items`);
             return result;
@@ -882,6 +906,9 @@ class Program {
         };
         const parseResults = sourceFileInfo.sourceFile.getParseResults();
         if ((parseResults === null || parseResults === void 0 ? void 0 : parseResults.parseTree) && (parseResults === null || parseResults === void 0 ? void 0 : parseResults.text)) {
+            const execEnv = this._configOptions.findExecEnvironment(filePath);
+            const sourceMapper = this._createSourceMapper(execEnv, token, sourceFileInfo, 
+            /* mapCompiled */ true);
             const offset = (0, positionUtils_1.convertPositionToOffset)(position, parseResults.tokenizerOutput.lines);
             if (offset !== undefined && sourceMapper) {
                 await Promise.all(extensibility_1.Extensions.getProgramExtensions(parseResults.parseTree).map((e) => {
@@ -899,284 +926,8 @@ class Program {
                 return;
             }
             this._bindFile(sourceFileInfo);
-            const execEnv = this._configOptions.findExecEnvironment(filePath);
-            sourceFileInfo.sourceFile.resolveCompletionItem(this._configOptions, this._importResolver, this._lookUpImport, this._evaluator, options, this._createSourceMapper(execEnv, token, sourceFileInfo, /* mapCompiled */ true), nameMap, libraryMap, () => this._buildModuleSymbolsMap(sourceFileInfo, libraryMap, options.includeUserSymbolsInAutoImport, token), completionItem, token);
-        });
-    }
-    renameModule(path, newPath, token) {
-        return this._runEvaluatorWithCancellationToken(token, () => {
-            if ((0, pathUtils_1.isFile)(this.fileSystem, path)) {
-                const fileInfo = this.getSourceFileInfo(path);
-                if (!fileInfo) {
-                    return undefined;
-                }
-            }
-            const renameModuleProvider = renameModuleProvider_1.RenameModuleProvider.createForModule(this._importResolver, this._configOptions, this._evaluator, path, newPath, token);
-            if (!renameModuleProvider) {
-                return undefined;
-            }
-            this._processModuleReferences(renameModuleProvider, renameModuleProvider.lastModuleName, path);
-            return { edits: renameModuleProvider.getEdits(), fileOperations: [] };
-        });
-    }
-    moveSymbolAtPosition(filePath, newFilePath, position, options, token) {
-        return this._runEvaluatorWithCancellationToken(token, () => {
-            var _a, _b;
-            const sourceFileExt = (0, pathUtils_1.getFileExtension)(filePath);
-            const destFileExt = (0, pathUtils_1.getFileExtension)(newFilePath);
-            if (sourceFileExt.toLowerCase() !== destFileExt.toLowerCase()) {
-                // Don't allow moving a symbol from py to pyi or vice versa.
-                return undefined;
-            }
-            const fileInfo = this.getSourceFileInfo(filePath);
-            if (!fileInfo) {
-                return undefined;
-            }
-            const newFileInfo = this.getBoundSourceFileInfo(newFilePath);
-            if (fileInfo === newFileInfo) {
-                // Can't move symbol to the same file.
-                return undefined;
-            }
-            this._bindFile(fileInfo);
-            const parseResults = fileInfo.sourceFile.getParseResults();
-            if (!parseResults) {
-                return undefined;
-            }
-            const offset = (0, positionUtils_1.convertPositionToOffset)(position, parseResults.tokenizerOutput.lines);
-            if (offset === undefined) {
-                return undefined;
-            }
-            const node = (0, parseTreeUtils_1.findNodeByOffset)(parseResults.parseTree, offset);
-            if (node === undefined) {
-                return undefined;
-            }
-            // If this isn't a name node, there are no references to be found.
-            if (node.nodeType !== 38 /* Name */ ||
-                !renameModuleProvider_1.RenameModuleProvider.canMoveSymbol(this._configOptions, this._evaluator, node)) {
-                return undefined;
-            }
-            // We will try to
-            // 1. Find symbol to move.
-            // 2. Update all references to the symbol to new location.
-            // 3. Remove the existing symbol.
-            // 4. Insert the symbol to the destination module.
-            // 5. Insert imports required for the symbol moved to the destination module.
-            // 6. Remove import no longer needed from the original module.
-            //
-            // Here all changes are done to edits, no features in LS will apply changes to
-            // program directly. All modification is done through LSP by a edit request so
-            // things like undo or edit stacks UI works.
-            // 1. Find symbol to move.
-            const execEnv = this._configOptions.findExecEnvironment(filePath);
-            const declarations = documentSymbolCollector_1.DocumentSymbolCollector.getDeclarationsForNode(node, this._evaluator, 
-            /* resolveLocalNames */ false, documentSymbolCollector_1.DocumentSymbolCollectorUseCase.Rename, token, this._createSourceMapper(execEnv, token, fileInfo));
-            const renameModuleProvider = renameModuleProvider_1.RenameModuleProvider.createForSymbol(this._importResolver, this._configOptions, this._evaluator, filePath, newFilePath, declarations, token);
-            if (!renameModuleProvider) {
-                return undefined;
-            }
-            // 2. Update affected references.
-            this._processModuleReferences(renameModuleProvider, node.value, filePath);
-            // 3. Remove existing symbols.
-            const sourceDecl = renameModuleProvider.declarations.find((d) => d.node && (0, pathUtils_1.getFileExtension)(d.path) === sourceFileExt);
-            if (!sourceDecl) {
-                // Can't find symbol we can move.
-                return undefined;
-            }
-            const symbolRange = renameModuleProvider_1.RenameModuleProvider.getSymbolTextRange(parseResults, sourceDecl);
-            const importAdder = new importAdder_1.ImportAdder(this._configOptions, this._importResolver, this._evaluator);
-            const collectedImports = importAdder.collectImportsForSymbolsUsed(parseResults, symbolRange, token);
-            let insertionPoint = 0;
-            let insertionIndentation = 0;
-            const newFileParseResults = newFileInfo === null || newFileInfo === void 0 ? void 0 : newFileInfo.sourceFile.getParseResults();
-            if (newFileParseResults) {
-                const insertBefore = renameModuleProvider.tryGetFirstSymbolUsage(newFileParseResults);
-                insertionPoint = (0, insertionPointUtils_1.getInsertionPointForSymbolUnderModule)(this._evaluator, newFileParseResults, node.value, {
-                    symbolDeclToIgnore: sourceDecl.path,
-                    insertBefore,
-                });
-                if (insertionPoint === undefined) {
-                    // No place to insert the symbol.
-                    return undefined;
-                }
-                insertionIndentation = (0, indentationUtils_1.getModuleStatementIndentation)(newFileParseResults);
-            }
-            const reindentResult = (0, indentationUtils_1.reindentSpan)(parseResults, symbolRange, insertionIndentation);
-            const fullRange = renameModuleProvider_1.RenameModuleProvider.getSymbolFullStatementTextRange(parseResults, sourceDecl);
-            renameModuleProvider.textEditTracker.addEdit(filePath, (0, positionUtils_1.convertTextRangeToRange)(textRange_1.TextRange.combine([reindentResult.originalSpan, fullRange]), parseResults.tokenizerOutput.lines), '');
-            // 4. Add the symbol to the destination file.
-            const fileOperations = [];
-            let codeSnippetToInsert = reindentResult.text;
-            if (newFileParseResults) {
-                const range = (0, positionUtils_1.convertTextRangeToRange)({ start: insertionPoint, length: 0 }, newFileParseResults.tokenizerOutput.lines);
-                // If we are adding at the end of line (ex, end of a file),
-                // add new lines.
-                const newLinesToAdd = _getNumberOfBlankLinesToInsert(newFileParseResults, sourceDecl, range.end);
-                codeSnippetToInsert = '\n'.repeat(newLinesToAdd) + codeSnippetToInsert;
-                renameModuleProvider.textEditTracker.addEdit(newFilePath, range, codeSnippetToInsert);
-            }
-            else {
-                fileOperations.push({ kind: 'create', filePath: newFilePath });
-                renameModuleProvider.textEditTracker.addEdit(newFilePath, (0, textRange_1.getEmptyRange)(), codeSnippetToInsert);
-            }
-            // 5. Insert imports required for the symbol moved to the destination module.
-            //
-            // Since step 5 and 6 can create nested edits, we clone the program and apply all changes to re-calculate
-            // edits we need to apply to the destination file. The same workflow as `fix all` but done in program level
-            // not service level.
-            const cloned = this.clone();
-            let edits = renameModuleProvider.getEdits();
-            const textAfterSymbolAdded = (0, workspaceEditUtils_1.applyTextEditsToString)(edits.filter((v) => v.filePath === newFilePath), (_a = newFileParseResults === null || newFileParseResults === void 0 ? void 0 : newFileParseResults.tokenizerOutput.lines) !== null && _a !== void 0 ? _a : new textRangeCollection_1.TextRangeCollection([]), (_b = newFileInfo === null || newFileInfo === void 0 ? void 0 : newFileInfo.sourceFile.getFileContent()) !== null && _b !== void 0 ? _b : '');
-            _updateFileContent(cloned, newFilePath, textAfterSymbolAdded);
-            const textAfterImportsAdded = _tryGetTextAfterImportsAdded(cloned, newFilePath, collectedImports, insertionPoint, token);
-            edits = _updateFileEditActions(edits, newFilePath, newFileParseResults, textAfterSymbolAdded, textAfterImportsAdded);
-            // 6. Remove imports no longer required from original module.
-            const textAfterSymbolRemoved = (0, workspaceEditUtils_1.applyTextEditsToString)(edits.filter((v) => v.filePath === filePath), parseResults.tokenizerOutput.lines, fileInfo.sourceFile.getFileContent());
-            _updateFileContent(cloned, filePath, textAfterSymbolRemoved);
-            const textAfterUnusedImportsRemoved = _tryGetTextAfterUnusedImportsRemoved(cloned, filePath, collectedImports, 0, token);
-            edits = _updateFileEditActions(edits, filePath, parseResults, textAfterSymbolRemoved, textAfterUnusedImportsRemoved);
-            cloned.dispose();
-            return {
-                edits,
-                fileOperations,
-            };
-            function _updateFileEditActions(edits, filePath, parseResults, oldText, newText) {
-                if (newText === undefined || oldText === newText) {
-                    return edits;
-                }
-                // There were nested edits. Replace whole file.
-                edits = edits.filter((v) => v.filePath !== filePath);
-                edits.push({
-                    filePath,
-                    range: parseResults
-                        ? (0, positionUtils_1.convertTextRangeToRange)(parseResults.parseTree, parseResults.tokenizerOutput.lines)
-                        : (0, textRange_1.getEmptyRange)(),
-                    replacementText: newText,
-                });
-                return edits;
-            }
-            function _tryGetTextAfterImportsAdded(cloned, filePath, importData, insertionPoint, token) {
-                const sourceFile = cloned.getBoundSourceFile(filePath);
-                const parseResults = sourceFile === null || sourceFile === void 0 ? void 0 : sourceFile.getParseResults();
-                if (!parseResults) {
-                    return undefined;
-                }
-                const insertAddEdits = importAdder.applyImports(importData, filePath, parseResults, insertionPoint, options.importFormat, token);
-                return (0, workspaceEditUtils_1.applyTextEditsToString)(insertAddEdits, parseResults.tokenizerOutput.lines, sourceFile.getFileContent());
-            }
-            function _tryGetTextAfterUnusedImportsRemoved(cloned, filePath, importData, attempt, token) {
-                var _a;
-                (0, cancellationUtils_1.throwIfCancellationRequested)(token);
-                cloned.analyzeFile(filePath, token);
-                const sourceFile = cloned.getBoundSourceFile(filePath);
-                const parseResults = sourceFile === null || sourceFile === void 0 ? void 0 : sourceFile.getParseResults();
-                if (!parseResults) {
-                    return undefined;
-                }
-                const tracker = new textEditTracker_1.TextEditTracker();
-                for (const diagnostic of cloned
-                    .getDiagnosticsForRange(filePath, (0, positionUtils_1.convertTextRangeToRange)(parseResults.parseTree, parseResults.tokenizerOutput.lines))
-                    .filter((d) => {
-                    var _a;
-                    return d.category === 3 /* UnusedCode */ &&
-                        ((_a = d.getActions()) === null || _a === void 0 ? void 0 : _a.some((a) => a.action === "pyright.unusedImport" /* unusedImport */));
-                })) {
-                    const nameNode = (0, parseTreeUtils_1.findNodeByPosition)(parseResults.parseTree, diagnostic.range.start, parseResults.tokenizerOutput.lines);
-                    if ((nameNode === null || nameNode === void 0 ? void 0 : nameNode.nodeType) !== 38 /* Name */) {
-                        continue;
-                    }
-                    // decl is synthesized. there is no node associated with the decl.
-                    // ex) import a or import a.b
-                    const dottedName1 = ((_a = nameNode.parent) === null || _a === void 0 ? void 0 : _a.nodeType) === 37 /* ModuleName */ ? nameNode.parent.nameParts : [nameNode];
-                    for (const [decl, names] of importData.declarations) {
-                        if (decl.node) {
-                            if (textRange_1.TextRange.containsRange(decl.node, nameNode)) {
-                                tracker.removeNodes({ node: nameNode, parseResults: parseResults });
-                                break;
-                            }
-                        }
-                        const dottedName2 = (0, parseTreeUtils_1.getDottedName)((0, parseTreeUtils_1.getDottedNameWithGivenNodeAsLastName)(names[0]));
-                        if (dottedName2 && (0, collectionUtils_1.arrayEquals)(dottedName1, dottedName2, (e1, e2) => e1.value === e2.value)) {
-                            tracker.removeNodes({ node: nameNode, parseResults: parseResults });
-                            break;
-                        }
-                    }
-                }
-                const oldText = sourceFile.getFileContent();
-                const newText = (0, workspaceEditUtils_1.applyTextEditsToString)(tracker.getEdits(token).filter((v) => v.filePath === filePath), parseResults.tokenizerOutput.lines, oldText);
-                // We will attempt to remove unused imports multiple times since removing 1 unused import
-                // could make another import unused. This is due to how we calculate which import is not used.
-                // ex) import os, os.path, os.path.xxx
-                // `os.path` and `os.path.xxx` will be marked as used due to `import os`.
-                // once `os` is removed `os.path` will be marked as unused and so on.
-                // We will attempt to remove those chained unused imports up to 10 chain.
-                if (attempt > 10 || oldText === newText) {
-                    return newText;
-                }
-                _updateFileContent(cloned, filePath, newText);
-                return _tryGetTextAfterUnusedImportsRemoved(cloned, filePath, importData, attempt + 1, token);
-            }
+            sourceFileInfo.sourceFile.resolveCompletionItem(this, this._lookUpImport, options, nameMap, libraryMap, () => this._buildModuleSymbolsMap(sourceFileInfo, libraryMap, options.includeUserSymbolsInAutoImport, token), completionItem, token);
         });
-        function _updateFileContent(cloned, filePath, text) {
-            var _a, _b;
-            const info = cloned.getSourceFileInfo(filePath);
-            const version = info ? ((_a = info.sourceFile.getClientVersion()) !== null && _a !== void 0 ? _a : 0) + 1 : 0;
-            const chainedFilePath = info ? (_b = info.chainedSourceFile) === null || _b === void 0 ? void 0 : _b.sourceFile.getFilePath() : undefined;
-            const ipythonMode = info ? info.sourceFile.getIPythonMode() : sourceFile_1.IPythonMode.None;
-            const isTracked = info ? info.isTracked : true;
-            const realFilePath = info ? info.sourceFile.getRealFilePath() : filePath;
-            cloned.setFileOpened(filePath, version, text, {
-                chainedFilePath,
-                ipythonMode,
-                isTracked,
-                realFilePath,
-            });
-        }
-        function _getNumberOfBlankLinesToInsert(parseResults, decl, position) {
-            if (position.line === 0 && position.character === 0) {
-                return 0;
-            }
-            let previousStatement;
-            const offset = (0, positionUtils_1.convertPositionToOffset)(position, parseResults.tokenizerOutput.lines);
-            if (offset && parseResults.parseTree.statements.length > 0) {
-                previousStatement = parseResults.parseTree.statements.reduce((prev, curr) => offset < curr.start ? prev : curr);
-            }
-            // This basically try to add some blank lines after the last line with text.
-            let linesToAdd = 0;
-            if (previousStatement) {
-                if ((0, declaration_1.isVariableDeclaration)(decl)) {
-                    switch (previousStatement.nodeType) {
-                        case 47 /* StatementList */:
-                            // Small statement such as call, assignment, etc.
-                            linesToAdd = 0;
-                            break;
-                        case 10 /* Class */:
-                        case 28 /* Function */:
-                            linesToAdd = 2;
-                            break;
-                        default:
-                            // any other statement such as if, while, etc. we will add 1 blank line.
-                            linesToAdd = 1;
-                    }
-                }
-                else {
-                    linesToAdd = 2;
-                }
-            }
-            // If the position is not at the beginning of the line, we need to add 1 more '\n'
-            // to start from blank line.
-            linesToAdd += position.character !== 0 ? 1 : 0;
-            // If there are already blank lines, we only add the difference.
-            const desiredBlankLines = linesToAdd;
-            const startingLine = position.character !== 0 ? position.line : position.line - 1;
-            for (let i = 0; i < desiredBlankLines; i++) {
-                const currentLine = startingLine - i;
-                if (currentLine < 0 || !(0, parseTreeUtils_1.isBlankLine)(parseResults, currentLine)) {
-                    break;
-                }
-                linesToAdd--;
-            }
-            return linesToAdd;
-        }
     }
     clone() {
         var _a, _b;
@@ -1200,123 +951,6 @@ class Program {
         }
         return program;
     }
-    canRenameSymbolAtPosition(filePath, position, isDefaultWorkspace, allowModuleRename, token) {
-        return this._runEvaluatorWithCancellationToken(token, () => {
-            const sourceFileInfo = this.getSourceFileInfo(filePath);
-            if (!sourceFileInfo) {
-                return undefined;
-            }
-            this._bindFile(sourceFileInfo);
-            const referencesResult = this._getReferenceResult(sourceFileInfo, filePath, position, allowModuleRename, token);
-            if (!referencesResult) {
-                return undefined;
-            }
-            if (referencesResult.containsOnlyImportDecls &&
-                !this._supportRenameModule(referencesResult.declarations, isDefaultWorkspace)) {
-                return undefined;
-            }
-            const renameMode = this._getRenameSymbolMode(sourceFileInfo, referencesResult, isDefaultWorkspace);
-            if (renameMode === 'none') {
-                return undefined;
-            }
-            // Return the range of the symbol.
-            const parseResult = sourceFileInfo.sourceFile.getParseResults();
-            return {
-                range: (0, positionUtils_1.convertTextRangeToRange)(referencesResult.nodeAtOffset, parseResult.tokenizerOutput.lines),
-                declarations: referencesResult.declarations,
-            };
-        });
-    }
-    renameSymbolAtPosition(filePath, position, newName, isDefaultWorkspace, allowModuleRename, token) {
-        return this._runEvaluatorWithCancellationToken(token, () => {
-            var _a;
-            const sourceFileInfo = this.getSourceFileInfo(filePath);
-            if (!sourceFileInfo) {
-                return undefined;
-            }
-            this._bindFile(sourceFileInfo);
-            const referencesResult = this._getReferenceResult(sourceFileInfo, filePath, position, allowModuleRename, token);
-            if (!referencesResult) {
-                return undefined;
-            }
-            if (referencesResult.containsOnlyImportDecls) {
-                // All decls must be on a user file.
-                if (!this._supportRenameModule(referencesResult.declarations, isDefaultWorkspace)) {
-                    return undefined;
-                }
-                const moduleInfo = renameModuleProvider_1.RenameModuleProvider.getRenameModulePathInfo(renameModuleProvider_1.RenameModuleProvider.getRenameModulePath(referencesResult.declarations), newName);
-                if (!moduleInfo) {
-                    // Can't figure out module to rename.
-                    return undefined;
-                }
-                const editActions = this.renameModule(moduleInfo.filePath, moduleInfo.newFilePath, token);
-                // Add file system rename.
-                editActions === null || editActions === void 0 ? void 0 : editActions.fileOperations.push({
-                    kind: 'rename',
-                    oldFilePath: moduleInfo.filePath,
-                    newFilePath: moduleInfo.newFilePath,
-                });
-                if ((0, sourceMapper_1.isStubFile)(moduleInfo.filePath)) {
-                    const matchingFiles = this._importResolver.getSourceFilesFromStub(moduleInfo.filePath, this._configOptions.findExecEnvironment(filePath), 
-                    /* mapCompiled */ false);
-                    for (const matchingFile of matchingFiles) {
-                        const matchingFileInfo = renameModuleProvider_1.RenameModuleProvider.getRenameModulePathInfo(matchingFile, newName);
-                        if (matchingFileInfo) {
-                            editActions === null || editActions === void 0 ? void 0 : editActions.fileOperations.push({
-                                kind: 'rename',
-                                oldFilePath: matchingFileInfo.filePath,
-                                newFilePath: matchingFileInfo.newFilePath,
-                            });
-                        }
-                    }
-                }
-                return editActions;
-            }
-            const referenceProvider = new referencesProvider_1.ReferencesProvider(this, token);
-            const renameMode = this._getRenameSymbolMode(sourceFileInfo, referencesResult, isDefaultWorkspace);
-            switch (renameMode) {
-                case 'singleFileMode':
-                    referenceProvider.addReferencesToResult(sourceFileInfo.sourceFile.getFilePath(), 
-                    /* includeDeclaration */ true, referencesResult);
-                    break;
-                case 'multiFileMode': {
-                    for (const curSourceFileInfo of this._sourceFileList) {
-                        // Make sure we only add user code to the references to prevent us
-                        // from accidentally changing third party library or type stub.
-                        if ((0, sourceFileInfoUtils_1.isUserCode)(curSourceFileInfo)) {
-                            // Make sure searching symbol name exists in the file.
-                            const content = (_a = curSourceFileInfo.sourceFile.getFileContent()) !== null && _a !== void 0 ? _a : '';
-                            if (!referencesResult.symbolNames.some((s) => content.search(s) >= 0)) {
-                                continue;
-                            }
-                            this._bindFile(curSourceFileInfo, content);
-                            referenceProvider.addReferencesToResult(curSourceFileInfo.sourceFile.getFilePath(), 
-                            /* includeDeclaration */ true, referencesResult);
-                        }
-                        // This operation can consume significant memory, so check
-                        // for situations where we need to discard the type cache.
-                        this._handleMemoryHighUsage();
-                    }
-                    break;
-                }
-                case 'none':
-                    // Rename is not allowed.
-                    // ex) rename symbols from libraries.
-                    return undefined;
-                default:
-                    (0, debug_1.assertNever)(renameMode);
-            }
-            const edits = [];
-            referencesResult.locations.forEach((loc) => {
-                edits.push({
-                    filePath: loc.path,
-                    range: loc.range,
-                    replacementText: newName,
-                });
-            });
-            return { edits, fileOperations: [] };
-        });
-    }
     // Returns a value from 0 to 1 (or more) indicating how "full" the cache is
     // relative to some predetermined high-water mark. We'll compute this value
     // based on two easy-to-compute metrics: the number of entries in the type
@@ -1337,87 +971,6 @@ class Program {
     test_createSourceMapper(execEnv, from) {
         return this._createSourceMapper(execEnv, vscode_languageserver_1.CancellationToken.None, /* from */ from, /* mapCompiled */ false);
     }
-    _getRenameSymbolMode(sourceFileInfo, referencesResult, isDefaultWorkspace) {
-        // We have 2 different cases
-        // Single file mode.
-        // 1. rename on default workspace (ex, standalone file mode).
-        // 2. rename local symbols.
-        // 3. rename symbols defined in the non user open file.
-        //
-        // and Multi file mode.
-        // 1. rename public symbols defined in user files on regular workspace (ex, open folder mode).
-        const userFile = (0, sourceFileInfoUtils_1.isUserCode)(sourceFileInfo);
-        if (isDefaultWorkspace ||
-            (userFile && !referencesResult.requiresGlobalSearch) ||
-            (!userFile &&
-                sourceFileInfo.isOpenByClient &&
-                referencesResult.declarations.every((d) => this.getSourceFileInfo(d.path) === sourceFileInfo))) {
-            return 'singleFileMode';
-        }
-        if (referencesResult.declarations.every((d) => (0, sourceFileInfoUtils_1.isUserCode)(this.getSourceFileInfo(d.path)))) {
-            return 'multiFileMode';
-        }
-        // Rename is not allowed.
-        // ex) rename symbols from libraries.
-        return 'none';
-    }
-    _supportRenameModule(declarations, isDefaultWorkspace) {
-        // Rename module is not supported for standalone file and all decls must be on a user file.
-        return !isDefaultWorkspace && declarations.every((d) => (0, sourceFileInfoUtils_1.isUserCode)(this.getSourceFileInfo(d.path)));
-    }
-    _getReferenceResult(sourceFileInfo, filePath, position, allowModuleRename, token) {
-        const execEnv = this._configOptions.findExecEnvironment(filePath);
-        const referencesResult = this._getDeclarationForPosition(sourceFileInfo, position, documentSymbolCollector_1.DocumentSymbolCollectorUseCase.Rename, this._createSourceMapper(execEnv, token), token);
-        if (!referencesResult) {
-            return undefined;
-        }
-        if (allowModuleRename && referencesResult.containsOnlyImportDecls) {
-            return referencesResult;
-        }
-        if (referencesResult.nonImportDeclarations.length === 0) {
-            // There is no symbol we can rename.
-            return undefined;
-        }
-        // Use declarations that doesn't contain import decls.
-        return new referencesProvider_1.ReferencesResult(referencesResult.requiresGlobalSearch, referencesResult.nodeAtOffset, referencesResult.symbolNames, referencesResult.nonImportDeclarations, referencesResult.useCase);
-    }
-    _getDeclarationForPosition(sourceFileInfo, position, useCase, sourceMapper, token, reporter) {
-        // If we have no completed analysis job, there's nothing to do.
-        const parseResults = sourceFileInfo.sourceFile.getParseResults();
-        if (!parseResults) {
-            return undefined;
-        }
-        return referencesProvider_1.ReferencesProvider.getDeclarationForPosition(sourceMapper, parseResults, sourceFileInfo.sourceFile.getFilePath(), position, this._evaluator, reporter, useCase, token, Array.from((0, sourceFileInfoUtils_1.collectImportedByFiles)(sourceFileInfo)).map((fileInfo) => fileInfo.sourceFile));
-    }
-    _processModuleReferences(renameModuleProvider, filteringText, currentFilePath) {
-        var _a;
-        // _sourceFileList contains every user files that match "include" pattern including
-        // py file even if corresponding pyi exists.
-        for (const currentFileInfo of this._sourceFileList) {
-            // Make sure we only touch user code to prevent us
-            // from accidentally changing third party library or type stub.
-            if (!(0, sourceFileInfoUtils_1.isUserCode)(currentFileInfo)) {
-                continue;
-            }
-            // If module name isn't mentioned in the current file, skip the file
-            // except the file that got actually renamed/moved.
-            // The file that got moved might have relative import paths we need to update.
-            const filePath = currentFileInfo.sourceFile.getFilePath();
-            const content = (_a = currentFileInfo.sourceFile.getFileContent()) !== null && _a !== void 0 ? _a : '';
-            if (filePath !== currentFilePath && content.indexOf(filteringText) < 0) {
-                continue;
-            }
-            this._bindFile(currentFileInfo, content);
-            const parseResult = currentFileInfo.sourceFile.getParseResults();
-            if (!parseResult) {
-                continue;
-            }
-            renameModuleProvider.renameReferences(parseResult);
-            // This operation can consume significant memory, so check
-            // for situations where we need to discard the type cache.
-            this._handleMemoryHighUsage();
-        }
-    }
     _handleMemoryHighUsage() {
         const cacheUsage = this._cacheManager.getCacheUsage();
         // If the total cache has exceeded 75%, determine whether we should empty
@@ -1769,7 +1322,7 @@ class Program {
                 let importedFileInfo = this.getSourceFileInfo(importInfo.path);
                 if (!importedFileInfo) {
                     const importName = this._getImportNameForFile(importInfo.path);
-                    const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, importInfo.path, importName, importInfo.isThirdPartyImport, importInfo.isPyTypedPresent, this._console, this._logTracker);
+                    const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, importInfo.path, importName, importInfo.isThirdPartyImport, importInfo.isPyTypedPresent, this._isEditMode, this._console, this._logTracker);
                     importedFileInfo = {
                         sourceFile,
                         isTracked: false,
@@ -1878,7 +1431,7 @@ class Program {
         const importName = this._getImportNameForFile(filePath);
         const sourceFile = new sourceFile_1.SourceFile(this.fileSystem, filePath, importName, 
         /* isThirdPartyImport */ false, 
-        /* isInPyTypedPackage */ false, this._console, this._logTracker);
+        /* isInPyTypedPackage */ false, this._isEditMode, this._console, this._logTracker);
         const sourceFileInfo = {
             sourceFile,
             isTracked: false,