strapi/packages/strapi-plugin-content-manager/admin/src/containers/EditView/utils.js

import { get, isBoolean, isNaN } from 'lodash';
import * as yup from 'yup';

const errorsTrads = {
  email: 'components.Input.error.validation.email',
  json: 'components.Input.error.validation.json',
  max: 'components.Input.error.validation.max',
  maxLength: 'components.Input.error.validation.maxLength',
  min: 'components.Input.error.validation.min',
  minLength: 'components.Input.error.validation.minLength',
  regex: 'components.Input.error.validation.regex',
  required: 'components.Input.error.validation.required',
};

const createYupSchema = (layout, groupLayoutsData) => {
  return yup.object().shape(
    Object.keys(get(layout, ['schema', 'attributes'], {})).reduce(
      (acc, current) => {
        const attribute = get(layout, ['schema', 'attributes', current], {});

        if (attribute.type !== 'relation' && attribute.type !== 'group') {
          const formatted = createYupSchemaAttribute(attribute.type, attribute);

          acc[current] = formatted;
        }

        if (attribute.type === 'relation') {
          acc[current] = [
            'oneWay',
            'oneToOne',
            'manyToOne',
            'oneToManyMorph',
            'oneToOneMorph',
          ].includes(attribute.relationType)
            ? yup.object()
            : yup.array();
        }

        if (attribute.type === 'group') {
          const getGroupSchema = (path = []) =>
            get(
              groupLayoutsData,
              [attribute.group, 'schema', 'attributes', ...path],
              {}
            );
          const groupAttributes = getGroupSchema();

          const groupSchema = Object.keys(groupAttributes).reduce(
            (acc2, curr) => {
              const groupAttribute = getGroupSchema([curr]);

              if (
                groupAttribute.type !== 'relation' &&
                groupAttribute.type !== 'group'
              ) {
                const formatted = createYupSchemaAttribute(
                  groupAttribute.type,
                  groupAttribute
                );

                acc2[curr] = formatted;
              }

              return acc2;
            },
            {}
          );

          let groupAttributeSchema =
            attribute.repeatable === true
              ? yup.array().of(yup.object().shape(groupSchema))
              : yup.object().shape(groupSchema);

          if (attribute.required === true) {
            groupAttributeSchema = groupAttributeSchema.required();
          }

          acc[current] = groupAttributeSchema;
        }

        return acc;
      },
      {}
    )
  );
};

const createYupSchemaAttribute = (type, validations) => {
  let schema = yup.mixed();

  if (['string', 'text', 'email', 'password', 'enumeration'].includes(type)) {
    schema = yup.string();
  }

  if (type === 'json') {
    schema = yup.object(errorsTrads.json).nullable();
  }

  if (type === 'email') {
    schema = schema.email(errorsTrads.email);
  }

  if (type === 'number') {
    schema = yup
      .number()
      .transform(cv => (isNaN(cv) ? undefined : cv))
      .typeError();
  }

  if (['date', 'datetime'].includes(type)) {
    schema = yup.date().typeError();
  }

  Object.keys(validations).forEach(validation => {
    const validationValue = validations[validation];

    if (
      !!validationValue ||
      ((!isBoolean(validationValue) &&
        Number.isInteger(Math.floor(validationValue))) ||
        validationValue === 0)
    ) {
      switch (validation) {
        case 'required':
          schema = schema.required(errorsTrads.required);
          break;
        case 'max':
          schema = schema.max(validationValue, errorsTrads.max);
          break;
        case 'maxLength':
          schema = schema.max(validationValue, errorsTrads.maxLength);
          break;
        case 'min':
          schema = schema.min(validationValue, errorsTrads.min);
          break;
        case 'minLength':
          schema = schema.min(validationValue, errorsTrads.minLength);
          break;
        case 'regex':
          schema = schema.matches(validationValue, errorsTrads.regex);
          break;
        case 'lowercase':
          if (['text', 'textarea', 'email', 'string'].includes(type)) {
            schema = schema.strict().lowercase();
          }
          break;
        case 'uppercase':
          if (['text', 'textarea', 'email', 'string'].includes(type)) {
            schema = schema.strict().uppercase();
          }
          break;
        case 'positive':
          if (
            ['number', 'integer', 'bigint', 'float', 'decimal'].includes(type)
          ) {
            schema = schema.positive();
          }
          break;
        case 'negative':
          if (
            ['number', 'integer', 'bigint', 'float', 'decimal'].includes(type)
          ) {
            schema = schema.negative();
          }
          break;
        default:
      }
    }
  });

  return schema;
};

export default createYupSchema;