prowler/ui/components/layout/app-sidebar/app-sidebar-mode-store.ts

import { create } from "zustand";
import { createJSONStorage, persist } from "zustand/middleware";

import { useStore } from "@/hooks/use-store";

import { APP_SIDEBAR_MODE, type AppSidebarMode } from "./types";

interface PersistedAppSidebarState {
  mode: AppSidebarMode;
}

interface AppSidebarModeStore extends PersistedAppSidebarState {
  setMode: (mode: AppSidebarMode) => void;
}

function isAppSidebarMode(value: unknown): value is AppSidebarMode {
  return Object.values(APP_SIDEBAR_MODE).some((mode) => mode === value);
}

export function migrateAppSidebarState(
  persistedState: unknown,
): PersistedAppSidebarState {
  if (typeof persistedState !== "object" || persistedState === null) {
    return { mode: APP_SIDEBAR_MODE.BROWSE };
  }

  if ("mode" in persistedState && isAppSidebarMode(persistedState.mode)) {
    return { mode: persistedState.mode };
  }

  if (
    "navigationMode" in persistedState &&
    isAppSidebarMode(persistedState.navigationMode)
  ) {
    return { mode: persistedState.navigationMode };
  }

  return { mode: APP_SIDEBAR_MODE.BROWSE };
}

export const useAppSidebarMode = create<AppSidebarModeStore>()(
  persist(
    (set) => ({
      mode: APP_SIDEBAR_MODE.BROWSE,
      setMode: (mode) => set({ mode }),
    }),
    {
      name: "sidebar",
      storage: createJSONStorage(() => localStorage),
      merge: (persistedState, currentState) => ({
        ...currentState,
        ...migrateAppSidebarState(persistedState),
      }),
    },
  ),
);

// The persisted mode is only known in the browser: `persist` rehydrates from
// localStorage before the first client render, while the server always
// rendered `browse`. Reading the store directly makes a tenant that last used
// the chat hydrate a different sidebar than the one in the HTML (React #418).
// This read answers `browse` until mounted, then the persisted value.
export function useHydratedAppSidebarMode(): AppSidebarMode {
  return (
    useStore(useAppSidebarMode, (state) => state.mode) ??
    APP_SIDEBAR_MODE.BROWSE
  );
}