Al-amin/handle_website_status_cookie_on_VPN_change (#2593)

alamin-deriv · web-flow · commit 0dcbcf2284b0 · 2022-03-10T13:05:50.000+08:00
* handle_website_status_cookie_on_VPN_change

* revert changes

* revert back my old changes

* revert back my old changes

* merged master
diff --git a/src/components/hooks/use-website-status.js b/src/components/hooks/use-website-status.js
@@ -1,4 +1,4 @@
-import { useState, useEffect } from 'react'
+import { useState, useLayoutEffect } from 'react'
 import { useCookieState } from './use-cookie-state'
 import { BinarySocketBase } from 'common/websocket/socket_base'
 import { getDateFromToday } from 'common/utility'
@@ -10,13 +10,13 @@ export const useWebsiteStatus = () => {
     const [website_status, setWebsiteStatus] = useCookieState(null, WEBSITE_STATUS_COUNTRY_KEY, {
         expires: getDateFromToday(COOKIE_EXPIRY_DAYS),
     })
+
     const [is_loading, setLoading] = useState(true)
 
-    useEffect(() => {
+    useLayoutEffect(() => {
         setLoading(true)
         if (!website_status) {
             const binary_socket = BinarySocketBase.init()
-
             binary_socket.onopen = () => {
                 binary_socket.send(JSON.stringify({ website_status: 1 }))
             }
@@ -26,13 +26,30 @@ export const useWebsiteStatus = () => {
 
                 if (!response.error) {
                     const { clients_country, crypto_config } = response.website_status
+
                     setWebsiteStatus({ clients_country, crypto_config })
                 }
                 setLoading(false)
                 binary_socket.close()
             }
         } else {
-            setLoading(false)
+            const binary_socket = BinarySocketBase.init()
+            binary_socket.onopen = () => {
+                binary_socket.send(JSON.stringify({ website_status: 1 }))
+            }
+
+            binary_socket.onmessage = (msg) => {
+                const response = JSON.parse(msg.data)
+
+                if (!response.error) {
+                    const { clients_country, crypto_config } = response.website_status
+                    if (clients_country !== website_status.clients_country) {
+                        setWebsiteStatus({ clients_country, crypto_config })
+                    }
+                }
+                binary_socket.close()
+                setLoading(false)
+            }
         }
     }, [website_status])
 
