s2graphql/src/main/scala/org/apache/s2graph/graphql/HttpServer.scala

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.s2graph.graphql

import java.nio.charset.Charset

import akka.actor.ActorSystem
import akka.http.scaladsl.Http
import akka.http.scaladsl.marshallers.sprayjson.SprayJsonSupport._
import akka.http.scaladsl.model._
import akka.http.scaladsl.server.Directives._
import akka.http.scaladsl.server.{Route, StandardRoute}
import akka.stream.ActorMaterializer
import org.slf4j.LoggerFactory
import sangria.parser.QueryParser
import spray.json._

import scala.concurrent.Await
import scala.language.postfixOps
import scala.util._
import akka.http.scaladsl.marshalling.{Marshaller, ToEntityMarshaller, ToResponseMarshallable}
import akka.http.scaladsl.unmarshalling.{FromEntityUnmarshaller, Unmarshaller}
import akka.util.ByteString
import sangria.ast.Document
import sangria.renderer.{QueryRenderer, QueryRendererConfig}

import scala.collection.immutable.Seq

object Server extends App {
  val logger = LoggerFactory.getLogger(this.getClass)

  implicit val system = ActorSystem("s2graphql-server")
  implicit val materializer = ActorMaterializer()

  import system.dispatcher
  import scala.concurrent.duration._

  import spray.json.DefaultJsonProtocol._

  val graphQLServer = new GraphQLServer()

  val route: Route =
    get {
      getFromResource("assets/graphiql.html")
    } ~ (post & path("updateEdgeFetcher")) {
      entity(as[JsValue]) { body =>
        graphQLServer.updateEdgeFetcher(body) match {
          case Success(_) => complete(StatusCodes.OK -> JsString("Update fetcher finished"))
          case Failure(e) =>
            logger.error("Error on execute", e)
            complete(StatusCodes.InternalServerError -> spray.json.JsObject("message" -> JsString(e.toString)))
        }
      }
    } ~ (post & path("graphql")) {
      parameters('operationName.?, 'variables.?) { (operationNameParam, variablesParam) =>
        entity(as[Document]) { document ⇒
          variablesParam.map(parseJson) match {
            case None ⇒ complete(graphQLServer.executeGraphQLQuery(document, operationNameParam, JsObject()))
            case Some(Right(js)) ⇒ complete(graphQLServer.executeGraphQLQuery(document, operationNameParam, js.asJsObject))
            case Some(Left(e)) ⇒
              logger.error("Error on execute", e)
              complete(StatusCodes.BadRequest -> graphQLServer.formatError(e))
          }
        } ~ entity(as[JsValue]) { body ⇒
          val fields = body.asJsObject.fields

          val query = fields.get("query").map(js => js.convertTo[String])
          val operationName = fields.get("operationName").filterNot(_ == JsNull).map(_.convertTo[String])
          val variables = fields.get("variables").filterNot(_ == JsNull)

          query.map(QueryParser.parse(_)) match {
            case None ⇒ complete(StatusCodes.BadRequest -> graphQLServer.formatError("No query to execute"))
            case Some(Failure(error)) ⇒
              logger.error("Error on execute", error)
              complete(StatusCodes.BadRequest -> graphQLServer.formatError(error))
            case Some(Success(document)) => variables match {
              case Some(js) ⇒ complete(graphQLServer.executeGraphQLQuery(document, operationName, js.asJsObject))
              case None ⇒ complete(graphQLServer.executeGraphQLQuery(document, operationName, JsObject()))
            }
          }
        }
      }
    }

  val port = sys.props.get("http.port").fold(8000)(_.toInt)

  logger.info(s"Starting GraphQL server... $port")

  Http().bindAndHandle(route, "0.0.0.0", port).foreach { binding =>
    logger.info(s"GraphQL server ready for connect")
  }

  def shutdown(): Unit = {
    logger.info("Terminating...")

    system.terminate()
    Await.result(system.whenTerminated, 30 seconds)

    logger.info("Terminated.")
  }

  // Unmarshaller

  def unmarshallerContentTypes: Seq[ContentTypeRange] = mediaTypes.map(ContentTypeRange.apply)

  def mediaTypes: Seq[MediaType.WithFixedCharset] =
    Seq(MediaType.applicationWithFixedCharset("graphql", HttpCharsets.`UTF-8`, "graphql"))

  implicit def documentMarshaller(implicit config: QueryRendererConfig = QueryRenderer.Compact): ToEntityMarshaller[Document] = {
    Marshaller.oneOf(mediaTypes: _*) {
      mediaType ⇒
        Marshaller.withFixedContentType(ContentType(mediaType)) {
          json ⇒ HttpEntity(mediaType, QueryRenderer.render(json, config))
        }
    }
  }

  implicit val documentUnmarshaller: FromEntityUnmarshaller[Document] = {
    Unmarshaller.byteStringUnmarshaller
      .forContentTypes(unmarshallerContentTypes: _*)
      .map {
        case ByteString.empty ⇒ throw Unmarshaller.NoContentException
        case data ⇒
          import sangria.parser.DeliveryScheme.Throw
          QueryParser.parse(data.decodeString(Charset.forName("UTF-8")))
      }
  }

  def parseJson(jsStr: String): Either[Throwable, JsValue] = {
    val parsed = Try(jsStr.parseJson)
    parsed match {
      case Success(js) => Right(js)
      case Failure(e) => Left(e)
    }
  }

}