izumi.fundamentals.platform.language.SourcePackageMaterializer.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fundamentals-language_2.13 Show documentation

fundamentals-language

The newest version!

package izumi.fundamentals.platform.language

import scala.language.experimental.macros
import scala.reflect.macros.blackbox

final case class SourcePackageMaterializer(get: SourcePackage) extends AnyVal

object SourcePackageMaterializer {
  @inline def apply()(implicit ev: SourcePackageMaterializer, dummy: DummyImplicit): SourcePackageMaterializer = ev
  @inline def thisPkg(implicit pkg: SourcePackageMaterializer): String = pkg.get.pkg

  implicit def materialize: SourcePackageMaterializer = macro SourcePackageMaterializerMacro.getSourcePackage

  object SourcePackageMaterializerMacro {
    // copypaste from lihaoyi 'sourcecode' https://github.com/lihaoyi/sourcecode/blob/d32637de59d5ecdd5040d1ef06a9370bc46a310f/sourcecode/shared/src/main/scala/sourcecode/SourceContext.scala
    def getSourcePackage(c: blackbox.Context): c.Expr[SourcePackageMaterializer] = {
      import c.universe._
      val renderedPath: String = getSourcePackageString(c)
      val pathExpr = c.Expr[String](q"$renderedPath")

      reify {
        SourcePackageMaterializer(SourcePackage(pathExpr.splice))
      }
    }

    def getSourcePackageString(c: blackbox.Context): String = {
      var current = c.internal.enclosingOwner
      var path = List.empty[String]
      while (current != c.universe.NoSymbol && current.toString != "package ") {
        if (current.isPackage) {
          path = current.name.decodedName.toString.trim :: path
        }
        current = current.owner
      }
      val renderedPath = path.mkString(".")
      renderedPath
    }
  }
}