Safe context hook creator function

index.ts

import { type Context, useContext } from "react";
import { invariant } from "../invariant";

export const createSafeContextHook = <TContextValue>(
  context: Context<TContextValue>,
  displayName: string
) => {
  return () => {
    const contextValue = useContext(context);

    invariant(
      contextValue,
      `${displayName} must be used within a ${displayName}Provider.`
    );

    return contextValue;
  };
};