fix: allow to call useKbsDisableGlobal outside of context

zakodium-oss · Feb 6, 2023 · 5ff073a · 5ff073a
1 parent dc43792
commit 5ff073a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/component/KbsProvider.tsx b/src/component/KbsProvider.tsx
@@ -43,6 +43,10 @@ export const kbsContext = createContext<KbsState>(initialKbsState);
 
 const kbsDispatchContext = createContext<Dispatch<KbsAction> | null>(null);
 
+export function useKbsUncheckedDispatch() {
+  return useContext(kbsDispatchContext);
+}
+
 export function useKbsDispatch() {
   const dispatch = useContext(kbsDispatchContext);
   if (!dispatch) {

diff --git a/src/component/hooks/useKbsDisableGlobal.ts b/src/component/hooks/useKbsDisableGlobal.ts
@@ -1,11 +1,11 @@
 import { useEffect } from 'react';
 
-import { useKbsDispatch } from '../KbsProvider';
+import { useKbsUncheckedDispatch } from '../KbsProvider';
 
 export function useKbsDisableGlobal(disabled = true) {
-  const kbsDispatch = useKbsDispatch();
+  const kbsDispatch = useKbsUncheckedDispatch();
   useEffect(() => {
-    if (!disabled) return;
+    if (!disabled || !kbsDispatch) return;
     kbsDispatch({ type: 'DISABLE_GLOBAL' });
     return () => kbsDispatch({ type: 'ENABLE_GLOBAL' });
   }, [kbsDispatch, disabled]);