Add version requirements for declarations with !! types

^KT-26245 Related

compiler/serialization/src/org/jetbrains/kotlin/serialization/DescriptorSerializer.kt

@@ -50,6 +50,13 @@ class DescriptorSerializer private constructor(
 ) {
     private val contractSerializer = ContractSerializer()
 
+    private var metDefinitelyNotNullTypeStack: MutableList<Boolean> = mutableListOf(false)
+    private var metDefinitelyNotNullType: Boolean
+        get() = metDefinitelyNotNullTypeStack.last()
+        set(value) {
+            metDefinitelyNotNullTypeStack[metDefinitelyNotNullTypeStack.lastIndex] = value
+        }
+
     private fun createChildSerializer(descriptor: DeclarationDescriptor): DescriptorSerializer =
         DescriptorSerializer(
             descriptor, Interner(typeParameters), extension, typeTable, versionRequirementTable,
@@ -61,7 +68,16 @@ class DescriptorSerializer private constructor(
 
     private fun useTypeTable(): Boolean = extension.shouldUseTypeTable()
 
-    fun classProto(classDescriptor: ClassDescriptor): ProtoBuf.Class.Builder {
+    private inline fun <T> withNewMetDefinitelyNotNullType(block: () -> T): T {
+        metDefinitelyNotNullTypeStack.add(false)
+        return try {
+            block()
+        } finally {
+            metDefinitelyNotNullTypeStack.removeAt(metDefinitelyNotNullTypeStack.lastIndex)
+        }
+    }
+
+    fun classProto(classDescriptor: ClassDescriptor): ProtoBuf.Class.Builder = withNewMetDefinitelyNotNullType {
         val builder = ProtoBuf.Class.newBuilder()
 
         val flags = Flags.getClassFlags(
@@ -174,10 +190,17 @@ class DescriptorSerializer private constructor(
 
         writeVersionRequirementForInlineClasses(classDescriptor, builder, versionRequirementTable)
 
+        if (metDefinitelyNotNullType) {
+            builder.addVersionRequirement(
+                writeLanguageVersionRequirement(LanguageFeature.DefinitelyNotNullTypeParameters, versionRequirementTable)
+            )
+        }
+
         val versionRequirementTableProto = versionRequirementTable.serialize()
         if (versionRequirementTableProto != null) {
             builder.versionRequirementTable = versionRequirementTableProto
         }
+
         return builder
     }
 
@@ -294,6 +317,10 @@ class DescriptorSerializer private constructor(
             if (descriptor.hasInlineClassTypesInSignature()) {
                 builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.InlineClasses))
             }
+
+            if (local.metDefinitelyNotNullType) {
+                builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.DefinitelyNotNullTypeParameters))
+            }
         }
 
         extension.serializeProperty(descriptor, builder, versionRequirementTable, local)
@@ -377,6 +404,10 @@ class DescriptorSerializer private constructor(
             if (descriptor.hasInlineClassTypesInSignature()) {
                 builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.InlineClasses))
             }
+
+            if (local.metDefinitelyNotNullType) {
+                builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.DefinitelyNotNullTypeParameters))
+            }
         }
 
         contractSerializer.serializeContractOfFunctionIfAny(descriptor, builder, this)
@@ -413,6 +444,10 @@ class DescriptorSerializer private constructor(
             if (descriptor.hasInlineClassTypesInSignature()) {
                 builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.InlineClasses))
             }
+
+            if (local.metDefinitelyNotNullType) {
+                builder.addVersionRequirement(writeVersionRequirement(LanguageFeature.DefinitelyNotNullTypeParameters))
+            }
         }
 
         extension.serializeConstructor(descriptor, builder, local)
@@ -442,7 +477,7 @@ class DescriptorSerializer private constructor(
         )
     }
 
-    private fun typeAliasProto(descriptor: TypeAliasDescriptor): ProtoBuf.TypeAlias.Builder? {
+    private fun typeAliasProto(descriptor: TypeAliasDescriptor): ProtoBuf.TypeAlias.Builder? = withNewMetDefinitelyNotNullType {
         if (!extension.shouldSerializeTypeAlias(descriptor)) return null
 
         val builder = ProtoBuf.TypeAlias.newBuilder()
@@ -475,6 +510,11 @@ class DescriptorSerializer private constructor(
 
         versionRequirementTable?.run {
             builder.addAllVersionRequirement(serializeVersionRequirements(descriptor))
+            if (local.metDefinitelyNotNullType) {
+                builder.addVersionRequirement(
+                    writeLanguageVersionRequirement(LanguageFeature.DefinitelyNotNullTypeParameters, versionRequirementTable)
+                )
+            }
         }
 
         for (annotation in descriptor.nonSourceAnnotations) {
@@ -600,7 +640,10 @@ class DescriptorSerializer private constructor(
                     builder.typeParameter = getTypeParameterId(descriptor)
                 }
 
-                builder.flags = Flags.getTypeFlags(false, type is DefinitelyNotNullType)
+                if (type.unwrap() is DefinitelyNotNullType) {
+                    metDefinitelyNotNullType = true
+                    builder.flags = Flags.getTypeFlags(false, true)
+                }
 
                 assert(type.arguments.isEmpty()) { "Found arguments for type constructor build on type parameter: $descriptor" }
             }