import * as React from 'react';
import { useLocale as useDumiLocale } from 'dumi';

export interface LocaleMap<Key extends string> {
  cn: Record<Key, string>;
  en: Record<Key, string>;
}

export default function useLocale<Key extends string>(
  localeMap?: LocaleMap<Key>,
): [Record<Key, string>, 'cn' | 'en'] {
  const { id } = useDumiLocale();
  const localeType = id === 'zh-CN' ? 'cn' : ('en' as const);
  return [localeMap?.[localeType]!, localeType];
}