Fix #2539: Error when adding refined terms to SAMs

Author: nicolasstucki

compiler/src/dotty/tools/dotc/transform/ExpandSAMs.scala

@@ -8,6 +8,7 @@ import TreeTransforms._
 import SymUtils._
 import ast.untpd
 import ast.Trees._
+import dotty.tools.dotc.util.Positions.Position
 
 /** Expand SAM closures that cannot be represented by the JVM as lambdas to anonymous classes.
  *  These fall into five categories
@@ -36,8 +37,10 @@ class ExpandSAMs extends MiniPhaseTransform { thisTransformer =>
         case tpe @ SAMType(_) if tpe.isRef(defn.PartialFunctionClass) =>
           toPartialFunction(tree)
         case tpe @ SAMType(_) if isPlatformSam(tpe.classSymbol.asClass) =>
+          checkRefinements(tpe, fn.pos)
           tree
         case tpe =>
+          checkRefinements(tpe, fn.pos)
           val Seq(samDenot) = tpe.abstractTermMembers.filter(!_.symbol.isSuperAccessor)
           cpy.Block(tree)(stats,
               AnonClass(tpe :: Nil, fn.symbol.asTerm :: Nil, samDenot.symbol.asTerm.name :: Nil))
@@ -83,4 +86,13 @@ class ExpandSAMs extends MiniPhaseTransform { thisTransformer =>
     val anonCls = AnonClass(tpt.tpe :: Nil, List(applyFn, isDefinedAtFn), List(nme.apply, nme.isDefinedAt))
     cpy.Block(tree)(List(applyDef, isDefinedAtDef), anonCls)
   }
+
+  private def checkRefinements(tpe: Type, pos: Position)(implicit ctx: Context): Unit = tpe match {
+    case RefinedType(parent, name, _) =>
+      if (name.isTermName && tpe.member(name).symbol.ownersIterator.isEmpty) // if member defined in the refinement
+        ctx.error(s"Cannot refine $name on a lambda", pos)
+      checkRefinements(parent, pos)
+    case _ =>
+  }
+
 }

tests/neg/i2539.scala

@@ -0,0 +1,15 @@
+object Foo {
+  val f0: Function0[Int]{ def foo(): Int } = () => 42 // error
+
+  val f1: Function1[Int, Int]{ def foo(): Int } = x1 => 42 // error
+
+  val f26: Function26[Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int, Int]{ def foo(): Int } =
+    (x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, x12, x13, x14, x15, x16, x17, x18, x19, x20, x21, x22, x23, x24, x25, x26) => 42 // error
+
+  val if1: ImplicitFunction1[Int, Int]{ def foo(): Int } = implicit x1 => 42 // error
+
+  abstract class Fun0[X] extends Function0[X]
+  val fun0a: Fun0[Int] = () => 42
+  val fun0b: Fun0[Int] { def foo(): Int } = () => 42 // error
+
+}