Function useWarehouseSessionStorage

src/client/hooks/useWarehouseSessionStorage.ts:32–104

(
  workspaceId: string
)

 * Stores messages, chart blocks, and selected database/table scopes
 */
export function useWarehouseSessionStorage(
  workspaceId: string
): UseWarehouseSessionStorageReturn {
  const storageKey = `warehouse-chat-${workspaceId}`;

  const [session, setSession] = useLocalStorageState<StoredSession | null>(
    storageKey,
    {
      defaultValue: null,
      serializer: (value) => JSON.stringify(value),
      deserializer: (value) => {
        try {
          return JSON.parse(value);
        } catch {
          return null;
        }
      },
    }
  );

  const saveSession = useCallback(
    (data: Partial<Omit<StoredSession, 'updatedAt'>>) => {
      setSession((prev) => {
        const currentSession = prev ?? {
          messages: [],
          chartBlocks: [],
          selectedScopes: [],
          updatedAt: Date.now(),
        };
        return {
          ...currentSession,
          ...data,
          updatedAt: Date.now(),
        };
      });
    },
    [setSession]
  );

  const saveMessages = useCallback(
    (messages: UIMessage[]) => {
      saveSession({ messages });
    },
    [saveSession]
  );

  const saveChartBlocks = useCallback(
    (chartBlocks: ChartBlock[]) => {
      saveSession({ chartBlocks });
    },
    [saveSession]
  );

  const saveSelectedScopes = useCallback(
    (selectedScopes: WarehouseScope[]) => {
      saveSession({ selectedScopes });
    },
    [saveSession]