strapi/packages/strapi-plugin-i18n/services/entity-service-decorator.js

'use strict';

const { has, omit } = require('lodash/fp');

const { getDefaultLocale } = require('./locales');
const { isLocalized } = require('./content-types');
const { syncLocalizations, updateNonLocalizedFields } = require('./localizations');

const LOCALE_QUERY_FILTER = '_locale';

/**
 * Adds default locale or replaces _locale by locale in query params
 * @param {object} params - query params
 */
const wrapParams = async (params = {}) => {
  if (has(LOCALE_QUERY_FILTER, params)) {
    return {
      ...omit(LOCALE_QUERY_FILTER, params),
      locale: params[LOCALE_QUERY_FILTER],
    };
  }

  return {
    ...params,
    locale: await getDefaultLocale(),
  };
};

/**
 * Decorates the entity service with I18N business logic
 * @param {object} service - entity service
 */
const decorator = service => ({
  /**
   * Wraps query options. In particular will add default locale to query params
   * @param {object} opts - Query options object (params, data, files, populate)
   * @param {object} ctx - Query context
   * @param {object} ctx.model - Model that is being used
   */
  async wrapOptions(opts = {}, ctx) {
    const wrappedOptions = await service.wrapOptions(opts, ctx);
    const model = strapi.db.getModel(ctx.model);

    if (!isLocalized(model)) {
      return wrappedOptions;
    }

    return {
      ...wrappedOptions,
      params: await wrapParams(wrappedOptions.params),
    };
  },

  /**
   * Creates an entry & make links between it and its related localizaionts
   * @param {object} opts - Query options object (params, data, files, populate)
   * @param {object} ctx - Query context
   * @param {object} ctx.model - Model that is being used
   */
  async create(opts, ctx) {
    const model = strapi.db.getModel(ctx.model);
    const entry = await service.create(opts, ctx);

    if (isLocalized(model)) {
      await syncLocalizations(entry, { model });
    }

    return entry;
  },

  /**
   * Updates an entry & update related localizations fields
   * @param {object} opts - Query options object (params, data, files, populate)
   * @param {object} ctx - Query context
   * @param {object} ctx.model - Model that is being used
   */
  async update(opts, ctx) {
    const model = strapi.db.getModel(ctx.model);
    const entry = await service.update(opts, ctx);

    if (isLocalized(model)) {
      await updateNonLocalizedFields(entry, { model });
    }

    return entry;
  },
});

module.exports = {
  decorator,
  wrapParams,
};