server.js

import cookieParser from 'cookie-parser'
import cors from 'cors'
import express from 'express'
import http from 'http'
import { ApolloServerPluginLandingPageGraphQLPlayground as enablePlayground } from 'apollo-server-core'
import { ApolloServer } from 'apollo-server-express'
import requestLanguage from 'express-request-language'
import { execute, subscribe, GraphQLSchema } from 'graphql'
import depthLimit from 'graphql-depth-limit'
import { createComplexityLimitRule } from 'graphql-validation-complexity'
import { SubscriptionServer } from 'subscriptions-transport-ws'

import { createContext } from './create-context'
import { createQuerySchema } from './query'
import { createMutationSchema } from './mutation'
import { createSubscriptionSchema } from './subscription'
import { createI18n } from './create-i18n'
import { verifyToken, userRequired, verifiedRequired } from './auth'
import { loadUserByKey } from './user/loaders'
import { customOnConnect } from './on-connect'
import { arangodb } from 'arango-express'

const createSchema = () =>
  new GraphQLSchema({
    query: createQuerySchema(),
    mutation: createMutationSchema(),
    subscription: createSubscriptionSchema(),
  })

const createValidationRules = (
  maxDepth,
  complexityCost,
  scalarCost,
  objectCost,
  listFactor,
) => {
  return [
    depthLimit(maxDepth),
    createComplexityLimitRule(complexityCost, {
      scalarCost,
      objectCost,
      listFactor,
      formatErrorMessage: (cost) => {
        console.warn(`User attempted a costly request: ${cost}`)
        return `Query error, query is too complex.`
      },
    }),
  ]
}

export const Server = async ({
  arango = {},
  maxDepth,
  complexityCost,
  scalarCost,
  objectCost,
  listFactor,
  tracing,
  context = {},
}) => {
  const app = express()

  app.use('*', cors())

  app.use(cookieParser())

  app.use(
    requestLanguage({
      languages: ['en', 'fr'],
    }),
  )

  app.use(arangodb(arango))

  app.get('/alive', (_req, res) => {
    res.json({ ok: 'yes' })
  })

  app.get('/ready', (_req, res) => {
    res.json({ ok: 'yes' })
  })

  // default error handler
  app.use(function (err, _req, res, _next) {
    res.status(200).json({
      error: {
        errors: [
          {
            message: err,
            locations: [
              {
                line: 1,
                column: 1,
              },
            ],
          },
        ],
      },
    })
  })

  const httpServer = http.createServer(app)

  const schema = createSchema()

  const server = new ApolloServer({
    schema,
    context: createContext(context),
    validationRules: createValidationRules(
      maxDepth,
      complexityCost,
      scalarCost,
      objectCost,
      listFactor,
    ),
    introspection: true,
    tracing,
    plugins: [enablePlayground()],
  })

  await server.start()
  server.applyMiddleware({ app })

  SubscriptionServer.create(
    {
      schema,
      execute,
      subscribe,
      onConnect: customOnConnect({
        createContext,
        serverContext: context,
        createI18n,
        verifyToken,
        userRequired,
        loadUserByKey,
        verifiedRequired,
      }),
    },
    {
      server: httpServer,
      path: server.graphqlPath,
    },
  )

  return httpServer
}