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

package/dist/typeServer/typeServerConversionUtils.js

@@ -0,0 +1,524 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProtocolTypeFactory = void 0;
+exports.toProtocolModuleName = toProtocolModuleName;
+exports.getDeclarationForTypeVar = getDeclarationForTypeVar;
+exports.findFirstExpression = findFirstExpression;
+const vscode_languageserver_protocol_1 = require("vscode-languageserver-protocol");
+const parseTreeWalker_1 = require("../analyzer/parseTreeWalker");
+const PyrightTypes = __importStar(require("../analyzer/types"));
+const debug_1 = require("../common/debug");
+const positionUtils_1 = require("../common/positionUtils");
+const uriUtils_1 = require("../common/uri/uriUtils");
+const PyrightNodes = __importStar(require("../parser/parseNodes"));
+const typeEvalUtils_1 = require("./typeEvalUtils");
+const stubGenerator_1 = require("./stubGenerator");
+const typeServerConversionTypes_1 = require("./typeServerConversionTypes");
+function toProtocolModuleName(moduleName) {
+    if (moduleName.length === 0) {
+        return { leadingDots: 0, nameParts: [] };
+    }
+    let startIndex = 0;
+    let leadingDots = 0;
+    for (; startIndex < moduleName.length; startIndex++) {
+        if (moduleName[startIndex] !== '.') {
+            break;
+        }
+        leadingDots++;
+    }
+    return {
+        leadingDots,
+        nameParts: moduleName.slice(startIndex).split('.'),
+    };
+}
+class ProtocolTypeFactory {
+    constructor(view, pythonVersion, declarationOrNode) {
+        this.view = view;
+        this.pythonVersion = pythonVersion;
+        this._id = 0;
+        this._cycleMap = new Map();
+        this._evaluator = view.createEvaluator();
+        // Extract source URI from declaration or node
+        this._sourceUri = this._extractSourceUri(declarationOrNode);
+    }
+    get sourceUri() {
+        return this._sourceUri;
+    }
+    get evaluator() {
+        return this._evaluator;
+    }
+    getNextId() {
+        return this._id++;
+    }
+    getModule(type) {
+        const moduleName = type.shared.moduleName;
+        const moduleDescriptor = toProtocolModuleName(moduleName);
+        const moduleUri = this.view.resolveImport(this.sourceUri, moduleDescriptor, vscode_languageserver_protocol_1.CancellationToken.None) || this.sourceUri;
+        const moduleNode = this.view.getParserOutput(moduleUri)?.parseTree;
+        (0, debug_1.assert)(moduleNode !== undefined, `Module node not found for URI: ${moduleUri.toString()}`);
+        const scope = this.view.symbolLookup.getScope(moduleNode);
+        const symbolTable = scope?.symbolTable;
+        return PyrightTypes.ModuleType.create(moduleName, moduleUri, symbolTable);
+    }
+    getType(type) {
+        const newId = this.getNextId();
+        const existingTypeId = this._cycleMap.get(type);
+        if (existingTypeId !== undefined) {
+            return {
+                id: newId,
+                kind: 9 /* TypeServerProtocol.TypeKind.TypeReference */,
+                flags: 0 /* TypeServerProtocol.TypeFlags.None */,
+                typeReferenceId: existingTypeId,
+            };
+        }
+        this._set(type, newId);
+        return toProtocolType(newId, type, this);
+    }
+    _set(type, id) {
+        this._cycleMap.set(type, id);
+    }
+    _extractSourceUri(declarationOrNode) {
+        if ((0, typeEvalUtils_1.isDeclaration)(declarationOrNode)) {
+            return declarationOrNode.uri;
+        }
+        else {
+            return this.view.getUri(declarationOrNode);
+        }
+    }
+}
+exports.ProtocolTypeFactory = ProtocolTypeFactory;
+function toProtocolTypesOrUndefined(types, factory) {
+    if (!types) {
+        return undefined;
+    }
+    return toProtocolTypes(types, factory);
+}
+function toProtocolTypes(types, factory) {
+    return (0, typeEvalUtils_1.map)(types, (type) => factory.getType(type));
+}
+function toProtocolTypeOrUndefined(type, factory) {
+    if (!type) {
+        return undefined;
+    }
+    return factory.getType(type);
+}
+function toProtocolTypeArgs(type, factory) {
+    // See if we have any type arguments. Set those on the handle too.
+    let typeArgs = undefined;
+    if (type.priv.tupleTypeArgs) {
+        typeArgs = toProtocolTypesOrUndefined(type.priv.tupleTypeArgs.map((t) => t.type), factory);
+        // Set the flags on each based on tuple information
+        for (let i = 0; i < type.priv.tupleTypeArgs.length; i++) {
+            if (typeArgs && typeArgs[i] && type.priv.tupleTypeArgs[i].isUnbounded) {
+                typeArgs[i].flags |= 512 /* TypeServerProtocol.TypeFlags.Unbound */;
+            }
+            if (typeArgs && typeArgs[i] && type.priv.tupleTypeArgs[i].isOptional) {
+                typeArgs[i].flags |= 256 /* TypeServerProtocol.TypeFlags.Optional */;
+            }
+        }
+    }
+    else if (type.priv.typeArgs) {
+        typeArgs = toProtocolTypesOrUndefined(type.priv.typeArgs, factory);
+    }
+    return typeArgs;
+}
+function toProtocolTypeAliasInfo(info, factory) {
+    if (!info) {
+        return undefined;
+    }
+    return {
+        name: info.shared.name,
+        fullName: info.shared.fullName,
+        moduleName: info.shared.moduleName,
+        fileUri: (0, uriUtils_1.convertUriToLspUriString)(factory.view.fs, info.shared.fileUri),
+        scopeId: info.shared.typeVarScopeId,
+        isTypeAliasType: info.shared.isTypeAliasType,
+        typeParams: toProtocolTypesOrUndefined(info.shared.typeParams, factory),
+        typeArgs: toProtocolTypesOrUndefined(info.typeArgs, factory),
+        computedVariance: info.shared.computedVariance?.map((v) => (0, typeServerConversionTypes_1.toProtocolVariance)(v) ?? 1 /* TypeServerProtocol.Variance.Unknown */),
+    };
+}
+// This should be only called from TspTypeFactory
+// Otherwise, cycle detection won't work.
+function toProtocolType(id, type, factory) {
+    switch (type.category) {
+        case 6 /* PyrightTypes.TypeCategory.Class */: {
+            const decl = type.shared.declaration;
+            if (decl) {
+                return {
+                    id,
+                    kind: 3 /* TypeServerProtocol.TypeKind.Class */,
+                    flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                    typeAliasInfo: toProtocolTypeAliasInfo(type.props?.typeAliasInfo, factory),
+                    declaration: (0, typeServerConversionTypes_1.toProtocolDecl)(decl, factory.view),
+                    literalValue: toProtocolLiteralValue(type, factory),
+                    typeArgs: toProtocolTypeArgs(type, factory),
+                };
+            }
+            return createSynthesizedType(id, type, factory);
+        }
+        case 4 /* PyrightTypes.TypeCategory.Function */: {
+            const decl = type.shared.declaration;
+            if (decl) {
+                // Compute boundToType: if not set but strippedFirstParamType exists, use it as boundToType
+                let boundToType = type.priv.boundToType;
+                if (!boundToType &&
+                    type.priv.strippedFirstParamType &&
+                    PyrightTypes.isClass(type.priv.strippedFirstParamType)) {
+                    boundToType = type.priv.strippedFirstParamType;
+                }
+                const returnType = PyrightTypes.FunctionType.getEffectiveReturnType(type, true);
+                return {
+                    id,
+                    kind: 2 /* TypeServerProtocol.TypeKind.Function */,
+                    flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                    typeAliasInfo: toProtocolTypeAliasInfo(type.props?.typeAliasInfo, factory),
+                    declaration: (0, typeServerConversionTypes_1.toProtocolDecl)(decl, factory.view),
+                    specializedTypes: toProtocolSpecializedFunctionTypesOrUndefined(type.priv.specializedTypes, factory),
+                    boundToType: toProtocolTypeOrUndefined(boundToType, factory),
+                    returnType: toProtocolTypeOrUndefined(returnType, factory),
+                };
+            }
+            return createSynthesizedType(id, type, factory);
+        }
+        case 5 /* PyrightTypes.TypeCategory.Overloaded */: {
+            const implementation = PyrightTypes.OverloadedType.getImplementation(type);
+            const overloaded = PyrightTypes.OverloadedType.getOverloads(type);
+            return {
+                id,
+                kind: 7 /* TypeServerProtocol.TypeKind.Overloaded */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                overloads: toProtocolTypes(overloaded, factory),
+                implementation: toProtocolTypeOrUndefined(implementation, factory),
+            };
+        }
+        case 9 /* PyrightTypes.TypeCategory.TypeVar */: {
+            const declaration = getDeclarationForTypeVar(type, factory);
+            if (declaration) {
+                return {
+                    id,
+                    kind: 6 /* TypeServerProtocol.TypeKind.TypeVar */,
+                    flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                    declaration: (0, typeServerConversionTypes_1.toProtocolDecl)(declaration, factory.view),
+                };
+            }
+            return createSynthesizedType(id, type, factory);
+        }
+        case 0 /* PyrightTypes.TypeCategory.Unbound */: {
+            const handle = {
+                id,
+                kind: 0 /* TypeServerProtocol.TypeKind.BuiltIn */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                name: 'unbound',
+            };
+            return handle;
+        }
+        case 1 /* PyrightTypes.TypeCategory.Unknown */: {
+            const possibleType = type.priv.possibleType;
+            const handle = {
+                id,
+                kind: 0 /* TypeServerProtocol.TypeKind.BuiltIn */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                name: 'unknown',
+                possibleType: toProtocolTypeOrUndefined(possibleType, factory),
+            };
+            return handle;
+        }
+        case 2 /* PyrightTypes.TypeCategory.Any */: {
+            const handle = {
+                id,
+                kind: 0 /* TypeServerProtocol.TypeKind.BuiltIn */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                name: type.priv.isEllipsis ? 'ellipsis' : 'any',
+            };
+            return handle;
+        }
+        case 3 /* PyrightTypes.TypeCategory.Never */: {
+            const handle = {
+                id,
+                kind: 0 /* TypeServerProtocol.TypeKind.BuiltIn */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                name: type.priv.isNoReturn ? 'noreturn' : 'never',
+            };
+            return handle;
+        }
+        case 8 /* PyrightTypes.TypeCategory.Union */:
+            return {
+                id,
+                kind: 4 /* TypeServerProtocol.TypeKind.Union */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                typeAliasInfo: toProtocolTypeAliasInfo(type.props?.typeAliasInfo, factory),
+                subTypes: (0, typeEvalUtils_1.map)(type.priv.subtypes, (t) => factory.getType(t)),
+            };
+        case 7 /* PyrightTypes.TypeCategory.Module */: {
+            return {
+                id,
+                kind: 5 /* TypeServerProtocol.TypeKind.Module */,
+                flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+                moduleName: type.priv.moduleName,
+                uri: (0, uriUtils_1.convertUriToLspUriString)(factory.view.fs, type.priv.fileUri),
+            };
+        }
+    }
+}
+function createSynthesizedType(id, type, factory) {
+    const options = { pythonVersion: factory.pythonVersion };
+    let moduleHandle;
+    let stubResult;
+    if (PyrightTypes.isTypeVar(type)) {
+        const moduleName = factory.view.getModuleName(factory.sourceUri) ?? '';
+        const moduleType = PyrightTypes.ModuleType.create(moduleName, factory.sourceUri, new Map());
+        moduleHandle = factory.getType(moduleType);
+        stubResult = (0, stubGenerator_1.generateStubFromTypeVar)(type, options);
+    }
+    else {
+        const module = factory.getModule(type);
+        moduleHandle = factory.getType(module);
+        stubResult = PyrightTypes.isFunction(type)
+            ? (0, stubGenerator_1.generateStubFromFunctionType)(factory.evaluator, type, options)
+            : (0, stubGenerator_1.generateStubFromClassType)(factory.evaluator, type, options);
+    }
+    const metadata = {
+        module: moduleHandle,
+        primaryDefinitionOffset: stubResult.primaryDefinitionOffset,
+    };
+    return {
+        id,
+        kind: 8 /* TypeServerProtocol.TypeKind.Synthesized */,
+        flags: (0, typeServerConversionTypes_1.toProtocolTypeFlags)(type),
+        typeAliasInfo: toProtocolTypeAliasInfo(type.props?.typeAliasInfo, factory),
+        stubContent: stubResult.stubContent,
+        metadata,
+    };
+}
+function toProtocolSpecializedFunctionTypesOrUndefined(specializedTypes, factory) {
+    if (!specializedTypes) {
+        return undefined;
+    }
+    return {
+        parameterTypes: (0, typeEvalUtils_1.map)(specializedTypes.parameterTypes, (t) => factory.getType(t)),
+        parameterDefaultTypes: specializedTypes.parameterDefaultTypes
+            ? (0, typeEvalUtils_1.map)(specializedTypes.parameterDefaultTypes, (t) => toProtocolTypeOrUndefined(t, factory))
+            : undefined,
+        returnType: toProtocolTypeOrUndefined(specializedTypes.returnType, factory),
+    };
+}
+function toProtocolLiteralValue(type, factory) {
+    const value = type.priv.literalValue;
+    if (value === undefined) {
+        return undefined;
+    }
+    if (value instanceof PyrightTypes.SentinelLiteral) {
+        return {
+            classNode: (0, typeServerConversionTypes_1.toProtocolNode)(type.shared.declaration.node, factory.view),
+            moduleName: type.shared.moduleName,
+            className: value.className,
+        };
+    }
+    if (value instanceof PyrightTypes.EnumLiteral) {
+        return {
+            className: value.className,
+            itemName: value.itemName,
+            itemType: factory.getType(value.itemType),
+        };
+    }
+    return value;
+}
+// Attempts to recover the original declaration node for a TypeVar so we can serialize
+// the same metadata the evaluator used when the symbol was first defined. Stub-based
+// reconstruction frequently loses this link, so we search for it on demand using the
+// scope id that is preserved on the TypeVar instance.
+function getDeclarationForTypeVar(typeVar, factory) {
+    // freeTypeVar stores the "canonical" definition for synthesized instances that were
+    // cloned for specialization. Always operate on that canonical TypeVar when available.
+    const canonicalTypeVar = typeVar.priv.freeTypeVar ?? typeVar;
+    if (canonicalTypeVar.shared.isSynthesized) {
+        return undefined;
+    }
+    // Without a scope id we have no way to map the TypeVar back to the parse tree.
+    const scopeId = canonicalTypeVar.priv.scopeId;
+    if (!scopeId) {
+        return undefined;
+    }
+    const scopeNode = findScopeNodeForId(scopeId, factory.view);
+    if (!scopeNode) {
+        return undefined;
+    }
+    // Start by inspecting the symbol table attached to the original scope; this covers
+    // most cases for function, class, and alias TypeVars.
+    let symbolTable = factory.view.symbolLookup.getSymbolsForNode(scopeNode);
+    if (!symbolTable && scopeNode.nodeType === 36 /* PyrightNodes.ParseNodeType.Module */) {
+        const fileInfo = factory.view.symbolLookup.getFileInfo(scopeNode);
+        symbolTable = factory.view.symbolLookup.getSymbolsForFile(fileInfo.fileUri);
+    }
+    const typeVarName = canonicalTypeVar.shared.name;
+    let symbol = symbolTable?.get(typeVarName) ?? factory.view.symbolLookup.lookupSymbol(scopeNode, typeVarName);
+    if ((!symbol || symbol.getDeclarations().length === 0) &&
+        scopeNode.nodeType !== 36 /* PyrightNodes.ParseNodeType.Module */) {
+        // For nested scopes we can fall back to the file-level module scope. This is
+        // common when using PEP 695 syntax where a type parameter is hoisted into the
+        // containing module table for reuse.
+        const moduleNode = findModuleAncestor(scopeNode);
+        if (moduleNode) {
+            const moduleInfo = factory.view.symbolLookup.getFileInfo(moduleNode);
+            const moduleSymbols = factory.view.symbolLookup.getSymbolsForFile(moduleInfo.fileUri);
+            const moduleSymbol = moduleSymbols?.get(typeVarName);
+            if (moduleSymbol) {
+                symbol = moduleSymbol;
+            }
+            else {
+                symbol = factory.view.symbolLookup.lookupSymbol(moduleNode, typeVarName);
+            }
+        }
+    }
+    const declarations = symbol?.getDeclarations();
+    if (!declarations || declarations.length === 0) {
+        // As a final attempt, locate the syntactic TypeParameter node and synthesize a
+        // declaration object pointing to it. This ensures we at least return file/offset
+        // information even when binding never populated a symbol.
+        const typeParamNode = findTypeParameterNode(scopeNode, typeVarName);
+        if (typeParamNode) {
+            const fileInfo = factory.view.symbolLookup.getFileInfo(typeParamNode);
+            return {
+                type: 3 /* DeclarationType.TypeParam */,
+                node: typeParamNode,
+                uri: fileInfo.fileUri,
+                range: (0, positionUtils_1.convertTextRangeToRange)(typeParamNode, fileInfo.lines),
+                moduleName: fileInfo.moduleName,
+                isInExceptSuite: false,
+            };
+        }
+        return undefined;
+    }
+    const typeParamDecl = declarations.find((decl) => decl.type === 3 /* DeclarationType.TypeParam */);
+    return typeParamDecl ?? declarations[0];
+}
+// Search all known parse trees to locate the node that owns the provided scope id.
+// Scope ids are assigned to functions, classes, modules, and type aliases; once we
+// find the matching node we can dig into its symbol table for declarations.
+function findScopeNodeForId(scopeId, view) {
+    const targetPrefix = scopeId.split('.', 1)[0];
+    const matchingFiles = view.symbolLookup.getMatchingFileInfos(targetPrefix);
+    for (const fileInfo of matchingFiles) {
+        const parseResults = view.getParserOutput(fileInfo.fileUri);
+        if (!parseResults) {
+            continue;
+        }
+        const stack = [parseResults.parseTree];
+        while (stack.length > 0) {
+            const node = stack.pop();
+            const nodeScopeId = view.symbolLookup.getScopeIdForNode(node);
+            if (isTypeVarScopeCandidate(node) && nodeScopeId === scopeId) {
+                return node;
+            }
+            // Expression nodes cannot contain declarations, so skip their children.
+            if (!PyrightNodes.isExpressionNode(node)) {
+                for (const child of (0, parseTreeWalker_1.getChildNodes)(node)) {
+                    if (child) {
+                        stack.push(child);
+                    }
+                }
+            }
+        }
+    }
+    return undefined;
+}
+// Returns the TypeParameter node with the requested name if the enclosing scope declares
+// type parameters (function/class/type alias). This gives us a concrete parse node even
+// when the TypeVar was not bound to a symbol table entry.
+function findTypeParameterNode(scopeNode, typeVarName) {
+    let typeParams;
+    switch (scopeNode.nodeType) {
+        case 31 /* PyrightNodes.ParseNodeType.Function */:
+        case 10 /* PyrightNodes.ParseNodeType.Class */:
+        case 77 /* PyrightNodes.ParseNodeType.TypeAlias */:
+            typeParams = scopeNode.d.typeParams;
+            break;
+        case 36 /* PyrightNodes.ParseNodeType.Module */:
+            typeParams = undefined;
+            break;
+    }
+    if (!typeParams) {
+        return undefined;
+    }
+    return typeParams.d.params.find((param) => param.d.name.d.value === typeVarName);
+}
+// Walks parent pointers up to the root module node. Used when we fall back to the
+// file-level scope to locate a TypeVar that was re-exported at module scope.
+function findModuleAncestor(node) {
+    let current = node;
+    while (current) {
+        if (current.nodeType === 36 /* PyrightNodes.ParseNodeType.Module */) {
+            return current;
+        }
+        current = current.parent;
+    }
+    return undefined;
+}
+// Only certain node kinds introduce scope ids; this helper keeps the search logic focused
+// on those nodes to avoid unnecessary traversal work during scope discovery.
+function isTypeVarScopeCandidate(node) {
+    switch (node.nodeType) {
+        case 36 /* PyrightNodes.ParseNodeType.Module */:
+        case 10 /* PyrightNodes.ParseNodeType.Class */:
+        case 31 /* PyrightNodes.ParseNodeType.Function */:
+        case 77 /* PyrightNodes.ParseNodeType.TypeAlias */:
+            return true;
+        default:
+            return false;
+    }
+}
+function findFirstExpression(node) {
+    if (PyrightNodes.isExpressionNode(node)) {
+        return node;
+    }
+    class ExpressionNodeWalker extends parseTreeWalker_1.ParseTreeWalker {
+        getResult() {
+            return this._result;
+        }
+        visitNode(node) {
+            if (PyrightNodes.isExpressionNode(node)) {
+                this._result = node;
+                return [];
+            }
+            return super.visitNode(node);
+        }
+    }
+    const walker = new ExpressionNodeWalker();
+    walker.walk(node);
+    return walker.getResult();
+}
+//# sourceMappingURL=typeServerConversionUtils.js.map

package/dist/typeServer/typeServerConversionUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeServerConversionUtils.js","sourceRoot":"","sources":["../../../../../src/typeServer/typeServerConversionUtils.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,oDAmBC;AAoXD,4DA4EC;AA2FD,kDAwBC;AA/lBD,mFAAmE;AAGnE,iEAA6E;AAC7E,gEAAkD;AAClD,2CAAyC;AACzC,2DAAkE;AAGlE,qDAAkE;AAClE,mEAAqD;AAGrD,mDAAqD;AAGrD,mDAMyB;AACzB,2EAAsH;AAEtH,SAAgB,oBAAoB,CAAC,UAAkB;IACnD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,OAAO,UAAU,GAAG,UAAU,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC;QAClD,IAAI,UAAU,CAAC,UAAU,CAAC,KAAK,GAAG,EAAE,CAAC;YACjC,MAAM;QACV,CAAC;QAED,WAAW,EAAE,CAAC;IAClB,CAAC;IAED,OAAO;QACH,WAAW;QACX,SAAS,EAAE,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC;KACrD,CAAC;AACN,CAAC;AAED,MAAa,mBAAmB;IAM5B,YACa,IAAc,EACd,aAA4B,EACrC,iBAAuD;QAF9C,SAAI,GAAJ,IAAI,CAAU;QACd,kBAAa,GAAb,aAAa,CAAe;QAPjC,QAAG,GAAG,CAAC,CAAC;QACC,cAAS,GAAG,IAAI,GAAG,EAA6B,CAAC;QAS9D,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACzC,8CAA8C;QAC9C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;IAChE,CAAC;IAED,IAAI,SAAS;QACT,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAED,IAAI,SAAS;QACT,OAAO,IAAI,CAAC,UAAU,CAAC;IAC3B,CAAC;IAED,SAAS;QACL,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;IACtB,CAAC;IAED,SAAS,CAAC,IAAwD;QAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAC1C,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,SAAS,GACX,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,EAAE,kDAAiB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC;QACxG,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC;QACnE,IAAA,cAAM,EAAC,UAAU,KAAK,SAAS,EAAE,kCAAkC,SAAS,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC3F,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,WAAW,GAAG,KAAK,EAAE,WAAW,CAAC;QACvC,OAAO,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;IAC9E,CAAC;IAED,OAAO,CAAC,IAAuB;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAE/B,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO;gBACH,EAAE,EAAE,KAAK;gBACT,IAAI,mDAA2C;gBAC/C,KAAK,2CAAmC;gBACxC,eAAe,EAAE,cAAc;aAClC,CAAC;QACN,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACvB,OAAO,cAAc,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;IAEO,IAAI,CAAC,IAAuB,EAAE,EAAU;QAC5C,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACjC,CAAC;IAEO,iBAAiB,CAAC,iBAAuD;QAC7E,IAAI,IAAA,6BAAa,EAAC,iBAAiB,CAAC,EAAE,CAAC;YACnC,OAAO,iBAAiB,CAAC,GAAG,CAAC;QACjC,CAAC;aAAM,CAAC;YACJ,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC;CACJ;AApED,kDAoEC;AAED,SAAS,0BAA0B,CAC/B,KAAsC,EACtC,OAA4B;IAE5B,IAAI,CAAC,KAAK,EAAE,CAAC;QACT,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;AAC3C,CAAC;AAED,SAAS,eAAe,CAAC,KAA0B,EAAE,OAA4B;IAC7E,OAAO,IAAA,mBAAG,EAAC,KAAK,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,yBAAyB,CAC9B,IAAmC,EACnC,OAA4B;IAE5B,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,kBAAkB,CAAC,IAA4B,EAAE,OAA4B;IAClF,kEAAkE;IAClE,IAAI,QAAQ,GAA0C,SAAS,CAAC;IAChE,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC1B,QAAQ,GAAG,0BAA0B,CACjC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAC1C,OAAO,CACV,CAAC;QAEF,mDAAmD;QACnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtD,IAAI,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;gBACnE,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAa,kDAAwC,CAAC;YACvE,CAAC;YACD,IAAI,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;gBAClE,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAa,mDAAyC,CAAC;YACxE,CAAC;QACL,CAAC;IACL,CAAC;SAAM,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC5B,QAAQ,GAAG,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACvE,CAAC;IAED,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,uBAAuB,CAAC,IAA4C,EAAE,OAA4B;IACvG,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO;QACH,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;QACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;QAC9B,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;QAClC,OAAO,EAAE,IAAA,mCAAwB,EAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACvE,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;QACnC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;QAC5C,UAAU,EAAE,0BAA0B,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC;QACvE,QAAQ,EAAE,0BAA0B,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC;QAC5D,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,GAAG,CAC/C,CAAC,CAAC,EAAE,EAAE,CAAC,IAAA,8CAAkB,EAAC,CAAC,CAAC,+CAAuC,CACtE;KACuC,CAAC;AACjD,CAAC;AAED,iDAAiD;AACjD,yCAAyC;AACzC,SAAS,cAAc,CAAC,EAAU,EAAE,IAAuB,EAAE,OAA4B;IACrF,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACpB,4CAAoC,CAAC,CAAC,CAAC;YACnC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YACrC,IAAI,IAAI,EAAE,CAAC;gBACP,OAAO;oBACH,EAAE;oBACF,IAAI,2CAAmC;oBACvC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;oBAChC,aAAa,EAAE,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC;oBAE1E,WAAW,EAAE,IAAA,0CAAc,EAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC;oBAC/C,YAAY,EAAE,sBAAsB,CAAC,IAAI,EAAE,OAAO,CAAC;oBACnD,QAAQ,EAAE,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC;iBACP,CAAC;YAC7C,CAAC;YAED,OAAO,qBAAqB,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,+CAAuC,CAAC,CAAC,CAAC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YACrC,IAAI,IAAI,EAAE,CAAC;gBACP,2FAA2F;gBAC3F,IAAI,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;gBACxC,IACI,CAAC,WAAW;oBACZ,IAAI,CAAC,IAAI,CAAC,sBAAsB;oBAChC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,EACxD,CAAC;oBACC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC;gBACnD,CAAC;gBAED,MAAM,UAAU,GAAG,YAAY,CAAC,YAAY,CAAC,sBAAsB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAChF,OAAO;oBACH,EAAE;oBACF,IAAI,8CAAsC;oBAC1C,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;oBAChC,aAAa,EAAE,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC;oBAE1E,WAAW,EAAE,IAAA,0CAAc,EAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC;oBAE/C,gBAAgB,EAAE,6CAA6C,CAC3D,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAC1B,OAAO,CACV;oBACD,WAAW,EAAE,yBAAyB,CAAC,WAAW,EAAE,OAAO,CAAC;oBAE5D,UAAU,EAAE,yBAAyB,CAAC,UAAU,EAAE,OAAO,CAAC;iBACnB,CAAC;YAChD,CAAC;YAED,OAAO,qBAAqB,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,iDAAyC,CAAC,CAAC,CAAC;YACxC,MAAM,cAAc,GAAG,YAAY,CAAC,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC3E,MAAM,UAAU,GAAG,YAAY,CAAC,cAAc,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAElE,OAAO;gBACH,EAAE;gBACF,IAAI,gDAAwC;gBAC5C,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,SAAS,EAAE,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC;gBAC/C,cAAc,EAAE,yBAAyB,CAAC,cAAc,EAAE,OAAO,CAAC;aACzB,CAAC;QAClD,CAAC;QACD,8CAAsC,CAAC,CAAC,CAAC;YACrC,MAAM,WAAW,GAAG,wBAAwB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAC5D,IAAI,WAAW,EAAE,CAAC;gBACd,OAAO;oBACH,EAAE;oBACF,IAAI,6CAAqC;oBACzC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;oBAChC,WAAW,EAAE,IAAA,0CAAc,EAAC,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC;iBAChB,CAAC;YAC/C,CAAC;YAED,OAAO,qBAAqB,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,8CAAsC,CAAC,CAAC,CAAC;YACrC,MAAM,MAAM,GAAmC;gBAC3C,EAAE;gBACF,IAAI,6CAAqC;gBACzC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,SAAS;aAClB,CAAC;YAEF,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,8CAAsC,CAAC,CAAC,CAAC;YACrC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;YAC5C,MAAM,MAAM,GAAmC;gBAC3C,EAAE;gBACF,IAAI,6CAAqC;gBACzC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,SAAS;gBACf,YAAY,EAAE,yBAAyB,CAAC,YAAY,EAAE,OAAO,CAAC;aACjE,CAAC;YACF,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,0CAAkC,CAAC,CAAC,CAAC;YACjC,MAAM,MAAM,GAAmC;gBAC3C,EAAE;gBACF,IAAI,6CAAqC;gBACzC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK;aAClD,CAAC;YACF,OAAO,MAAM,CAAC;QAClB,CAAC;QACD,4CAAoC,CAAC,CAAC,CAAC;YACnC,MAAM,MAAM,GAAmC;gBAC3C,EAAE;gBACF,IAAI,6CAAqC;gBACzC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO;aACpD,CAAC;YACF,OAAO,MAAM,CAAC;QAClB,CAAC;QACD;YACI,OAAO;gBACH,EAAE;gBACF,IAAI,2CAAmC;gBACvC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,aAAa,EAAE,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC;gBAE1E,QAAQ,EAAE,IAAA,mBAAG,EAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;aACxB,CAAC;QAE7C,6CAAqC,CAAC,CAAC,CAAC;YACpC,OAAO;gBACH,EAAE;gBACF,IAAI,4CAAoC;gBACxC,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;gBAChC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;gBAChC,GAAG,EAAE,IAAA,mCAAwB,EAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;aAC5B,CAAC;QAC9C,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,qBAAqB,CAC1B,EAAU,EACV,IAAmF,EACnF,OAA4B;IAE5B,MAAM,OAAO,GAAG,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,EAAkC,CAAC;IACzF,IAAI,YAA2C,CAAC;IAChD,IAAI,UAAgC,CAAC;IAErC,IAAI,YAAY,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACvE,MAAM,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QAC5F,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAkC,CAAC;QAE5E,UAAU,GAAG,IAAA,uCAAuB,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;SAAM,CAAC;QACJ,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACvC,YAAY,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAkC,CAAC;QAExE,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC;YACtC,CAAC,CAAC,IAAA,4CAA4B,EAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC;YAChE,CAAC,CAAC,IAAA,yCAAyB,EAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,QAAQ,GAA+C;QACzD,MAAM,EAAE,YAAY;QACpB,uBAAuB,EAAE,UAAU,CAAC,uBAAuB;KAC9D,CAAC;IAEF,OAAO;QACH,EAAE;QACF,IAAI,iDAAyC;QAC7C,KAAK,EAAE,IAAA,+CAAmB,EAAC,IAAI,CAAC;QAChC,aAAa,EAAE,uBAAuB,CAAC,IAAI,CAAC,KAAK,EAAE,aAAa,EAAE,OAAO,CAAC;QAC1E,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,QAAQ;KACkC,CAAC;AACnD,CAAC;AAED,SAAS,6CAA6C,CAClD,gBAAmE,EACnE,OAA4B;IAE5B,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACpB,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO;QACH,cAAc,EAAE,IAAA,mBAAG,EAAC,gBAAgB,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC/E,qBAAqB,EAAE,gBAAgB,CAAC,qBAAqB;YACzD,CAAC,CAAC,IAAA,mBAAG,EAAC,gBAAgB,CAAC,qBAAqB,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,yBAAyB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAC3F,CAAC,CAAC,SAAS;QACf,UAAU,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC;KAC9E,CAAC;AACN,CAAC;AAED,SAAS,sBAAsB,CAC3B,IAA4B,EAC5B,OAA4B;IAE5B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC;IACrC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,KAAK,YAAY,YAAY,CAAC,eAAe,EAAE,CAAC;QAChD,OAAO;YACH,SAAS,EAAE,IAAA,0CAAc,EAAC,IAAI,CAAC,MAAM,CAAC,WAAY,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC;YACtE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;YAClC,SAAS,EAAE,KAAK,CAAC,SAAS;SACgB,CAAC;IACnD,CAAC;IAED,IAAI,KAAK,YAAY,YAAY,CAAC,WAAW,EAAE,CAAC;QAC5C,OAAO;YACH,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;SACH,CAAC;IAC/C,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,sFAAsF;AACtF,qFAAqF;AACrF,qFAAqF;AACrF,sDAAsD;AACtD,SAAgB,wBAAwB,CACpC,OAAiC,EACjC,OAA4B;IAE5B,oFAAoF;IACpF,sFAAsF;IACtF,MAAM,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC;IAE7D,IAAI,gBAAgB,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,+EAA+E;IAC/E,MAAM,OAAO,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC;IAC9C,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5D,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,mFAAmF;IACnF,sDAAsD;IACtD,IAAI,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACzE,IAAI,CAAC,WAAW,IAAI,SAAS,CAAC,QAAQ,+CAAsC,EAAE,CAAC;QAC3E,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAClE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChF,CAAC;IAED,MAAM,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC;IACjD,IAAI,MAAM,GAAG,WAAW,EAAE,GAAG,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;IAE7G,IACI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC,MAAM,KAAK,CAAC,CAAC;QAClD,SAAS,CAAC,QAAQ,+CAAsC,EAC1D,CAAC;QACC,6EAA6E;QAC7E,8EAA8E;QAC9E,qCAAqC;QACrC,MAAM,UAAU,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,UAAU,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YACrE,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YACtF,MAAM,YAAY,GAAG,aAAa,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;YACrD,IAAI,YAAY,EAAE,CAAC;gBACf,MAAM,GAAG,YAAY,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACJ,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;YAC7E,CAAC;QACL,CAAC;IACL,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,EAAE,eAAe,EAAE,CAAC;IAC/C,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7C,+EAA+E;QAC/E,iFAAiF;QACjF,0DAA0D;QAC1D,MAAM,aAAa,GAAG,qBAAqB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;QACpE,IAAI,aAAa,EAAE,CAAC;YAChB,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;YACtE,OAAO;gBACH,IAAI,mCAA2B;gBAC/B,IAAI,EAAE,aAAa;gBACnB,GAAG,EAAE,QAAQ,CAAC,OAAO;gBACrB,KAAK,EAAE,IAAA,uCAAuB,EAAC,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC;gBAC7D,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,eAAe,EAAE,KAAK;aACH,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,sCAA8B,CAAC,CAAC;IAC3F,OAAO,aAAa,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,mFAAmF;AACnF,mFAAmF;AACnF,4EAA4E;AAC5E,SAAS,kBAAkB,CAAC,OAAe,EAAE,IAAc;IACvD,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAE3E,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,SAAS;QACb,CAAC;QAED,MAAM,KAAK,GAA6B,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACjE,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,EAAG,CAAC;YAC1B,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC9D,IAAI,uBAAuB,CAAC,IAAI,CAAC,IAAI,WAAW,KAAK,OAAO,EAAE,CAAC;gBAC3D,OAAO,IAAI,CAAC;YAChB,CAAC;YAED,wEAAwE;YACxE,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvC,KAAK,MAAM,KAAK,IAAI,IAAA,+BAAa,EAAC,IAAI,CAAC,EAAE,CAAC;oBACtC,IAAI,KAAK,EAAE,CAAC;wBACR,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACtB,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,yFAAyF;AACzF,wFAAwF;AACxF,0DAA0D;AAC1D,SAAS,qBAAqB,CAC1B,SAAiC,EACjC,WAAmB;IAEnB,IAAI,UAA0D,CAAC;IAE/D,QAAQ,SAAS,CAAC,QAAQ,EAAE,CAAC;QACzB,kDAAyC;QACzC,+CAAsC;QACtC;YACI,UAAU,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;YACpC,MAAM;QACV;YACI,UAAU,GAAG,SAAS,CAAC;YACvB,MAAM;IACd,CAAC;IAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,KAAK,WAAW,CAAC,CAAC;AACrF,CAAC;AAED,kFAAkF;AAClF,6EAA6E;AAC7E,SAAS,kBAAkB,CAAC,IAA4B;IACpD,IAAI,OAAO,GAAuC,IAAI,CAAC;IACvD,OAAO,OAAO,EAAE,CAAC;QACb,IAAI,OAAO,CAAC,QAAQ,+CAAsC,EAAE,CAAC;YACzD,OAAO,OAAkC,CAAC;QAC9C,CAAC;QACD,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAC7B,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,0FAA0F;AAC1F,6EAA6E;AAC7E,SAAS,uBAAuB,CAAC,IAA4B;IACzD,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACpB,gDAAuC;QACvC,+CAAsC;QACtC,kDAAyC;QACzC;YACI,OAAO,IAAI,CAAC;QAChB;YACI,OAAO,KAAK,CAAC;IACrB,CAAC;AACL,CAAC;AAED,SAAgB,mBAAmB,CAAC,IAA4B;IAC5D,IAAI,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;QACtC,OAAO,IAAmC,CAAC;IAC/C,CAAC;IAED,MAAM,oBAAqB,SAAQ,iCAAe;QAG9C,SAAS;YACL,OAAO,IAAI,CAAC,OAAO,CAAC;QACxB,CAAC;QAEQ,SAAS,CAAC,IAA4B;YAC3C,IAAI,YAAY,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,OAAO,GAAG,IAAmC,CAAC;gBACnD,OAAO,EAAE,CAAC;YACd,CAAC;YACD,OAAO,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;KACJ;IACD,MAAM,MAAM,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC1C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAElB,OAAO,MAAM,CAAC,SAAS,EAAE,CAAC;AAC9B,CAAC"}

package/dist/typeServer/typeServerEvaluator.d.ts

@@ -0,0 +1,16 @@
+import { Program } from '../analyzer/program';
+import { TypeEvaluator } from '../analyzer/typeEvaluatorTypes';
+import { ISymbolLookup } from './programTypes';
+export interface ITypeServerEvaluator extends TypeEvaluator {
+    getSymbolLookup(): ISymbolLookup;
+}
+/**
+ * Build an `ITypeServerEvaluator` backed by the `Program`'s synchronous `TypeEvaluator`.
+ *
+ * Pyright's evaluator is a closure-backed object whose methods don't depend on `this`, so we
+ * can prototype-delegate to it and add `getSymbolLookup` without copying every method. The
+ * evaluator is resolved from `program.evaluator` at creation time; callers create a fresh
+ * evaluator per request (the program recreates its evaluator when configuration or imports
+ * change), so the captured reference always matches the snapshot being queried.
+ */
+export declare function createTypeServerEvaluator(program: Program, symbolLookup: ISymbolLookup): ITypeServerEvaluator;

package/dist/typeServer/typeServerEvaluator.js

@@ -0,0 +1,36 @@
+"use strict";
+/*
+ * typeServerEvaluator.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * The type server answers type queries with Pyright's ordinary synchronous `TypeEvaluator`.
+ * It needs exactly one capability beyond the stock evaluator: access to the snapshot's symbol
+ * lookup. This module defines that thin extension (`ITypeServerEvaluator`) and a factory that
+ * layers it onto the program's current evaluator.
+ *
+ * Cancellation works exactly as it does everywhere else in Pyright: the evaluator polls the
+ * active cancellation token via `runWithCancellationToken` / `throwIfCancellationRequested`,
+ * so a long-running synchronous query can still be interrupted in the middle of a request.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTypeServerEvaluator = createTypeServerEvaluator;
+/**
+ * Build an `ITypeServerEvaluator` backed by the `Program`'s synchronous `TypeEvaluator`.
+ *
+ * Pyright's evaluator is a closure-backed object whose methods don't depend on `this`, so we
+ * can prototype-delegate to it and add `getSymbolLookup` without copying every method. The
+ * evaluator is resolved from `program.evaluator` at creation time; callers create a fresh
+ * evaluator per request (the program recreates its evaluator when configuration or imports
+ * change), so the captured reference always matches the snapshot being queried.
+ */
+function createTypeServerEvaluator(program, symbolLookup) {
+    const evaluator = program.evaluator;
+    if (!evaluator) {
+        throw new Error('Type evaluator is not available for the current program.');
+    }
+    const wrapper = Object.create(evaluator);
+    wrapper.getSymbolLookup = () => symbolLookup;
+    return wrapper;
+}
+//# sourceMappingURL=typeServerEvaluator.js.map

package/dist/typeServer/typeServerEvaluator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeServerEvaluator.js","sourceRoot":"","sources":["../../../../../src/typeServer/typeServerEvaluator.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;GAaG;;AAoBH,8DASC;AAlBD;;;;;;;;GAQG;AACH,SAAgB,yBAAyB,CAAC,OAAgB,EAAE,YAA2B;IACnF,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAChF,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,SAAS,CAAyB,CAAC;IACjE,OAAO,CAAC,eAAe,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC;IAC7C,OAAO,OAAO,CAAC;AACnB,CAAC"}

package/dist/typeServer/typeServerFileSystem.d.ts

@@ -0,0 +1,47 @@
+import type { Dirent, ReadStream, WriteStream } from 'fs';
+import { Disposable } from 'vscode-jsonrpc';
+import { FileSystem, MkDirOptions, Stats } from '../common/fileSystem';
+import { FileWatcher, FileWatcherEventHandler } from '../common/fileWatcher';
+import { Uri } from '../common/uri/uri';
+import { IPyrightFileSystem } from '../pyrightFileSystem';
+import { INotebookUriMapper } from './notebookUriMapper';
+import { VirtualFileOverlayFileSystem } from './virtualFileOverlayFileSystem';
+export declare class TypeServerFileSystem implements IPyrightFileSystem {
+    private readonly _uriMapper?;
+    private readonly _fs;
+    private readonly _virtualOverlay;
+    constructor(realFileSystem: FileSystem, _uriMapper?: INotebookUriMapper | undefined);
+    /**
+     * Returns the virtual file overlay layer. The server's virtual-file-redirect handlers use
+     * this to add/remove per-file redirects that transparently redirect read operations to
+     * alternate on-disk locations.
+     */
+    get virtualOverlay(): VirtualFileOverlayFileSystem;
+    static is(obj: any): obj is TypeServerFileSystem;
+    mkdirSync(uri: Uri, options?: MkDirOptions): void;
+    writeFileSync(uri: Uri, data: string | Buffer, encoding: BufferEncoding | null): void;
+    unlinkSync(uri: Uri): void;
+    rmdirSync(uri: Uri): void;
+    createFileSystemWatcher(uris: Uri[], listener: FileWatcherEventHandler): FileWatcher;
+    createReadStream(uri: Uri): ReadStream;
+    createWriteStream(uri: Uri): WriteStream;
+    copyFileSync(src: Uri, dst: Uri): void;
+    existsSync(uri: Uri): boolean;
+    chdir(uri: Uri): void;
+    readdirEntriesSync(uri: Uri): Dirent[];
+    readdirSync(uri: Uri): string[];
+    readFileSync(uri: Uri, encoding?: null): Buffer;
+    readFileSync(uri: Uri, encoding: BufferEncoding): string;
+    readFileSync(uri: Uri, encoding?: BufferEncoding | null): string | Buffer;
+    statSync(uri: Uri): Stats;
+    realpathSync(uri: Uri): Uri;
+    getModulePath(): Uri;
+    readFile(uri: Uri): Promise<Buffer>;
+    readFileText(uri: Uri, encoding?: BufferEncoding): Promise<string>;
+    realCasePath(uri: Uri): Uri;
+    isMappedUri(fileUri: Uri): boolean;
+    getOriginalUri(mappedFileUri: Uri): Uri;
+    getMappedUri(originalFileUri: Uri): Uri;
+    isInZip(uri: Uri): boolean;
+    mapDirectory(mappedUri: Uri, originalUri: Uri, filter?: (originalUri: Uri, fs: FileSystem) => boolean): Disposable;
+}