| 1 |
import create from 'zustand'; |
| 2 |
import { devtools, persist } from 'zustand/middleware'; |
| 3 |
import { Lang } from '../types'; |
| 4 |
|
| 5 |
type LanguageType = { |
| 6 |
previousLanguage: Lang; |
| 7 |
setPreviousLanguage: (language: Lang) => void; |
| 8 |
recentLanguages: Lang[]; |
| 9 |
addRecentLanguage: (language: Lang) => void; |
| 10 |
}; |
| 11 |
export const useLanguageStore = create<LanguageType>()( |
| 12 |
persist( |
| 13 |
devtools( |
| 14 |
(set, get) => ({ |
| 15 |
previousLanguage: 'javascript' as Lang, |
| 16 |
setPreviousLanguage(language: Lang) { |
| 17 |
set({ previousLanguage: language }); |
| 18 |
get().addRecentLanguage(language); |
| 19 |
}, |
| 20 |
recentLanguages: [] as Lang[], |
| 21 |
addRecentLanguage(language: Lang) { |
| 22 |
// Limit to the last 3 |
| 23 |
set((state) => { |
| 24 |
if (state.recentLanguages.includes(language)) { |
| 25 |
return state; |
| 26 |
} |
| 27 |
return { |
| 28 |
recentLanguages: [ |
| 29 |
...(state.recentLanguages?.slice(-2) ?? []), |
| 30 |
language, |
| 31 |
], |
| 32 |
}; |
| 33 |
}); |
| 34 |
}, |
| 35 |
}), |
| 36 |
{ name: 'Code Block Pro Language' }, |
| 37 |
), |
| 38 |
{ |
| 39 |
name: 'code-block-pro-last-language', |
| 40 |
getStorage: () => localStorage, |
| 41 |
partialize: (state) => ({ |
| 42 |
previousLanguage: state?.previousLanguage ?? null, |
| 43 |
recentLanguages: state?.recentLanguages ?? [], |
| 44 |
}), |
| 45 |
}, |
| 46 |
), |
| 47 |
); |
| 48 |
|
