packages/plugins/i18n/server/src/services/locales.ts - develop - githubmirror/strapi

packages/plugins/i18n/server/src/services/locales.ts
103 строки3 KB
Alexandre Bodin
chore: database enhancements
14 мар 2024, 00:13
14 мар 2024, 00:13a952855
import { isNil } from 'lodash/fp';
import { DEFAULT_LOCALE } from '../constants';
import { getService, getCoreStore } from '../utils';

const find = (params: any = {}) =>
  strapi.db.query('plugin::i18n.locale').findMany({ where: params });

const findById = (id: any) => strapi.db.query('plugin::i18n.locale').findOne({ where: { id } });

const findByCode = (code: any) =>
  strapi.db.query('plugin::i18n.locale').findOne({ where: { code } });

const count = (params: any = {}) => strapi.db.query('plugin::i18n.locale').count({ where: params });

const create = async (locale: any) => {
  const result = await strapi.db.query('plugin::i18n.locale').create({ data: locale });

  getService('metrics').sendDidUpdateI18nLocalesEvent();

  return result;
};

const update = async (params: any, updates: any) => {
  const result = await strapi.db
    .query('plugin::i18n.locale')
    .update({ where: params, data: updates });

  getService('metrics').sendDidUpdateI18nLocalesEvent();

  return result;
};

const deleteFn = async ({ id }: any) => {
  const localeToDelete = await findById(id);

  if (localeToDelete) {
    await deleteAllLocalizedEntriesFor({ locale: localeToDelete.code });
    const result = await strapi.db.query('plugin::i18n.locale').delete({ where: { id } });

    getService('metrics').sendDidUpdateI18nLocalesEvent();

    return result;
  }

  return localeToDelete;
};

const setDefaultLocale = ({ code }: any) =>
  getCoreStore().set({ key: 'default_locale', value: code });

const getDefaultLocale = () => getCoreStore().get({ key: 'default_locale' });

const setIsDefault = async (locales: any) => {
  if (isNil(locales)) {
    return locales;
  }

  const actualDefault = await getDefaultLocale();

  if (Array.isArray(locales)) {
    return locales.map((locale) => ({ ...locale, isDefault: actualDefault === locale.code }));
  }
  // single locale
  return { ...locales, isDefault: actualDefault === locales.code };
};

const initDefaultLocale = async () => {
  const existingLocalesNb = await strapi.db.query('plugin::i18n.locale').count();
  if (existingLocalesNb === 0) {
    await create(DEFAULT_LOCALE);
    await setDefaultLocale({ code: DEFAULT_LOCALE.code });
  }
};

const deleteAllLocalizedEntriesFor = async ({ locale }: any) => {
  const { isLocalizedContentType } = getService('content-types');

  const localizedModels = Object.values(strapi.contentTypes).filter(isLocalizedContentType);

  for (const model of localizedModels) {
    // FIXME: delete many content & their associations
    await strapi.db.query(model.uid).deleteMany({ where: { locale } });
  }
};

const locales = () => ({
  find,
  findById,
  findByCode,
  create,
  update,
  count,
  setDefaultLocale,
  getDefaultLocale,
  setIsDefault,
  delete: deleteFn,
  initDefaultLocale,
});

type LocaleService = typeof locales;

export default locales;
export type { LocaleService };