support multiple rex store instances in a project

Author: DaniAkash

src/index.tsx

@@ -4,36 +4,34 @@ import React, {
   createContext,
   useContext,
   Dispatch,
-  SetStateAction
+  SetStateAction,
+  Context
 } from "react";
 
-const createStore = (store: object) => {
-  const intializedStore = {};
-  for (const item in store) {
-    intializedStore[item] = new store[item]();
-  }
-  return intializedStore;
-};
-
-export interface RexProviderProps {
-  children: ReactNode;
-  store: object;
-}
-
-const initalAppState = {};
-
-const AppContext = createContext(initalAppState);
-
-const { Provider } = AppContext;
-
-export const useRex = () => {
-  const store = useContext(AppContext);
-  return store;
-};
+export const createRexStore: <T extends { [key: string]: Rex<any> }>(
+  store: T
+) => {
+  RexProvider: ({ children }: { children: React.ReactNode }) => JSX.Element;
+  useRex: () => T;
+} = store => {
+  let RexContext: Context<typeof store>;
+
+  const useRex = () => {
+    const store = useContext(RexContext);
+    return store;
+  };
+
+  const RexProvider = ({ children }: { children: ReactNode }) => {
+    const intializedStore: typeof store = {};
+    for (const item in store) {
+      intializedStore[item] = new store[item]();
+    }
+    RexContext = createContext(intializedStore);
+    const { Provider } = RexContext;
+    return <Provider value={intializedStore}>{children}</Provider>;
+  };
 
-export const RexProvider = ({ children, store }: RexProviderProps) => {
-  const initializedStore = createStore(store);
-  return <Provider value={initializedStore}>{children}</Provider>;
+  return { RexProvider, useRex };
 };
 
 class Rex<S> {