v10/packages/store/src/react/context.tsx

/* eslint-disable react-refresh/only-export-components */

import type { ReactNode } from 'react';
import type { AnyStore } from '../core/store';

import { createContext, useContext } from 'react';

/**
 * Internal shared context for store instances.
 * All Providers write to this context.
 */
const StoreContext = createContext<AnyStore | null>(null);

/**
 * Internal hook for primitive UI components.
 * Accesses the nearest store from context without type information.
 *
 * @throws If used outside of a Provider
 */
export function useStoreContext(): AnyStore {
  const store = useContext(StoreContext);

  if (!store) {
    throw new Error('useStoreContext must be used within a Provider');
  }

  return store;
}

/**
 * Internal hook to get parent store from context.
 * Returns null if no parent Provider exists.
 * Used by Provider to implement the `inherit` prop.
 */
export function useParentStore(): AnyStore | null {
  return useContext(StoreContext);
}

/**
 * Internal provider component that wraps children with store context.
 */
export function StoreContextProvider({ store, children }: { store: AnyStore; children: ReactNode }): ReactNode {
  return <StoreContext.Provider value={store}>{children}</StoreContext.Provider>;
}