@zzzen/pyright-internal 1.2.0-dev.20260705 → 1.2.0-dev.20260719

package/dist/typeServer/notebookUriMapper.js

@@ -0,0 +1,164 @@
+"use strict";
+/*
+ * notebookUriMapper.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Maps notebook cell URIs to file-like URIs so notebook cells can be analyzed as if
+ * they were regular Python files.
+ *
+ * In a virtual workspace, a notebook might have a path like:
+ *   vscode-vfs://path/to/notebook.ipynb
+ * but its cells arrive with paths like:
+ *   vscode-notebook-cell://path/to/notebook.ipynb#cellName
+ * meaning the 'virtual' portion of the cell path is lost in the cell URI. This class
+ * resolves that by turning cell URIs into cell URIs that carry the virtual information:
+ *   vscode-notebook-cell://path/to/notebook.ipynb#cellName
+ * becomes
+ *   vscode-vfs://path/to/notebook.ipynb#cellName
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NotebookUriMapper = exports.INotebookUriMapper = void 0;
+const uri_1 = require("../common/uri/uri");
+var INotebookUriMapper;
+(function (INotebookUriMapper) {
+    function is(obj) {
+        return obj && typeof obj.parseNotebookOpen === 'function';
+    }
+    INotebookUriMapper.is = is;
+})(INotebookUriMapper || (exports.INotebookUriMapper = INotebookUriMapper = {}));
+const notebookPrefixCellFragment = 'pylancePrefixCell';
+const NotebookCellScheme = 'vscode-notebook-cell';
+const InteractiveWindowInputBoxScheme = 'vscode-interactive-input';
+const VSCodeChatCodeBlockScheme = 'vscode-chat-code-block';
+const VSCodeCopilotChatCodeBlockScheme = 'vscode-copilot-chat-code-block';
+class NotebookUriMapper {
+    constructor(_caseDetector) {
+        this._caseDetector = _caseDetector;
+        this._originalToMapped = new Map();
+        this._mappedToOriginal = new Map();
+        this._pendingDeletes = new Set();
+        this._inUseUris = new Set();
+        // Empty
+    }
+    parseNotebookOpen(params) {
+        // Clear out any pending deletes that are no longer in use. This might
+        // include the cells we're about to open.
+        this._clearPendingDeletes();
+        // First get the uri to the notebook. This is used as the base for the cells.
+        const notebookUri = uri_1.Uri.parse(params.notebookDocument.uri, this._caseDetector);
+        // If there are any cell uris in the params, stick them in the map.
+        params.cellTextDocuments.forEach((c) => {
+            const cellOriginal = uri_1.Uri.parse(c.uri, this._caseDetector);
+            const cellMapped = this._generateCellMappedUri(notebookUri, cellOriginal);
+            this._save(cellOriginal, cellMapped);
+        });
+        return notebookUri;
+    }
+    parseNotebookChange(params) {
+        // Any cell closures should be added to the pending deletes.
+        params.change.cells?.structure?.didClose?.forEach((c) => {
+            const cellOriginal = uri_1.Uri.parse(c.uri, this._caseDetector);
+            this._pendDelete(cellOriginal);
+        });
+        // Any new cell adds should be added to the cell URI map.
+        const notebookUri = uri_1.Uri.parse(params.notebookDocument.uri, this._caseDetector);
+        params.change.cells?.structure?.didOpen?.forEach((c) => {
+            const cellOriginal = uri_1.Uri.parse(c.uri, this._caseDetector);
+            const cellMapped = this._generateCellMappedUri(notebookUri, cellOriginal);
+            this._save(cellOriginal, cellMapped);
+        });
+        return notebookUri;
+    }
+    parseNotebookClose(params) {
+        // Add all of the current cells to the list of pending deletes.
+        params.cellTextDocuments.forEach((c) => {
+            const cellUri = uri_1.Uri.parse(c.uri, this._caseDetector);
+            this._pendDelete(cellUri);
+        });
+        return uri_1.Uri.parse(params.notebookDocument.uri, this._caseDetector);
+    }
+    markUsing(uri, using) {
+        if (NotebookUriMapper.isNotebookCell(uri)) {
+            // Remove or add to the in use set.
+            if (using) {
+                this._inUseUris.add(uri.key);
+            }
+            else {
+                this._inUseUris.delete(uri.key);
+            }
+        }
+    }
+    getOriginalCellUri(cellUri) {
+        // Should be in the map. Cell paths are added to
+        // the map when opening/changing a notebook.
+        return this._mappedToOriginal.get(cellUri.key) || cellUri;
+    }
+    getMappedCellUri(cellUri) {
+        // Should be in the map. Cell paths are added to the map
+        // when opening/changing a notebook.
+        return this._originalToMapped.get(cellUri.key) || cellUri;
+    }
+    getNotebookUriFromCell(cellUri) {
+        // If this is the original cell, we need the mapped cell.
+        const mappedCell = this.getMappedCellUri(cellUri);
+        // The mapped cell contains the URI from the notebook in it. Just
+        // need to remove the cell fragment and the last extension
+        return mappedCell.replaceExtension('').withQuery(mappedCell.query);
+    }
+    static isNotebookPrefixCell(uri) {
+        return uri.fragment === notebookPrefixCellFragment;
+    }
+    static isNotebookFile(uri) {
+        if (uri.scheme === VSCodeChatCodeBlockScheme || uri.scheme === VSCodeCopilotChatCodeBlockScheme) {
+            return false;
+        }
+        return uri.hasExtension('.ipynb') && uri.fragment.length <= 0;
+    }
+    static isNotebookCell(uri) {
+        if (uri.scheme === InteractiveWindowInputBoxScheme) {
+            return true;
+        }
+        // A URI with a fragment is a notebook cell if it contains the .ipynb extension (regular notebook
+        // mapped cell), uses the vscode-notebook-cell scheme (original cell URI), or contains the
+        // .interactive extension (interactive window mapped cell, e.g. untitled:/Foo.interactive.py#cellN).
+        return uri.fragment
+            ? uri.containsExtension('.ipynb') ||
+                uri.scheme === NotebookCellScheme ||
+                uri.containsExtension('.interactive')
+            : false;
+    }
+    static getNotebookPrefixUri(notebookUri) {
+        return notebookUri.addExtension('.py').withFragment(notebookPrefixCellFragment);
+    }
+    _generateCellMappedUri(notebookUri, cellUri) {
+        // Should just be the notebook URI with the cell fragment with a '.py' extension.
+        return notebookUri.addExtension('.py').withFragment(cellUri.fragment);
+    }
+    _clearPendingDeletes() {
+        const pending = [...this._pendingDeletes];
+        pending.forEach((key) => {
+            if (!this._inUseUris.has(key)) {
+                this._delete(key);
+            }
+        });
+    }
+    _pendDelete(original) {
+        this._pendingDeletes.add(original.key);
+    }
+    _save(original, mapped) {
+        this._originalToMapped.set(original.key, mapped);
+        this._mappedToOriginal.set(mapped.key, original);
+        this._pendingDeletes.delete(original.key);
+    }
+    _delete(originalKey) {
+        const mapped = this._originalToMapped.get(originalKey);
+        if (mapped) {
+            this._originalToMapped.delete(originalKey);
+            this._mappedToOriginal.delete(mapped.key);
+            this._pendingDeletes.delete(originalKey);
+        }
+    }
+}
+exports.NotebookUriMapper = NotebookUriMapper;
+//# sourceMappingURL=notebookUriMapper.js.map

package/dist/typeServer/notebookUriMapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"notebookUriMapper.js","sourceRoot":"","sources":["../../../../../src/typeServer/notebookUriMapper.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;GAiBG;;;AASH,2CAAwC;AAYxC,IAAiB,kBAAkB,CAIlC;AAJD,WAAiB,kBAAkB;IAC/B,SAAgB,EAAE,CAAC,GAAQ;QACvB,OAAO,GAAG,IAAI,OAAO,GAAG,CAAC,iBAAiB,KAAK,UAAU,CAAC;IAC9D,CAAC;IAFe,qBAAE,KAEjB,CAAA;AACL,CAAC,EAJgB,kBAAkB,kCAAlB,kBAAkB,QAIlC;AAED,MAAM,0BAA0B,GAAG,mBAAmB,CAAC;AACvD,MAAM,kBAAkB,GAAG,sBAAsB,CAAC;AAClD,MAAM,+BAA+B,GAAG,0BAA0B,CAAC;AACnE,MAAM,yBAAyB,GAAG,wBAAwB,CAAC;AAC3D,MAAM,gCAAgC,GAAG,gCAAgC,CAAC;AAE1E,MAAa,iBAAiB;IAM1B,YAA6B,aAAsC;QAAtC,kBAAa,GAAb,aAAa,CAAyB;QAL3D,sBAAiB,GAAG,IAAI,GAAG,EAAe,CAAC;QAC3C,sBAAiB,GAAG,IAAI,GAAG,EAAe,CAAC;QAC3C,oBAAe,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,eAAU,GAAG,IAAI,GAAG,EAAU,CAAC;QAGnC,QAAQ;IACZ,CAAC;IAED,iBAAiB,CAAC,MAAqC;QACnD,sEAAsE;QACtE,yCAAyC;QACzC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAE5B,6EAA6E;QAC7E,MAAM,WAAW,GAAG,SAAG,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAE/E,mEAAmE;QACnE,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnC,MAAM,YAAY,GAAG,SAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC1E,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,OAAO,WAAW,CAAC;IACvB,CAAC;IAED,mBAAmB,CAAC,MAAuC;QACvD,4DAA4D;QAC5D,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACpD,MAAM,YAAY,GAAG,SAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1D,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,yDAAyD;QACzD,MAAM,WAAW,GAAG,SAAG,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAC/E,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnD,MAAM,YAAY,GAAG,SAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,sBAAsB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;YAC1E,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,OAAO,WAAW,CAAC;IACvB,CAAC;IAED,kBAAkB,CAAC,MAAsC;QACrD,+DAA+D;QAC/D,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnC,MAAM,OAAO,GAAG,SAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YACrD,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,OAAO,SAAG,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IACtE,CAAC;IAED,SAAS,CAAC,GAAQ,EAAE,KAAc;QAC9B,IAAI,iBAAiB,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;YACxC,mCAAmC;YACnC,IAAI,KAAK,EAAE,CAAC;gBACR,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACpC,CAAC;QACL,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,OAAY;QAC3B,gDAAgD;QAChD,4CAA4C;QAC5C,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC;IAC9D,CAAC;IAED,gBAAgB,CAAC,OAAY;QACzB,wDAAwD;QACxD,oCAAoC;QACpC,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,OAAO,CAAC;IAC9D,CAAC;IAED,sBAAsB,CAAC,OAAY;QAC/B,yDAAyD;QACzD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAElD,iEAAiE;QACjE,0DAA0D;QAC1D,OAAO,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,CAAC,oBAAoB,CAAC,GAAQ;QAChC,OAAO,GAAG,CAAC,QAAQ,KAAK,0BAA0B,CAAC;IACvD,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,GAAQ;QAC1B,IAAI,GAAG,CAAC,MAAM,KAAK,yBAAyB,IAAI,GAAG,CAAC,MAAM,KAAK,gCAAgC,EAAE,CAAC;YAC9F,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,OAAO,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,GAAQ;QAC1B,IAAI,GAAG,CAAC,MAAM,KAAK,+BAA+B,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,iGAAiG;QACjG,0FAA0F;QAC1F,oGAAoG;QACpG,OAAO,GAAG,CAAC,QAAQ;YACf,CAAC,CAAC,GAAG,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAC3B,GAAG,CAAC,MAAM,KAAK,kBAAkB;gBACjC,GAAG,CAAC,iBAAiB,CAAC,cAAc,CAAC;YAC3C,CAAC,CAAC,KAAK,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,oBAAoB,CAAC,WAAgB;QACxC,OAAO,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,0BAA0B,CAAC,CAAC;IACpF,CAAC;IAEO,sBAAsB,CAAC,WAAgB,EAAE,OAAY;QACzD,iFAAiF;QACjF,OAAO,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC1E,CAAC;IAEO,oBAAoB;QACxB,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1C,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACpB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACtB,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,WAAW,CAAC,QAAa;QAC7B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,QAAa,EAAE,MAAW;QACpC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;IAEO,OAAO,CAAC,WAAmB;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC3C,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1C,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC;IACL,CAAC;CACJ;AAvJD,8CAuJC"}

package/dist/typeServer/profilingStub.d.ts

@@ -0,0 +1,10 @@
+import { Uri } from '../common/uri/uri';
+export interface ProfilingInfo {
+    profileId: number;
+    profilingFolderUri: Uri;
+    outputFileUris?: Uri[];
+}
+export interface IProfilingService {
+    startProfiling(): ProfilingInfo | undefined;
+    stopProfiling(): ProfilingInfo | undefined;
+}

package/dist/typeServer/profilingStub.js

@@ -0,0 +1,12 @@
+"use strict";
+/*
+ * profilingStub.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * The Pylance type server integrated with a profiling service. Profiling is a
+ * Pylance-only feature, so the standalone Pyright type server keeps only the shape
+ * (`ProfilingInfo`) that the program interfaces reference; there is no profiler.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=profilingStub.js.map

package/dist/typeServer/profilingStub.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"profilingStub.js","sourceRoot":"","sources":["../../../../../src/typeServer/profilingStub.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG"}

package/dist/typeServer/programTypes.d.ts

@@ -0,0 +1,136 @@
+import { TypeServerProtocol } from './protocol/typeServerProtocol';
+import { CancellationToken, Disposable, DocumentDiagnosticReport, FileEvent } from 'vscode-languageserver-protocol';
+import { AbsoluteModuleDescriptor, AnalyzerFileInfo, ImportLookupResult, LookupImportOptions } from '../analyzer/analyzerFileInfo';
+import { DunderAllInfo } from '../analyzer/analyzerNodeInfo';
+import { FlowNode } from '../analyzer/codeFlowTypes';
+import { Declaration, FunctionDeclaration } from '../analyzer/declaration';
+import { ImportResult } from '../analyzer/importResult';
+import { Scope } from '../analyzer/scope';
+import { Symbol, SymbolTable } from '../analyzer/symbol';
+import { ExpectedTypeResult } from '../analyzer/typeEvaluatorTypes';
+import { ClassType, Type } from '../analyzer/types';
+import { ConfigOptions } from '../common/configOptions';
+import { ConsoleInterface } from '../common/console';
+import { Diagnostic as PyrightDiagnostic } from '../common/diagnostic';
+import { FileEditAction } from '../common/editAction';
+import { SourceFileInfo } from '../common/extensibility';
+import { ReadOnlyFileSystem } from '../common/fileSystem';
+import { ServiceProvider } from '../common/serviceProvider';
+import { Range } from '../common/textRange';
+import { Uri } from '../common/uri/uri';
+import { ModuleNode, ParseNode } from '../parser/parseNodes';
+import { ParseFileResults, ParserOutput } from '../parser/parser';
+import { IParserOutputProvider } from './typeServerConversionTypes';
+import { ITypeServerEvaluator } from './typeServerEvaluator';
+import { ProfilingInfo } from './profilingStub';
+export interface ParseResults extends ParseFileResults {
+    moduleName: string;
+    uri: Uri;
+}
+export interface ISourceFileInfo extends SourceFileInfo {
+    getImports(): ISourceFileInfo[];
+    getImportedBy(): ISourceFileInfo[];
+    getBuiltinsImportForBinding(): ISourceFileInfo | undefined;
+    getImplicitImportForBinding(): ISourceFileInfo | undefined;
+}
+export interface ISymbolLookup {
+    getFileInfo(node: ParseNode): AnalyzerFileInfo;
+    getDeclaration(node: ParseNode): Declaration | undefined;
+    getScope(node: ParseNode): Scope | undefined;
+    getDunderAllInfo(node: ParseNode): DunderAllInfo | undefined;
+    getImportInfo(node: ParseNode): ImportResult | undefined;
+    getFlowNode(node: ParseNode): FlowNode | undefined;
+    getSymbolsForFile(fileUri: Uri, skipFileNeededCheck?: boolean, token?: CancellationToken): SymbolTable | undefined;
+    getSymbolsForNode(node: ParseNode, token?: CancellationToken): SymbolTable | undefined;
+    lookupSymbol(scopingNode: ParseNode, name: string, skipFileNeededCheck?: boolean, token?: CancellationToken): Symbol | undefined;
+    getScopeIdForNode(node: ParseNode): string;
+    getMatchingFileInfos(fileId: string): AnalyzerFileInfo[];
+}
+export interface ISourceMapper {
+    findDeclarations(decl: Declaration): Declaration[];
+    findDeclarationsByType(originatedPath: Uri, type: ClassType, useTypeAlias: boolean): Declaration[];
+    findFunctionDeclarations(decl: FunctionDeclaration): FunctionDeclaration[];
+    findClassDeclarationsByType(uri: Uri, type: ClassType): Declaration[];
+    getSourcePathsFromStub(stubUri: Uri, fromFile: Uri | undefined): Uri[];
+    getModuleNode(uri: Uri): ModuleNode | undefined;
+    findModules(stubFile: Uri): ModuleNode[];
+    getFileInfo(node: ParseNode): AnalyzerFileInfo;
+}
+export interface IProgramBase extends IParserOutputProvider {
+    readonly id: string;
+    readonly rootPath: Uri;
+    readonly console: ConsoleInterface;
+    readonly configOptions: ConfigOptions;
+    readonly fileSystem: ReadOnlyFileSystem;
+    readonly serviceProvider: ServiceProvider;
+    readonly isAlive: boolean;
+    readonly isPyright: boolean;
+    readonly performsAnalysis: boolean;
+    readonly supportsPullDiagnostics: boolean;
+    owns(uri: Uri): boolean;
+    getParserOutput(fileUri: Uri): ParserOutput | undefined;
+    getParseResults(fileUri: Uri): ParseResults | undefined;
+    getModuleName(fileUri: Uri): string | undefined;
+    /**
+     * Non-side-effecting existence probe. Returns true only if the program
+     * already has an `ISourceFileInfo` registered for `fileUri`. Unlike
+     * `getSourceFileInfo`, this does NOT lazily materialize a new entry, so
+     * it is safe to use when callers (e.g. workspace ownership checks) only
+     * want to know whether a file is already tracked.
+     */
+    hasSourceFile(fileUri: Uri): boolean;
+    getSourceFileInfo(fileUri: Uri): ISourceFileInfo | undefined;
+    getTrackedFileList(): readonly ISourceFileInfo[];
+    lookupImport(fileUriOrModule: Uri | AbsoluteModuleDescriptor, options?: LookupImportOptions): ImportLookupResult | undefined;
+}
+export interface ITypeProvider {
+    getComputedType(arg: ParseNode | Declaration, token: CancellationToken): Type | undefined;
+    getExpectedType(arg: ParseNode | Declaration, token: CancellationToken): ExpectedTypeResult | undefined;
+    getDeclaredType(arg: ParseNode | Declaration, token: CancellationToken): Type | undefined;
+}
+export interface IProgram extends IProgramBase, ITypeProvider, Disposable {
+    readonly symbolLookup: ISymbolLookup;
+    createEvaluator(): ITypeServerEvaluator;
+    getSnapshot(token: CancellationToken): number;
+    run<T>(callback: (p: IProgram) => T, token: CancellationToken): T;
+    runEditMode<T>(callback: (p: IProgram) => T, token: CancellationToken): FileEditAction[];
+    enterEditMode(): void;
+    exitEditMode(): FileEditAction[];
+    addInterimFile(uri: Uri): IProgram;
+    resolveImport(sourceUri: Uri, moduleDescriptor: TypeServerProtocol.ModuleName, token: CancellationToken): Uri | undefined;
+    getPythonSearchPaths(token: CancellationToken): Uri[] | undefined;
+    getSourceMapper(fileUri: Uri, mapCompiled: boolean, preferStubs: boolean, token: CancellationToken): ISourceMapper | undefined;
+    getDocumentDiagnostics(uri: Uri, previousResultId: string | undefined, token: CancellationToken): DocumentDiagnosticReport;
+    getDiagnosticsForRangeWithoutFileIgnore?(fileUri: Uri, range: Range): readonly PyrightDiagnostic[];
+    getCachedTypeForDeclaration(decl: Declaration): Type | undefined;
+    setCachedTypeForDeclaration(decl: Declaration, type: Type): void;
+    getCachedProtocolDecl(tspDecl: TypeServerProtocol.Declaration): Declaration | undefined;
+    setCachedProtocolDecl(tspDecl: TypeServerProtocol.Declaration, decl: Declaration): void;
+    startProfiling(): ProfilingInfo | undefined;
+    stopProfiling(): ProfilingInfo | undefined;
+    updateFileContents(uri: Uri, newContents: string): void;
+    startWorkspaceDiagnostics(partialResultToken: string): void;
+    stopWorkspaceDiagnostics(): void;
+    /**
+     * Add a file to be tracked by the program for indexing purposes.
+     * Creates a source file info entry if it doesn't exist.
+     */
+    addTrackedFile(uri: Uri, isThirdPartyImport: boolean, isInPyTypedPackage: boolean): void;
+    /**
+     * Set the list of tracked files for the program.
+     * Clears existing tracked state and sets the specified files as tracked.
+     */
+    setTrackedFiles(uris: Uri[]): void;
+    /**
+     * Handle high memory usage by clearing caches or performing cleanup.
+     * Used by long workspace-wide walks (e.g. find-all-references / rename
+     * transitive seed discovery) to shed the type cache between files, mirroring
+     * the sync `ProgramView.handleMemoryHighUsage`. May be a no-op for
+     * implementations that don't manage their own in-process memory.
+     */
+    handleMemoryHighUsage(): void;
+    /**
+     * Notify the program that watched files have changed.
+     */
+    changedWatchedFiles(changes: FileEvent[]): void;
+}

package/dist/typeServer/programTypes.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=programTypes.js.map

package/dist/typeServer/programTypes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"programTypes.js","sourceRoot":"","sources":["../../../../../src/typeServer/programTypes.ts"],"names":[],"mappings":""}

package/dist/typeServer/programWrapper.d.ts

@@ -0,0 +1,134 @@
+/**
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * programWrapper.ts
+ *
+ * Wraps Pyright's `Program` in the `IProgram` interface the type server consumes.
+ *
+ * The wrapper is thin: every method delegates synchronously to the underlying `Program`.
+ * It exists to (a) present the `IProgram` contract the conversion layer is written against,
+ * (b) reshape Pyright types (`SourceFileInfo`, `ParseFileResults`, `SourceMapper`) into the
+ * type-server-facing shapes, and (c) maintain a monotonic `snapshot` content-version counter
+ * (on the shared `ITypeCache`) that the server exposes to clients for cache coherence.
+ *
+ * Cancellation is handled the same way as everywhere else in Pyright: work runs inside
+ * `runWithCancellationToken`, and the evaluator polls the token, so long-running synchronous
+ * queries can be interrupted mid-request.
+ */
+import { TypeServerProtocol } from './protocol/typeServerProtocol';
+import { CancellationToken, DocumentDiagnosticReport, FileEvent } from 'vscode-languageserver-protocol';
+import { AbsoluteModuleDescriptor, ImportLookupResult, LookupImportOptions } from '../analyzer/analyzerFileInfo';
+import { Declaration } from '../analyzer/declaration';
+import { Program } from '../analyzer/program';
+import { ExpectedTypeResult } from '../analyzer/typeEvaluatorTypes';
+import { Type } from '../analyzer/types';
+import { ConfigOptions } from '../common/configOptions';
+import { ConsoleInterface } from '../common/console';
+import { FileEditAction } from '../common/editAction';
+import { ProgramView } from '../common/extensibility';
+import { ReadOnlyFileSystem } from '../common/fileSystem';
+import { ServiceProvider } from '../common/serviceProvider';
+import { Uri } from '../common/uri/uri';
+import { ParseNode } from '../parser/parseNodes';
+import { ParserOutput } from '../parser/parser';
+import { ITypeServerEvaluator } from './typeServerEvaluator';
+import { INotebookUriMapper } from './notebookUriMapper';
+import { ProfilingInfo } from './profilingStub';
+import { IProgram, ISourceFileInfo, ISourceMapper, ISymbolLookup, ParseResults } from './programTypes';
+import { ITypeCache } from './typeCache';
+/**
+ * Adapter that lets the in-proc Pyright `Program` (which is mutable and has no
+ * native snapshot concept) satisfy the `IProgram` contract the type server's
+ * conversion layer is written against.
+ *
+ *   - `run` runs the callback over the live `Program` via `_program.run(...)`.
+ *     Because Pyright's evaluator is synchronous the callback executes atomically,
+ *     so it simply receives `this` — there is no copy-on-write snapshot view.
+ *   - A monotonic `snapshot` counter (on the shared `ITypeCache`) is bumped by the
+ *     mutating methods below. It is a protocol-level content version: the server
+ *     hands it to clients via `typeServer/getSnapshot`, rejects stale requests at
+ *     the request boundary, and fires `SnapshotChangedNotification` on change.
+ *   - Cross-request state (`_declTypeCache`, `_protocolDeclCache`, `_addedStubs`, …)
+ *     lives on the wrapper, not on the `Program`, which has nowhere to put it, and
+ *     is invalidated when the snapshot increments.
+ */
+export declare class ProgramWrapper implements IProgram {
+    private readonly _program;
+    private _cache;
+    private _cachedSearchPaths;
+    private _declTypeCache;
+    private _protocolDeclCache;
+    private _disableSnapshotIncrement;
+    private _addedStubs;
+    private _inEditMode;
+    constructor(_program: Program, _cache: ITypeCache);
+    get id(): string;
+    get typeCache(): ITypeCache;
+    get rootPath(): Uri;
+    get console(): ConsoleInterface;
+    get configOptions(): ConfigOptions;
+    get fs(): import("../common/fileSystem").FileSystem;
+    get fileSystem(): ReadOnlyFileSystem;
+    get serviceProvider(): ServiceProvider;
+    get uriMapper(): INotebookUriMapper | undefined;
+    get isAlive(): boolean;
+    get isPyright(): boolean;
+    get performsAnalysis(): boolean;
+    get supportsPullDiagnostics(): boolean;
+    get symbolLookup(): ISymbolLookup;
+    runEditMode<T>(callback: (p: IProgram) => T, token: CancellationToken): FileEditAction[];
+    enterEditMode(): void;
+    exitEditMode(): FileEditAction[];
+    run<T>(callback: (p: IProgram) => T, token: CancellationToken): T;
+    createEvaluator(): ITypeServerEvaluator;
+    getCachedTypeForDeclaration(decl: Declaration): Type | undefined;
+    setCachedTypeForDeclaration(decl: Declaration, type: Type): void;
+    getCachedProtocolDecl(tspDecl: TypeServerProtocol.Declaration): Declaration | undefined;
+    setCachedProtocolDecl(tspDecl: TypeServerProtocol.Declaration, decl: Declaration): void;
+    getSourceMapper(fileUri: Uri, mapCompiled: boolean, preferStubs: boolean, token: CancellationToken): ISourceMapper | undefined;
+    getUri(node: ParseNode): Uri;
+    isCaseSensitive(uri: string): boolean;
+    /**
+     * Ensures the file is analyzed and the evaluator's prefetched type cache is warmed up.
+     * This avoids expensive re-evaluation when the cache is cold (e.g., first request for a file
+     * that imports large libraries).
+     */
+    ensureFileAnalyzed(fileUri: Uri, token: CancellationToken): void;
+    getComputedType(arg: ParseNode | Declaration, token: CancellationToken): Type | undefined;
+    getExpectedType(arg: ParseNode | Declaration, token: CancellationToken): ExpectedTypeResult | undefined;
+    getDeclaredType(arg: ParseNode | Declaration, token: CancellationToken): Type | undefined;
+    startProfiling(): ProfilingInfo | undefined;
+    stopProfiling(): ProfilingInfo | undefined;
+    startWorkspaceDiagnostics(partialResultToken: string): void;
+    stopWorkspaceDiagnostics(): void;
+    getPythonSearchPaths(token: CancellationToken): Uri[] | undefined;
+    lookupImport(fileUriOrModule: Uri | AbsoluteModuleDescriptor, options?: LookupImportOptions): ImportLookupResult | undefined;
+    owns(uri: Uri): boolean;
+    getParserOutput(fileUri: Uri): ParserOutput | undefined;
+    getParseResults(fileUri: Uri): ParseResults | undefined;
+    addStubCode(code: string, directoryUri?: Uri): {
+        uri: Uri;
+        parseResults: ParseResults;
+    };
+    getModuleName(fileUri: Uri): string | undefined;
+    getSourceFileInfo(fileUri: Uri): ISourceFileInfo | undefined;
+    hasSourceFile(fileUri: Uri): boolean;
+    getTrackedFileList(): readonly ISourceFileInfo[];
+    updateFileContents(uri: Uri, newContents: string): void;
+    addTrackedFile(uri: Uri, isThirdPartyImport: boolean, isInPyTypedPackage: boolean): void;
+    setTrackedFiles(uris: Uri[]): void;
+    handleMemoryHighUsage(): void;
+    getSnapshot(token: CancellationToken): number;
+    getDocumentDiagnostics(uri: Uri, previousResultId: string | undefined, token: CancellationToken): DocumentDiagnosticReport;
+    resolveImport(source: Uri, moduleDescriptor: TypeServerProtocol.ModuleName, token: CancellationToken): Uri | undefined;
+    changedWatchedFiles(changes: FileEvent[]): void;
+    addInterimFile(uri: Uri): IProgram;
+    dispose(): void;
+    private _massageTypeResult;
+    private _massageExpectedTypeResult;
+    private _makeSourceFileInfo;
+    private _incrementSnapshot;
+    private _getExecEnv;
+}
+export declare function makeProgram(program: ProgramView, cache?: ITypeCache): IProgram;