diff --git a/src/hooks/use-translate.ts b/src/hooks/use-translate.ts index 92c4569..9c69bef 100644 --- a/src/hooks/use-translate.ts +++ b/src/hooks/use-translate.ts @@ -1,5 +1,5 @@ import { Language } from "@/store/lang"; -export const useTranslate = (lang: Language): number => { +export const useTranslate = (lang: "en" | "ru"): number => { return lang === Language.RU ? 0 : 1; }; diff --git a/src/store/lang.ts b/src/store/lang.ts index 72318c1..9c707af 100644 --- a/src/store/lang.ts +++ b/src/store/lang.ts @@ -1,4 +1,5 @@ import { create } from "zustand"; +import { persist } from "zustand/middleware"; export enum Language { EN = "en", @@ -6,11 +7,16 @@ export enum Language { } export interface LangState { - lang: Language; + lang: "ru" | "en; setLang: (lang: Language) => void; } -export const useLangStore = create()((set) => ({ - lang: Language.RU, - setLang: (lang) => set({ lang }), -})); +export const useLangStore = create()( + persist( + (set) => ({ + lang: Language.RU, + setLang: (lang) => set({ lang }), + }), + { name: "lang-storage" } + ) +);