Merge pull request #1341 from ethereum-push-notification-service/chor…

…e/read-only-feat-yf chore: protect yield farming fns in read only mode
push-protocol · Dec 18, 2023 · fa2bbdd · fa2bbdd
2 parents 3d00aee + 9c6b5f6
commit fa2bbdd
Show file tree

Hide file tree

Showing 2 changed files with 92 additions and 3 deletions.
diff --git a/src/components/yield/StakingModalComponent.tsx b/src/components/yield/StakingModalComponent.tsx
@@ -2,20 +2,21 @@ import { useClickAway } from 'react-use';
 import { Input, Span } from 'primaries/SharedStyling';
 
 // React + Web3 Essentials
-import React, { useState } from 'react';
+import React, { useContext, useState } from 'react';
 import { ethers } from 'ethers';
 
 // External Packages
 import styled, { useTheme } from 'styled-components';
 import { MdCheckCircle, MdError } from 'react-icons/md';
-
+import { useSelector } from 'react-redux';
 
 // Internal Compoonents
 import { ReactComponent as Close } from 'assets/chat/group-chat/close.svg';
 import LoaderSpinner, { LOADER_TYPE } from 'components/reusables/loaders/LoaderSpinner';
 import { bn, bnToInt, formatTokens } from 'helpers/StakingHelper';
 import { P } from 'components/SharedStyling';
 import { ButtonV2, H2V2, ItemHV2, ItemVV2, SpanV2 } from 'components/reusables/SharedStylingV2';
+import { AppContext } from 'contexts/AppContext';
 
 // Internal Configs
 import { abis, addresses } from 'config';
@@ -42,6 +43,11 @@ const StakingModalComponent = ({ onClose, InnerComponentProps, toastObject }) =>
 
     const [txnMessage, setTxnMessage] = useState(null);
 
+    const { userPushSDKInstance } = useSelector((state: any) => {
+        return state.user;
+    });
+    const { handleConnectWallet } = useContext(AppContext);
+
     const [depositAmount, setDepositAmount] = useState(0);
 
     const handleClose = () => {
@@ -88,6 +94,17 @@ const StakingModalComponent = ({ onClose, InnerComponentProps, toastObject }) =>
     }, [])
 
     const approveDeposit = async () => {
+        if(!userPushSDKInstance.signer) {
+            toastObject.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         if (depositApproved || txInProgressApprDep) {
             return
         }
@@ -158,6 +175,17 @@ const StakingModalComponent = ({ onClose, InnerComponentProps, toastObject }) =>
     }
 
     const depositAmountTokenFarmSingleTx = async () => {
+        if(!userPushSDKInstance.signer) {
+            toastObject.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         if (txInProgressDep || !depositApproved) {
             return
         }

diff --git a/src/components/yield/YieldPoolCard.tsx b/src/components/yield/YieldPoolCard.tsx
@@ -1,10 +1,11 @@
 // React + Web3 Essentials
-import React, { useEffect, useState } from 'react';
+import React, { useContext, useState } from 'react';
 import { ethers } from 'ethers';
 
 // External Packages
 import styled, { useTheme } from 'styled-components';
 import { MdCheckCircle, MdError, MdWarning } from 'react-icons/md';
+import { useSelector } from 'react-redux';
 
 // Internal Compoonents
 import useToast from 'hooks/useToast';
@@ -16,6 +17,7 @@ import { Button, Span } from 'primaries/SharedStyling';
 import LoaderSpinner, { LOADER_TYPE } from 'components/reusables/loaders/LoaderSpinner';
 import { formatTokens, numberWithCommas } from 'helpers/StakingHelper';
 import { ButtonV2, H2V2, ImageV2, ItemHV2, ItemVV2, SectionV2, Skeleton, SkeletonLine, SpanV2 } from 'components/reusables/SharedStylingV2';
+import { AppContext } from 'contexts/AppContext';
 
 // Internal Configs
 import { abis, addresses } from 'config';
@@ -41,13 +43,29 @@ const YieldPoolCard = ({
     const [unstakeErrorMessage, setUnstakeErrorMessage] = useState(null);
     const [withdrawErrorMessage, setWithdrawErrorMessage] = useState(null);
 
+    const { userPushSDKInstance } = useSelector((state: any) => {
+        return state.user;
+    });
+    const { handleConnectWallet } = useContext(AppContext);
+
     const [filled, setFilled] = useState(0);
 
     const yieldFarmToast = useToast();
 
     const theme = useTheme();
 
     const massClaimRewardsTokensAll = async () => {
+        if(!userPushSDKInstance.signer) {
+            yieldFarmToast.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         if (txInProgressClaimRewards) {
             return;
         }
@@ -126,6 +144,17 @@ const YieldPoolCard = ({
     };
 
     const withdrawTokens = async () => {
+        if(!userPushSDKInstance.signer) {
+            yieldFarmToast.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         if (txInProgressWithdraw) {
             return;
         }
@@ -197,6 +226,17 @@ const YieldPoolCard = ({
     };
 
     const migrateToNewPool = async () => {
+        if(!userPushSDKInstance.signer) {
+            yieldFarmToast.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         if (txInProgressMigrate) {
             return;
         }
@@ -421,6 +461,17 @@ const YieldPoolCard = ({
 
 
     const depositLpToken = async (tx, withdrawAmount, totalTxnSteps) => {
+        if(!userPushSDKInstance.signer) {
+            yieldFarmToast.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
+
         var signer = provider.getSigner(account);
         var stakingV2 = new ethers.Contract(addresses.stakingV2, abis.stakingV2, signer);
 
@@ -471,6 +522,16 @@ const YieldPoolCard = ({
     }
 
     const depositPushToken = async (tx, withdrawAmount, totalTxnSteps) => {
+        if(!userPushSDKInstance.signer) {
+            yieldFarmToast.showMessageToast({
+                toastTitle: 'Error',
+                toastMessage: `Please connect your wallet`,
+                toastType: 'ERROR',
+                getToastIcon: (size) => <MdError size={size} color="red" />,
+            });
+            handleConnectWallet();
+            return;
+        }
 
         var signer = provider.getSigner(account);
         let pushCoreV2 = new ethers.Contract(addresses.pushCoreV2, abis.pushCoreV2, signer);