refactor: use custom hook in a right way

juspay · Nov 7, 2024 · 0724fa9 · 0724fa9
1 parent d39aeaf
commit 0724fa9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 11 deletions.
diff --git a/src/Hooks/UtilityHooks.res b/src/Hooks/UtilityHooks.res
@@ -61,3 +61,17 @@ let useSendEventsToParent = eventsToSendToParent => {
     Some(() => {Window.removeEventListener("message", handle)})
   })
 }
+
+let useUpdateShouldUseTopRedirection = () => {
+  let setShouldUseTopRedirection = Recoil.useSetRecoilState(RecoilAtoms.shouldUseTopRedirectionAtom)
+  let updateTopRedirectionAtom = paymentOptions => {
+    paymentOptions
+    ->Dict.get("shouldUseTopRedirection")
+    ->Option.flatMap(JSON.Decode.bool)
+    ->Option.map(useTop => {
+      setShouldUseTopRedirection(_ => useTop)
+    })
+    ->ignore
+  }
+  updateTopRedirectionAtom
+}
diff --git a/src/LoaderController.res b/src/LoaderController.res
@@ -9,7 +9,6 @@ let make = (~children, ~paymentMode, ~setIntegrateErrorError, ~logger, ~initTime
   let (_, setSessions) = Recoil.useRecoilState(sessions)
   let (options, setOptions) = Recoil.useRecoilState(elementOptions)
   let (optionsPayment, setOptionsPayment) = Recoil.useRecoilState(optionAtom)
-  let setShouldUseTopRedirection = Recoil.useSetRecoilState(shouldUseTopRedirectionAtom)
   let setSessionId = Recoil.useSetRecoilState(sessionId)
   let setBlockConfirm = Recoil.useSetRecoilState(isConfirmBlocked)
   let setCustomPodUri = Recoil.useSetRecoilState(customPodUri)
@@ -147,14 +146,7 @@ let make = (~children, ~paymentMode, ~setIntegrateErrorError, ~logger, ~initTime
     }
   }
 
-  let updateTopRedirectionAtom = paymentOptions =>
-    paymentOptions
-    ->Dict.get("shouldUseTopRedirection")
-    ->Option.flatMap(JSON.Decode.bool)
-    ->Option.map(useTop => {
-      setShouldUseTopRedirection(_ => useTop)
-    })
-    ->ignore
+  let updateShouldUseTopRedirection = UtilityHooks.useUpdateShouldUseTopRedirection()
 
   React.useEffect0(() => {
     messageParentWindow([("iframeMounted", true->JSON.Encode.bool)])
@@ -286,7 +278,7 @@ let make = (~children, ~paymentMode, ~setIntegrateErrorError, ~logger, ~initTime
                 logger.setClientSecret(clientSecret)
 
                 // Update top redirection atom
-                updateTopRedirectionAtom(paymentOptions)
+                updateShouldUseTopRedirection(paymentOptions)
 
                 switch getThemePromise(paymentOptions) {
                 | Some(promise) =>
@@ -336,7 +328,7 @@ let make = (~children, ~paymentMode, ~setIntegrateErrorError, ~logger, ~initTime
             logger.setClientSecret(clientSecret)
 
             // Update top redirection atom
-            updateTopRedirectionAtom(paymentOptions)
+            updateShouldUseTopRedirection(paymentOptions)
 
             switch getThemePromise(paymentOptions) {
             | Some(promise) =>