update balance after upgrade nft

zkLinkProtocol · Jul 30, 2024 · c20c62c · c20c62c
1 parent cf411b8
commit c20c62c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/src/components/DashboardS2/DailyRoulette/DailyDrawModal.tsx b/src/components/DashboardS2/DailyRoulette/DailyDrawModal.tsx
@@ -74,7 +74,7 @@ const DailyDrawModal: React.FC<IProps> = (props: IProps) => {
   const [mintStatus, setMintStatus] = useState<MintStatus | undefined>();
   const [failMessage, setFailMessage] = useState("");
   const { switchChain } = useSwitchChain();
-  const { updateFactory } = useUpdateNftBalanceStore();
+  const { updateFactor } = useUpdateNftBalanceStore();
 
   const { sendTrademarkMintTx } = useNovaNFT();
   const { modalInstance, onDrawed, remain } = props;
@@ -159,7 +159,7 @@ const DailyDrawModal: React.FC<IProps> = (props: IProps) => {
       mintResultModal.onOpen();
       onDrawed(); // update remain times after mint tx
       setMintParams(undefined);
-      updateFactory();
+      updateFactor();
     } catch (e) {
       console.log(e);
       setMintStatus(MintStatus.Failed);
@@ -178,7 +178,7 @@ const DailyDrawModal: React.FC<IProps> = (props: IProps) => {
     mintResultModal,
     onDrawed,
     sendTrademarkMintTx,
-    updateFactory,
+    updateFactor,
   ]);
 
   const handleSkip = async () => {

diff --git a/src/components/DashboardS2/Tabs/PortfolioComponents/LynksNFT.tsx b/src/components/DashboardS2/Tabs/PortfolioComponents/LynksNFT.tsx
@@ -8,17 +8,15 @@ import { Abi } from "viem";
 import SbtUpgradeModal from "@/components/Dashboard/NovaCharacterComponents/SbtUpgradeModal";
 import { useUpdateNftBalanceStore } from "@/hooks/useUpdateNftBalanceStore";
 
-
 export default function SbtNFT() {
   const { address } = useAccount();
   const [lynksBalance, setLynksBalance] = useState(0);
   const { publicClient, trademarkNFT } = useNovaNFT();
   const { getLynksNFT } = useNovaNFT();
   const { nft, loading: mintLoading, fetchLoading } = useSbtNft();
-  const [update, setUpdate] = useState(0);
   const upgradeModal = useDisclosure();
   const [upgradable, setUpgradable] = useState(false);
-  const { factor } = useUpdateNftBalanceStore();
+  const { factor, updateFactor } = useUpdateNftBalanceStore();
 
   useEffect(() => {
     (async () => {
@@ -49,7 +47,7 @@ export default function SbtNFT() {
         }
       }
     })();
-  }, [address, getLynksNFT, publicClient, trademarkNFT, update,factor]);
+  }, [address, getLynksNFT, publicClient, trademarkNFT, factor]);
 
   useEffect(() => {
     (async () => {
@@ -65,7 +63,7 @@ export default function SbtNFT() {
         setLynksBalance(balance);
       }
     })();
-  }, [address, getLynksNFT]);
+  }, [address, getLynksNFT, factor]);
 
   const handleMintNow = useCallback(() => {
     if (fetchLoading) {
@@ -112,7 +110,9 @@ export default function SbtNFT() {
         nft={nft}
         mintLoading={mintLoading}
         upgradeModal={upgradeModal}
-        onUpgraded={() => setUpdate((v) => v + 1)}
+        onUpgraded={() => {
+          updateFactor();
+        }}
       />
     </NftContainer>
   );

diff --git a/src/hooks/useUpdateNftBalanceStore.ts b/src/hooks/useUpdateNftBalanceStore.ts
@@ -2,12 +2,12 @@ import { create } from "zustand";
 
 interface UpdateNftBalanceStore {
   factor: number;
-  updateFactory: () => void;
+  updateFactor: () => void;
 }
 
 export const useUpdateNftBalanceStore = create<UpdateNftBalanceStore>(
   (set) => ({
     factor: 0,
-    updateFactory: () => set((state) => ({ factor: state.factor + 1 })),
+    updateFactor: () => set((state) => ({ factor: state.factor + 1 })),
   })
 );