gatsby-source-strapi/dist/clean-data.js

Gatsby source plugin for building websites using Strapi as a data source

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
exports.__esModule = true;
exports.cleanData = exports.cleanAttributes = void 0;
var _pick2 = _interopRequireDefault(require("lodash/pick"));
var _snakeCase2 = _interopRequireDefault(require("lodash/snakeCase"));
var _helpers = require("./helpers");
const MEDIA_FIELDS = ["id", "name", "alternativeText", "caption", "width", "height", "formats", "hash", "ext", "mime", "size", "url", "previewUrl", "createdAt", "updatedAt", "documentId", "publishedAt"];
const restrictedFields = new Set(["__component", `children`, `fields`, `internal`, `parent`]);
const getValue = (value, version) => {
  if (!value) {
    return;
  }
  if (version === 4) {
    return value === null || value === void 0 ? void 0 : value.data;
  }
  // assume v5
  return value;
};
const getAttributes = (data, version) => {
  if (!data) {
    return;
  }
  if (version === 4) {
    return {
      id: data.id,
      ...data.attributes
    };
  }
  // assume v5
  return data;
};

/**
 * Removes the attribute key in the entire data.
 * @param {Object} attributes response from the API
 * @param {Object} currentSchema
 * @param {*} schemas
 * @returns
 */
const cleanAttributes = (attributes, currentSchema, schemas, version = 5) => {
  if (!attributes) {
    return;
  }

  // eslint-disable-next-line unicorn/no-array-reduce
  return Object.entries(attributes).reduce((accumulator, [name, value]) => {
    const attribute = currentSchema.schema.attributes[name];
    const attributeName = restrictedFields.has(name) ? (0, _snakeCase2.default)(`strapi_${name}`) : name;
    if (!(attribute !== null && attribute !== void 0 && attribute.type)) {
      accumulator[attributeName] = value;
      return accumulator;
    }

    // Changing the format in order to extract images from the richtext field
    // NOTE: We could add an option to disable the extraction
    if (attribute.type === "richtext") {
      return {
        ...accumulator,
        [attributeName]: {
          data: value || "",
          medias: []
        }
      };
    }
    if (!value) {
      accumulator[attributeName] = value;
      return accumulator;
    }
    if (attribute.type === "dynamiczone") {
      return {
        ...accumulator,
        [attributeName]: value.map(v => {
          const compoSchema = (0, _helpers.getContentTypeSchema)(schemas, v.__component);
          return cleanAttributes(v, compoSchema, schemas, version);
        })
      };
    }
    if (attribute.type === "component") {
      const isRepeatable = attribute.repeatable;
      const compoSchema = (0, _helpers.getContentTypeSchema)(schemas, attribute.component);
      if (isRepeatable) {
        return {
          ...accumulator,
          [attributeName]: value.map(v => {
            return cleanAttributes(v, compoSchema, schemas, version);
          })
        };
      }
      return {
        ...accumulator,
        [attributeName]: cleanAttributes(value, compoSchema, schemas, version)
      };
    }

    // make sure we can use both v4 and v5 outputs
    const valueData = getValue(value, version);
    if (attribute.type === "media") {
      if (Array.isArray(valueData)) {
        return {
          ...accumulator,
          [attributeName]: valueData ? valueData.map(data => ({
            ...(0, _pick2.default)(getAttributes(data, version), MEDIA_FIELDS)
          })) : undefined
        };
      }
      return {
        ...accumulator,
        [attributeName]: valueData ? {
          ...(0, _pick2.default)(getAttributes(valueData, version), MEDIA_FIELDS)
        } : undefined
      };
    }
    if (attribute.type === "relation") {
      const relationSchema = (0, _helpers.getContentTypeSchema)(schemas, attribute.target);
      if (Array.isArray(valueData)) {
        return {
          ...accumulator,
          [attributeName]: valueData.map(data => cleanAttributes(getAttributes(data, version), relationSchema, schemas, version))
        };
      }
      return {
        ...accumulator,
        [attributeName]: cleanAttributes(getAttributes(valueData, version) || undefined, relationSchema, schemas, version)
      };
    }
    accumulator[attributeName] = value;
    return accumulator;
  }, {});
};

/**
 * Transform v4 API response to the v3 format, remove data and attributes key
 * Transform richtext field to prepare media extraction
 * @param {Object} data
 * @param {Object} ctx
 * @returns {Object}
 */
exports.cleanAttributes = cleanAttributes;
const cleanData = (data, context, version = 5) => {
  const {
    schemas,
    contentTypeUid
  } = context;
  const currentContentTypeSchema = (0, _helpers.getContentTypeSchema)(schemas, contentTypeUid);
  return {
    ...cleanAttributes(getAttributes(data, version), currentContentTypeSchema, schemas, version)
  };
};
exports.cleanData = cleanData;