• Home
• com.softwaremill.macwire
• macrosautocats_sjs1_2.13
• 2.6.2
• source code
• MacwireAutoCatsMacros.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.softwaremill.macwire.autocats.MacwireAutoCatsMacros.scala Maven / Gradle / Ivy

package com.softwaremill.macwire.autocats

import scala.reflect.macros.blackbox
import cats.effect.{IO, Resource => CatsResource}
import com.softwaremill.macwire.internals._
import com.softwaremill.macwire.autocats.internals._

object MacwireAutoCatsMacros {
  private val log = new Logger()

  def autowire_impl[T: c.WeakTypeTag](
      c: blackbox.Context
  )(dependencies: c.Expr[Any]*): c.Expr[CatsResource[IO, T]] = {
    import c.universe._

    val targetType = implicitly[c.WeakTypeTag[T]]

    val graphContext = new CatsProvidersGraphContext[c.type](c, log)
    val graph = graphContext.buildGraph(dependencies.toList, targetType.tpe)
    val sortedProviders = graph.topologicalSort()

    log(s"Sorted providers [${sortedProviders.mkString(", ")}]")

    val code = sortedProviders
      .map {
        case fm: graphContext.FactoryMethod => fm.result
        case p                              => p
      }
      .collect { case p @ (_: graphContext.Effect | _: graphContext.Resource) => p }
      .foldRight(
        q"cats.effect.Resource.pure[cats.effect.IO, $targetType](${graph.root.ident})"
      ) { case (resource, acc) =>
        q"${resource.value}.flatMap((${resource.ident}: ${resource.resultType}) => $acc)"
      }

    log(s"CODE: [$code]")

    c.Expr[CatsResource[IO, T]](code)
  }

}