wvlet.airframe.surface.reflect.TastySurfaceFactory.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of airframe-surface_3 Show documentation

A library for extracting object structure surface

There is a newer version: 24.9.0

package wvlet.airframe.surface.reflect

import wvlet.log.LogSupport
import wvlet.airframe.surface.{MethodSurface, Surface, Primitive}
import wvlet.airframe.surface.CompileTimeSurfaceFactory

import java.util.concurrent.ConcurrentHashMap
import scala.quoted.*
import scala.tasty.inspector.*

object TastySurfaceFactory extends LogSupport {

  given staging.Compiler = staging.Compiler.make(getClass.getClassLoader)

  inline def of[A]: Surface = ${ CompileTimeSurfaceFactory.surfaceOf[A] }

  import scala.jdk.CollectionConverters.*
  private val cache = new ConcurrentHashMap[Class[_], Surface]().asScala

  def ofClass(cl: Class[?]): Surface = {
    debug(s"ofClass: ${cl}")
    cache.getOrElseUpdate(
      cl, {
        debug(s"Update cache for ${cl}")
        // Generates Surface from a runtime class
        staging.run { (quotes: Quotes) ?=>
          import quotes.reflect.*
          val tastyType = quotes.reflect.TypeRepr.typeConstructorOf(cl)
          debug(tastyType)
          val f = new CompileTimeSurfaceFactory(using quotes)
          tastyType match {
            case t if t.show.endsWith(".") =>
              // Example use case is MessageCodec.of[Any] or case objects
              f.surfaceFromClass(cl)
            case _ =>
              f.surfaceOf(tastyType.asType)
          }
        }
      }
    )
  }

  def methodsOfClass(cl: Class[?]): Seq[MethodSurface] = {
    // Generates Surface from a runtime class
    val code: Seq[MethodSurface] = staging.run { (quotes: Quotes) ?=>
      import quotes.reflect.*
      val tastyType = quotes.reflect.TypeRepr.typeConstructorOf(cl)
      debug(tastyType)
      val f = new CompileTimeSurfaceFactory(using quotes)
      f.methodsOf(tastyType.asType)
    }
    code
  }

}