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

package/dist/tests/typeServer/typeServer.virtualFileRedirect.test.js

@@ -0,0 +1,298 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+/*
+ * typeServer.virtualFileRedirect.test.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Tests for the TspSupplemental virtual file redirect notifications in the TypeServer.
+ * Verifies that the type server correctly handles setVirtualFileRedirect /
+ * removeVirtualFileRedirect notifications, re-reads content, and triggers reanalysis,
+ * including observable type changes.
+ */
+const assert_1 = __importDefault(require("assert"));
+const tspSupplemental_1 = require("../../typeServer/protocol/tspSupplemental");
+const typeServerProtocol_1 = require("../../typeServer/protocol/typeServerProtocol");
+const inProcTypeServerTestUtils_1 = require("./inProcTypeServerTestUtils");
+jest.setTimeout(120000);
+/** Extract the class name from a protocol Type, if it's a ClassType with a regular declaration. */
+function getClassTypeName(type) {
+    if (!type || type.kind !== 3 /* TypeServerProtocol.TypeKind.Class */) {
+        return undefined;
+    }
+    const classType = type;
+    if (classType.declaration.kind === 0 /* TypeServerProtocol.DeclarationKind.Regular */) {
+        return classType.declaration.name;
+    }
+    return undefined;
+}
+describe('TypeServer virtual file redirect (TspSupplemental)', () => {
+    beforeAll(async () => {
+        await (0, inProcTypeServerTestUtils_1.initializeDependenciesForInProcTests)();
+    });
+    test('setVirtualFileRedirect triggers reanalysis and changes inferred types', async () => {
+        const code = `
+// @filename: mymod.py
+//// x: int = 1
+//// # /*mod_marker*/
+
+// @filename: virtual_mymod.py
+//// x: str = "hello"
+
+// @filename: test.py
+//// from mymod import x
+//// [|/*y*/y|] = x
+//// # /*test_marker*/
+`;
+        await (0, inProcTypeServerTestUtils_1.withInProcTypeServer)(code, async (context) => {
+            // Open the module file and test file in the type server.
+            await context.openFileForMarker('test_marker');
+            await context.openFileForMarker('mod_marker');
+            // Get the node for y so we can query its computed type.
+            const yNode = context.getNodeForMarker('y');
+            // Get the initial snapshot (ensures analysis has completed).
+            const initialSnapshot = await context.refreshSnapshot();
+            // Verify y's type is int before the redirect.
+            const typeBefore = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: yNode,
+            });
+            (0, assert_1.default)(typeBefore, 'y should have a computed type before redirect');
+            assert_1.default.strictEqual(getClassTypeName(typeBefore), 'int', 'y should be int before redirect');
+            // Listen for snapshot change — this proves the type server re-analyzed.
+            const snapshotChanged = new Promise((resolve) => {
+                context.onNotification(typeServerProtocol_1.TypeServerProtocol.SnapshotChangedNotification.type, (params) => {
+                    resolve(params);
+                });
+            });
+            // Get the file URIs from fourslash data.
+            const mymodFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/mymod.py'));
+            (0, assert_1.default)(mymodFile, 'mymod.py should exist in fourslash files');
+            const virtualFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/virtual_mymod.py'));
+            (0, assert_1.default)(virtualFile, 'virtual_mymod.py should exist in fourslash files');
+            // Send the virtual file redirect notification to the type server.
+            context.sendNotification(tspSupplemental_1.TspSupplemental.SetVirtualFileRedirectNotification.type, {
+                realUri: mymodFile.fileUri.toString(),
+                virtualUri: virtualFile.fileUri.toString(),
+            });
+            // Wait for the snapshot change (indicates type server processed the notification
+            // and re-analyzed).
+            const params = await snapshotChanged;
+            (0, assert_1.default)(params.new > params.old, 'Snapshot should have incremented after redirect');
+            (0, assert_1.default)(params.old >= initialSnapshot, 'Old snapshot should be >= initial');
+            // Verify y's type changed to str after the redirect (mymod.py now serves virtual content).
+            const typeAfter = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: yNode,
+            });
+            (0, assert_1.default)(typeAfter, 'y should have a computed type after redirect');
+            assert_1.default.strictEqual(getClassTypeName(typeAfter), 'str', 'y should be str after redirect');
+        });
+    });
+    test('removeVirtualFileRedirect triggers reanalysis and restores original types', async () => {
+        const code = `
+// @filename: mymod.py
+//// x: int = 1
+//// # /*mod_marker*/
+
+// @filename: virtual_mymod.py
+//// x: str = "hello"
+
+// @filename: test.py
+//// from mymod import x
+//// [|/*y*/y|] = x
+//// # /*test_marker*/
+`;
+        await (0, inProcTypeServerTestUtils_1.withInProcTypeServer)(code, async (context) => {
+            // Open the module file and test file.
+            await context.openFileForMarker('test_marker');
+            await context.openFileForMarker('mod_marker');
+            // Get the node for y so we can query its computed type.
+            const yNode = context.getNodeForMarker('y');
+            const mymodFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/mymod.py'));
+            (0, assert_1.default)(mymodFile, 'mymod.py should exist in fourslash files');
+            const virtualFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/virtual_mymod.py'));
+            (0, assert_1.default)(virtualFile, 'virtual_mymod.py should exist in fourslash files');
+            // Verify y's type is int before any redirect.
+            await context.refreshSnapshot();
+            const typeOriginal = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: yNode,
+            });
+            assert_1.default.strictEqual(getClassTypeName(typeOriginal), 'int', 'y should be int before redirect');
+            // Add the redirect (mymod.py → virtual_mymod.py).
+            context.sendNotification(tspSupplemental_1.TspSupplemental.SetVirtualFileRedirectNotification.type, {
+                realUri: mymodFile.fileUri.toString(),
+                virtualUri: virtualFile.fileUri.toString(),
+            });
+            // Wait for the snapshot from the add.
+            await context.refreshSnapshot();
+            // Verify y's type is now str.
+            const typeRedirected = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, { arg: yNode });
+            assert_1.default.strictEqual(getClassTypeName(typeRedirected), 'str', 'y should be str after redirect');
+            // Now listen for the next snapshot change.
+            const snapshotChanged = new Promise((resolve) => {
+                context.onNotification(typeServerProtocol_1.TypeServerProtocol.SnapshotChangedNotification.type, (params) => {
+                    resolve(params);
+                });
+            });
+            // Remove the redirect.
+            context.sendNotification(tspSupplemental_1.TspSupplemental.RemoveVirtualFileRedirectNotification.type, {
+                realUri: mymodFile.fileUri.toString(),
+            });
+            // Wait for the snapshot change from the removal.
+            const params = await snapshotChanged;
+            (0, assert_1.default)(params.new > params.old, 'Snapshot should have incremented after removing redirect');
+            // Verify y's type is back to int after removal.
+            const typeRestored = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: yNode,
+            });
+            assert_1.default.strictEqual(getClassTypeName(typeRestored), 'int', 'y should be int after redirect removal');
+        });
+    });
+    test('redirect with annotated class fields updates field types', async () => {
+        // Simulates what happens when the Django sidecar produces a virtual
+        // file with type-annotated fields: the original has plain assignments,
+        // the virtual file has annotations.
+        const code = `
+// @filename: models.py
+//// class Author:
+////     name = "default"
+//// # /*mod_marker*/
+
+// @filename: virtual_models.py
+//// class Author:
+////     name: str = "default"
+
+// @filename: test.py
+//// from models import Author
+//// a = Author()
+//// [|/*result*/result|] = a.name
+//// # /*test_marker*/
+`;
+        await (0, inProcTypeServerTestUtils_1.withInProcTypeServer)(code, async (context) => {
+            await context.openFileForMarker('test_marker');
+            await context.openFileForMarker('mod_marker');
+            const resultNode = context.getNodeForMarker('result');
+            await context.refreshSnapshot();
+            // Before redirect: `name = "default"` is inferred as str.
+            const typeBefore = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: resultNode,
+            });
+            (0, assert_1.default)(typeBefore, 'result should have a computed type before redirect');
+            assert_1.default.strictEqual(getClassTypeName(typeBefore), 'str', 'result should be str before redirect');
+            // Apply redirect to the annotated virtual file.
+            const modelsFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/models.py'));
+            (0, assert_1.default)(modelsFile, 'models.py should exist');
+            const virtualFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/virtual_models.py'));
+            (0, assert_1.default)(virtualFile, 'virtual_models.py should exist');
+            context.sendNotification(tspSupplemental_1.TspSupplemental.SetVirtualFileRedirectNotification.type, {
+                realUri: modelsFile.fileUri.toString(),
+                virtualUri: virtualFile.fileUri.toString(),
+            });
+            await context.refreshSnapshot();
+            // After redirect: `name: str = "default"` — still str, but now explicitly annotated.
+            const typeAfter = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: resultNode,
+            });
+            (0, assert_1.default)(typeAfter, 'result should have a computed type after redirect');
+            assert_1.default.strictEqual(getClassTypeName(typeAfter), 'str', 'result should still be str after redirect');
+        });
+    });
+    test('redirect changes field type when annotation differs from inferred', async () => {
+        // Virtual file adds a type annotation that widens the field type,
+        // simulating the sidecar annotating a field as Any.
+        const code = `
+// @filename: models.py
+//// class Author:
+////     name = "default"
+//// # /*mod_marker*/
+
+// @filename: virtual_models.py
+//// from typing import Any
+//// class Author:
+////     name: Any = "default"
+
+// @filename: test.py
+//// from models import Author
+//// a = Author()
+//// [|/*result*/result|] = a.name
+//// # /*test_marker*/
+`;
+        await (0, inProcTypeServerTestUtils_1.withInProcTypeServer)(code, async (context) => {
+            await context.openFileForMarker('test_marker');
+            await context.openFileForMarker('mod_marker');
+            const resultNode = context.getNodeForMarker('result');
+            await context.refreshSnapshot();
+            // Before redirect: inferred as str.
+            const typeBefore = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: resultNode,
+            });
+            assert_1.default.strictEqual(getClassTypeName(typeBefore), 'str', 'result should be str before redirect');
+            // Apply redirect to virtual file where name is annotated as Any.
+            const modelsFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/models.py'));
+            const virtualFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/virtual_models.py'));
+            (0, assert_1.default)(modelsFile && virtualFile, 'both files should exist');
+            context.sendNotification(tspSupplemental_1.TspSupplemental.SetVirtualFileRedirectNotification.type, {
+                realUri: modelsFile.fileUri.toString(),
+                virtualUri: virtualFile.fileUri.toString(),
+            });
+            await context.refreshSnapshot();
+            // After redirect: name is annotated as Any, so result should no longer be str.
+            const typeAfter = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: resultNode,
+            });
+            (0, assert_1.default)(typeAfter, 'result should have a computed type after redirect');
+            // Any is not a ClassType, it has its own kind.
+            assert_1.default.notStrictEqual(getClassTypeName(typeAfter), 'str', 'result should no longer be str after redirect');
+        });
+    });
+    test('redirect with synthetic members makes new fields visible', async () => {
+        // Simulates the sidecar injecting synthetic members (id, pk)
+        // into the virtual file.
+        const code = `
+// @filename: models.py
+//// class Author:
+////     name = "default"
+//// # /*mod_marker*/
+
+// @filename: virtual_models.py
+//// class Author:
+////     name: str = "default"
+////     id: int
+////     pk: int
+
+// @filename: test.py
+//// from models import Author
+//// a = Author()
+//// [|/*id_val*/id_val|] = a.id
+//// # /*test_marker*/
+`;
+        await (0, inProcTypeServerTestUtils_1.withInProcTypeServer)(code, async (context) => {
+            await context.openFileForMarker('test_marker');
+            await context.openFileForMarker('mod_marker');
+            const idNode = context.getNodeForMarker('id_val');
+            // Apply redirect before checking — `id` doesn't exist in the original.
+            const modelsFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/models.py'));
+            const virtualFile = context.fourslash.files.find((f) => f.fileUri.toString().endsWith('/virtual_models.py'));
+            (0, assert_1.default)(modelsFile && virtualFile, 'both files should exist');
+            const snapshotChangedPromise = context.waitForSnapshotChanged();
+            context.sendNotification(tspSupplemental_1.TspSupplemental.SetVirtualFileRedirectNotification.type, {
+                realUri: modelsFile.fileUri.toString(),
+                virtualUri: virtualFile.fileUri.toString(),
+            });
+            // Wait for the snapshot changed event. The redirect (models.py is open, so the
+            // server reads the virtual content and updates the in-memory buffer) triggers a
+            // reanalysis that makes the new `id` field visible.
+            await snapshotChangedPromise;
+            await context.refreshSnapshot();
+            // After redirect: id should be int (from virtual file's `id: int` declaration).
+            const typeAfter = await context.sendRequestWithSnapshot(typeServerProtocol_1.TypeServerProtocol.GetComputedTypeRequest.type, {
+                arg: idNode,
+            });
+            (0, assert_1.default)(typeAfter, 'id_val should have a computed type after redirect');
+            assert_1.default.strictEqual(getClassTypeName(typeAfter), 'int', 'id_val should be int after redirect');
+        });
+    });
+});
+//# sourceMappingURL=typeServer.virtualFileRedirect.test.js.map

package/dist/tests/typeServer/typeServer.virtualFileRedirect.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeServer.virtualFileRedirect.test.js","sourceRoot":"","sources":["../../../../../../src/tests/typeServer/typeServer.virtualFileRedirect.test.ts"],"names":[],"mappings":";;;;;AAAA;;;;;;;;;GASG;AACH,oDAA4B;AAE5B,+EAA4E;AAC5E,qFAAkF;AAClF,2EAAyG;AAEzG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;AAExB,mGAAmG;AACnG,SAAS,gBAAgB,CAAC,IAAyC;IAC/D,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,8CAAsC,EAAE,CAAC;QAC3D,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,MAAM,SAAS,GAAG,IAAoC,CAAC;IACvD,IAAI,SAAS,CAAC,WAAW,CAAC,IAAI,uDAA+C,EAAE,CAAC;QAC5E,OAAO,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC;IACtC,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,QAAQ,CAAC,oDAAoD,EAAE,GAAG,EAAE;IAChE,SAAS,CAAC,KAAK,IAAI,EAAE;QACjB,MAAM,IAAA,gEAAoC,GAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,IAAI,GAAG;;;;;;;;;;;;CAYpB,CAAC;QAEM,MAAM,IAAA,gDAAoB,EAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC/C,yDAAyD;YACzD,MAAM,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE9C,wDAAwD;YACxD,MAAM,KAAK,GAA4B,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;YAErE,6DAA6D;YAC7D,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAExD,8CAA8C;YAC9C,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACrG,GAAG,EAAE,KAAK;aACb,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,UAAU,EAAE,+CAA+C,CAAC,CAAC;YACpE,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,iCAAiC,CAAC,CAAC;YAE3F,wEAAwE;YACxE,MAAM,eAAe,GAAG,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,EAAE;gBAC1E,OAAO,CAAC,cAAc,CAAC,uCAAkB,CAAC,2BAA2B,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;oBACnF,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpB,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;YAEH,yCAAyC;YACzC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;YAClG,IAAA,gBAAM,EAAC,SAAS,EAAE,0CAA0C,CAAC,CAAC;YAE9D,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAC5G,IAAA,gBAAM,EAAC,WAAW,EAAE,kDAAkD,CAAC,CAAC;YAExE,kEAAkE;YAClE,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,kCAAkC,CAAC,IAAI,EAAE;gBAC9E,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE;gBACrC,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC,CAAC;YAEH,iFAAiF;YACjF,oBAAoB;YACpB,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC;YAErC,IAAA,gBAAM,EAAC,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,iDAAiD,CAAC,CAAC;YACnF,IAAA,gBAAM,EAAC,MAAM,CAAC,GAAG,IAAI,eAAe,EAAE,mCAAmC,CAAC,CAAC;YAE3E,2FAA2F;YAC3F,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACpG,GAAG,EAAE,KAAK;aACb,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,SAAS,EAAE,8CAA8C,CAAC,CAAC;YAClE,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,gCAAgC,CAAC,CAAC;QAC7F,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,IAAI,GAAG;;;;;;;;;;;;CAYpB,CAAC;QAEM,MAAM,IAAA,gDAAoB,EAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC/C,sCAAsC;YACtC,MAAM,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE9C,wDAAwD;YACxD,MAAM,KAAK,GAA4B,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;YAErE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;YAClG,IAAA,gBAAM,EAAC,SAAS,EAAE,0CAA0C,CAAC,CAAC;YAE9D,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAC5G,IAAA,gBAAM,EAAC,WAAW,EAAE,kDAAkD,CAAC,CAAC;YAExE,8CAA8C;YAC9C,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAChC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACvG,GAAG,EAAE,KAAK;aACb,CAAC,CAAC;YACH,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,iCAAiC,CAAC,CAAC;YAE7F,kDAAkD;YAClD,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,kCAAkC,CAAC,IAAI,EAAE;gBAC9E,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE;gBACrC,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC,CAAC;YAEH,sCAAsC;YACtC,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,8BAA8B;YAC9B,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,uBAAuB,CACxD,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAC9C,EAAE,GAAG,EAAE,KAAK,EAAE,CACjB,CAAC;YACF,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAAE,KAAK,EAAE,gCAAgC,CAAC,CAAC;YAE9F,2CAA2C;YAC3C,MAAM,eAAe,GAAG,IAAI,OAAO,CAA+B,CAAC,OAAO,EAAE,EAAE;gBAC1E,OAAO,CAAC,cAAc,CAAC,uCAAkB,CAAC,2BAA2B,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;oBACnF,OAAO,CAAC,MAAM,CAAC,CAAC;gBACpB,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;YAEH,uBAAuB;YACvB,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,qCAAqC,CAAC,IAAI,EAAE;gBACjF,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,QAAQ,EAAE;aACxC,CAAC,CAAC;YAEH,iDAAiD;YACjD,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC;YACrC,IAAA,gBAAM,EAAC,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,0DAA0D,CAAC,CAAC;YAE5F,gDAAgD;YAChD,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACvG,GAAG,EAAE,KAAK;aACb,CAAC,CAAC;YACH,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,wCAAwC,CAAC,CAAC;QACxG,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,oEAAoE;QACpE,uEAAuE;QACvE,oCAAoC;QACpC,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;CAepB,CAAC;QAEM,MAAM,IAAA,gDAAoB,EAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE9C,MAAM,UAAU,GAA4B,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC/E,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,0DAA0D;YAC1D,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACrG,GAAG,EAAE,UAAU;aAClB,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,UAAU,EAAE,oDAAoD,CAAC,CAAC;YACzE,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,sCAAsC,CAAC,CAAC;YAEhG,gDAAgD;YAChD,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YACpG,IAAA,gBAAM,EAAC,UAAU,EAAE,wBAAwB,CAAC,CAAC;YAC7C,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CACtD,CAAC;YACF,IAAA,gBAAM,EAAC,WAAW,EAAE,gCAAgC,CAAC,CAAC;YAEtD,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,kCAAkC,CAAC,IAAI,EAAE;gBAC9E,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE;gBACtC,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,qFAAqF;YACrF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACpG,GAAG,EAAE,UAAU;aAClB,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,SAAS,EAAE,mDAAmD,CAAC,CAAC;YACvE,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,2CAA2C,CAAC,CAAC;QACxG,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,kEAAkE;QAClE,oDAAoD;QACpD,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;CAgBpB,CAAC;QAEM,MAAM,IAAA,gDAAoB,EAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE9C,MAAM,UAAU,GAA4B,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC/E,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,oCAAoC;YACpC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACrG,GAAG,EAAE,UAAU;aAClB,CAAC,CAAC;YACH,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,sCAAsC,CAAC,CAAC;YAEhG,iEAAiE;YACjE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YACpG,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CACtD,CAAC;YACF,IAAA,gBAAM,EAAC,UAAU,IAAI,WAAW,EAAE,yBAAyB,CAAC,CAAC;YAE7D,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,kCAAkC,CAAC,IAAI,EAAE;gBAC9E,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE;gBACtC,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,+EAA+E;YAC/E,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACpG,GAAG,EAAE,UAAU;aAClB,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,SAAS,EAAE,mDAAmD,CAAC,CAAC;YACvE,+CAA+C;YAC/C,gBAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,+CAA+C,CAAC,CAAC;QAC/G,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,6DAA6D;QAC7D,yBAAyB;QACzB,MAAM,IAAI,GAAG;;;;;;;;;;;;;;;;;CAiBpB,CAAC;QAEM,MAAM,IAAA,gDAAoB,EAAC,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAE9C,MAAM,MAAM,GAA4B,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAE3E,uEAAuE;YACvE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YACpG,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACnD,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CACtD,CAAC;YACF,IAAA,gBAAM,EAAC,UAAU,IAAI,WAAW,EAAE,yBAAyB,CAAC,CAAC;YAE7D,MAAM,sBAAsB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;YAChE,OAAO,CAAC,gBAAgB,CAAC,iCAAe,CAAC,kCAAkC,CAAC,IAAI,EAAE;gBAC9E,OAAO,EAAE,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE;gBACtC,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC,CAAC;YAEH,+EAA+E;YAC/E,gFAAgF;YAChF,oDAAoD;YACpD,MAAM,sBAAsB,CAAC;YAE7B,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEhC,gFAAgF;YAChF,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,uBAAuB,CAAC,uCAAkB,CAAC,sBAAsB,CAAC,IAAI,EAAE;gBACpG,GAAG,EAAE,MAAM;aACd,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,SAAS,EAAE,mDAAmD,CAAC,CAAC;YACvE,gBAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,qCAAqC,CAAC,CAAC;QAClG,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

package/dist/typeServer/cancellation.d.ts

@@ -0,0 +1,5 @@
+import { ResponseError } from 'vscode-languageserver';
+export declare class ServerCanceledException extends ResponseError<void> {
+    constructor(message?: string | undefined);
+    static is(e: any): e is ServerCanceledException;
+}

package/dist/typeServer/cancellation.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ServerCanceledException = void 0;
+const vscode_languageserver_1 = require("vscode-languageserver");
+class ServerCanceledException extends vscode_languageserver_1.ResponseError {
+    constructor(message) {
+        super(vscode_languageserver_1.LSPErrorCodes.ServerCancelled, message || 'server cancelled');
+    }
+    static is(e) {
+        return e.code === vscode_languageserver_1.LSPErrorCodes.ServerCancelled;
+    }
+}
+exports.ServerCanceledException = ServerCanceledException;
+//# sourceMappingURL=cancellation.js.map

package/dist/typeServer/cancellation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancellation.js","sourceRoot":"","sources":["../../../../../src/typeServer/cancellation.ts"],"names":[],"mappings":";;;AAAA,iEAAqE;AAErE,MAAa,uBAAwB,SAAQ,qCAAmB;IAC5D,YAAY,OAA4B;QACpC,KAAK,CAAC,qCAAa,CAAC,eAAe,EAAE,OAAO,IAAI,kBAAkB,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,CAAC,EAAE,CAAC,CAAM;QACZ,OAAO,CAAC,CAAC,IAAI,KAAK,qCAAa,CAAC,eAAe,CAAC;IACpD,CAAC;CACJ;AARD,0DAQC"}

package/dist/typeServer/diagnosticUtils.d.ts

@@ -0,0 +1,4 @@
+import { Diagnostic } from 'vscode-languageserver';
+import { Diagnostic as PyrightDiagnostic } from '../common/diagnostic';
+import { ReadOnlyFileSystem } from '../common/fileSystem';
+export declare function convertFromPyrightDiagnostic(diag: PyrightDiagnostic, fs: ReadOnlyFileSystem, supportsUnnecessaryDiagnosticTag: boolean, supportsTaskItemDiagnosticTag: boolean): Diagnostic | undefined;

package/dist/typeServer/diagnosticUtils.js

@@ -0,0 +1,71 @@
+"use strict";
+/*
+ * diagnosticUtils.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Converts Pyright's internal diagnostics into LSP diagnostics for the type server.
+ * This mirrors `LanguageServerBase._convertDiagnostics`; the Pylance-only VS "task item"
+ * ranking is omitted (task-item diagnostics are dropped, matching Pyright's own server).
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.convertFromPyrightDiagnostic = convertFromPyrightDiagnostic;
+const vscode_languageserver_1 = require("vscode-languageserver");
+const uriUtils_1 = require("../common/uri/uriUtils");
+const navigationUtils_1 = require("../languageService/navigationUtils");
+function convertCategoryToSeverity(category) {
+    switch (category) {
+        case 0 /* DiagnosticCategory.Error */:
+            return vscode_languageserver_1.DiagnosticSeverity.Error;
+        case 1 /* DiagnosticCategory.Warning */:
+            return vscode_languageserver_1.DiagnosticSeverity.Warning;
+        case 2 /* DiagnosticCategory.Information */:
+        case 6 /* DiagnosticCategory.TaskItem */:
+            return vscode_languageserver_1.DiagnosticSeverity.Information;
+        case 3 /* DiagnosticCategory.UnusedCode */:
+        case 4 /* DiagnosticCategory.UnreachableCode */:
+        case 5 /* DiagnosticCategory.Deprecated */:
+            return vscode_languageserver_1.DiagnosticSeverity.Hint;
+    }
+}
+function convertFromPyrightDiagnostic(diag, fs, supportsUnnecessaryDiagnosticTag, supportsTaskItemDiagnosticTag) {
+    const severity = convertCategoryToSeverity(diag.category);
+    const rule = diag.getRule();
+    const vsDiag = vscode_languageserver_1.Diagnostic.create(diag.range, diag.message, severity, rule, 'pyright');
+    // Save all of the actions in the data.
+    const actions = diag.getActions();
+    if (actions?.length) {
+        vsDiag.data = { ...vsDiag.data, actions: actions };
+    }
+    if (diag.category === 3 /* DiagnosticCategory.UnusedCode */ || diag.category === 4 /* DiagnosticCategory.UnreachableCode */) {
+        vsDiag.tags = [vscode_languageserver_1.DiagnosticTag.Unnecessary];
+        vsDiag.severity = vscode_languageserver_1.DiagnosticSeverity.Hint;
+        vsDiag.data = { ...vsDiag.data, category: diag.category, rule: rule };
+        // If the client doesn't support "unnecessary" tags, don't report unused code.
+        if (!supportsUnnecessaryDiagnosticTag) {
+            return undefined;
+        }
+    }
+    else if (diag.category === 5 /* DiagnosticCategory.Deprecated */) {
+        vsDiag.tags = [vscode_languageserver_1.DiagnosticTag.Deprecated];
+        vsDiag.severity = vscode_languageserver_1.DiagnosticSeverity.Hint;
+        // If the client doesn't support "deprecated" tags, don't report.
+        if (!supportsUnnecessaryDiagnosticTag) {
+            return undefined;
+        }
+    }
+    else if (diag.category === 6 /* DiagnosticCategory.TaskItem */) {
+        // Task items are a Pylance/VS-only concept; drop them unless the client opted in.
+        if (!supportsTaskItemDiagnosticTag) {
+            return undefined;
+        }
+    }
+    const relatedInfo = diag.getRelatedInfo();
+    if (relatedInfo.length > 0) {
+        vsDiag.relatedInformation = relatedInfo
+            .filter((info) => (0, navigationUtils_1.canNavigateToFile)(fs, info.uri))
+            .map((info) => vscode_languageserver_1.DiagnosticRelatedInformation.create(vscode_languageserver_1.Location.create((0, uriUtils_1.convertUriToLspUriString)(fs, info.uri), info.range), info.message));
+    }
+    return vsDiag;
+}
+//# sourceMappingURL=diagnosticUtils.js.map

package/dist/typeServer/diagnosticUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"diagnosticUtils.js","sourceRoot":"","sources":["../../../../../src/typeServer/diagnosticUtils.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;AAkCH,oEAqDC;AArFD,iEAM+B;AAI/B,qDAAkE;AAClE,wEAAuE;AAEvE,SAAS,yBAAyB,CAAC,QAA4B;IAC3D,QAAQ,QAAQ,EAAE,CAAC;QACf;YACI,OAAO,0CAAkB,CAAC,KAAK,CAAC;QAEpC;YACI,OAAO,0CAAkB,CAAC,OAAO,CAAC;QAEtC,4CAAoC;QACpC;YACI,OAAO,0CAAkB,CAAC,WAAW,CAAC;QAE1C,2CAAmC;QACnC,gDAAwC;QACxC;YACI,OAAO,0CAAkB,CAAC,IAAI,CAAC;IACvC,CAAC;AACL,CAAC;AAED,SAAgB,4BAA4B,CACxC,IAAuB,EACvB,EAAsB,EACtB,gCAAyC,EACzC,6BAAsC;IAEtC,MAAM,QAAQ,GAAG,yBAAyB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAG,kCAAU,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAEtF,uCAAuC;IACvC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IAClC,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;QAClB,MAAM,CAAC,IAAI,GAAG,EAAE,GAAG,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;IACvD,CAAC;IAED,IAAI,IAAI,CAAC,QAAQ,0CAAkC,IAAI,IAAI,CAAC,QAAQ,+CAAuC,EAAE,CAAC;QAC1G,MAAM,CAAC,IAAI,GAAG,CAAC,qCAAa,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,GAAG,0CAAkB,CAAC,IAAI,CAAC;QAC1C,MAAM,CAAC,IAAI,GAAG,EAAE,GAAG,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAEtE,8EAA8E;QAC9E,IAAI,CAAC,gCAAgC,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACrB,CAAC;IACL,CAAC;SAAM,IAAI,IAAI,CAAC,QAAQ,0CAAkC,EAAE,CAAC;QACzD,MAAM,CAAC,IAAI,GAAG,CAAC,qCAAa,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,GAAG,0CAAkB,CAAC,IAAI,CAAC;QAE1C,iEAAiE;QACjE,IAAI,CAAC,gCAAgC,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACrB,CAAC;IACL,CAAC;SAAM,IAAI,IAAI,CAAC,QAAQ,wCAAgC,EAAE,CAAC;QACvD,kFAAkF;QAClF,IAAI,CAAC,6BAA6B,EAAE,CAAC;YACjC,OAAO,SAAS,CAAC;QACrB,CAAC;IACL,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;IAC1C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,MAAM,CAAC,kBAAkB,GAAG,WAAW;aAClC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAA,mCAAiB,EAAC,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;aACjD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACV,oDAA4B,CAAC,MAAM,CAC/B,gCAAQ,CAAC,MAAM,CAAC,IAAA,mCAAwB,EAAC,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,EACnE,IAAI,CAAC,OAAO,CACf,CACJ,CAAC;IACV,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/dist/typeServer/enums.d.ts

@@ -0,0 +1,14 @@
+import { VariableDeclaration } from '../analyzer/declaration';
+import { Arg, TypeResult } from '../analyzer/typeEvaluatorTypes';
+import { ClassType, Type } from '../analyzer/types';
+import { ExpressionNode, ParseNode } from '../parser/parseNodes';
+import { ITypeServerEvaluator } from './typeServerEvaluator';
+export declare function isEnumMetaclass(classType: ClassType): boolean;
+export declare function isEnumClassWithMembers(evaluator: ITypeServerEvaluator, classType: ClassType): boolean;
+export declare function createEnumType(evaluator: ITypeServerEvaluator, errorNode: ExpressionNode, enumClass: ClassType, argList: Arg[]): ClassType | undefined;
+export declare function transformTypeForEnumMember(evaluator: ITypeServerEvaluator, classType: ClassType, memberName: string, ignoreAnnotation?: boolean, recursionCount?: number): Type | undefined;
+export declare function isDeclInEnumClass(evaluator: ITypeServerEvaluator, decl: VariableDeclaration): boolean;
+export declare function getEnumDeclaredValueType(evaluator: ITypeServerEvaluator, classType: ClassType, declaredTypesOnly?: boolean): Type | undefined;
+export declare function getTypeOfEnumMember(evaluator: ITypeServerEvaluator, errorNode: ParseNode, classType: ClassType, memberName: string, isIncomplete: boolean): TypeResult | undefined;
+export declare function getEnumAutoValueType(evaluator: ITypeServerEvaluator, node: ExpressionNode): Type;
+export declare function isReprEnumClass(enumClass: ClassType): boolean;