From 5a6b8a4d4430d25c8cc47ba2d9c4ee36e7208ae0 Mon Sep 17 00:00:00 2001
From: JayJay1024 <lygm1062@gmail.com>
Date: Wed, 13 Mar 2024 11:58:09 +0800
Subject: [PATCH] Fix deposit id type

---
 src/components/bond-more-deposit-modal.tsx |  13 +-
 src/components/bond-more-kton-modal.tsx    |  11 +-
 src/components/bond-more-ring-modal.tsx    |  13 +-
 src/components/records-bonded-tokens.tsx   |  12 +-
 src/components/unbond-all-staked.tsx       |   8 +-
 src/components/unbond-deposit-modal.tsx    |  13 +-
 src/components/unbond-kton-modal.tsx       |  11 +-
 src/components/unbond-ring-modal.tsx       |  13 +-
 src/config/abi/staking-v2.json             | 167 +++++++++++++++++++++
 9 files changed, 233 insertions(+), 28 deletions(-)
 create mode 100644 src/config/abi/staking-v2.json

diff --git a/src/components/bond-more-deposit-modal.tsx b/src/components/bond-more-deposit-modal.tsx
index f1ae3a3..da7cf19 100644
--- a/src/components/bond-more-deposit-modal.tsx
+++ b/src/components/bond-more-deposit-modal.tsx
@@ -6,6 +6,7 @@ import { ExtraPower } from "./balance-input";
 import { useApp, useStaking } from "@/hooks";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function BondMoreDepositModal({
   commission,
@@ -35,17 +36,21 @@ export default function BondMoreDepositModal({
   );
 
   const availableDeposits = deposits.filter(({ id }) => !stakedDeposits.includes(id));
-  const { nativeToken, contract, explorer } = getChainConfig(activeChain);
+  const { nativeToken } = getChainConfig(activeChain);
 
   const handleBond = useCallback(async () => {
     setBusy(true);
+    const { contract, explorer } = getChainConfig(activeChain);
 
     try {
-      const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+      const abi =
+        activeChain === ChainID.CRAB
+          ? (await import("@/config/abi/staking-v2.json")).default
+          : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
       const { hash } = await writeContract({
         address: contract.staking.address,
-        abi: contractAbi,
+        abi,
         functionName: "stake",
         args: [0n, 0n, checkedDeposits],
       });
@@ -62,7 +67,7 @@ export default function BondMoreDepositModal({
     }
 
     setBusy(false);
-  }, [explorer, contract.staking, checkedDeposits, onClose]);
+  }, [activeChain, checkedDeposits, onClose]);
 
   return (
     <Modal
diff --git a/src/components/bond-more-kton-modal.tsx b/src/components/bond-more-kton-modal.tsx
index 98c91f7..ef7f894 100644
--- a/src/components/bond-more-kton-modal.tsx
+++ b/src/components/bond-more-kton-modal.tsx
@@ -5,6 +5,7 @@ import { useAccount, useBalance } from "wagmi";
 import { useCallback, useState } from "react";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function BondMoreKtonModal({
   commission,
@@ -30,13 +31,17 @@ export default function BondMoreKtonModal({
       notification.warn({ description: "Your balance is insufficient." });
     } else {
       setBusy(true);
+      const { contract, explorer } = getChainConfig(activeChain);
 
       try {
-        const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const abi =
+          activeChain === ChainID.CRAB
+            ? (await import("@/config/abi/staking-v2.json")).default
+            : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
         const { hash } = await writeContract({
           address: contract.staking.address,
-          abi: contractAbi,
+          abi,
           functionName: "stake",
           args: [0n, inputAmount, []],
         });
@@ -54,7 +59,7 @@ export default function BondMoreKtonModal({
 
       setBusy(false);
     }
-  }, [contract.staking, explorer, inputAmount, ktonBalance?.value, onClose]);
+  }, [activeChain, inputAmount, ktonBalance?.value, onClose]);
 
   return (
     ktonToken && (
diff --git a/src/components/bond-more-ring-modal.tsx b/src/components/bond-more-ring-modal.tsx
index 1d5f1c8..f891b0f 100644
--- a/src/components/bond-more-ring-modal.tsx
+++ b/src/components/bond-more-ring-modal.tsx
@@ -5,6 +5,7 @@ import { useApp, useStaking } from "@/hooks";
 import { useCallback, useState } from "react";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function BondMoreRingModal({
   commission,
@@ -23,20 +24,24 @@ export default function BondMoreRingModal({
   const [inputAmount, setInputAmount] = useState(0n);
   const [busy, setBusy] = useState(false);
 
-  const { nativeToken, contract, explorer } = getChainConfig(activeChain);
+  const { nativeToken } = getChainConfig(activeChain);
 
   const handleBond = useCallback(async () => {
     if ((ringBalance?.value || 0n) < inputAmount) {
       notification.warn({ description: "Your balance is insufficient." });
     } else {
       setBusy(true);
+      const { contract, explorer } = getChainConfig(activeChain);
 
       try {
-        const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const abi =
+          activeChain === ChainID.CRAB
+            ? (await import("@/config/abi/staking-v2.json")).default
+            : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
         const { hash } = await writeContract({
           address: contract.staking.address,
-          abi: contractAbi,
+          abi,
           functionName: "stake",
           args: [inputAmount, 0n, []],
         });
@@ -53,7 +58,7 @@ export default function BondMoreRingModal({
 
       setBusy(false);
     }
-  }, [contract.staking, explorer, inputAmount, ringBalance?.value, onClose]);
+  }, [activeChain, inputAmount, ringBalance?.value, onClose]);
 
   return (
     <BondMoreTokenModal
diff --git a/src/components/records-bonded-tokens.tsx b/src/components/records-bonded-tokens.tsx
index e30bfb9..e2e6df6 100644
--- a/src/components/records-bonded-tokens.tsx
+++ b/src/components/records-bonded-tokens.tsx
@@ -1,5 +1,5 @@
 import { useApp } from "@/hooks";
-import { StakingRecordsDataSource } from "@/types";
+import { ChainID, StakingRecordsDataSource } from "@/types";
 import { formatBlanace, getChainConfig, notifyTransaction } from "@/utils";
 import UnbondingTokenTooltip from "./unbonding-token-tooltip";
 import UnbondingDepositTooltip from "./unbonding-deposit-tooltip";
@@ -35,13 +35,17 @@ export default function RecordsBondedTokens({ row }: { row: StakingRecordsDataSo
       const { contract, explorer } = getChainConfig(activeChain);
 
       try {
-        const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const abi =
+          activeChain === ChainID.CRAB
+            ? (await import("@/config/abi/staking-v2.json")).default
+            : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const args = activeChain === ChainID.CRAB ? [ring, depositIds] : [ring, kton, depositIds];
 
         const { hash } = await writeContract({
           address: contract.staking.address,
-          abi: contractAbi,
+          abi,
           functionName: "restake",
-          args: [ring, kton, depositIds],
+          args,
         });
         const receipt = await waitForTransaction({ hash });
 
diff --git a/src/components/unbond-all-staked.tsx b/src/components/unbond-all-staked.tsx
index 742ec90..1797f2e 100644
--- a/src/components/unbond-all-staked.tsx
+++ b/src/components/unbond-all-staked.tsx
@@ -4,6 +4,7 @@ import { useCallback, useState } from "react";
 import { writeContract, waitForTransaction } from "@wagmi/core";
 import { notification } from "./notification";
 import { getChainConfig, notifyTransaction } from "@/utils";
+import { ChainID } from "@/types";
 
 export default function UnbondAllStaked() {
   const { stakedRing, stakedKton, stakedDeposits } = useStaking();
@@ -15,11 +16,14 @@ export default function UnbondAllStaked() {
     setBusy(true);
 
     try {
-      const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+      const abi =
+        activeChain === ChainID.CRAB
+          ? (await import("@/config/abi/staking-v2.json")).default
+          : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
       const { hash } = await writeContract({
         address: contract.staking.address,
-        abi: contractAbi,
+        abi,
         functionName: "unstake",
         args: [stakedRing, stakedKton, stakedDeposits],
       });
diff --git a/src/components/unbond-deposit-modal.tsx b/src/components/unbond-deposit-modal.tsx
index 93e8fa2..2d08852 100644
--- a/src/components/unbond-deposit-modal.tsx
+++ b/src/components/unbond-deposit-modal.tsx
@@ -6,6 +6,7 @@ import { ExtraPower } from "./balance-input";
 import { useApp, useStaking } from "@/hooks";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function UnbondDepositModal({
   commission,
@@ -35,17 +36,21 @@ export default function UnbondDepositModal({
   );
 
   const availableDeposits = deposits.filter(({ id }) => stakedDeposits.includes(id));
-  const { nativeToken, contract, explorer } = getChainConfig(activeChain);
+  const { nativeToken } = getChainConfig(activeChain);
 
   const handleUnbond = useCallback(async () => {
     setBusy(true);
+    const { contract, explorer } = getChainConfig(activeChain);
 
     try {
-      const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+      const abi =
+        activeChain === ChainID.CRAB
+          ? (await import("@/config/abi/staking-v2.json")).default
+          : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
       const { hash } = await writeContract({
         address: contract.staking.address,
-        abi: contractAbi,
+        abi,
         functionName: "unstake",
         args: [0n, 0n, checkedDeposits],
       });
@@ -62,7 +67,7 @@ export default function UnbondDepositModal({
     }
 
     setBusy(false);
-  }, [explorer, contract.staking, checkedDeposits, onClose]);
+  }, [activeChain, checkedDeposits, onClose]);
 
   return (
     <Modal
diff --git a/src/components/unbond-kton-modal.tsx b/src/components/unbond-kton-modal.tsx
index 8fb15a9..5cd43f5 100644
--- a/src/components/unbond-kton-modal.tsx
+++ b/src/components/unbond-kton-modal.tsx
@@ -4,6 +4,7 @@ import { useApp, useStaking } from "@/hooks";
 import { useCallback, useState } from "react";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function UnbondKtonModal({
   commission,
@@ -32,13 +33,17 @@ export default function UnbondKtonModal({
       });
     } else {
       setBusy(true);
+      const { contract, explorer } = getChainConfig(activeChain);
 
       try {
-        const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const abi =
+          activeChain === ChainID.CRAB
+            ? (await import("@/config/abi/staking-v2.json")).default
+            : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
         const { hash } = await writeContract({
           address: contract.staking.address,
-          abi: contractAbi,
+          abi,
           functionName: "unstake",
           args: [0n, inputAmount, []],
         });
@@ -56,7 +61,7 @@ export default function UnbondKtonModal({
 
       setBusy(false);
     }
-  }, [explorer, contract.staking, stakedKton, inputAmount, ktonToken, onClose]);
+  }, [activeChain, stakedKton, inputAmount, ktonToken, onClose]);
 
   return (
     ktonToken && (
diff --git a/src/components/unbond-ring-modal.tsx b/src/components/unbond-ring-modal.tsx
index 80c009c..9ecde7a 100644
--- a/src/components/unbond-ring-modal.tsx
+++ b/src/components/unbond-ring-modal.tsx
@@ -4,6 +4,7 @@ import { useApp, useStaking } from "@/hooks";
 import { useCallback, useState } from "react";
 import { notification } from "./notification";
 import { writeContract, waitForTransaction } from "@wagmi/core";
+import { ChainID } from "@/types";
 
 export default function UnbondRingModal({
   commission,
@@ -20,7 +21,7 @@ export default function UnbondRingModal({
   const [inputAmount, setInputAmount] = useState(0n);
   const [busy, setBusy] = useState(false);
 
-  const { nativeToken, contract, explorer } = getChainConfig(activeChain);
+  const { nativeToken } = getChainConfig(activeChain);
 
   const handleUnbond = useCallback(async () => {
     if (stakedRing < inputAmount) {
@@ -32,13 +33,17 @@ export default function UnbondRingModal({
       });
     } else {
       setBusy(true);
+      const { contract, explorer } = getChainConfig(activeChain);
 
       try {
-        const contractAbi = (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
+        const abi =
+          activeChain === ChainID.CRAB
+            ? (await import("@/config/abi/staking-v2.json")).default
+            : (await import(`@/config/abi/${contract.staking.abiFile}`)).default;
 
         const { hash } = await writeContract({
           address: contract.staking.address,
-          abi: contractAbi,
+          abi,
           functionName: "unstake",
           args: [inputAmount, 0n, []],
         });
@@ -56,7 +61,7 @@ export default function UnbondRingModal({
 
       setBusy(false);
     }
-  }, [explorer, contract.staking, stakedRing, inputAmount, nativeToken, onClose]);
+  }, [activeChain, stakedRing, inputAmount, nativeToken, onClose]);
 
   return (
     <UnbondTokenModal
diff --git a/src/config/abi/staking-v2.json b/src/config/abi/staking-v2.json
new file mode 100644
index 0000000..9dbec17
--- /dev/null
+++ b/src/config/abi/staking-v2.json
@@ -0,0 +1,167 @@
+[
+  {
+    "inputs": [],
+    "name": "chill",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [],
+    "name": "claim",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "uint32",
+        "name": "commission",
+        "type": "uint32"
+      }
+    ],
+    "name": "collect",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "target",
+        "type": "address"
+      }
+    ],
+    "name": "nominate",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "who",
+        "type": "address"
+      }
+    ],
+    "name": "payout",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "ringAmount",
+        "type": "uint256"
+      },
+      {
+        "internalType": "uint16[]",
+        "name": "depositIds",
+        "type": "uint16[]"
+      }
+    ],
+    "name": "restake",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "ringAmount",
+        "type": "uint256"
+      },
+      {
+        "internalType": "uint256",
+        "name": "ktonAmount",
+        "type": "uint256"
+      },
+      {
+        "internalType": "uint16[]",
+        "name": "depositIds",
+        "type": "uint16[]"
+      }
+    ],
+    "name": "stake",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "ringAmount",
+        "type": "uint256"
+      },
+      {
+        "internalType": "uint256",
+        "name": "ktonAmount",
+        "type": "uint256"
+      },
+      {
+        "internalType": "uint16[]",
+        "name": "depositIds",
+        "type": "uint16[]"
+      }
+    ],
+    "name": "unstake",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  }
+]