persist viewer preferences

2026-02-24 16:13:37 +09:00
parent 14708c041a
commit 63cf407cac
1 changed files with 63 additions and 48 deletions
@@ -10,6 +10,7 @@ import type {
 import type { Object3D } from "three";

 import { create } from "zustand";
+import { persist } from "zustand/middleware";

 type SelectionPath = {
  buildingId: BuildingNode["id"] | null;
@@ -57,62 +58,76 @@ type ViewerState = {
  setCameraDragging: (dragging: boolean) => void
 }

-const useViewer = create<ViewerState>()((set, get) => ({
-  selection: { buildingId: null, levelId: null, zoneId: null, selectedIds: [] },
-  hoveredId: null,
-  setHoveredId: (id) => set({ hoveredId: id }),
+const useViewer = create<ViewerState>()(
+  persist(
+    (set) => ({
+      selection: { buildingId: null, levelId: null, zoneId: null, selectedIds: [] },
+      hoveredId: null,
+      setHoveredId: (id) => set({ hoveredId: id }),

-  cameraMode: "perspective",
-  setCameraMode: (mode) => set({ cameraMode: mode }),
+      cameraMode: "perspective",
+      setCameraMode: (mode) => set({ cameraMode: mode }),

-  levelMode: "stacked",
-  setLevelMode: (mode) => set({ levelMode: mode }),
+      levelMode: "stacked",
+      setLevelMode: (mode) => set({ levelMode: mode }),

-  wallMode: 'cutaway',
-  setWallMode: (mode) => set({ wallMode: mode }),
+      wallMode: 'cutaway',
+      setWallMode: (mode) => set({ wallMode: mode }),

-  showScans: true,
-  setShowScans: (show) => set({ showScans: show }),
+      showScans: true,
+      setShowScans: (show) => set({ showScans: show }),

-  showGuides: true,
-  setShowGuides: (show) => set({ showGuides: show }),
+      showGuides: true,
+      setShowGuides: (show) => set({ showGuides: show }),

-  setSelection: (updates) =>
-    set((state) => {
-      const newSelection = { ...state.selection, ...updates };
+      setSelection: (updates) =>
+        set((state) => {
+          const newSelection = { ...state.selection, ...updates };

-      // Hierarchy Guard: If we change a high-level parent, reset the children
-      if (updates.buildingId !== undefined) {
-        newSelection.levelId = null;
-        newSelection.zoneId = null;
-        newSelection.selectedIds = [];
-      } else if (updates.levelId !== undefined) {
-        newSelection.zoneId = null;
-        newSelection.selectedIds = [];
-      } else if (updates.zoneId !== undefined) {
-        newSelection.selectedIds = [];
-      }
+          // Hierarchy Guard: If we change a high-level parent, reset the children
+          if (updates.buildingId !== undefined) {
+            newSelection.levelId = null;
+            newSelection.zoneId = null;
+            newSelection.selectedIds = [];
+          } else if (updates.levelId !== undefined) {
+            newSelection.zoneId = null;
+            newSelection.selectedIds = [];
+          } else if (updates.zoneId !== undefined) {
+            newSelection.selectedIds = [];
+          }

-      return { selection: newSelection };
+          return { selection: newSelection };
+        }),
+
+      resetSelection: () =>
+        set({
+          selection: {
+            buildingId: null,
+            levelId: null,
+            zoneId: null,
+            selectedIds: [],
+          },
+        }),
+
+      outliner: { selectedObjects: [], hoveredObjects: [] },
+
+      exportScene: null,
+      setExportScene: (fn) => set({ exportScene: fn }),
+
+      cameraDragging: false,
+      setCameraDragging: (dragging) => set({ cameraDragging: dragging }),
    }),
-
-  resetSelection: () =>
-    set({
-      selection: {
-        buildingId: null,
-        levelId: null,
-        zoneId: null,
-        selectedIds: [],
-      },
-    }),
-
-  outliner: { selectedObjects: [], hoveredObjects: [] },
-
-  exportScene: null,
-  setExportScene: (fn) => set({ exportScene: fn }),
-
-  cameraDragging: false,
-  setCameraDragging: (dragging) => set({ cameraDragging: dragging }),
-}));
+    {
+      name: 'viewer-preferences',
+      partialize: (state) => ({
+        cameraMode: state.cameraMode,
+        levelMode: state.levelMode,
+        wallMode: state.wallMode,
+        showScans: state.showScans,
+        showGuides: state.showGuides,
+      }),
+    },
+  ),
+);

 export default useViewer;