sttp.tapir.internal.SNameMacros.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tapir-core_sjs1_3 Show documentation

tapir-core

There is a newer version: 1.11.9

package sttp.tapir.internal

import scala.quoted.*

// TODO: make private[tapir] once Scala3 compilation is fixed
object SNameMacros {

  inline def typeFullName[T] = ${ typeFullNameImpl[T] }
  private def typeFullNameImpl[T: Type](using q: Quotes): Expr[String] = {
    import q.reflect.*
    val tpe = TypeRepr.of[T]
    Expr(typeFullNameFromTpe(tpe))
  }
  def typeFullNameFromTpe(using q: Quotes)(tpe: q.reflect.TypeRepr): String = {
    import q.reflect.*

    def normalizedName(s: Symbol): String = if s.flags.is(Flags.Module) then s.name.stripSuffix("$") else s.name
    def nameChain(sym: Symbol): List[String] =
      if sym.isNoSymbol then List.empty
      else if sym == defn.EmptyPackageClass then List.empty
      else if sym == defn.RootPackage then List.empty
      else if sym == defn.RootClass then List.empty
      else if sym.name.matches("<[^>]+>") then nameChain(sym.owner)
      else nameChain(sym.owner) :+ normalizedName(sym)

    nameChain(tpe.typeSymbol).mkString(".") match {
      case "scala.Predef.String" => "java.lang.String"
      case other                 => other
    }
  }

  def extractTypeArguments(using q: Quotes)(tpe: q.reflect.TypeRepr): List[String] = {
    import q.reflect.*

    def allTypeArguments(tr: TypeRepr): Seq[TypeRepr] = tr match {
      case at: AppliedType => at.args.flatMap(tn2 => tn2 +: allTypeArguments(tn2))
      case _               => List.empty[TypeRepr]
    }

    allTypeArguments(tpe).map(typeFullNameFromTpe).toList
  }

}