@zzzen/pyright-internal 1.2.0-dev.20260823 → 1.2.0-dev.20260906

package/dist/analyzer/typeStubGeneration.js

@@ -0,0 +1,226 @@
+"use strict";
+/*
+ * typeStubGeneration.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Planning and semantic generation for type stub files.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TypeStubGenerationConflictError = void 0;
+exports.createTypeStubGenerationPlan = createTypeStubGenerationPlan;
+exports.resolveTypeStubTarget = resolveTypeStubTarget;
+exports.runTypeStubGeneration = runTypeStubGeneration;
+exports.collectTypeStubSourceFileUris = collectTypeStubSourceFileUris;
+exports.createPartialTypeStubGenerationPlan = createPartialTypeStubGenerationPlan;
+exports.generateTypeStubFiles = generateTypeStubFiles;
+const cancellationUtils_1 = require("../common/cancellationUtils");
+const pathConsts_1 = require("../common/pathConsts");
+const uri_1 = require("../common/uri/uri");
+const uriUtils_1 = require("../common/uri/uriUtils");
+const tokenizer_1 = require("../parser/tokenizer");
+const analyzerNodeInfo_1 = require("./analyzerNodeInfo");
+const importResolver_1 = require("./importResolver");
+const typeStubRenderer_1 = require("./typeStubRenderer");
+function createTypeStubGenerationPlan(target) {
+    return {
+        source: {
+            targetImportPath: target.targetImportPath,
+            targetIsSingleFile: target.targetIsSingleFile,
+            outputPath: target.outputPath,
+        },
+        additionalFiles: [],
+    };
+}
+function resolveTypeStubTarget(importResolver, configOptions, importName, sourceFileUri) {
+    const moduleNameParts = validateImportName(importName);
+    const fs = importResolver.fileSystem;
+    const importResult = importResolver.resolveImport(uri_1.Uri.empty(), configOptions.findExecEnvironment(sourceFileUri ?? configOptions.projectRoot), (0, importResolver_1.createImportedModuleDescriptor)(importName));
+    if (!importResult.isImportFound) {
+        throw new Error(`Import '${importName}' could not be resolved`);
+    }
+    const firstConcretePartIndex = importResult.resolvedUris.findIndex((uri) => !uri.isEmpty());
+    if (firstConcretePartIndex < 0) {
+        throw new Error(`Import '${importName}' resolves through a namespace package, which is not supported`);
+    }
+    const finalResolvedPath = importResult.resolvedUris[importResult.resolvedUris.length - 1];
+    const isFinalPathFile = (0, uriUtils_1.isFile)(fs, finalResolvedPath);
+    const isFinalPathInitFile = isFinalPathFile && finalResolvedPath.stripAllExtensions().fileName === '__init__';
+    const targetIsSingleFile = importResult.resolvedUris.length === 1 && !isFinalPathInitFile;
+    const firstConcretePath = importResult.resolvedUris[firstConcretePartIndex];
+    const isFirstConcretePathFile = (0, uriUtils_1.isFile)(fs, firstConcretePath);
+    const isFirstConcretePathInitFile = isFirstConcretePathFile && firstConcretePath.stripAllExtensions().fileName === '__init__';
+    const targetImportPath = isFirstConcretePathInitFile || targetIsSingleFile ? firstConcretePath.getDirectory() : firstConcretePath;
+    if (!(0, uriUtils_1.isFile)(fs, targetImportPath) && !(0, uriUtils_1.isDirectory)(fs, targetImportPath)) {
+        throw new Error(`Import '${importName}' could not be resolved`);
+    }
+    const sourceFileUris = [];
+    if (!finalResolvedPath.isEmpty()) {
+        sourceFileUris.push(finalResolvedPath);
+    }
+    importResult.filteredImplicitImports?.forEach((implicitImport) => {
+        if (importResolver_1.ImportResolver.isSupportedImportSourceFile(implicitImport.uri)) {
+            sourceFileUris.push(implicitImport.uri);
+        }
+    });
+    const stubPath = getTypeStubPath(fs, configOptions);
+    const outputPartCount = targetIsSingleFile ? 1 : firstConcretePartIndex + (isFirstConcretePathInitFile ? 1 : 0);
+    return {
+        outputPath: stubPath.resolvePaths(...moduleNameParts.slice(0, outputPartCount)),
+        stubPath,
+        targetImportPath,
+        targetIsSingleFile,
+        sourceFileUris,
+    };
+}
+class TypeStubGenerationConflictError extends Error {
+    constructor(message, uri) {
+        super(message);
+        this.uri = uri;
+        this.name = 'TypeStubGenerationConflictError';
+    }
+}
+exports.TypeStubGenerationConflictError = TypeStubGenerationConflictError;
+async function runTypeStubGeneration(createService, request, handleResult, token) {
+    (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+    const service = await createService();
+    try {
+        let plan;
+        let sourceFileUris;
+        if (request.kind === 'partial') {
+            plan = createPartialTypeStubGenerationPlan(service.fs, getTypeStubPath(service.fs, service.getConfigOptions()), request);
+            sourceFileUris = [request.targetFileUri];
+        }
+        else {
+            const target = resolveTypeStubTarget(service.getImportResolver(), service.getConfigOptions(), request.importName, request.sourceFileUri);
+            plan = createTypeStubGenerationPlan(target);
+            sourceFileUris = collectTypeStubSourceFileUris(service.fs, target, token);
+        }
+        service.backgroundAnalysisProgram.setAllowedThirdPartyImports([request.importName]);
+        service.backgroundAnalysisProgram.setTrackedFiles([...sourceFileUris]);
+        const result = await service.backgroundAnalysisProgram.generateTypeStubFiles(plan, token);
+        return await handleResult(service, result);
+    }
+    finally {
+        service.dispose();
+    }
+}
+function collectTypeStubSourceFileUris(fs, target, token) {
+    (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+    return target.targetIsSingleFile || (0, uriUtils_1.isFile)(fs, target.targetImportPath)
+        ? target.sourceFileUris
+        : collectPackageSourceFileUris(fs, target.targetImportPath, token);
+}
+function collectPackageSourceFileUris(fs, root, token) {
+    const sourceFileUrisByStubUri = new Map();
+    const pending = [root];
+    while (pending.length > 0) {
+        (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+        const directory = pending.pop();
+        const entries = fs.readdirEntriesSync(directory).sort((left, right) => left.name.localeCompare(right.name));
+        for (const entry of entries) {
+            const uri = directory.resolvePaths(entry.name);
+            if (entry.isDirectory()) {
+                pending.push(uri);
+            }
+            else if (entry.isFile() && importResolver_1.ImportResolver.isSupportedImportSourceFile(uri)) {
+                const stubUriKey = uri.replaceExtension('.pyi').key;
+                const existingSource = sourceFileUrisByStubUri.get(stubUriKey);
+                if (!existingSource ||
+                    (existingSource.lastExtension.toLowerCase() !== '.pyi' &&
+                        uri.lastExtension.toLowerCase() === '.pyi')) {
+                    sourceFileUrisByStubUri.set(stubUriKey, uri);
+                }
+            }
+        }
+    }
+    return [...sourceFileUrisByStubUri.values()].sort((left, right) => left.toString().localeCompare(right.toString()));
+}
+function createPartialTypeStubGenerationPlan(fs, stubPath, options) {
+    const moduleNameParts = validateImportName(options.importName);
+    const partialStubRoot = stubPath.resolvePaths(`${moduleNameParts[0]}${pathConsts_1.stubsSuffix}`);
+    return {
+        source: createPartialTypeStubSource(partialStubRoot, options),
+        additionalFiles: createPartialStubMarker(fs, partialStubRoot),
+    };
+}
+function getTypeStubPath(fs, configOptions) {
+    return configOptions.stubPath ?? fs.realCasePath(configOptions.projectRoot.resolvePaths(pathConsts_1.defaultStubsDirectory));
+}
+function validateImportName(importName) {
+    const moduleNameParts = importName.split('.');
+    if (moduleNameParts.some((part) => part.length === 0 || !tokenizer_1.Tokenizer.isPythonIdentifier(part))) {
+        throw new Error(`Invalid import name: ${importName}`);
+    }
+    return moduleNameParts;
+}
+function generateTypeStubFiles(program, plan, token) {
+    const files = [];
+    const fileSystem = program.serviceProvider.fs();
+    if ((0, uriUtils_1.isFile)(fileSystem, plan.source.targetImportPath)) {
+        const sourceFileInfo = program.getSourceFileInfo(plan.source.targetImportPath);
+        if (!sourceFileInfo) {
+            throw new Error(`Could not find source file '${plan.source.targetImportPath.toUserVisibleString()}'`);
+        }
+        files.push(generateTypeStubFile(program, sourceFileInfo, sourceFileInfo.uri.fileName, plan, token));
+    }
+    else {
+        for (const sourceFileInfo of program.getSourceFileInfoList()) {
+            (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+            const relativePath = plan.source.targetImportPath.getRelativePath(sourceFileInfo.uri);
+            if (relativePath !== undefined) {
+                files.push(generateTypeStubFile(program, sourceFileInfo, relativePath, plan, token));
+            }
+        }
+    }
+    if (files.length === 0) {
+        throw new Error(`No source files found for '${plan.source.targetImportPath.toUserVisibleString()}'`);
+    }
+    files.push(...plan.additionalFiles);
+    return { files };
+}
+function generateTypeStubFile(program, sourceFileInfo, relativePath, plan, token) {
+    (0, cancellationUtils_1.throwIfCancellationRequested)(token);
+    program.analyzeFile(sourceFileInfo.uri, token);
+    let stubUri = plan.source.outputPath.resolvePaths(relativePath);
+    if (plan.source.targetIsSingleFile) {
+        stubUri = stubUri.getDirectory().initPyiUri;
+    }
+    else {
+        stubUri = stubUri.replaceExtension('.pyi');
+    }
+    const parseResults = program.getParseResults(sourceFileInfo.uri);
+    if (!parseResults) {
+        throw new Error(`Could not bind file '${sourceFileInfo.uri.toUserVisibleString()}' for stub generation`);
+    }
+    const evaluator = program.evaluator;
+    if (!evaluator) {
+        throw new Error('Type evaluator unavailable for stub generation');
+    }
+    const contents = (0, typeStubRenderer_1.renderTypeStubFile)(stubUri, parseResults, evaluator, (0, analyzerNodeInfo_1.getInfoReader)(program));
+    program.handleMemoryHighUsage();
+    return { uri: stubUri, contents, kind: 'stub' };
+}
+function createPartialTypeStubSource(partialStubRoot, options) {
+    const importNameParts = options.importName.split('.');
+    const relativeImportParts = importNameParts.slice(1);
+    const isPackage = options.targetFileUri.stripAllExtensions().fileName === '__init__';
+    const relativeOutputPathParts = isPackage ? relativeImportParts : relativeImportParts.slice(0, -1);
+    return {
+        targetImportPath: options.targetFileUri,
+        targetIsSingleFile: importNameParts.length === 1 && !isPackage,
+        outputPath: partialStubRoot.resolvePaths(...relativeOutputPathParts),
+    };
+}
+function createPartialStubMarker(fs, partialStubRoot) {
+    const markerUri = partialStubRoot.pytypedUri;
+    if (!fs.existsSync(markerUri)) {
+        return [{ uri: markerUri, contents: 'partial\n', kind: 'partialMarker' }];
+    }
+    const contents = fs.statSync(markerUri).isFile() ? fs.readFileSync(markerUri, 'utf8') : undefined;
+    if (contents !== 'partial\n' && contents !== 'partial\r\n') {
+        throw new TypeStubGenerationConflictError(`Existing py.typed file '${markerUri.toUserVisibleString()}' is incompatible with partial stub generation`, markerUri);
+    }
+    return [];
+}
+//# sourceMappingURL=typeStubGeneration.js.map

package/dist/analyzer/typeStubGeneration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeStubGeneration.js","sourceRoot":"","sources":["../../../../../src/analyzer/typeStubGeneration.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;AAwCH,oEASC;AAED,sDAyDC;AA2BD,sDAqCC;AAED,sEASC;AA6BD,kFAWC;AAeD,sDAgCC;AA1QD,mEAA2E;AAI3E,qDAA0E;AAC1E,2CAAwC;AACxC,qDAA6D;AAC7D,mDAAgD;AAChD,yDAAmD;AACnD,qDAAkF;AAClF,yDAAwD;AA0BxD,SAAgB,4BAA4B,CAAC,MAA8B;IACvE,OAAO;QACH,MAAM,EAAE;YACJ,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;YACzC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB;YAC7C,UAAU,EAAE,MAAM,CAAC,UAAU;SAChC;QACD,eAAe,EAAE,EAAE;KACtB,CAAC;AACN,CAAC;AAED,SAAgB,qBAAqB,CACjC,cAA8B,EAC9B,aAA4B,EAC5B,UAAkB,EAClB,aAAmB;IAEnB,MAAM,eAAe,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;IAEvD,MAAM,EAAE,GAAG,cAAc,CAAC,UAAU,CAAC;IACrC,MAAM,YAAY,GAAG,cAAc,CAAC,aAAa,CAC7C,SAAG,CAAC,KAAK,EAAE,EACX,aAAa,CAAC,mBAAmB,CAAC,aAAa,IAAI,aAAa,CAAC,WAAW,CAAC,EAC7E,IAAA,+CAA8B,EAAC,UAAU,CAAC,CAC7C,CAAC;IACF,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,yBAAyB,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,sBAAsB,GAAG,YAAY,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5F,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,gEAAgE,CAAC,CAAC;IAC3G,CAAC;IAED,MAAM,iBAAiB,GAAG,YAAY,CAAC,YAAY,CAAC,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC1F,MAAM,eAAe,GAAG,IAAA,iBAAM,EAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC;IACtD,MAAM,mBAAmB,GAAG,eAAe,IAAI,iBAAiB,CAAC,kBAAkB,EAAE,CAAC,QAAQ,KAAK,UAAU,CAAC;IAC9G,MAAM,kBAAkB,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC;IAC1F,MAAM,iBAAiB,GAAG,YAAY,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC;IAC5E,MAAM,uBAAuB,GAAG,IAAA,iBAAM,EAAC,EAAE,EAAE,iBAAiB,CAAC,CAAC;IAC9D,MAAM,2BAA2B,GAC7B,uBAAuB,IAAI,iBAAiB,CAAC,kBAAkB,EAAE,CAAC,QAAQ,KAAK,UAAU,CAAC;IAC9F,MAAM,gBAAgB,GAClB,2BAA2B,IAAI,kBAAkB,CAAC,CAAC,CAAC,iBAAiB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;IAE7G,IAAI,CAAC,IAAA,iBAAM,EAAC,EAAE,EAAE,gBAAgB,CAAC,IAAI,CAAC,IAAA,sBAAW,EAAC,EAAE,EAAE,gBAAgB,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,KAAK,CAAC,WAAW,UAAU,yBAAyB,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,cAAc,GAAU,EAAE,CAAC;IACjC,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,EAAE,CAAC;QAC/B,cAAc,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC3C,CAAC;IACD,YAAY,CAAC,uBAAuB,EAAE,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;QAC7D,IAAI,+BAAc,CAAC,2BAA2B,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;YACjE,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,eAAe,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;IACpD,MAAM,eAAe,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,GAAG,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChH,OAAO;QACH,UAAU,EAAE,QAAQ,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC;QAC/E,QAAQ;QACR,gBAAgB;QAChB,kBAAkB;QAClB,cAAc;KACjB,CAAC;AACN,CAAC;AAOD,MAAa,+BAAgC,SAAQ,KAAK;IACtD,YAAY,OAAe,EAAW,GAAQ;QAC1C,KAAK,CAAC,OAAO,CAAC,CAAC;QADmB,QAAG,GAAH,GAAG,CAAK;QAE1C,IAAI,CAAC,IAAI,GAAG,iCAAiC,CAAC;IAClD,CAAC;CACJ;AALD,0EAKC;AAeM,KAAK,UAAU,qBAAqB,CACvC,aAAiD,EACjD,OAAkC,EAClC,YAAiG,EACjG,KAAwB;IAExB,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,MAAM,aAAa,EAAE,CAAC;IAEtC,IAAI,CAAC;QACD,IAAI,IAA4B,CAAC;QACjC,IAAI,cAA8B,CAAC;QACnC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,GAAG,mCAAmC,CACtC,OAAO,CAAC,EAAE,EACV,eAAe,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,gBAAgB,EAAE,CAAC,EACvD,OAAO,CACV,CAAC;YACF,cAAc,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,MAAM,MAAM,GAAG,qBAAqB,CAChC,OAAO,CAAC,iBAAiB,EAAE,EAC3B,OAAO,CAAC,gBAAgB,EAAE,EAC1B,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,aAAa,CACxB,CAAC;YACF,IAAI,GAAG,4BAA4B,CAAC,MAAM,CAAC,CAAC;YAC5C,cAAc,GAAG,6BAA6B,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,CAAC,yBAAyB,CAAC,2BAA2B,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;QACpF,OAAO,CAAC,yBAAyB,CAAC,eAAe,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,yBAAyB,CAAC,qBAAqB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC1F,OAAO,MAAM,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;YAAS,CAAC;QACP,OAAO,CAAC,OAAO,EAAE,CAAC;IACtB,CAAC;AACL,CAAC;AAED,SAAgB,6BAA6B,CACzC,EAAsB,EACtB,MAAyC,EACzC,KAAwB;IAExB,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;IACpC,OAAO,MAAM,CAAC,kBAAkB,IAAI,IAAA,iBAAM,EAAC,EAAE,EAAE,MAAM,CAAC,gBAAgB,CAAC;QACnE,CAAC,CAAC,MAAM,CAAC,cAAc;QACvB,CAAC,CAAC,4BAA4B,CAAC,EAAE,EAAE,MAAM,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;AAC3E,CAAC;AAED,SAAS,4BAA4B,CAAC,EAAsB,EAAE,IAAS,EAAE,KAAwB;IAC7F,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAe,CAAC;IACvD,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC;IACvB,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;QACpC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,EAAG,CAAC;QACjC,MAAM,OAAO,GAAG,EAAE,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5G,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,GAAG,GAAG,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC/C,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtB,CAAC;iBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,IAAI,+BAAc,CAAC,2BAA2B,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3E,MAAM,UAAU,GAAG,GAAG,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC;gBACpD,MAAM,cAAc,GAAG,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC/D,IACI,CAAC,cAAc;oBACf,CAAC,cAAc,CAAC,aAAa,CAAC,WAAW,EAAE,KAAK,MAAM;wBAClD,GAAG,CAAC,aAAa,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,EACjD,CAAC;oBACC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;gBACjD,CAAC;YACL,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,CAAC,GAAG,uBAAuB,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;AACxH,CAAC;AAED,SAAgB,mCAAmC,CAC/C,EAAsB,EACtB,QAAa,EACb,OAAyC;IAEzC,MAAM,eAAe,GAAG,kBAAkB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC/D,MAAM,eAAe,GAAG,QAAQ,CAAC,YAAY,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC,GAAG,wBAAW,EAAE,CAAC,CAAC;IACrF,OAAO;QACH,MAAM,EAAE,2BAA2B,CAAC,eAAe,EAAE,OAAO,CAAC;QAC7D,eAAe,EAAE,uBAAuB,CAAC,EAAE,EAAE,eAAe,CAAC;KAChE,CAAC;AACN,CAAC;AAED,SAAS,eAAe,CAAC,EAAsB,EAAE,aAA4B;IACzE,OAAO,aAAa,CAAC,QAAQ,IAAI,EAAE,CAAC,YAAY,CAAC,aAAa,CAAC,WAAW,CAAC,YAAY,CAAC,kCAAqB,CAAC,CAAC,CAAC;AACpH,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAkB;IAC1C,MAAM,eAAe,GAAG,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC9C,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,qBAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC3F,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO,eAAe,CAAC;AAC3B,CAAC;AAED,SAAgB,qBAAqB,CACjC,OAAoB,EACpB,IAA4B,EAC5B,KAAwB;IAExB,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,eAAe,CAAC,EAAE,EAAE,CAAC;IAEhD,IAAI,IAAA,iBAAM,EAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACnD,MAAM,cAAc,GAAG,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;QAC1G,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,cAAc,EAAE,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IACxG,CAAC;SAAM,CAAC;QACJ,KAAK,MAAM,cAAc,IAAI,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;YAC3D,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;YAEpC,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;YACtF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACzF,CAAC;QACL,CAAC;IACL,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,8BAA8B,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;IACzG,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC;IACpC,OAAO,EAAE,KAAK,EAAE,CAAC;AACrB,CAAC;AAED,SAAS,oBAAoB,CACzB,OAAoB,EACpB,cAA8B,EAC9B,YAAoB,EACpB,IAA4B,EAC5B,KAAwB;IAExB,IAAA,gDAA4B,EAAC,KAAK,CAAC,CAAC;IACpC,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAE/C,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;IAChE,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;QACjC,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC;IAChD,CAAC;SAAM,CAAC;QACJ,OAAO,GAAG,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;IACjE,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,wBAAwB,cAAc,CAAC,GAAG,CAAC,mBAAmB,EAAE,uBAAuB,CAAC,CAAC;IAC7G,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACpC,IAAI,CAAC,SAAS,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,QAAQ,GAAG,IAAA,qCAAkB,EAAC,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,IAAA,gCAAa,EAAC,OAAO,CAAC,CAAC,CAAC;IAC9F,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAChC,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;AACpD,CAAC;AAED,SAAS,2BAA2B,CAChC,eAAoB,EACpB,OAAyC;IAEzC,MAAM,eAAe,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,mBAAmB,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACrD,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,kBAAkB,EAAE,CAAC,QAAQ,KAAK,UAAU,CAAC;IACrF,MAAM,uBAAuB,GAAG,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEnG,OAAO;QACH,gBAAgB,EAAE,OAAO,CAAC,aAAa;QACvC,kBAAkB,EAAE,eAAe,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS;QAC9D,UAAU,EAAE,eAAe,CAAC,YAAY,CAAC,GAAG,uBAAuB,CAAC;KACvE,CAAC;AACN,CAAC;AAED,SAAS,uBAAuB,CAAC,EAAsB,EAAE,eAAoB;IACzE,MAAM,SAAS,GAAG,eAAe,CAAC,UAAU,CAAC;IAC7C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,MAAM,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClG,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QACzD,MAAM,IAAI,+BAA+B,CACrC,2BAA2B,SAAS,CAAC,mBAAmB,EAAE,gDAAgD,EAC1G,SAAS,CACZ,CAAC;IACN,CAAC;IAED,OAAO,EAAE,CAAC;AACd,CAAC"}

package/dist/analyzer/typeStubMessages.d.ts

@@ -0,0 +1,4 @@
+export declare function getTypeStubSuccessMessage(kind: TypeStubOperationKind, importName: string): string;
+export declare function getTypeStubCancellationMessage(kind: TypeStubOperationKind, importName: string): string;
+export declare function getTypeStubErrorPrefix(kind: TypeStubOperationKind, importName: string): string;
+export type TypeStubOperationKind = 'full' | 'partial';

package/dist/analyzer/typeStubMessages.js

@@ -0,0 +1,25 @@
+"use strict";
+/*
+ * typeStubMessages.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Shared user-facing messages for type stub operations.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTypeStubSuccessMessage = getTypeStubSuccessMessage;
+exports.getTypeStubCancellationMessage = getTypeStubCancellationMessage;
+exports.getTypeStubErrorPrefix = getTypeStubErrorPrefix;
+function getTypeStubSuccessMessage(kind, importName) {
+    const prefix = kind === 'partial' ? 'Partial type stub' : 'Type stub';
+    return `${prefix} was successfully created for '${importName}'.`;
+}
+function getTypeStubCancellationMessage(kind, importName) {
+    const prefix = kind === 'partial' ? 'Partial type stub creation' : 'Type stub creation';
+    return `${prefix} for '${importName}' was canceled`;
+}
+function getTypeStubErrorPrefix(kind, importName) {
+    const stubKind = kind === 'partial' ? 'partial type stub' : 'type stub';
+    return `An error occurred when creating ${stubKind} for '${importName}'`;
+}
+//# sourceMappingURL=typeStubMessages.js.map

package/dist/analyzer/typeStubMessages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeStubMessages.js","sourceRoot":"","sources":["../../../../../src/analyzer/typeStubMessages.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AAEH,8DAGC;AAED,wEAGC;AAED,wDAGC;AAbD,SAAgB,yBAAyB,CAAC,IAA2B,EAAE,UAAkB;IACrF,MAAM,MAAM,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,WAAW,CAAC;IACtE,OAAO,GAAG,MAAM,kCAAkC,UAAU,IAAI,CAAC;AACrE,CAAC;AAED,SAAgB,8BAA8B,CAAC,IAA2B,EAAE,UAAkB;IAC1F,MAAM,MAAM,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,oBAAoB,CAAC;IACxF,OAAO,GAAG,MAAM,SAAS,UAAU,gBAAgB,CAAC;AACxD,CAAC;AAED,SAAgB,sBAAsB,CAAC,IAA2B,EAAE,UAAkB;IAClF,MAAM,QAAQ,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,WAAW,CAAC;IACxE,OAAO,mCAAmC,QAAQ,SAAS,UAAU,GAAG,CAAC;AAC7E,CAAC"}

package/dist/analyzer/typeStubOutput.d.ts

@@ -0,0 +1,9 @@
+import { WorkspaceEdit } from 'vscode-languageserver-types';
+import { FileSystem, ReadOnlyFileSystem } from '../common/fileSystem';
+import { GeneratedTypeStubFile } from './typeStubGeneration';
+export declare function writeGeneratedTypeStubFiles(fs: FileSystem, files: readonly GeneratedTypeStubFile[]): void;
+export declare function convertGeneratedTypeStubFilesToWorkspaceEdit(fs: ReadOnlyFileSystem, files: readonly GeneratedTypeStubFile[], getOpenDocument?: (uri: GeneratedTypeStubFile['uri']) => OpenDocumentSnapshot | undefined): WorkspaceEdit;
+export interface OpenDocumentSnapshot {
+    version: number;
+    contents: string;
+}

package/dist/analyzer/typeStubOutput.js

@@ -0,0 +1,52 @@
+"use strict";
+/*
+ * typeStubOutput.ts
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT license.
+ *
+ * Output adapters for generated type stub files.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.writeGeneratedTypeStubFiles = writeGeneratedTypeStubFiles;
+exports.convertGeneratedTypeStubFilesToWorkspaceEdit = convertGeneratedTypeStubFilesToWorkspaceEdit;
+const vscode_languageserver_types_1 = require("vscode-languageserver-types");
+const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
+const uriUtils_1 = require("../common/uri/uriUtils");
+function writeGeneratedTypeStubFiles(fs, files) {
+    for (const file of files) {
+        fs.mkdirSync(file.uri.getDirectory(), { recursive: true });
+        fs.writeFileSync(file.uri, file.contents, 'utf8');
+    }
+}
+function convertGeneratedTypeStubFilesToWorkspaceEdit(fs, files, getOpenDocument) {
+    const documentChanges = [];
+    for (const file of files) {
+        const uri = (0, uriUtils_1.convertUriToLspUriString)(fs, file.uri);
+        const openDocument = getOpenDocument?.(file.uri);
+        const fileExists = fs.existsSync(file.uri);
+        const existingContents = fileExists ? fs.readFileSync(file.uri, 'utf8') : undefined;
+        if (openDocument && openDocument.contents !== existingContents) {
+            throw new Error(`Cannot update generated type stub '${file.uri.toUserVisibleString()}' because it has unsaved changes`);
+        }
+        let range = {
+            start: { line: 0, character: 0 },
+            end: { line: 0, character: 0 },
+        };
+        if (existingContents !== undefined) {
+            const document = vscode_languageserver_textdocument_1.TextDocument.create(uri, 'plaintext', openDocument?.version ?? 0, existingContents);
+            range = {
+                start: range.start,
+                end: document.positionAt(existingContents.length),
+            };
+        }
+        else {
+            documentChanges.push(vscode_languageserver_types_1.CreateFile.create(uri, { overwrite: false }));
+        }
+        documentChanges.push(vscode_languageserver_types_1.TextDocumentEdit.create({
+            uri,
+            version: openDocument?.version ?? null,
+        }, [{ range, newText: file.contents }]));
+    }
+    return { documentChanges };
+}
+//# sourceMappingURL=typeStubOutput.js.map

package/dist/analyzer/typeStubOutput.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeStubOutput.js","sourceRoot":"","sources":["../../../../../src/analyzer/typeStubOutput.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;AASH,kEAKC;AAED,oGA6CC;AA3DD,6EAA0F;AAC1F,2FAAkE;AAGlE,qDAAkE;AAGlE,SAAgB,2BAA2B,CAAC,EAAc,EAAE,KAAuC;IAC/F,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACtD,CAAC;AACL,CAAC;AAED,SAAgB,4CAA4C,CACxD,EAAsB,EACtB,KAAuC,EACvC,eAAyF;IAEzF,MAAM,eAAe,GAAkD,EAAE,CAAC;IAE1E,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAA,mCAAwB,EAAC,EAAE,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QACnD,MAAM,YAAY,GAAG,eAAe,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,gBAAgB,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACpF,IAAI,YAAY,IAAI,YAAY,CAAC,QAAQ,KAAK,gBAAgB,EAAE,CAAC;YAC7D,MAAM,IAAI,KAAK,CACX,sCAAsC,IAAI,CAAC,GAAG,CAAC,mBAAmB,EAAE,kCAAkC,CACzG,CAAC;QACN,CAAC;QAED,IAAI,KAAK,GAAG;YACR,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;YAChC,GAAG,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE;SACjC,CAAC;QAEF,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,iDAAY,CAAC,MAAM,CAAC,GAAG,EAAE,WAAW,EAAE,YAAY,EAAE,OAAO,IAAI,CAAC,EAAE,gBAAgB,CAAC,CAAC;YACrG,KAAK,GAAG;gBACJ,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,GAAG,EAAE,QAAQ,CAAC,UAAU,CAAC,gBAAgB,CAAC,MAAM,CAAC;aACpD,CAAC;QACN,CAAC;aAAM,CAAC;YACJ,eAAe,CAAC,IAAI,CAAC,wCAAU,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACvE,CAAC;QAED,eAAe,CAAC,IAAI,CAChB,8CAAgB,CAAC,MAAM,CACnB;YACI,GAAG;YACH,OAAO,EAAE,YAAY,EAAE,OAAO,IAAI,IAAI;SACzC,EACD,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CACtC,CACJ,CAAC;IACN,CAAC;IAED,OAAO,EAAE,eAAe,EAAE,CAAC;AAC/B,CAAC"}

package/dist/analyzer/typeStubRenderer.d.ts

@@ -0,0 +1,5 @@
+import { Uri } from '../common/uri/uri';
+import { ParseFileResults } from '../parser/parser';
+import * as AnalyzerNodeInfo from './analyzerNodeInfo';
+import { TypeEvaluator } from './typeEvaluatorTypes';
+export declare function renderTypeStubFile(stubPath: Uri, parseResults: ParseFileResults, evaluator: TypeEvaluator, nodeInfo: AnalyzerNodeInfo.AnalyzerNodeInfoReader): string;

package/dist/analyzer/{typeStubWriter.js → typeStubRenderer.js}

@@ -42,12 +42,9 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TypeStubWriter = void 0;
-const cancellationUtils_1 = require("../common/cancellationUtils");
-const uriUtils_1 = require("../common/uri/uriUtils");
+exports.renderTypeStubFile = renderTypeStubFile;
 const parseNodes_1 = require("../parser/parseNodes");
 const AnalyzerNodeInfo = __importStar(require("./analyzerNodeInfo"));
-const analyzerNodeInfo_1 = require("./analyzerNodeInfo");
 const ParseTreeUtils = __importStar(require("./parseTreeUtils"));
 const parseTreeWalker_1 = require("./parseTreeWalker");
 const scopeUtils_1 = require("./scopeUtils");
@@ -130,93 +127,14 @@ class ImportSymbolWalker extends parseTreeWalker_1.ParseTreeWalker {
         return undefined;
     }
 }
-class TypeStubWriter {
-    constructor(_program) {
-        this._program = _program;
-    }
-    writeTypeStub(request, token) {
-        const fileSystem = this._program.serviceProvider.fs();
-        const stubPath = request.stubPath ?? request.outputPath;
-        this._ensureTypeStubOutputPath(fileSystem, request.outputPath, stubPath);
-        if ((0, uriUtils_1.isFile)(fileSystem, request.targetImportPath)) {
-            const sourceFileInfo = this._program.getSourceFileInfo(request.targetImportPath);
-            if (!sourceFileInfo) {
-                throw new Error(`Could not find source file '${request.targetImportPath.toUserVisibleString()}'`);
-            }
-            this._writeTypeStubForSourceFile(sourceFileInfo, sourceFileInfo.uri.fileName, request, fileSystem, token);
-            return;
-        }
-        for (const sourceFileInfo of this._program.getSourceFileInfoList()) {
-            (0, cancellationUtils_1.throwIfCancellationRequested)(token);
-            const relativePath = request.targetImportPath.getRelativePath(sourceFileInfo.uri);
-            if (relativePath === undefined) {
-                continue;
-            }
-            this._writeTypeStubForSourceFile(sourceFileInfo, relativePath, request, fileSystem, token);
-        }
-    }
-    _writeTypeStubForSourceFile(sourceFileInfo, relativePath, request, fileSystem, token) {
-        (0, cancellationUtils_1.throwIfCancellationRequested)(token);
-        this._program.analyzeFile(sourceFileInfo.uri, token);
-        let typeStubPath = request.outputPath.resolvePaths(relativePath);
-        if (request.targetIsSingleFile) {
-            typeStubPath = typeStubPath.getDirectory().initPyiUri;
-        }
-        else {
-            typeStubPath = typeStubPath.replaceExtension('.pyi');
-        }
-        const typeStubDir = typeStubPath.getDirectory();
-        try {
-            (0, uriUtils_1.makeDirectories)(fileSystem, typeStubDir, request.outputPath);
-        }
-        catch {
-            throw new Error(`Could not create directory for '${typeStubDir}'`);
-        }
-        const parseResults = this._program.getParseResults(sourceFileInfo.uri);
-        if (!parseResults) {
-            throw new Error(`Could not bind file '${sourceFileInfo.uri.toUserVisibleString()}' for stub generation`);
-        }
-        const evaluator = this._program.evaluator;
-        if (!evaluator) {
-            throw new Error('Type evaluator unavailable for stub generation');
-        }
-        const treeWalker = new TypeStubTreeWalker(typeStubPath, parseResults, fileSystem, evaluator, (0, analyzerNodeInfo_1.getInfoReader)(this._program));
-        treeWalker.write();
-        this._program.handleMemoryHighUsage();
-    }
-    _ensureTypeStubOutputPath(fileSystem, outputPath, stubPath) {
-        try {
-            if (!fileSystem.existsSync(stubPath)) {
-                // Use recursive to handle callers that pass outputPath as
-                // stubPath (e.g. BackgroundAnalysisProgram.writeTypeStub)
-                // where parent directories may not yet exist.
-                fileSystem.mkdirSync(stubPath, { recursive: true });
-            }
-        }
-        catch {
-            const errMsg = `Could not create typings directory '${stubPath.toUserVisibleString()}'`;
-            this._program.console.error(errMsg);
-            throw new Error(errMsg);
-        }
-        try {
-            if (!fileSystem.existsSync(outputPath)) {
-                (0, uriUtils_1.makeDirectories)(fileSystem, outputPath, stubPath);
-            }
-        }
-        catch {
-            const errMsg = `Could not create typings subdirectory '${outputPath.toUserVisibleString()}'`;
-            this._program.console.error(errMsg);
-            throw new Error(errMsg);
-        }
-    }
+function renderTypeStubFile(stubPath, parseResults, evaluator, nodeInfo) {
+    return new TypeStubRenderer(stubPath, parseResults, evaluator, nodeInfo).render();
 }
-exports.TypeStubWriter = TypeStubWriter;
-class TypeStubTreeWalker extends parseTreeWalker_1.ParseTreeWalker {
-    constructor(_stubPath, _parseResults, _fileSystem, _evaluator, _nodeInfo) {
-        super();
+class TypeStubRenderer extends parseTreeWalker_1.ParseTreeWalker {
+    constructor(_stubPath, _parseResults, _evaluator, _nodeInfo) {
+        super(_nodeInfo);
         this._stubPath = _stubPath;
         this._parseResults = _parseResults;
-        this._fileSystem = _fileSystem;
         this._evaluator = _evaluator;
         this._nodeInfo = _nodeInfo;
         this._indentAmount = 0;
@@ -239,11 +157,11 @@ class TypeStubTreeWalker extends parseTreeWalker_1.ParseTreeWalker {
             this._includeAllImports = true;
         }
     }
-    write() {
+    render() {
         this._lineEnd = this._parseResults.tokenizerOutput.predominantEndOfLineSequence;
         this._tab = this._parseResults.tokenizerOutput.predominantTabSequence;
         this.walk(this._parseResults.parserOutput.parseTree);
-        this._writeFile();
+        return this._getFileContents();
     }
     walk(node) {
         if (!AnalyzerNodeInfo.isCodeUnreachable(node, this._nodeInfo)) {
@@ -747,11 +665,11 @@ class TypeStubTreeWalker extends parseTreeWalker_1.ParseTreeWalker {
         }
         return importStr;
     }
-    _writeFile() {
+    _getFileContents() {
         let finalText = this._printHeaderDocString();
         finalText += this._printTrackedImports();
         finalText += this._typeStubText;
-        this._fileSystem.writeFileSync(this._stubPath, finalText, 'utf8');
+        return finalText;
     }
 }
-//# sourceMappingURL=typeStubWriter.js.map
+//# sourceMappingURL=typeStubRenderer.js.map

package/dist/analyzer/typeStubRenderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeStubRenderer.js","sourceRoot":"","sources":["../../../../../src/analyzer/typeStubRenderer.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAsJH,gDAOC;AA1JD,qDA6B8B;AAG9B,qEAAuD;AACvD,iEAAmD;AACnD,uDAAoD;AACpD,6CAA+C;AAE/C,mEAAqD;AAErD,mCAQiB;AAEjB,MAAM,aAAa;IAGf,YAAmB,UAAkB;QAAlB,eAAU,GAAV,UAAU,CAAQ;QAFrC,eAAU,GAAG,KAAK,CAAC;IAEqB,CAAC;CAC5C;AAED,MAAM,eAAgB,SAAQ,aAAa;IACvC,YAAY,UAAkB,EAAS,KAAyB,EAAS,MAAc;QACnF,KAAK,CAAC,UAAU,CAAC,CAAC;QADiB,UAAK,GAAL,KAAK,CAAoB;QAAS,WAAM,GAAN,MAAM,CAAQ;IAEvF,CAAC;CACJ;AASD,MAAM,iBAAkB,SAAQ,aAAa;IAGzC,YAAY,UAAkB,EAAS,gBAAyB,EAAS,IAAqB;QAC1F,KAAK,CAAC,UAAU,CAAC,CAAC;QADiB,qBAAgB,GAAhB,gBAAgB,CAAS;QAAS,SAAI,GAAJ,IAAI,CAAiB;QAF9F,YAAO,GAA0B,EAAE,CAAC;IAIpC,CAAC;IAED,SAAS,CAAC,MAA0B,EAAE,IAAY,EAAE,KAAyB,EAAE,UAAU,GAAG,KAAK;QAC7F,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;YAC7C,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;gBACd,MAAM;gBACN,IAAI;gBACJ,KAAK;gBACL,UAAU;aACb,CAAC,CAAC;QACP,CAAC;IACL,CAAC;CACJ;AAED,MAAM,kBAAmB,SAAQ,iCAAe;IAC5C,YACY,wBAAqC,EACrC,sBAA+B,EAC/B,SAAkD;QAE1D,KAAK,CAAC,SAAS,CAAC,CAAC;QAJT,6BAAwB,GAAxB,wBAAwB,CAAa;QACrC,2BAAsB,GAAtB,sBAAsB,CAAS;QAC/B,cAAS,GAAT,SAAS,CAAyC;IAG9D,CAAC;IAED,OAAO,CAAC,IAAoB;QACxB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAEQ,IAAI,CAAC,IAAe;QACzB,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;IACL,CAAC;IAEQ,SAAS,CAAC,IAAc;QAC7B,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEQ,iBAAiB,CAAC,IAAsB;QAC7C,MAAM,cAAc,GAAG,IAAI,CAAC,mCAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEjF,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,GAAG,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEQ,WAAW,CAAC,IAAgB;QACjC,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAC9B,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACpD,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,mCAAmC,CAAC,IAAoB;QAC5D,IAAI,IAAI,CAAC,QAAQ,gCAAuB,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,wCAA+B,EAAE,CAAC;YAC/C,MAAM,cAAc,GAAG,IAAI,CAAC,mCAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACjF,IAAI,CAAC,cAAc,EAAE,CAAC;gBAClB,OAAO,SAAS,CAAC;YACrB,CAAC;YAED,OAAO,GAAG,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;QACxD,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AAED,SAAgB,kBAAkB,CAC9B,QAAa,EACb,YAA8B,EAC9B,SAAwB,EACxB,QAAiD;IAEjD,OAAO,IAAI,gBAAgB,CAAC,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,MAAM,EAAE,CAAC;AACtF,CAAC;AAED,MAAM,gBAAiB,SAAQ,iCAAe;IAe1C,YACqB,SAAc,EACd,aAA+B,EAC/B,UAAyB,EACzB,SAAkD;QAEnE,KAAK,CAAC,SAAS,CAAC,CAAC;QALA,cAAS,GAAT,SAAS,CAAK;QACd,kBAAa,GAAb,aAAa,CAAkB;QAC/B,eAAU,GAAV,UAAU,CAAe;QACzB,cAAS,GAAT,SAAS,CAAyC;QAlB/D,kBAAa,GAAG,CAAC,CAAC;QAClB,uBAAkB,GAAG,KAAK,CAAC;QAC3B,kBAAa,GAAG,EAAE,CAAC;QACnB,aAAQ,GAAG,IAAI,CAAC;QAChB,SAAI,GAAG,MAAM,CAAC;QACd,oBAAe,GAAG,CAAC,CAAC;QACpB,uBAAkB,GAAG,CAAC,CAAC;QACvB,iBAAY,GAAG,CAAC,CAAC;QACjB,kBAAa,GAAG,KAAK,CAAC;QACtB,mBAAc,GAAG,IAAI,CAAC;QACtB,qBAAgB,GAAG,IAAI,GAAG,EAA2B,CAAC;QACtD,uBAAkB,GAAG,IAAI,GAAG,EAA6B,CAAC;QAC1D,6BAAwB,GAAG,IAAI,GAAG,EAAU,CAAC;QAUjD,6DAA6D;QAC7D,iEAAiE;QACjE,6DAA6D;QAC7D,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,KAAK,cAAc,EAAE,CAAC;YAC7C,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;QACnC,CAAC;IACL,CAAC;IAED,MAAM;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,4BAA4B,CAAC;QAChF,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,sBAAsB,CAAC;QAEtE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACnC,CAAC;IAEQ,IAAI,CAAC,IAAe;QACzB,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;IACL,CAAC;IAEQ,UAAU,CAAC,IAAe;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QAEtC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACxC,IAAI,IAAI,GAAG,SAAS,SAAS,EAAE,CAAC;QAEhC,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC;QAED,oDAAoD;QACpD,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAChC,CAAC,GAAG,EAAE,EAAE,CACJ,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS;YACxB,GAAG,CAAC,CAAC,CAAC,WAAW,+BAAuB;YACxC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,gCAAuB;YAC/C,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAC3C,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAClB,IAAI,IAAI,IAAI,IAAI;iBACX,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBACT,IAAI,SAAS,GAAG,EAAE,CAAC;gBACnB,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACb,SAAS,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC;gBACzC,CAAC;gBACD,SAAS,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACpD,OAAO,SAAS,CAAC;YACrB,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;QACvB,CAAC;QACD,IAAI,IAAI,GAAG,CAAC;QACZ,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAErB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE;YACjB,IAAI,CAAC,eAAe,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACxB,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACnB,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAEnB,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,aAAa,CAAC,IAAkB;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QAEzC,+EAA+E;QAC/E,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,wBAAwB,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3F,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACxC,IAAI,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1C,IAAI,IAAI,OAAO,YAAY,EAAE,CAAC;YAE9B,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;gBACpB,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACrD,CAAC;YAED,IAAI,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;YAEpG,IAAI,gBAAoC,CAAC;YACzC,IAAI,IAAI,CAAC,CAAC,CAAC,gBAAgB,EAAE,CAAC;gBAC1B,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,EAAE,2BAA2B,CAAC,IAAI,CAAC,CAAC;YACxG,CAAC;iBAAM,IAAI,IAAI,CAAC,CAAC,CAAC,qBAAqB,EAAE,CAAC;gBACtC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CACpC,IAAI,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,gBAAgB;gBAC/C,2BAA2B,CAAC,IAAI,CACnC,CAAC;YACN,CAAC;iBAAM,CAAC;gBACJ,6DAA6D;gBAC7D,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;oBACrC,gBAAgB,GAAG,MAAM,CAAC;gBAC9B,CAAC;qBAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;oBAC3C,gBAAgB,GAAG,KAAK,CAAC;gBAC7B,CAAC;qBAAM,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC9E,gBAAgB,GAAG,KAAK,CAAC;gBAC7B,CAAC;qBAAM,IACH,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAC7D,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CACzC,EACH,CAAC;oBACC,gBAAgB,GAAG,MAAM,CAAC;gBAC9B,CAAC;YACL,CAAC;YAED,IAAI,gBAAgB,EAAE,CAAC;gBACnB,IAAI,IAAI,MAAM,GAAG,gBAAgB,CAAC;YACtC,CAAC;YAED,IAAI,IAAI,GAAG,CAAC;YAEZ,+DAA+D;YAC/D,sDAAsD;YACtD,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBAC7D,IAAI,YAAY,IAAI,IAAA,kBAAU,EAAC,YAAY,CAAC,YAAY,CAAC,EAAE,CAAC;oBACxD,IAAI,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;oBAClF,UAAU,GAAG,IAAA,8BAAsB,EAAC,UAAU,CAAC,CAAC;oBAChD,IAAI,CAAC,IAAA,eAAO,EAAC,UAAU,CAAC,IAAI,CAAC,IAAA,iBAAS,EAAC,UAAU,CAAC,EAAE,CAAC;wBACjD,IAAI,IAAI,SAAS,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,mBAAmB,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC;oBAC7F,CAAC;gBACL,CAAC;YACL,CAAC;YAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAErB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE;gBACjB,mCAAmC;gBACnC,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACxB,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACvB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,UAAU,CAAC,IAAe;QAC/B,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,QAAQ,CAAC,IAAa;QAC3B,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,QAAQ,CAAC,IAAa;QAC3B,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAE5B,6DAA6D;QAC7D,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC3B,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,SAAS,CAAC,IAAc;QAC7B,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,OAAO,CAAC,IAAY;QACzB,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAE5B,4CAA4C;QAC5C,uBAAuB;QACvB,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,SAAS,CAAC,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC;YACrE,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE;gBACjB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YACnC,IAAI,SAAS,EAAE,CAAC;gBACZ,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACxB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE;oBACjB,IAAI,SAAS,CAAC,QAAQ,8BAAqB,EAAE,CAAC;wBAC1C,IAAI,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;oBAC1F,CAAC;yBAAM,CAAC;wBACJ,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;oBAC9C,CAAC;gBACL,CAAC,CAAC,CAAC;YACP,CAAC;YACD,IAAI,CAAC,YAAY,EAAE,CAAC;QACxB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,cAAc,CAAC,IAAmB;QACvC,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAE1C,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,IAAI,KAAK,CAAC;QACd,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAErB,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,eAAe,CAAC,IAAoB;QACzC,IAAI,WAAW,GAAG,KAAK,CAAC;QACxB,IAAI,IAAI,GAAG,EAAE,CAAC;QAEd,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,gCAAuB,EAAE,CAAC;YAClD,sCAAsC;YACtC,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxC,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;oBAC3D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;oBAE1B,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBAC9C,IAAI,IAAI,KAAK,CAAC;oBACd,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAChD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC;gBAED,OAAO,KAAK,CAAC;YACjB,CAAC;YAED,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;gBAChC,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAC9C,IAAI,IAAI,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;oBAC3B,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,EAAE,2BAA2B,CAAC,IAAI,CAAC,CAAC;gBACrG,CAAC;gBAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBAC3D,IAAI,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAClC,WAAW,GAAG,IAAI,CAAC;gBACvB,CAAC;qBAAM,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,+BAAuB,EAAE,CAAC;oBAC1D,yEAAyE;oBACzE,0BAA0B;oBAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBAC1E,IACI,YAAY;wBACZ,IAAA,2BAAmB,EAAC,YAAY,CAAC;wBACjC,iBAAS,CAAC,SAAS,CAAC,YAAY,EAAE,CAAC,SAAS,EAAE,cAAc,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC,EACxF,CAAC;wBACC,WAAW,GAAG,IAAI,CAAC;oBACvB,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;aAAM,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,0CAAiC,EAAE,CAAC;YACnE,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YAE9C,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,EAAE;gBACnF,iBAAiB,EAAE,IAAI;gBACvB,aAAa,EAAE,IAAI;aACtB,CAAC,CAAC;YAEH,6CAA6C;YAC7C,IAAI,IAAA,uBAAe,EAAC,YAAY,CAAC,IAAI,iBAAS,CAAC,SAAS,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,CAAC;gBAClF,WAAW,GAAG,IAAI,CAAC;YACvB,CAAC;YAED,IAAI,SAAS,CAAC,QAAQ,gCAAuB,EAAE,CAAC;gBAC5C,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;oBAChC,IAAI,GAAG,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,gBAAgB,CAChE,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU;oBAC5B,2BAA2B,CAAC,IAAI,CACnC,EAAE,CAAC;gBACR,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,IAAI,EAAE,CAAC;YACP,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAE1B,IAAI,IAAI,KAAK,CAAC;YAEd,IAAI,WAAW,EAAE,CAAC;gBACd,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACJ,IAAI,IAAI,KAAK,CAAC;YAClB,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,wBAAwB,CAAC,IAA6B;QAC3D,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,gCAAuB,EAAE,CAAC;YAClD,yCAAyC;YACzC,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,QAAQ,kCAA0B,EAAE,CAAC;gBACrF,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;oBAC3D,IAAI,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;oBAClD,IAAI,IAAI,MAAM,CAAC;oBACf,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAChD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,mBAAmB,CAAC,IAAwB;QACjD,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;YAChC,IAAI,IAAI,GAAG,EAAE,CAAC;YACd,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,gCAAuB,EAAE,CAAC;gBACnD,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACnD,CAAC;iBAAM,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,wCAA+B,EAAE,CAAC;gBAClE,MAAM,cAAc,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACnD,IAAI,cAAc,CAAC,QAAQ,gCAAuB,EAAE,CAAC;oBACjD,IAAI,cAAc,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;wBACpC,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;wBACrD,IAAI,CAAC,eAAe,CAAC,wBAAwB,CAAC,UAAU,CAAC,EAAE,CAAC;4BACxD,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;wBACnD,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAED,IAAI,IAAI,EAAE,CAAC;gBACP,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,2BAA2B,CAAC,IAAI,CAAC,CAAC;gBAC1F,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;QACL,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,WAAW,CAAC,IAAgB;QACjC,IAAI,IAAI,CAAC,kBAAkB,GAAG,CAAC,IAAI,IAAI,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YAC1D,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,YAAY,GAAG,IAAA,4BAAe,EAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,YAAY,EAAE,CAAC;YACf,8BAA8B;YAC9B,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBACxB,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;gBACvD,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACzC,MAAM,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK;wBAC1B,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK;wBACrB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC;4BACrC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK;4BACrC,CAAC,CAAC,EAAE,CAAC;oBACT,MAAM,UAAU,GAAG,YAAY,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;oBAClE,IAAI,UAAU,EAAE,CAAC;wBACb,MAAM,eAAe,GAAG,IAAI,eAAe,CACvC,UAAU,EACV,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,EAC7C,UAAU,CAAC,MAAM,CACpB,CAAC;wBACF,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;oBAC3D,CAAC;gBACL,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,eAAe,CAAC,IAAoB;QACzC,IAAI,IAAI,CAAC,kBAAkB,GAAG,CAAC,IAAI,IAAI,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YAC1D,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,YAAY,GAAG,IAAA,4BAAe,EAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,YAAY,EAAE,CAAC;YACf,8BAA8B;YAC9B,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACxD,IAAI,iBAAiB,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAChE,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACrB,iBAAiB,GAAG,IAAI,iBAAiB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;gBACrF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;YAC/D,CAAC;YAED,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC3B,MAAM,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;gBAC1E,MAAM,UAAU,GAAG,YAAY,CAAC,qBAAqB,CAAC,UAAU,CAAC,CAAC;gBAClE,IAAI,UAAU,EAAE,CAAC;oBACb,iBAAkB,CAAC,SAAS,CACxB,UAAU,CAAC,MAAM,EACjB,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,EAClB,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,EAC7C,KAAK,CACR,CAAC;gBACN,CAAC;YACL,CAAC,CAAC,CAAC;QACP,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAEQ,kBAAkB,CAAC,IAAuB;QAC/C,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,sCAA6B,EAAE,CAAC;YAC7F,2DAA2D;YAC3D,iDAAiD;YACjD,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC7C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChE,CAAC;QACL,CAAC;QAED,qDAAqD;QACrD,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAE5B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACrC,OAAO,KAAK,CAAC;IACjB,CAAC;IAEO,UAAU,CAAC,QAAoB;QACnC,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE;YACtB,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC;YAC5C,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;YAE3B,QAAQ,EAAE,CAAC;YAEX,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;gBACtB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;YAED,IAAI,CAAC,aAAa,GAAG,gBAAgB,CAAC;QAC1C,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,eAAe,CAAC,QAAoB;QACxC,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,QAAQ,EAAE,CAAC;QACX,IAAI,CAAC,aAAa,EAAE,CAAC;IACzB,CAAC;IAEO,eAAe,CAAC,UAA2B;QAC/C,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC7B,IAAI,CAAC,SAAS,CAAC,GAAG,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,qBAAqB;QACzB,OAAO,CACH,KAAK;YACL,IAAI,CAAC,QAAQ;YACb,+CAA+C;YAC/C,IAAI,CAAC,QAAQ;YACb,KAAK;YACL,IAAI,CAAC,QAAQ;YACb,IAAI,CAAC,QAAQ,CAChB,CAAC;IACN,CAAC;IAEO,SAAS,CAAC,IAAY;QAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,aAAa,IAAI,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC;IAC/C,CAAC;IAEO,gBAAgB,CAAC,IAA2B;QAChD,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IAC9F,CAAC;IAEO,eAAe,CAAC,IAAuB;QAC3C,IAAI,IAAI,GAAG,EAAE,CAAC;QAEd,IAAI,IAAI,CAAC,CAAC,CAAC,aAAa,KAAK,0BAAa,CAAC,YAAY,EAAE,CAAC;YACtD,IAAI,IAAI,GAAG,CAAC;QAChB,CAAC;aAAM,IAAI,IAAI,CAAC,CAAC,CAAC,aAAa,KAAK,0BAAa,CAAC,SAAS,EAAE,CAAC;YAC1D,IAAI,IAAI,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QAE5B,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,IAAI,IAAI,CAAC;YACb,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,IAAI,KAAK,CAAC;YACd,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,gBAAgB,CAAC,IAAoB;QACzC,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,IAAI,GAAG,CAAC;QAChB,CAAC;QACD,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,WAAW,CAAC,SAAwB,EAAE,YAA0B,EAAE,UAAkB;QACxF,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,SAAS,CAAC,CAAC,CAAC,QAAQ,mCAA2B,EAAE,CAAC;YAClD,IAAI,IAAI,GAAG,CAAC;QAChB,CAAC;aAAM,IAAI,SAAS,CAAC,CAAC,CAAC,QAAQ,qCAA6B,EAAE,CAAC;YAC3D,IAAI,IAAI,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACnB,IAAI,IAAI,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC;QACrC,CAAC;aAAM,IAAI,SAAS,CAAC,CAAC,CAAC,QAAQ,iCAAyB,EAAE,CAAC;YACvD,IAAI,IAAI,GAAG,CAAC;QAChB,CAAC;QAED,MAAM,mBAAmB,GAAG,cAAc,CAAC,yBAAyB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAC/F,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,mBAAmB,EAAE,CAAC;YACtB,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,mBAAmB,EAAE,2BAA2B,CAAC,IAAI,CAAC,CAAC;QAC7F,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACZ,IAAI,IAAI,IAAI,GAAG,SAAS,CAAC;QAC7B,CAAC;QAED,IAAI,SAAS,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;YAC3B,oDAAoD;YACpD,6CAA6C;YAC7C,IAAI,SAAS,EAAE,CAAC;gBACZ,IAAI,IAAI,QAAQ,CAAC;YACrB,CAAC;iBAAM,CAAC;gBACJ,IAAI,IAAI,MAAM,CAAC;YACnB,CAAC;QACL,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,gBAAgB,CAAC,IAAoB,EAAE,MAAM,GAAG,KAAK,EAAE,qBAAqB,GAAG,KAAK;QACxF,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,CAC7C,IAAI,CAAC,wBAAwB,EAC7B,qBAAqB,EACrB,IAAI,CAAC,SAAS,CACjB,CAAC;QACF,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAEjC,IAAI,eAAe,GAAG,MAAM;YACxB,CAAC;YACD,CAAC,iDAAyC,CAAC;QAC/C,eAAe,sEAA8D,CAAC;QAE9E,OAAO,cAAc,CAAC,eAAe,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IACjE,CAAC;IAEO,oBAAoB;QACxB,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,gCAAgC;QAChC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAClC,IAAI,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBACjE,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC;YAC1B,CAAC;YAED,IAAI,GAAG,CAAC,UAAU,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5C,SAAS,IAAI,UAAU,GAAG,CAAC,UAAU,EAAE,CAAC;gBACxC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACZ,SAAS,IAAI,OAAO,GAAG,CAAC,KAAK,EAAE,CAAC;gBACpC,CAAC;gBACD,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC;gBAC3B,WAAW,GAAG,IAAI,CAAC;YACvB,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,qCAAqC;QACrC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACpC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBACtB,IAAI,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;oBACvD,CAAC,CAAC,UAAU,GAAG,IAAI,CAAC;gBACxB,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBACvB,SAAS,IAAI,QAAQ,GAAG,CAAC,UAAU,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC;gBAC/D,WAAW,GAAG,IAAI,CAAC;YACvB,CAAC;YAED,MAAM,aAAa,GAAG,GAAG,CAAC,OAAO;iBAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,kBAAkB,CAAC;iBACtD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACX,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;oBAClB,OAAO,CAAC,CAAC,CAAC;gBACd,CAAC;qBAAM,IAAI,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;oBACzB,OAAO,CAAC,CAAC;gBACb,CAAC;gBACD,OAAO,CAAC,CAAC;YACb,CAAC,CAAC,CAAC;YAEP,2DAA2D;YAC3D,8DAA8D;YAC9D,wBAAwB;YACxB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,UAAU,KAAK,YAAY,EAAE,CAAC;gBAC9D,SAAS,IAAI,QAAQ,GAAG,CAAC,UAAU,UAAU,CAAC;gBAE9C,SAAS,IAAI,aAAa;qBACrB,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;oBACZ,IAAI,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC;oBACzB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;wBACf,MAAM,IAAI,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;oBACpC,CAAC;oBACD,OAAO,MAAM,CAAC;gBAClB,CAAC,CAAC;qBACD,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEhB,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC;gBAC3B,WAAW,GAAG,IAAI,CAAC;YACvB,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,WAAW,EAAE,CAAC;YACd,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC;QAC/B,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,gBAAgB;QACpB,IAAI,SAAS,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7C,SAAS,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACzC,SAAS,IAAI,IAAI,CAAC,aAAa,CAAC;QAChC,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ"}

package/dist/analyzer/typeUtils.d.ts

@@ -193,6 +193,7 @@ export declare function computeMroLinearization(classType: ClassType): boolean;
 export declare function getDeclaringModulesForType(type: Type): string[];
 export declare function convertTypeToParamSpecValue(type: Type): FunctionType;
 export declare function simplifyFunctionToParamSpec(type: FunctionType): FunctionType | ParamSpecType;
+export declare function transformTypePair(sourceType: Type, targetType: Type, transformNode: (sourceNode: Type, targetNode: Type) => Type | undefined): Type;
 export declare class TypeVarTransformer {
     private _pendingTypeVarTransformations;
     private _pendingFunctionTransformations;