Merge pull request #4159 from dotty-staging/fix-4157

Fix #4157: Broken error message when trying to eta-expand

Author: smarter

compiler/src/dotty/tools/dotc/reporting/diagnostic/messages.scala

@@ -1837,10 +1837,10 @@ object messages {
     }
   }
 
-  case class OnlyFunctionsCanBeFollowedByUnderscore(pt: Type)(implicit ctx: Context)
+  case class OnlyFunctionsCanBeFollowedByUnderscore(tp: Type)(implicit ctx: Context)
     extends Message(OnlyFunctionsCanBeFollowedByUnderscoreID) {
     val kind = "Syntax"
-    val msg = hl"Not a function: $pt: cannot be followed by ${"_"}"
+    val msg = hl"Only function types can be followed by ${"_"} but the current expression has type $tp"
     val explanation =
       hl"""The syntax ${"x _"} is no longer supported if ${"x"} is not a function.
           |To convert to a function value, you need to explicitly write ${"() => x"}"""

compiler/src/dotty/tools/dotc/typer/Typer.scala

@@ -1661,9 +1661,10 @@ class Typer extends Namer
     val nestedCtx = ctx.fresh.setNewTyperState()
     val res = typed(qual, pt1)(nestedCtx)
     res match {
-      case res @ closure(_, _, _) =>
+      case closure(_, _, _) =>
       case _ =>
-        ctx.errorOrMigrationWarning(OnlyFunctionsCanBeFollowedByUnderscore(res.tpe), tree.pos)
+        val recovered = typed(qual)(ctx.fresh.setExploreTyperState())
+        ctx.errorOrMigrationWarning(OnlyFunctionsCanBeFollowedByUnderscore(recovered.tpe.widen), tree.pos)
         if (ctx.scala2Mode) {
           // Under -rewrite, patch `x _` to `(() => x)`
           patch(Position(tree.pos.start), "(() => ")

compiler/test-resources/repl/errmsgs

@@ -72,3 +72,11 @@ scala> val a: iDontExist = 1
 1 | val a: iDontExist = 1
   |        ^^^^^^^^^^
   |        not found: type iDontExist
+scala> def foo1(x: => Int) = x _
+1 | def foo1(x: => Int) = x _
+  |                       ^^^
+  |Only function types can be followed by _ but the current expression has type Int
+scala> def foo2(x: => Int): () => Int = x _
+1 | def foo2(x: => Int): () => Int = x _
+  |                                  ^^^
+  |Only function types can be followed by _ but the current expression has type Int

compiler/test/dotty/tools/dotc/reporting/ErrorMessagesTests.scala

@@ -1050,8 +1050,8 @@ class ErrorMessagesTests extends ErrorMessagesTest {
       implicit val ctx: Context = ictx
 
       assertMessageCount(1, messages)
-      val OnlyFunctionsCanBeFollowedByUnderscore(pt) :: Nil = messages
-      assertEquals("String(n)", pt.show)
+      val OnlyFunctionsCanBeFollowedByUnderscore(tp) :: Nil = messages
+      assertEquals("String", tp.show)
     }
 
   @Test def missingEmptyArgumentList =