chore: feature-flag for geo-block feature (cowprotocol#3940)

xyzseer · Feb 28, 2024 · fe9501e · fe9501e
1 parent bce2f42
commit fe9501e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/apps/cowswap-frontend/src/modules/application/containers/App/Updaters.tsx b/apps/cowswap-frontend/src/modules/application/containers/App/Updaters.tsx
@@ -10,6 +10,7 @@ import { EthFlowDeadlineUpdater, EthFlowSlippageUpdater } from 'modules/swap/sta
 import { useOnTokenListAddingError } from 'modules/tokensList'
 import { UsdPricesUpdater } from 'modules/usdAmount'
 
+import { useFeatureFlags } from 'common/hooks/featureFlags/useFeatureFlags'
 import { TotalSurplusUpdater } from 'common/state/totalSurplusState'
 import { ApplicationUpdater } from 'common/updaters/ApplicationUpdater'
 import { CancelReplaceTxUpdater } from 'common/updaters/CancelReplaceTxUpdater'
@@ -34,6 +35,7 @@ export function Updaters() {
   const { chainId, account } = useWalletInfo()
   const { tokenLists, appCode } = useInjectedWidgetParams()
   const onTokenListAddingError = useOnTokenListAddingError()
+  const { isGeoBlockEnabled } = useFeatureFlags()
 
   return (
     <>
@@ -62,7 +64,7 @@ export function Updaters() {
       <CowEventsUpdater />
       <TotalSurplusUpdater />
       <UsdPricesUpdater />
-      <TokensListsUpdater chainId={chainId} />
+      <TokensListsUpdater chainId={chainId} isGeoBlockEnabled={isGeoBlockEnabled} />
       <WidgetTokensListsUpdater
         tokenLists={tokenLists}
         appCode={appCode}

diff --git a/libs/tokens/src/updaters/TokensListsUpdater/index.ts b/libs/tokens/src/updaters/TokensListsUpdater/index.ts
@@ -32,9 +32,10 @@ const NETWORKS_WITHOUT_RESTRICTIONS = [SupportedChainId.SEPOLIA, SupportedChainI
 
 interface TokensListsUpdaterProps {
   chainId: SupportedChainId
+  isGeoBlockEnabled: boolean
 }
 
-export function TokensListsUpdater({ chainId: currentChainId }: TokensListsUpdaterProps) {
+export function TokensListsUpdater({ chainId: currentChainId, isGeoBlockEnabled }: TokensListsUpdaterProps) {
   const { chainId } = useAtomValue(environmentAtom)
   const setEnvironment = useSetAtom(updateEnvironmentAtom)
   const allTokensLists = useAtomValue(allListsSourcesAtom)
@@ -72,7 +73,7 @@ export function TokensListsUpdater({ chainId: currentChainId }: TokensListsUpdat
 
   // Check if a user is from US and use Uniswap list, because of the SEC regulations
   useEffect(() => {
-    if (isInjectedWidget()) return
+    if (!isGeoBlockEnabled || isInjectedWidget()) return
 
     if (NETWORKS_WITHOUT_RESTRICTIONS.includes(chainId)) {
       setEnvironment({ useCuratedListOnly: false })
@@ -101,7 +102,7 @@ export function TokensListsUpdater({ chainId: currentChainId }: TokensListsUpdat
         })
       })
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [chainId])
+  }, [chainId, isGeoBlockEnabled])
 
   return null
 }