strapi/packages/strapi-plugin-graphql/hooks/graphql/index.js

'use strict';

/**
 * Module dependencies
 */

// Public node modules.
const _ = require('lodash');
const path = require('path');
const glob = require('glob');
const { graphqlKoa, graphiqlKoa } = require('apollo-server-koa');

module.exports = strapi => {
  return {
    beforeInitialize: async function() {
      // Try to inject this hook just after the others hooks to skip the router processing.
      if (!_.get(strapi.config.hook.load, 'after')) {
        _.set(strapi.config.hook.load, 'after', []);
      }

      strapi.config.hook.load.after.push('graphql');

      // Load core utils.
      const utils = require(path.resolve(strapi.config.appPath, 'node_modules', 'strapi', 'lib', 'utils'));

      // Set '*.graphql' files configurations in the global variable.
      await Promise.all([
        // Load root configurations.
        new Promise((resolve, reject) => {
          glob('./config/*.graphql', {
            cwd: strapi.config.appPath
          }, (err, files) => {
            if (err) {
              return reject(err);
            }

            utils.loadConfig.call(strapi, files, true).then(resolve).catch(reject);
          });
        }),
        // Load APIs configurations.
        new Promise((resolve, reject) => {
          glob('./api/*/config/*.graphql', {
            cwd: strapi.config.appPath
          }, (err, files) => {
            if (err) {
              return reject(err);
            }

            utils.loadConfig.call(strapi, files, true).then(resolve).catch(reject);
          });
        }),
        // Load plugins configurations.
        new Promise((resolve, reject) => {
          glob('./plugins/*/config/*.graphql', {
            cwd: strapi.config.appPath
          }, (err, files) => {
            if (err) {
              return reject(err);
            }

            utils.loadConfig.call(strapi, files, true).then(resolve).catch(reject);
          });
        })
      ]);

      /*
       * Create a merge of all the GraphQL configuration.
       */

      // Set path with initial state.
      _.set(strapi.plugins.graphql, 'config._schema.graphql', { definition: ``, query: ``, type : {}, resolver: {} });

      // Merge user API.
      Object.keys(strapi.api).reduce((acc, current) => {
        const { definition, query, type, resolver } = _.get(strapi.api[current], 'config.schema.graphql', {});

        acc.definition += definition || ``;
        acc.query += query || ``;

        return _.merge(acc, {
          type,
          resolver
        });
      }, strapi.plugins.graphql.config._schema.graphql);

      // Merge plugins API.
      Object.keys(strapi.plugins).reduce((acc, current) => {
        const { definition, query, type, resolver } = _.get(strapi.plugins[current], 'config.schema.graphql', {});

        acc.definition += definition || ``;
        acc.query += query || ``;

        return _.merge(acc, {
          type,
          resolver
        });
      }, strapi.plugins.graphql.config._schema.graphql);
    },

    initialize: function(cb) {
      const schema = strapi.plugins.graphql.services.graphql.generateSchema();

      if (_.isEmpty(schema)) {
        strapi.log.warn(`GraphQL schema has not been generated because it's empty`);

        return cb();
      }

      const router = strapi.koaMiddlewares.routerJoi();

      router.post(strapi.plugins.graphql.config.endpoint, async (ctx, next) => graphqlKoa({ schema, context: ctx })(ctx, next));
      router.get(strapi.plugins.graphql.config.endpoint, async (ctx, next) => graphqlKoa({ schema, context: ctx })(ctx, next));

      // Disable GraphiQL in production environment.
      if (strapi.config.environment !== 'production') {
        router.get('/graphiql', graphiqlKoa({ endpointURL: strapi.plugins.graphql.config.endpoint }));
      }

      strapi.app.use(router.middleware());

      cb();
    }
  };
};