support run method without extension syntax

zio · Jun 2, 2023 · 848cc1a · 848cc1a
1 parent 4868054
commit 848cc1a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/zio-direct-test/src/test/scala/zio/direct/VariaSpec.scala b/zio-direct-test/src/test/scala/zio/direct/VariaSpec.scala
@@ -199,6 +199,14 @@ object VariaSpec extends DeferRunSpec { //
           lazy val x = 123
           """
         }
+      },
+      test("supports run method without extension syntax") {
+        @directRunCall
+        def await[T](v: Task[T]): T = ???
+        class Blah(val value: Int)
+        runLiftTest(1) {
+          await(ZIO.succeed(1))
+        }
       }
     )
   )

diff --git a/zio-direct/src/main/scala-3.x/zio/direct/core/metaprog/Extractor.scala b/zio-direct/src/main/scala-3.x/zio/direct/core/metaprog/Extractor.scala
@@ -133,6 +133,8 @@ object Extractors {
       tree match
         case DottyExtensionCall.OneArg(invocation @ DirectRunCallAnnotated.Term(), effect) =>
           Some(effect.asExpr)
+        case quotes.reflect.Apply(quotes.reflect.TypeApply(invocation @ DirectRunCallAnnotated.Term(), _), effect :: Nil) =>
+          Some(effect.asExpr)
         case _ =>
           None
     }