Adapt the compiler to generate non-nullable types.

sjrd · sjrd · commit 972effed1b88 · 2024-08-22T22:33:11.000+02:00
For `This` nodes and `IsInstanceOf` nodes.

We only active the relevant deserialization hacks when reading IR
from before 1.17.
diff --git a/compiler/src/main/scala/org/scalajs/nscplugin/GenJSCode.scala b/compiler/src/main/scala/org/scalajs/nscplugin/GenJSCode.scala
@@ -152,12 +152,19 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
     private val fieldsMutatedInCurrentClass = new ScopedVar[mutable.Set[Name]]
     private val generatedSAMWrapperCount = new ScopedVar[VarBox[Int]]
 
+    def currentThisTypeNullable: jstpe.Type =
+      encodeClassType(currentClassSym)
+
     def currentThisType: jstpe.Type = {
-      encodeClassType(currentClassSym) match {
+      currentThisTypeNullable match {
         case tpe @ jstpe.ClassType(cls, _) =>
-          jstpe.BoxedClassToPrimType.getOrElse(cls, tpe)
-        case tpe =>
+          jstpe.BoxedClassToPrimType.getOrElse(cls, tpe.toNonNullable)
+        case tpe @ jstpe.AnyType =>
+          // We are in a JS class, in which even `this` is nullable
           tpe
+        case tpe =>
+          throw new AssertionError(
+              s"Unexpected IR this type $tpe for class ${currentClassSym.get}")
       }
     }
 
@@ -2125,7 +2132,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
             mutableLocalVars += thisSym
 
           val thisLocalIdent = encodeLocalSym(thisSym)
-          val thisLocalType = currentThisType
+          val thisLocalType = currentThisTypeNullable
 
           val genRhs = {
             /* #3267 In default methods, scalac will type its _$this
@@ -2222,7 +2229,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
      *  @param tree
      *    The tree to adapt.
      *  @param tpe
-     *    The target type, which must be either `AnyType` or `ClassType(_)`.
+     *    The target type, which must be either `AnyType` or `ClassType`.
      */
     private def forceAdapt(tree: js.Tree, tpe: jstpe.Type): js.Tree = {
       if (tree.tpe == tpe || tpe == jstpe.AnyType) {
@@ -2670,7 +2677,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
         js.This()(currentThisType)
       } { thisLocalIdent =>
         // .copy() to get the correct position
-        js.VarRef(thisLocalIdent.copy())(currentThisType)
+        js.VarRef(thisLocalIdent.copy())(currentThisTypeNullable)
       }
     }
 
@@ -3333,7 +3340,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
           val isTailJumpThisLocalVar = formalArgSym.name == nme.THIS
 
           val tpe =
-            if (isTailJumpThisLocalVar) currentThisType
+            if (isTailJumpThisLocalVar) currentThisTypeNullable
             else toIRType(formalArgSym.tpe)
 
           val fixedActualArg =
@@ -3561,7 +3568,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
         // The Scala type system prevents x.isInstanceOf[Null] and ...[Nothing]
         assert(sym != NullClass && sym != NothingClass,
             s"Found a .isInstanceOf[$sym] at $pos")
-        js.IsInstanceOf(value, toIRType(to))
+        js.IsInstanceOf(value, toIRType(to).toNonNullable)
       }
     }
 
@@ -6334,7 +6341,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
       }
       val className = encodeClassName(currentClassSym).withSuffix(suffix)
 
-      val classType = jstpe.ClassType(className)
+      val thisType = jstpe.ClassType(className, nullable = false)
 
       // val f: Any
       val fFieldIdent = js.FieldIdent(FieldName(className, SimpleFieldName("f")))
@@ -6353,10 +6360,10 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
             jstpe.NoType,
             Some(js.Block(List(
                 js.Assign(
-                    js.Select(js.This()(classType), fFieldIdent)(jstpe.AnyType),
+                    js.Select(js.This()(thisType), fFieldIdent)(jstpe.AnyType),
                     fParamDef.ref),
                 js.ApplyStatically(js.ApplyFlags.empty.withConstructor(true),
-                    js.This()(classType),
+                    js.This()(thisType),
                     ir.Names.ObjectClass,
                     js.MethodIdent(ir.Names.NoArgConstructorName),
                     Nil)(jstpe.NoType)))))(
@@ -6404,7 +6411,7 @@ abstract class GenJSCode[G <: Global with Singleton](val global: G)
         }.map((ensureBoxed _).tupled)
 
         val call = js.JSFunctionApply(
-            js.Select(js.This()(classType), fFieldIdent)(jstpe.AnyType),
+            js.Select(js.This()(thisType), fFieldIdent)(jstpe.AnyType),
             actualParams)
 
         val body = fromAny(call, enteringPhase(currentRun.posterasurePhase) {
diff --git a/compiler/src/main/scala/org/scalajs/nscplugin/GenJSExports.scala b/compiler/src/main/scala/org/scalajs/nscplugin/GenJSExports.scala
@@ -930,7 +930,7 @@ trait GenJSExports[G <: Global with Singleton] extends SubComponent {
 
   private sealed abstract class RTTypeTest
 
-  private case class PrimitiveTypeTest(tpe: jstpe.Type, rank: Int)
+  private case class PrimitiveTypeTest(tpe: jstpe.PrimType, rank: Int)
       extends RTTypeTest
 
   // scalastyle:off equals.hash.code
diff --git a/ir/shared/src/main/scala/org/scalajs/ir/Serializers.scala b/ir/shared/src/main/scala/org/scalajs/ir/Serializers.scala
@@ -1234,7 +1234,7 @@ object Serializers {
         case TagIsInstanceOf =>
           val expr = readTree()
           val testType0 = readType()
-          val testType = if (true /* hacks.use16 */) { // scalastyle:ignore
+          val testType = if (hacks.use16) {
             testType0 match {
               case ClassType(className, true)    => ClassType(className, nullable = false)
               case ArrayType(arrayTypeRef, true) => ArrayType(arrayTypeRef, nullable = false)
@@ -1298,7 +1298,7 @@ object Serializers {
 
         case TagThis =>
           val tpe = readType()
-          if (true /* hacks.use16 */) // scalastyle:ignore
+          if (hacks.use16)
             This()(thisTypeForHack)
           else
             This()(tpe)
@@ -1307,7 +1307,7 @@ object Serializers {
           val arrow = readBoolean()
           val captureParams = readParamDefs()
           val (params, restParam) = readParamDefsWithRest()
-          val body = if (false /* !hacks.use16 */) { // scalastyle:ignore
+          val body = if (!hacks.use16) {
             readTree()
           } else {
             val prevThisTypeForHack = thisTypeForHack
@@ -1416,7 +1416,7 @@ object Serializers {
       val originalName = readOriginalName()
       val kind = ClassKind.fromByte(readByte())
 
-      if (true /* hacks.use16 */) { // scalastyle:ignore
+      if (hacks.use16) {
         thisTypeForHack = {
           if (kind.isJSType)
             AnyType
diff --git a/project/JavaLangObject.scala b/project/JavaLangObject.scala
@@ -26,7 +26,7 @@ object JavaLangObject {
     implicit val DummyPos = NoPosition
 
     // ClassType(Object) is normally invalid, but not in this class def
-    val ThisType = ClassType(ObjectClass)
+    val ThisType = ClassType(ObjectClass, nullable = false)
 
     val ObjectClassRef = ClassRef(ObjectClass)
     val ClassClassRef = ClassRef(ClassClass)
@@ -101,7 +101,7 @@ object JavaLangObject {
           Nil,
           AnyType,
           Some {
-            If(IsInstanceOf(This()(ThisType), ClassType(CloneableClass)), {
+            If(IsInstanceOf(This()(ThisType), ClassType(CloneableClass, nullable = false)), {
               Clone(AsInstanceOf(This()(ThisType), ClassType(CloneableClass)))
             }, {
               Throw(New(ClassName("java.lang.CloneNotSupportedException"),
