umami/hooks/useLocale.js

import { useEffect } from 'react';
import { useDispatch, useSelector } from 'react-redux';
import { setLocale } from 'redux/actions/app';
import { useRouter } from 'next/router';
import { get, setItem } from 'lib/web';
import { LOCALE_CONFIG } from 'lib/constants';
import { languages } from 'lib/lang';
import useForceUpdate from 'hooks/useForceUpdate';
import enUS from 'public/lang/en-US.json';

const messages = {
  'en-US': enUS,
};

export default function useLocale() {
  const locale = useSelector(state => state.app.locale);
  const dispatch = useDispatch();
  const { basePath } = useRouter();
  const forceUpdate = useForceUpdate();
  const dir = languages[locale]?.dir || 'ltr';

  async function loadMessages(locale) {
    const { ok, data } = await get(`${basePath}/lang/${locale}.json`);

    if (ok) {
      messages[locale] = data;
    }
  }

  async function saveLocale(value) {
    if (!messages[value]) {
      await loadMessages(value);
    }

    setItem(LOCALE_CONFIG, value);

    if (locale !== value) {
      dispatch(setLocale(value));
    } else {
      forceUpdate();
    }
  }

  useEffect(() => {
    if (!messages[locale]) {
      saveLocale(locale);
    }
  }, [locale]);

  return { locale, saveLocale, messages, dir };
}