34 lines
1.2 KiB
TypeScript
34 lines
1.2 KiB
TypeScript
import { create } from "zustand";
|
||
import { persist } from "zustand/middleware";
|
||
|
||
export type ThemeName = "light" | "dark";
|
||
|
||
type ThemeStore = {
|
||
theme: ThemeName;
|
||
setTheme: (theme: ThemeName) => void;
|
||
};
|
||
|
||
const VALID_THEMES: ThemeName[] = ["light", "dark"];
|
||
|
||
export const useThemeStore = create<ThemeStore>()(
|
||
persist(
|
||
(set) => ({
|
||
theme: "light",
|
||
// 写入前校验：非法 theme 直接忽略，防止 syncRemoteUserData 等路径绕过 merge 写入坏值
|
||
setTheme: (next) => set((state) => (VALID_THEMES.includes(next) ? { theme: next } : state)),
|
||
}),
|
||
{
|
||
name: "infinite-canvas:theme_store",
|
||
// 持久化恢复校验：旧版本/坏 session 写入的非法值回退到 light，
|
||
// 避免 canvasThemes[非法值] = undefined 触发 "reading 'node'" 崩溃
|
||
merge: (persisted, current) => {
|
||
const stored = (persisted || {}) as Partial<ThemeStore>;
|
||
const theme = VALID_THEMES.includes(stored.theme as ThemeName)
|
||
? (stored.theme as ThemeName)
|
||
: "light";
|
||
return { ...current, theme };
|
||
},
|
||
},
|
||
),
|
||
);
|