contextFactory.ts

import React, { useContext } from "react";
import { createContext } from "react";

export const contextFactory = <TParams, TStore>(useStore: (args: TParams) => TStore) => {
  const Context = createContext<TStore>(null as TStore);

  const useStoreContext = () => useContext(Context);

  const Provider = (
    props:
      | { store: TStore; args?: never; children: React.ReactNode }
      | { store?: never; args: TParams; children: React.ReactNode }
  ) => {
    if (props.store)
      return <Context.Provider value={props.store}>{props.children}</Context.Provider>;

    const store = useStore(props.args as TParams);
    return <Context.Provider value={store}>{props.children}</Context.Provider>;
  };

  return { Context, useStoreContext, Provider };
};