Intlayer

/** * Creates a new datastore for a given server context. * Attempts to closely mimic the `createContext` API. * * @example * const IntlayerServer = createServerContext<string | null>(null); * * <IntlayerServer value={locale}> * {children} * </IntlayerServer> */ import react, { type FC, type PropsWithChildren, type ReactNode } from 'react'; type CacheType<T> = () => { value: T | undefined }; const cacheFallback = () => () => ({ value: undefined }); export const createServerContext = <T,>(defaultValue?: T): ServerContext<T> => { throwInClient(); /** @ts-ignore remove error Property 'cache' does not exist on type 'typeof React'. */ const cache = react.cache<CacheType<T>> ?? cacheFallback; const getCache = cache(() => ({ value: undefined, })); const Provider: FC<PropsWithChildren<{ value?: T }>> = ({ children, value, }) => { getCache().value = value; return children; }; const ServerContext = Provider as ServerContext<T>; ServerContext.Provider = Provider; ServerContext.Consumer = (props) => { const store = getCache(); return props.children(store ? store.value : defaultValue); }; ServerContext._storage = getCache; ServerContext._defaultValue = defaultValue; return ServerContext; }; /** * Fetches a value present in a given server context. * Attempts to closely mimic the `useContext` API. * * @example * getServerContext(IntlayerServer); */ export const getServerContext = <T,>({ _storage, _defaultValue, }: ServerContext<T>) => { // throwInClient(); const store = _storage(); if (!store) return _defaultValue; return store.value; }; type ServerContext<T> = FC<PropsWithChildren<{ value?: T }>> & { Provider: FC<PropsWithChildren<{ value?: T }>>; Consumer: FC< PropsWithChildren<{ children: (context: T | undefined) => ReactNode }> >; _storage: () => { value: T | undefined }; _defaultValue: T | undefined; }; /** * Throws if called within a client component environment. * Useful to help prevent mistakes. */ const throwInClient = (): undefined | never => { // If window.document is defined we're in a client component if (typeof window !== 'undefined') { throw new Error(`createServerContext only works in Server Components`); } };