refactor: useStore & error checking

Author: animify

package.json

@@ -28,7 +28,7 @@
     "react": "^16.7.0-alpha.0",
     "react-dom": "^16.7.0-alpha.0",
     "redux": "^4.0.1",
-    "use-restate": "^0.0.9"
+    "use-restate": "^0.0.10"
   },
   "devDependencies": {
     "@types/jest": "^23.3.10",
@@ -63,4 +63,4 @@
   "lint-staged": {
     "*.ts": "yarn run lint"
   }
-}
+}

src/index.ts

@@ -5,9 +5,21 @@ import shallowEqual from './shallowEqual';
 export const RestateContext: React.Context<Store<any> | undefined> = React.createContext(undefined);
 export const RestateProvider = RestateContext.Provider;
 
-export function useRestate<TState, TSelector>(selectFrom: (state: TState) => TSelector): TSelector {
+const useStore = () => {
     const store = useContext(RestateContext);
 
+    if (!store) {
+        throw new Error(
+            'use-restate requires you to initiate your redux-like store - e.g <RestateProvider value={store} />',
+        );
+    }
+
+    return store;
+};
+
+export function useRestate<TState, TSelector>(selectFrom: (state: TState) => TSelector): TSelector {
+    const store = useStore();
+
     const [restate, setRestate] = useState(() => selectFrom(store.getState()));
     const previousRestate = useRef({});
 
@@ -39,7 +51,7 @@ export function useRestate<TState, TSelector>(selectFrom: (state: TState) => TSe
 }
 
 export function useDispatch<TAction extends Action>(): Dispatch<TAction> {
-    const store = useContext(RestateContext);
+    const store = useStore();
 
     return store.dispatch;
 }

yarn.lock

@@ -8819,6 +8819,16 @@ url@^0.11.0:
     punycode "1.3.2"
     querystring "0.2.0"
 
+use-restate@^0.0.10:
+  version "0.0.10"
+  resolved "https://registry.yarnpkg.com/use-restate/-/use-restate-0.0.10.tgz#2f2bceb5f569f3961ae1ef0ef326b8b6370c644d"
+  integrity sha512-w2aheIL5NXORBE77E0Qv0DZ1p6zSXY3J5rLqU0wK00yb9kbBnd5CATF3YF5HjuFpqnqZ7ttzXMOv4YfxEIBeOQ==
+  dependencies:
+    react "^16.7.0-alpha.0"
+    react-dom "^16.7.0-alpha.0"
+    redux "^4.0.1"
+    use-restate "^0.0.9"
+
 use-restate@^0.0.9:
   version "0.0.9"
   resolved "https://registry.yarnpkg.com/use-restate/-/use-restate-0.0.9.tgz#cce76d72a7e88139b07bd7d730f4771465eed9ed"