#9 Add Caller

sourcecode/shared/src/main/scala/sourcecode/SourceContext.scala

@@ -1,6 +1,7 @@
 package sourcecode
 
 import language.experimental.macros
+import scala.reflect.macros.whitebox
 
 
 object Util{
@@ -105,6 +106,15 @@ object Pkg extends SourceCompanion[String, Pkg](new Pkg(_)){
   def impl(c: Compat.Context): c.Expr[Pkg] = Impls.enclosing[Pkg](c)(_.isPackage)
 }
 
+case class Caller[A](value: A)
+object Caller {
+  implicit def generate: Caller[_] = macro impl
+  def impl(c: whitebox.Context): c.Tree = {
+    import c.universe._
+    q"new _root_.sourcecode.Caller(this)"
+  }
+}
+
 case class Text[T](value: T, source: String)
 object Text{
   implicit def generate[T](v: T): Text[T] = macro Impls.text[T]

sourcecode/shared/src/test/scala/sourcecode/CallerTest.scala

@@ -0,0 +1,16 @@
+package sourcecode
+
+/**
+  * @author 张志豪 (izhangzhihao) <[email protected]>
+  */
+object CallerTest {
+  def run(): Unit = {
+    assert(Foo.call == CallerTest.getClass.getName)
+  }
+}
+
+object Foo {
+  def call(implicit caller: Caller[_]): String = {
+    caller.value.getClass.getName
+  }
+}

sourcecode/shared/src/test/scala/sourcecode/Tests.scala

@@ -21,6 +21,7 @@ object Tests{
     ManualImplicit()
     TextTests()
     ArgsTests()
+    CallerTest.run()
 
     println("================LogExample================")
     logExample()