1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2020 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.technical.graphql
21 
22 import akka.http.scaladsl.server
23 import akka.http.scaladsl.server.{Directives, Route}
24 import caliban.AkkaHttpAdapter.ContextWrapper
25 import caliban.interop.circe.AkkaHttpCirceAdapter
26 import caliban.{CalibanError, GraphQLInterpreter}
27 import org.make.api.technical.MakeDirectives.MakeDirectivesDependencies
28 import org.make.api.technical.graphql.GraphQLRuntimeComponent.{EnvType, RuntimeType}
29 import org.make.api.technical.{ActorSystemComponent, MakeDirectives}
30 import org.make.core.RequestContext
31 import zio.clock.Clock
32 import zio.internal.Platform
33 import zio.{FiberRef, Runtime, URIO, ZIO, ZLayer}
34 
35 trait GraphQLApi extends Directives {
36 
37   def form: Route
38   def graphQL: Route
39   def graphQLWebSocket: Route
40 
41   def routes: Route = form ~ graphQL ~ graphQLWebSocket
42 }
43 
44 trait GraphQLApiComponent {
45   def graphQLApi: GraphQLApi
46 }
47 
48 trait DefaultGraphQLApiComponent extends GraphQLApiComponent with MakeDirectives {
49   self: MakeDirectivesDependencies with GraphQLRuntimeComponent with ActorSystemComponent =>
50 
51   override lazy val graphQLApi: GraphQLApi = new DefaultGraphQLApi
52 
53   class DefaultGraphQLApi extends GraphQLApi with AkkaHttpCirceAdapter with Directives {
54 
55     implicit private val runtime: Runtime[RuntimeType] =
56       Runtime.unsafeFromLayer(
57         ZLayer.fromEffect(FiberRef.make(RequestContext.empty)) ++ GraphQLUtils.ConsoleLoggingService.live ++ Clock.live,
58         Platform.default
59       )
60 
61     private val interpreter: GraphQLInterpreter[RuntimeType, CalibanError] =
62       runtime.unsafeRun(graphQLRuntime.graphQLService.interpreter)
63 
64     override def form: Route = path("graphiql") {
65       getFromResource("graphiql.html")
66     }
67 
68     override def graphQL: Route = path("api" / "graphql") {
69       extractExecutionContext { implicit executionContext =>
70         makeOperation("graphQL") { requestContext =>
71           adapter.makeHttpService(interpreter, contextWrapper = new RequestContextWrapper(requestContext))
72         }
73       }
74     }
75 
76     override def graphQLWebSocket: Route = path("ws" / "graphql") {
77       extractExecutionContext { implicit executionContext =>
78         makeOperation("graphQL") { requestContext =>
79           adapter.makeWebSocketService(interpreter, contextWrapper = new RequestContextWrapper(requestContext))
80         }
81       }
82     }
83   }
84 }
85 
86 class RequestContextWrapper[T](context: RequestContext) extends ContextWrapper[RuntimeType, T] {
87 
88   override def apply[R1 <: RuntimeType, A1 >: T](ctx: server.RequestContext)(effect: URIO[R1, A1]): URIO[R1, A1] = {
89     ZIO.accessM[RuntimeType](_.get[EnvType].set(context)) *> effect
90   }
91 }