Carlos Florêncio
32 lines
1003 B
TypeScript
32 lines
1003 B
TypeScript
"use client";
|
|
|
|
import { createContext, useContext, useState } from "react";
|
|
import type { StoreApi } from "zustand";
|
|
import { useStore } from "zustand";
|
|
import type { AppState, StoreProviderProps } from "@/lib/state/store";
|
|
import { createAppStore } from "@/lib/state/store";
|
|
|
|
const StoreContext = createContext<StoreApi<AppState> | null>(null);
|
|
|
|
export function StateProvider({ children, initialState }: StoreProviderProps) {
|
|
const [store] = useState(() => createAppStore(initialState));
|
|
|
|
return <StoreContext.Provider value={store}>{children}</StoreContext.Provider>;
|
|
}
|
|
|
|
export function useAppStore<T>(selector: (state: AppState) => T) {
|
|
const store = useContext(StoreContext);
|
|
if (!store) {
|
|
throw new Error("useAppStore must be used within StateProvider");
|
|
}
|
|
return useStore(store, selector);
|
|
}
|
|
|
|
export function useAppStoreApi() {
|
|
const store = useContext(StoreContext);
|
|
if (!store) {
|
|
throw new Error("useAppStoreApi must be used within StateProvider");
|
|
}
|
|
return store;
|
|
}
|