packages/components/hooks/useMyCountry.tsx

   1 import { useEffect, useState } from 'react';
   2
   3 import { getSilentApi } from '@proton/shared/lib/api/helpers/customConfig';
   4 import { getLocation } from '@proton/shared/lib/api/vpn';
   5 import { singleCountryTimezoneDatabase } from '@proton/shared/lib/date/singleCountryTimezoneDatabase';
   6 import { manualFindTimeZone } from '@proton/shared/lib/date/timezoneDatabase';
   7 import { getNaiveCountryCode } from '@proton/shared/lib/i18n/helper';
   8 import type { Api, MyLocationResponse } from '@proton/shared/lib/interfaces';
   9
  10 import useApi from './useApi';
  11
  12 const tryTimezone = (tz: string): string | undefined =>
  13     singleCountryTimezoneDatabase[tz as keyof typeof singleCountryTimezoneDatabase];
  14
  15 const getCountryFromTimezone = () => {
  16     try {
  17         const timezone = Intl.DateTimeFormat().resolvedOptions().timeZone;
  18
  19         if (timezone) {
  20             return tryTimezone(timezone) || tryTimezone(manualFindTimeZone(timezone));
  21         }
  22     } catch (e) {
  23         // undefined
  24     }
  25 };
  26
  27 export const getCountryFromLanguage = () => {
  28     const language = navigator.languages.find((language) => /[_-]/.test(language));
  29     if (language) {
  30         return getNaiveCountryCode(language);
  31     }
  32 };
  33
  34 const getStaticState = () => {
  35     return (getCountryFromTimezone() || getCountryFromLanguage())?.toUpperCase();
  36 };
  37
  38 const getMyCountry = async (api: Api): Promise<string | undefined> => {
  39     // TODO: Have a non-VPN dedicated API for that purpose
  40     const value = await api<MyLocationResponse>(getLocation());
  41     return value?.Country?.toUpperCase();
  42 };
  43
  44 const state: { initialized: boolean; promise: null | Promise<string | undefined>; value: string | undefined } = {
  45     promise: null,
  46     value: undefined,
  47     initialized: false,
  48 };
  49
  50 const getInitialValue = () => {
  51     if (!state.initialized) {
  52         state.initialized = true;
  53         state.value = getStaticState();
  54     }
  55     return state.value;
  56 };
  57
  58 const getCountryPromise = (api: Api) => {
  59     if (state.promise) {
  60         return state.promise;
  61     }
  62     state.promise = getMyCountry(getSilentApi(api))
  63         .then((value) => {
  64             state.value = value;
  65             return value;
  66         })
  67         .catch(() => {
  68             return undefined;
  69         });
  70     return state.promise;
  71 };
  72
  73 const useMyCountry = (): string | undefined => {
  74     const [country, setMyCountry] = useState<string | undefined>(getInitialValue);
  75     const api = useApi();
  76     useEffect(() => {
  77         if (country) {
  78             return;
  79         }
  80         void getCountryPromise(api).then(setMyCountry);
  81     }, []);
  82     return country;
  83 };
  84
  85 export default useMyCountry;