Merge pull request #24 from The-Poolz/issue-23

Issue #23
The-Poolz · Oct 23, 2023 · 4b3b787 · 4b3b787
2 parents 7e9ffd3 + 11ec25f
commit 4b3b787
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/src/contracts/configs/BinanceTest.config.ts b/src/contracts/configs/BinanceTest.config.ts
@@ -13,5 +13,9 @@ export const BinanceTestChainConfig: TChainConfig = {
   delayVault: {
     address: "0x607155A953d5f598d2F7CcD9a6395Af389cfecE5",
     nameVersion: "[email protected]"
+  },
+  lockDealNFT: {
+    address: "0xD40e523BCb4230FFA1126E301f4CA0294B8CF180",
+    nameVersion: "[email protected]"
   }
 }
diff --git a/src/index.ts b/src/index.ts
@@ -8,3 +8,4 @@ export { ThePoolzProvider, useThePoolz, useConnectWallet, useSiwe, useSidNameFor
 export type { IThePoolzInterface, IERC20Info } from "./types/IThePoolzInterface"
 export * from "./utils"
 export * from "./constants"
+export type { Contract } from "web3-eth-contract"
diff --git a/src/poolz/ThePoolz.ts b/src/poolz/ThePoolz.ts
@@ -29,6 +29,7 @@ class ThePoolz implements IThePoolzInterface {
   public poolzBackWithdrawContract: IThePoolzInterface["poolzBackWithdrawContract"]
   public lockedDealV2: IThePoolzInterface["lockedDealV2"]
   public delayVaultContract: IThePoolzInterface["delayVaultContract"]
+  public lockDealNFTContract: IThePoolzInterface["lockDealNFTContract"]
 
   #provider: typeof Web3.givenProvider
   #contracts = new Map<string, Contract>()
@@ -70,7 +71,8 @@ class ThePoolz implements IThePoolzInterface {
   private async initPoolzContracts() {
     const chainConfig = AVAILABLE_CHAINS.get(this.chainId)
     if (!chainConfig) return
-    const { poolzTokenAddress, poolzAddress, whiteListAddress, lockedDealV2, poolzBackWithdraw, signUpAddress, delayVault } = chainConfig
+    const { poolzTokenAddress, poolzAddress, whiteListAddress, lockedDealV2, poolzBackWithdraw, signUpAddress, delayVault, lockDealNFT } =
+      chainConfig
 
     this.poolzTokenAddress = poolzTokenAddress
 
@@ -125,6 +127,12 @@ class ThePoolz implements IThePoolzInterface {
         this.delayVaultContract = { ...delayVault, contract: new this.web3.eth.Contract(abi as AbiItem[], delayVault.address) }
       } catch (e) {}
     }
+    if (lockDealNFT) {
+      try {
+        const abi = await this.fetchContractAbi(lockDealNFT.nameVersion)
+        this.lockDealNFTContract = { ...lockDealNFT, contract: new this.web3.eth.Contract(abi as AbiItem[]) }
+      } catch (e) {}
+    }
   }
 
   async getChaincoinInfo(chainId?: typeof this.chainId) {

diff --git a/src/types/IThePoolzInterface.ts b/src/types/IThePoolzInterface.ts
@@ -62,6 +62,7 @@ export interface IThePoolzInterface {
 
   poolzTokenAddress?: string
   delayVaultContract?: IContractInfo
+  lockDealNFTContract?: IContractInfo
 
   getChaincoinInfo(k?: number): Promise<IChainInfo | undefined>
   ERC20Balance(token: string, account: string): Promise<string>
@@ -106,6 +107,7 @@ export interface IChainConfig {
 
   poolzTokenAddress: IThePoolzInterface["poolzTokenAddress"]
   delayVault: Omit<NonNullable<IThePoolzInterface["delayVaultContract"]>, "contract">
+  lockDealNFT: Omit<NonNullable<IThePoolzInterface["lockDealNFTContract"]>, "contract">
 }
 
 export type TChainConfig = Partial<IChainConfig>