From 366809dd16cb1a3f8bc0aaceeb968a4476daae39 Mon Sep 17 00:00:00 2001
From: Nicho <nichosystem@protonmail.com>
Date: Tue, 28 Nov 2023 14:46:20 -0600
Subject: [PATCH] Limit geoblocked users to legacy withdraw page

---
 .../components/Layouts/CoreLayout/Account.tsx |  5 -----
 .../Layouts/V2Layout/Nav/WalletNav.tsx        | 12 +-----------
 .../components/Layouts/V2Layout/V2Account.tsx |  6 ------
 .../src/components/Layouts/V2Layout/index.tsx | 19 +++++++++++++++++--
 .../Core/DappPages/Legacy/ClaimFromVaults.tsx | 12 +++++++++++-
 apps/dapp/src/hooks/use-geo-blocked.tsx       |  5 ++++-
 6 files changed, 33 insertions(+), 26 deletions(-)

diff --git a/apps/dapp/src/components/Layouts/CoreLayout/Account.tsx b/apps/dapp/src/components/Layouts/CoreLayout/Account.tsx
index f88c5415c..e446963e6 100644
--- a/apps/dapp/src/components/Layouts/CoreLayout/Account.tsx
+++ b/apps/dapp/src/components/Layouts/CoreLayout/Account.tsx
@@ -21,16 +21,11 @@ export const Account = () => {
   const isSmallDesktop = useMediaQuery({
     query: queryVerySmallDesktop,
   });
-  const { isBlocked } = useGeoBlocked();
 
   if (connecting) {
     return <Loader />;
   }
 
-  if (isBlocked) {
-    return <ConnectButton disabled label="Restricted Jurisdiction" isSmall isActive isUppercase role="button" />;
-  }
-
   if (wallet) {
     const disconnectButton = (
       <ConnectButton
diff --git a/apps/dapp/src/components/Layouts/V2Layout/Nav/WalletNav.tsx b/apps/dapp/src/components/Layouts/V2Layout/Nav/WalletNav.tsx
index cfa2841b9..fabd287e6 100644
--- a/apps/dapp/src/components/Layouts/V2Layout/Nav/WalletNav.tsx
+++ b/apps/dapp/src/components/Layouts/V2Layout/Nav/WalletNav.tsx
@@ -15,7 +15,6 @@ import { useState } from 'react';
 import { useConnectWallet, useSetChain } from '@web3-onboard/react';
 import { useWallet } from 'providers/WalletProvider';
 import TruncatedAddress from 'components/TruncatedAddress';
-import { useGeoBlocked } from 'hooks/use-geo-blocked';
 
 type WalletNavProps = {
   isNavCollapsed: boolean;
@@ -26,7 +25,6 @@ export const WalletNav = (props: WalletNavProps) => {
   const { isNavCollapsed, onClickHandler } = props;
   const [{ wallet, connecting }, connect, disconnect] = useConnectWallet();
   const { walletAddress } = useWallet();
-  const { isBlocked } = useGeoBlocked();
 
   const [isWalletDropdownOpen, setIsWalletDropdownOpen] = useState(false);
   const [{ connectedChain }] = useSetChain();
@@ -46,8 +44,6 @@ export const WalletNav = (props: WalletNavProps) => {
   const explorerUrl = getChainExplorerURL(walletAddress || '', currentChainId);
 
   const connectWallet = async () => {
-    // TODO: What to do in this case?
-    if (isBlocked) return alert('Restricted Jurisdiction');
     if (onClickHandler) onClickHandler();
     await connect();
   };
@@ -60,7 +56,7 @@ export const WalletNav = (props: WalletNavProps) => {
   return (
     <WalletContainer onMouseLeave={() => setIsWalletDropdownOpen(false)}>
       {connecting && <Loader iconSize={24} />}
-      {!connecting && !isBlocked && wallet ? (
+      {!connecting && wallet ? (
         <FlexColumnContainer>
           <WalletConnectedContainer onClick={() => setIsWalletDropdownOpen(!isWalletDropdownOpen)}>
             <WalletIcon />
@@ -92,12 +88,6 @@ export const WalletNav = (props: WalletNavProps) => {
           {!isNavCollapsed && <NavLinkText style={{ cursor: 'pointer' }}>Connect</NavLinkText>}
         </div>
       )}
-      {!connecting && isBlocked && (
-        <>
-          <WalletIcon fill={theme.palette.grayOpaque} />
-          {!isNavCollapsed && <NavLinkText small>Blocked</NavLinkText>}
-        </>
-      )}
     </WalletContainer>
   );
 };
diff --git a/apps/dapp/src/components/Layouts/V2Layout/V2Account.tsx b/apps/dapp/src/components/Layouts/V2Layout/V2Account.tsx
index 83dd3e3fd..051863127 100644
--- a/apps/dapp/src/components/Layouts/V2Layout/V2Account.tsx
+++ b/apps/dapp/src/components/Layouts/V2Layout/V2Account.tsx
@@ -18,21 +18,15 @@ export const V2Account = () => {
   const { walletAddress } = useWallet();
   const [{ connectedChain }] = useSetChain();
   const currentChainId = useMemo(() => parseInt(connectedChain?.id || '', 16), [connectedChain]);
-  const { isBlocked } = useGeoBlocked();
 
   const isSmallDesktop = useMediaQuery({
     query: queryVerySmallDesktop,
   });
 
-
   if (connecting) {
     return <Loader />;
   }
 
-  if (isBlocked) {
-    return <ConnectButton disabled label="Restricted Jurisdiction" isSmall isActive isUppercase role="button" />;
-  }
-
   if (wallet) {
     const disconnectButton = (
       <ConnectButton
diff --git a/apps/dapp/src/components/Layouts/V2Layout/index.tsx b/apps/dapp/src/components/Layouts/V2Layout/index.tsx
index ffb2e6019..a4baf80ac 100644
--- a/apps/dapp/src/components/Layouts/V2Layout/index.tsx
+++ b/apps/dapp/src/components/Layouts/V2Layout/index.tsx
@@ -1,7 +1,7 @@
-import { FunctionComponent, SVGProps, useState } from 'react';
+import { FunctionComponent, SVGProps, useEffect, useState } from 'react';
 import * as breakpoints from 'styles/breakpoints';
 import { useMediaQuery } from 'react-responsive';
-import { Outlet, useLocation } from 'react-router-dom';
+import { Outlet, useLocation, useNavigate } from 'react-router-dom';
 
 import styled from 'styled-components';
 import { queryMinTablet } from 'styles/breakpoints';
@@ -14,6 +14,7 @@ import CurrencyExchange from 'assets/icons/currency_exchange.svg?react';
 import Payments from 'assets/icons/payments.svg?react';
 import Candle from 'assets/icons/candle.svg?react';
 import Restore from 'assets/icons/restore.svg?react';
+import { useGeoBlocked } from 'hooks/use-geo-blocked';
 
 export type MenuNavItem = {
   label: string;
@@ -37,6 +38,8 @@ const V2Layout = () => {
     query: queryMinTablet,
   });
   const loc = useLocation();
+  const navigate = useNavigate();
+  const { isBlocked, loading } = useGeoBlocked();
   const [menuNavItems, setMenuNavItems] = useState<Array<MenuNavItem>>([
     {
       label: 'Trade',
@@ -71,6 +74,18 @@ const V2Layout = () => {
     },
   ]);
 
+  // Handle geoblocked users
+  useEffect(() => {
+    // Define all permitted paths
+    const permittedPaths = ['/dapp/legacy'];
+    if (loading || (!loading && !isBlocked)) return;
+    // Force redirect to permitted path
+    if (!permittedPaths.find((path) => path === loc.pathname)) navigate(permittedPaths[0]);
+    // Remove nav items that are not permitted
+    const newMenuNavItems = menuNavItems.filter((menuItem) => permittedPaths.find((path) => path === menuItem.linkTo));
+    setMenuNavItems(newMenuNavItems);
+  }, [loading, isBlocked, loc]);
+
   const onSelectMenuNavItems = async (selectedMenuItem: MenuNavItem) => {
     await setMenuNavItems((prevSelectedMenuNavItems) => {
       const newSelectedMenuNavItems = prevSelectedMenuNavItems.map((prevMenuItem) => {
diff --git a/apps/dapp/src/components/Pages/Core/DappPages/Legacy/ClaimFromVaults.tsx b/apps/dapp/src/components/Pages/Core/DappPages/Legacy/ClaimFromVaults.tsx
index 1454c416d..1d0c0f14c 100644
--- a/apps/dapp/src/components/Pages/Core/DappPages/Legacy/ClaimFromVaults.tsx
+++ b/apps/dapp/src/components/Pages/Core/DappPages/Legacy/ClaimFromVaults.tsx
@@ -10,6 +10,8 @@ import { VaultButton } from '../../VaultPages/VaultContent';
 import { useTokenContractAllowance } from 'hooks/core/use-token-contract-allowance';
 import env from 'constants/env';
 import _ from 'lodash';
+import { useConnectWallet } from '@web3-onboard/react';
+import { TradeButton } from '../../NewUI/Home';
 
 const EMPTY_CLAIM_STATE = {
   claimSubvaultAddress: '',
@@ -17,6 +19,7 @@ const EMPTY_CLAIM_STATE = {
 };
 
 export const ClaimFromVaults = () => {
+  const [{ wallet }, connect] = useConnectWallet();
   const {
     balances: { balances, isLoading: balancesIsLoading },
     vaultGroups: { vaultGroups, isLoading: vaultGroupsIsLoading },
@@ -103,7 +106,14 @@ export const ClaimFromVaults = () => {
       )}
 
       <ButtonContainer>
-        {Number(claimState.claimAmount) === 0 ? (
+        {!wallet ? (
+          <TradeButton
+            label={`Connect`}
+            onClick={() => {
+              connect();
+            }}
+          />
+        ) : Number(claimState.claimAmount) === 0 ? (
           <ClaimButton
             label={`Nothing to Claim`}
             disabled={true}
diff --git a/apps/dapp/src/hooks/use-geo-blocked.tsx b/apps/dapp/src/hooks/use-geo-blocked.tsx
index bf79adf14..4ce1e802f 100644
--- a/apps/dapp/src/hooks/use-geo-blocked.tsx
+++ b/apps/dapp/src/hooks/use-geo-blocked.tsx
@@ -2,7 +2,8 @@ import { useEffect, useState } from 'react';
 
 export const useGeoBlocked = () => {
   const [isBlocked, setIsBlocked] = useState(false);
-  
+  const [loading, setLoading] = useState(true);
+
   useEffect(() => {
     const checkBlocked = async () => {
       const blocked = await fetch(`${window.location.href}api/geoblock`)
@@ -10,11 +11,13 @@ export const useGeoBlocked = () => {
         .then((res) => res.blocked)
         .catch(() => false);
       setIsBlocked(blocked);
+      setLoading(false);
     };
     checkBlocked();
   }, []);
 
   return {
     isBlocked,
+    loading,
   };
 };