@zzzen/pyright-internal 1.2.0-dev.20240825 → 1.2.0-dev.20240908

package/dist/analyzer/typeUtils.js

@@ -36,7 +36,6 @@ exports.isTypeAliasRecursive = isTypeAliasRecursive;
 exports.transformPossibleRecursiveTypeAlias = transformPossibleRecursiveTypeAlias;
 exports.getTypeVarScopeId = getTypeVarScopeId;
 exports.getTypeVarScopeIds = getTypeVarScopeIds;
-exports.specializeWithDefaultTypeArgs = specializeWithDefaultTypeArgs;
 exports.specializeWithUnknownTypeArgs = specializeWithUnknownTypeArgs;
 exports.getUnknownForTypeVar = getUnknownForTypeVar;
 exports.getUnknownForTypeVarTuple = getUnknownForTypeVarTuple;
@@ -45,8 +44,11 @@ exports.selfSpecializeClass = selfSpecializeClass;
 exports.getSpecializedTupleType = getSpecializedTupleType;
 exports.isLiteralType = isLiteralType;
 exports.isLiteralTypeOrUnion = isLiteralTypeOrUnion;
+exports.isLiteralLikeType = isLiteralLikeType;
 exports.containsLiteralType = containsLiteralType;
 exports.getLiteralTypeClassName = getLiteralTypeClassName;
+exports.stripTypeForm = stripTypeForm;
+exports.stripTypeFormRecursive = stripTypeFormRecursive;
 exports.getUnionSubtypeCount = getUnionSubtypeCount;
 exports.isEllipsisType = isEllipsisType;
 exports.isProperty = isProperty;
@@ -72,11 +74,12 @@ exports.getContainerDepth = getContainerDepth;
 exports.lookUpObjectMember = lookUpObjectMember;
 exports.lookUpClassMember = lookUpClassMember;
 exports.getClassMemberIterator = getClassMemberIterator;
+exports.isMemberReadOnly = isMemberReadOnly;
 exports.getClassIterator = getClassIterator;
 exports.getClassFieldsRecursive = getClassFieldsRecursive;
 exports.addTypeVarsToListIfUnique = addTypeVarsToListIfUnique;
 exports.getTypeVarArgsRecursive = getTypeVarArgsRecursive;
-exports.specializeClassType = specializeClassType;
+exports.specializeWithDefaultTypeArgs = specializeWithDefaultTypeArgs;
 exports.buildSolutionFromSpecializedClass = buildSolutionFromSpecializedClass;
 exports.buildSolution = buildSolution;
 exports.specializeForBaseClass = specializeForBaseClass;
@@ -689,6 +692,11 @@ function preserveUnknown(type1, type2) {
 // Determines whether the specified type is a type that can be
 // combined with other types for a union.
 function isUnionableType(subtypes) {
+    // If all of the subtypes are TypeForm types, we know that they
+    // are unionable.
+    if (subtypes.every((t) => { var _a; return ((_a = t.props) === null || _a === void 0 ? void 0 : _a.typeForm) !== undefined; })) {
+        return true;
+    }
     let typeFlags = 2 /* TypeFlags.Instance */ | 1 /* TypeFlags.Instantiable */;
     for (const subtype of subtypes) {
         typeFlags &= subtype.flags;
@@ -871,17 +879,6 @@ function getTypeVarScopeIds(type) {
     }
     return scopeIds;
 }
-// If the class type is generic and does not already have type arguments
-// specified, specialize it with default type arguments (Unknown or the
-// default type if provided).
-function specializeWithDefaultTypeArgs(type) {
-    if (type.shared.typeParams.length === 0 || type.priv.typeArgs) {
-        return type;
-    }
-    return types_1.ClassType.specialize(type, type.shared.typeParams.map((param) => param.shared.defaultType), 
-    /* isTypeArgExplicit */ false, 
-    /* includeSubclasses */ type.priv.includeSubclasses);
-}
 // Specializes the class with "Unknown" type args (or the equivalent for ParamSpecs
 // or TypeVarTuples).
 function specializeWithUnknownTypeArgs(type, tupleClassType) {
@@ -982,6 +979,15 @@ function isLiteralTypeOrUnion(type, allowNone = false) {
     }
     return false;
 }
+function isLiteralLikeType(type) {
+    if (type.priv.literalValue !== undefined) {
+        return true;
+    }
+    if (types_1.ClassType.isBuiltIn(type, 'LiteralString')) {
+        return true;
+    }
+    return false;
+}
 function containsLiteralType(type, includeTypeArgs = false) {
     class ContainsLiteralTypeWalker extends typeWalker_1.TypeWalker {
         constructor(_includeTypeArgs) {
@@ -991,7 +997,7 @@ function containsLiteralType(type, includeTypeArgs = false) {
         }
         visitClass(classType) {
             if ((0, types_1.isClassInstance)(classType)) {
-                if (isLiteralType(classType) || types_1.ClassType.isBuiltIn(classType, 'LiteralString')) {
+                if (isLiteralLikeType(classType)) {
                     this.foundLiteral = true;
                     this.cancelWalk();
                 }
@@ -1032,6 +1038,24 @@ function getLiteralTypeClassName(type) {
     }
     return undefined;
 }
+function stripTypeForm(type) {
+    var _a;
+    if ((_a = type.props) === null || _a === void 0 ? void 0 : _a.typeForm) {
+        return types_1.TypeBase.cloneWithTypeForm(type, undefined);
+    }
+    return type;
+}
+function stripTypeFormRecursive(type, recursionCount = 0) {
+    var _a;
+    if (recursionCount > types_1.maxTypeRecursionCount) {
+        return type;
+    }
+    recursionCount++;
+    if ((_a = type.props) === null || _a === void 0 ? void 0 : _a.typeForm) {
+        type = types_1.TypeBase.cloneWithTypeForm(type, undefined);
+    }
+    return mapSubtypes(type, (subtype) => stripTypeFormRecursive(subtype, recursionCount));
+}
 function getUnionSubtypeCount(type) {
     if ((0, types_1.isUnion)(type)) {
         return type.priv.subtypes.length;
@@ -1250,6 +1274,7 @@ function getProtocolSymbolsRecursive(classType, symbolMap, classFlags = 512 /* C
                 isInstanceMember: symbol.isInstanceMember(),
                 isClassMember: symbol.isClassMember(),
                 isClassVar: (0, symbolUtils_1.isEffectivelyClassVar)(symbol, /* isDataclass */ false),
+                isReadOnly: false,
                 isTypeDeclared: symbol.hasTypedDeclarations(),
                 skippedUndeclaredType: false,
             });
@@ -1360,6 +1385,7 @@ function* getClassMemberIterator(classType, memberName, flags = 0 /* MemberAcces
                         isClassVar: false,
                         classType,
                         unspecializedClassType: classType,
+                        isReadOnly: false,
                         isTypeDeclared: false,
                         skippedUndeclaredType: false,
                     };
@@ -1384,6 +1410,7 @@ function* getClassMemberIterator(classType, memberName, flags = 0 /* MemberAcces
                             isClassVar: (0, symbolUtils_1.isEffectivelyClassVar)(symbol, types_1.ClassType.isDataClass(specializedMroClass)),
                             classType: specializedMroClass,
                             unspecializedClassType: mroClass,
+                            isReadOnly: isMemberReadOnly(specializedMroClass, memberName),
                             isTypeDeclared: hasDeclaredType,
                             skippedUndeclaredType,
                         };
@@ -1423,6 +1450,7 @@ function* getClassMemberIterator(classType, memberName, flags = 0 /* MemberAcces
                             isClassVar: (0, symbolUtils_1.isEffectivelyClassVar)(symbol, isDataclass),
                             classType: specializedMroClass,
                             unspecializedClassType: mroClass,
+                            isReadOnly: false,
                             isTypeDeclared: hasDeclaredType,
                             skippedUndeclaredType,
                         };
@@ -1445,6 +1473,7 @@ function* getClassMemberIterator(classType, memberName, flags = 0 /* MemberAcces
             isClassVar: false,
             classType,
             unspecializedClassType: classType,
+            isReadOnly: false,
             isTypeDeclared: false,
             skippedUndeclaredType: false,
         };
@@ -1452,6 +1481,21 @@ function* getClassMemberIterator(classType, memberName, flags = 0 /* MemberAcces
     }
     return undefined;
 }
+// Checks for whether the member is effectively read only because it
+// belongs to a frozen dataclass or a named tuple.
+function isMemberReadOnly(classType, name) {
+    var _a;
+    if (types_1.ClassType.hasNamedTupleEntry(classType, name)) {
+        return true;
+    }
+    if (types_1.ClassType.isDataClassFrozen(classType)) {
+        const dcEntries = (_a = classType.shared) === null || _a === void 0 ? void 0 : _a.dataClassEntries;
+        if (dcEntries === null || dcEntries === void 0 ? void 0 : dcEntries.some((entry) => entry.name === name)) {
+            return true;
+        }
+    }
+    return false;
+}
 function* getClassIterator(classType, flags = 0 /* ClassIteratorFlags.Default */, skipMroClass) {
     if ((0, types_1.isClass)(classType)) {
         let foundSkipMroClass = skipMroClass === undefined;
@@ -1511,6 +1555,7 @@ function getClassFieldsRecursive(classType) {
                         isInstanceMember: symbol.isInstanceMember(),
                         isClassMember: symbol.isClassMember(),
                         isClassVar: (0, symbolUtils_1.isEffectivelyClassVar)(symbol, types_1.ClassType.isDataClass(specializedMroClass)),
+                        isReadOnly: isMemberReadOnly(specializedMroClass, name),
                         isTypeDeclared: true,
                         skippedUndeclaredType: false,
                     });
@@ -1598,8 +1643,11 @@ function getTypeVarArgsRecursive(type, recursionCount = 0) {
     return [];
 }
 // Creates a specialized version of the class, filling in any unspecified
-// type arguments with Unknown.
-function specializeClassType(type) {
+// type arguments with Unknown or default value.
+function specializeWithDefaultTypeArgs(type) {
+    if (type.shared.typeParams.length === 0 || type.priv.typeArgs) {
+        return type;
+    }
     const solution = new constraintSolution_1.ConstraintSolution();
     const typeParams = types_1.ClassType.getTypeParams(type);
     typeParams.forEach((typeParam) => {
@@ -1679,8 +1727,9 @@ function synthesizeTypeVarForSelfCls(classType, isClsParam) {
     const scopeId = (_a = getTypeVarScopeId(classType)) !== null && _a !== void 0 ? _a : '';
     selfType.shared.isSynthesized = true;
     selfType.shared.isSynthesizedSelf = true;
-    selfType.priv.nameWithScope = types_1.TypeVarType.makeNameWithScope(selfType.shared.name, scopeId);
     selfType.priv.scopeId = scopeId;
+    selfType.priv.scopeName = '';
+    selfType.priv.nameWithScope = types_1.TypeVarType.makeNameWithScope(selfType.shared.name, scopeId, selfType.priv.scopeName);
     const boundType = types_1.ClassType.specialize(classType, 
     /* typeArgs */ undefined, 
     /* isTypeArgExplicit */ false, 
@@ -3283,6 +3332,11 @@ class ApplySolvedTypeVarsTransformer extends TypeVarTransformer {
         if (solutionSets.length <= 1) {
             return callback();
         }
+        // Handle the case where we're already processing one of the signature contexts
+        // and are called recursively. Don't loop over all the signature contexts again.
+        if (this._activeConstraintSetIndex !== undefined) {
+            return callback();
+        }
         // Loop through all of the signature contexts in the type var context
         // to create an overload type.
         const overloadTypes = solutionSets.map((_, index) => {