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;
|
|
}
|