From b85d2d081f9c140a56639207566dcb1c87502eae Mon Sep 17 00:00:00 2001
From: Daniel Dimitrov <daniel@compojoom.com>
Date: Thu, 19 Sep 2024 10:08:34 +0200
Subject: [PATCH] feat: add ofac blocking for the stake page

---
 .../common/BlockedAddress/index.tsx           |  4 +--
 .../stake/components/StakePage/index.tsx      | 25 +++++++++++++++++++
 src/features/swap/index.tsx                   |  2 +-
 3 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/src/components/common/BlockedAddress/index.tsx b/src/components/common/BlockedAddress/index.tsx
index 4e2dc6d8ee..3d271d4228 100644
--- a/src/components/common/BlockedAddress/index.tsx
+++ b/src/components/common/BlockedAddress/index.tsx
@@ -5,7 +5,7 @@ import { useRouter } from 'next/router'
 import Disclaimer from '@/components/common/Disclaimer'
 import { AppRoutes } from '@/config/routes'
 
-export const BlockedAddress = ({ address }: { address?: string }): ReactElement => {
+export const BlockedAddress = ({ address, featureTitle }: { address: string; featureTitle: string }): ReactElement => {
   const theme = useTheme()
   const isMobile = useMediaQuery(theme.breakpoints.down('sm'))
   const displayAddress = address && isMobile ? shortenAddress(address) : address
@@ -19,7 +19,7 @@ export const BlockedAddress = ({ address }: { address?: string }): ReactElement
     <Disclaimer
       title="Blocked address"
       subtitle={displayAddress}
-      content="The above address is part of the OFAC SDN list and the embedded swaps feature with CoW Swap is unavailable for sanctioned addresses."
+      content={`The above address is part of the OFAC SDN list and the ${featureTitle} is unavailable for sanctioned addresses.`}
       onAccept={handleAccept}
     />
   )
diff --git a/src/features/stake/components/StakePage/index.tsx b/src/features/stake/components/StakePage/index.tsx
index 7ab3382e1e..59243e5c93 100644
--- a/src/features/stake/components/StakePage/index.tsx
+++ b/src/features/stake/components/StakePage/index.tsx
@@ -3,9 +3,34 @@ import Disclaimer from '@/components/common/Disclaimer'
 import WidgetDisclaimer from '@/components/common/WidgetDisclaimer'
 import useStakeConsent from '@/features/stake/useStakeConsent'
 import StakingWidget from '../StakingWidget'
+import { useGetIsSanctionedQuery } from '@/store/ofac'
+import { skipToken } from '@reduxjs/toolkit/query/react'
+import useWallet from '@/hooks/wallets/useWallet'
+import useSafeInfo from '@/hooks/useSafeInfo'
+import { getKeyWithTrueValue } from '@/utils/helpers'
+import BlockedAddress from '@/components/common/BlockedAddress'
 
 const StakePage = () => {
   const { isConsentAccepted, onAccept } = useStakeConsent()
+  const { safeAddress, safeLoading } = useSafeInfo()
+  const wallet = useWallet()
+
+  const { data: isSafeAddressBlocked } = useGetIsSanctionedQuery(safeAddress || skipToken)
+  const { data: isWalletAddressBlocked } = useGetIsSanctionedQuery(wallet?.address || skipToken)
+  const blockedAddresses = {
+    [safeAddress]: !!isSafeAddressBlocked,
+    [wallet?.address || '']: !!isWalletAddressBlocked,
+  }
+
+  const blockedAddress = getKeyWithTrueValue(blockedAddresses)
+
+  if (blockedAddress) {
+    return (
+      <Stack direction="column" alignItems="center" justifyContent="center" flex={1}>
+        <BlockedAddress address={blockedAddress} featureTitle="stake feature with Kiln" />
+      </Stack>
+    )
+  }
 
   return (
     <>
diff --git a/src/features/swap/index.tsx b/src/features/swap/index.tsx
index b2ed794996..9ac20f238d 100644
--- a/src/features/swap/index.tsx
+++ b/src/features/swap/index.tsx
@@ -294,7 +294,7 @@ const SwapWidget = ({ sell }: Params) => {
   useCustomAppCommunicator(iframeRef, appData, chain)
 
   if (blockedAddress) {
-    return <BlockedAddress address={blockedAddress} />
+    return <BlockedAddress address={blockedAddress} featureTitle="embedded swaps feature with CoW Swap" />
   }
 
   if (!isConsentAccepted) {