| 1 |
import { Theme } from 'shiki'; |
| 2 |
import create from 'zustand'; |
| 3 |
import { devtools, persist } from 'zustand/middleware'; |
| 4 |
|
| 5 |
type ThemeType = { |
| 6 |
previousTheme: Theme; |
| 7 |
setPreviousTheme: (theme: Theme) => void; |
| 8 |
}; |
| 9 |
export const useThemeStore = create<ThemeType>()( |
| 10 |
persist( |
| 11 |
devtools( |
| 12 |
(set) => ({ |
| 13 |
previousTheme: 'nord', |
| 14 |
setPreviousTheme(theme: Theme) { |
| 15 |
set({ previousTheme: theme }); |
| 16 |
}, |
| 17 |
}), |
| 18 |
{ name: 'Code Block Pro Theme' }, |
| 19 |
), |
| 20 |
{ |
| 21 |
name: 'code-block-pro-last-theme', |
| 22 |
getStorage: () => localStorage, |
| 23 |
partialize: (state) => ({ |
| 24 |
previousTheme: state?.previousTheme ?? null, |
| 25 |
}), |
| 26 |
}, |
| 27 |
), |
| 28 |
); |
| 29 |
|