From 6868a12294ed2fb6244485005c059324e37729fd Mon Sep 17 00:00:00 2001
From: samuelea <batonenbois@gmail.com>
Date: Thu, 5 Dec 2024 11:57:27 -0500
Subject: [PATCH] sale details hook

---
 .../src/hooks/useSaleContractCheckout.ts      | 96 ++++++++++++++++---
 1 file changed, 85 insertions(+), 11 deletions(-)

diff --git a/packages/checkout/src/hooks/useSaleContractCheckout.ts b/packages/checkout/src/hooks/useSaleContractCheckout.ts
index bb6f686d..8cae246d 100644
--- a/packages/checkout/src/hooks/useSaleContractCheckout.ts
+++ b/packages/checkout/src/hooks/useSaleContractCheckout.ts
@@ -8,11 +8,17 @@ import { ERC_1155_SALE_CONTRACT, ERC_721_SALE_CONTRACT } from '../constants/abi'
 import { Abi, Hex } from 'viem'
 import { useReadContract, useReadContracts } from 'wagmi'
 
-export interface UseSaleContractCheckoutArgs extends CheckoutOptionsSalesContractArgs {
+interface UseSaleContractCheckoutArgs extends CheckoutOptionsSalesContractArgs {
   chain: number | string,
   tokenType: 'ERC1155' | 'ERC721'
 }
 
+interface UseSaleContractCheckoutReturn {
+  data?: undefined,
+  isLoading: boolean,
+  isError: boolean
+}
+
 export const useSaleContractCheckout = ({
   chain,
   tokenType,
@@ -20,7 +26,7 @@ export const useSaleContractCheckout = ({
   wallet,
   collectionAddress,
   items,
-}: UseSaleContractCheckoutArgs) => {
+}: UseSaleContractCheckoutArgs): UseSaleContractCheckoutReturn => {
   const { data: checkoutOptions, isLoading: isLoadingCheckoutOptions, isError: isErrorCheckoutOtions } = useCheckoutOptionsSalesContract(chain, {
     contractAddress,
     wallet,
@@ -30,13 +36,19 @@ export const useSaleContractCheckout = ({
   const network = findSupportedNetwork(chain)
   const chainId = network?.chainId || 137
 
-  const data = useSaleContractConfig({ chainId, tokenType, contractAddress, tokenIds: items.map(i => i.tokenId) })
+  const { data: saleConfigData, isLoading: isLoadingSaleConfig, isError: isErrorSaleConfig } = useSaleContractConfig({ chainId, tokenType, contractAddress, tokenIds: items.map(i => i.tokenId) })
 
   // Get sale contract currency data...
   // Get token sale data
   // Pass wrapper to open select modal
-  const isLoading = isLoadingCheckoutOptions
-  const error = isErrorCheckoutOtions
+  const isLoading = isLoadingCheckoutOptions || isLoadingSaleConfig
+  const error = isErrorCheckoutOtions || isErrorSaleConfig
+
+  return ({
+    data: undefined,
+    isLoading,
+    isError: error
+  })
 }
 
 interface UseSaleContractConfigArgs {
@@ -46,12 +58,23 @@ interface UseSaleContractConfigArgs {
   tokenIds: string[]
 }
 
-interface UseSaleContractConfigReturn {
+interface SaleConfig {
+  tokenId: string
+  price: string
+}
+
+interface UseSaleContractConfigData {
   currencyAddress: string
-  prices: string[]
+  saleConfigs: SaleConfig[]
 }
 
-export const useSaleContractConfig = ({ chainId, tokenType, contractAddress, tokenIds, }: UseSaleContractConfigArgs) => {
+interface UseSaleContractConfigReturn {
+  data?: UseSaleContractConfigData,
+  isLoading: boolean,
+  isError: boolean
+}
+
+export const useSaleContractConfig = ({ chainId, tokenType, contractAddress, tokenIds, }: UseSaleContractConfigArgs): UseSaleContractConfigReturn => {
   const { data: paymentTokenERC1155, isLoading: isLoadingPaymentTokenERC1155, isError: isErrorPaymentTokenERC1155 } = useReadContract({
     chainId,
     abi: ERC_1155_SALE_CONTRACT,
@@ -62,6 +85,9 @@ export const useSaleContractConfig = ({ chainId, tokenType, contractAddress, tok
     }
   })
 
+  // [cost, supplyCap, startTime, endTime, merkleRoot]
+  type SaleDetailsERC1155 = [bigint, bigint, bigint, bigint, string]
+
   const { data: globalSaleDetailsERC1155, isLoading: isLoadingGlobalSaleDetailsERC1155, isError: isErrorGlobalSaleDetailsERC1155 } = useReadContract({
     chainId,
     abi: ERC_1155_SALE_CONTRACT,
@@ -91,6 +117,8 @@ export const useSaleContractConfig = ({ chainId, tokenType, contractAddress, tok
     }
   })
 
+  // [supplyCap, cost, paymentToken, startTime, endTime, merkleRoot]
+  type SaleDetailsERC721 = [bigint, bigint, string, bigint, string, string]
 
   const { data: saleDetailsERC721, isLoading: isLoadingSaleDetailsERC721, isError: isErrorSaleDetailsERC721 } = useReadContract({
     chainId,
@@ -102,7 +130,53 @@ export const useSaleContractConfig = ({ chainId, tokenType, contractAddress, tok
     }
   })
 
-  // get currency
-  
-  // get price
+  const isLoadingERC1155 = isLoadingPaymentTokenERC1155 || isLoadingGlobalSaleDetailsERC1155 || isLoadingTokenSaleDetailsERC1155
+  const isLoadingERC721 = isLoadingSaleDetailsERC721
+  const isErrorERC1155 = isErrorPaymentTokenERC1155 || isErrorGlobalSaleDetailsERC1155 || isErrorTokenSaleDetailsERC1155
+  const isErrorERC721 = isErrorSaleDetailsERC721
+
+  let saleInfos: SaleConfig[] = []
+
+  if (isLoadingERC1155 || isLoadingERC721 || isErrorERC1155 || isErrorERC721) {
+    return ({
+      data: undefined,
+      isLoading: tokenType === 'ERC1155' ? isLoadingERC1155 : isLoadingERC721,
+      isError: tokenType === 'ERC1155' ? isErrorERC1155 : isErrorERC721
+    })
+  }
+
+  const getPrice = () => {
+    if (isLoadingERC1155 || isLoadingERC721 || isErrorERC1155 || isErrorERC721) return
+    if (tokenType === 'ERC1155') {
+      // In the sale contract, the global sale has priority over the token sale
+      // So we need to check if the global sale is set, and if it is, use that
+      // Otherwise, we use the token sale
+      const [costERC721, _, startTime, endTime] = globalSaleDetailsERC1155 as SaleDetailsERC1155
+      const isGlobalSaleValid = endTime === BigInt(0) || (startTime <= BigInt(Math.floor(Date.now() / 1000)) || endTime >= BigInt(Math.floor(Date.now() / 1000)))
+      saleInfos = tokenIds.map((tokenId, index) => {
+        const tokenPrice = (tokenSaleDetailsERC1155?.[index].result as SaleDetailsERC1155)[0] || BigInt(0)
+        return ({
+          tokenId,
+          price: (isGlobalSaleValid ? costERC721 : tokenPrice).toString()
+        })
+      })
+    } else {
+      saleInfos = tokenIds.map((tokenId, index) => {
+        const tokenPrice = (saleDetailsERC721 as SaleDetailsERC721)[1] || BigInt(0)
+        return ({
+          tokenId,
+          price: tokenPrice.toString()
+        })
+      })
+    }
+  }  
+
+  return ({
+    data: {
+      currencyAddress: tokenType === 'ERC1155' ? (paymentTokenERC1155 as string) : (saleDetailsERC721 as SaleDetailsERC721)[2],
+      saleConfigs: saleInfos
+    },
+    isLoading: tokenType === 'ERC1155' ? isLoadingERC1155 : isLoadingERC721,
+    isError: tokenType === 'ERC1155' ? isErrorERC1155 : isErrorERC721
+  })
 }
\ No newline at end of file