strapi/packages/plugins/graphql/server/services/builders/enums.js

'use strict';

const { enumType } = require('nexus');
const { set } = require('lodash/fp');

/**
 * Build a Nexus enum type from a Strapi enum attribute
 * @param {object} definition - The definition of the enum
 * @param {string[]} definition.enum - The params of the enum
 * @param {string} [definition.enumName] - The name of the enum
 * @param {string?} defaultName - Optional, the name of the enum that will be used if config.enumName is nil
 * @return {NexusEnumTypeDef}
 */
const buildEnumTypeDefinition = (definition, defaultName) => {
  const name = defaultName || definition.enumName;

  if (!name) {
    throw new Error('Cannot build an enum type without a name');
  }

  return enumType({
    name,
    // In Strapi V3, the key of an enum is also its value
    // todo[V4]: allow passing an object of key/value instead of an array
    members: definition.enum.reduce((acc, value) => set(value, value, acc), {}),
  });
};

module.exports = () => ({
  buildEnumTypeDefinition,
});