An Unapplied extractor.

This makes it a lot less error prone and redundant to find the
part you need when unwrapping an UnApply tree.

1 files changed, 11 insertions, 0 deletions

diff --git a/src/reflect/scala/reflect/internal/TreeInfo.scala b/src/reflect/scala/reflect/internal/TreeInfo.scala
index d01f1ce681..34fe0afb1a 100644
--- a/src/reflect/scala/reflect/internal/TreeInfo.scala
+++ b/src/reflect/scala/reflect/internal/TreeInfo.scala
@@ -773,6 +773,17 @@ abstract class TreeInfo {
       unapply(dissectApplied(tree))
   }
 
+  /** Locates the synthetic Apply node corresponding to an extractor's call to
+   *  unapply (unwrapping nested Applies) and returns the fun part of that Apply.
+   */
+  object Unapplied {
+    def unapply(tree: Tree): Option[Tree] = tree match {
+      case Apply(fun, Ident(nme.SELECTOR_DUMMY) :: Nil) => Some(fun)
+      case Apply(fun, _)                                => unapply(fun)
+      case _                                            => None
+    }
+  }
+
   /** Is this file the body of a compilation unit which should not
    *  have Predef imported? This is the case iff the first import in the
    *  unit explicitly refers to Predef.