From c823b99ef7e68249e784ce71b954acb808b56f91 Mon Sep 17 00:00:00 2001
From: Evan Salzbrenner <evan.salzbrenner@gmail.com>
Date: Tue, 17 Sep 2024 15:33:36 -0700
Subject: [PATCH] =?UTF-8?q?SpaceDapp:=20getJoinPriceDetails,=20add=20prepa?=
 =?UTF-8?q?idSupply=20and=20remainingFreeSu=E2=80=A6=20(#1069)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

…pply to getMembershipInfo
---
 packages/web3/src/ContractTypes.ts           |  5 +-
 packages/web3/src/ISpaceDapp.ts              |  7 ++-
 packages/web3/src/v3/PlatformRequirements.ts |  4 ++
 packages/web3/src/v3/SpaceDapp.ts            | 65 +++++++++++++++-----
 4 files changed, 63 insertions(+), 18 deletions(-)

diff --git a/packages/web3/src/ContractTypes.ts b/packages/web3/src/ContractTypes.ts
index f980e2389..c58103eba 100644
--- a/packages/web3/src/ContractTypes.ts
+++ b/packages/web3/src/ContractTypes.ts
@@ -188,7 +188,10 @@ export type MembershipInfo = Pick<
     MembershipInfoStruct,
     'maxSupply' | 'currency' | 'feeRecipient' | 'price' | 'duration' | 'pricingModule'
 > &
-    TotalSupplyInfo
+    TotalSupplyInfo & {
+        prepaidSupply: number
+        remainingFreeSupply: number
+    }
 
 export type TotalSupplyInfo = Pick<TotalSupplyOutputStruct, 'totalSupply'>
 
diff --git a/packages/web3/src/ISpaceDapp.ts b/packages/web3/src/ISpaceDapp.ts
index 3dffb1f07..14a3d21fb 100644
--- a/packages/web3/src/ISpaceDapp.ts
+++ b/packages/web3/src/ISpaceDapp.ts
@@ -269,7 +269,11 @@ export interface ISpaceDapp {
     ) => Promise<TransactionType>
     getSpace(spaceId: string): Space | undefined
     getSpaceMembershipTokenAddress: (spaceId: string) => Promise<string>
-    getJoinSpacePrice: (spaceId: string) => Promise<BigNumber>
+    getJoinSpacePriceDetails: (spaceId: string) => Promise<{
+        price: ethers.BigNumber
+        prepaidSupply: ethers.BigNumber
+        remainingFreeSupply: ethers.BigNumber
+    }>
     joinSpace: (
         spaceId: string,
         recipient: string,
@@ -312,4 +316,5 @@ export interface ISpaceDapp {
         receiver: string,
         abortController?: AbortController,
     ) => Promise<{ issued: true; tokenId: string } | { issued: false; tokenId: undefined }>
+    getMembershipFreeAllocation: (spaceId: string) => Promise<BigNumber>
 }
diff --git a/packages/web3/src/v3/PlatformRequirements.ts b/packages/web3/src/v3/PlatformRequirements.ts
index 84254f2d6..222e4bf88 100644
--- a/packages/web3/src/v3/PlatformRequirements.ts
+++ b/packages/web3/src/v3/PlatformRequirements.ts
@@ -13,6 +13,10 @@ export class PlatformRequirements extends BaseContractShim<LocalhostContract, Lo
         super(address, provider, LocalhostAbi)
     }
 
+    public getMembershipMintLimit() {
+        return this.read.getMembershipMintLimit()
+    }
+
     public getMembershipFee() {
         return this.read.getMembershipFee()
     }
diff --git a/packages/web3/src/v3/SpaceDapp.ts b/packages/web3/src/v3/SpaceDapp.ts
index 89b5635d9..8027f451a 100644
--- a/packages/web3/src/v3/SpaceDapp.ts
+++ b/packages/web3/src/v3/SpaceDapp.ts
@@ -4,6 +4,7 @@ import {
     ChannelMetadata,
     EntitlementModuleType,
     isPermission,
+    MembershipInfo,
     Permission,
     PricingModuleStruct,
     RoleDetails,
@@ -1277,17 +1278,39 @@ export class SpaceDapp implements ISpaceDapp {
         return space.Membership.address
     }
 
-    public async getJoinSpacePrice(spaceId: string): Promise<ethers.BigNumber> {
+    public async getJoinSpacePriceDetails(spaceId: string): Promise<{
+        price: ethers.BigNumber
+        prepaidSupply: ethers.BigNumber
+        remainingFreeSupply: ethers.BigNumber
+    }> {
         const space = this.getSpace(spaceId)
         if (!space) {
             throw new Error(`Space with spaceId "${spaceId}" is not found.`)
         }
         const prepaidSupply = await space.Prepay.read.prepaidMembershipSupply()
+        const membershipPrice = await space.Membership.read.getMembershipPrice()
+        const freeAllocation = await this.getMembershipFreeAllocation(spaceId)
+        const totalSupply = await space.ERC721A.read.totalSupply()
+        // totalSupply = number of memberships minted
+        // freeAllocation = number of memberships that are free to mint, set during space creation
+        // prepaidSupply = number of additional prepaid memberships
+        const remainingFreeSupply = totalSupply.lt(freeAllocation)
+            ? freeAllocation.add(prepaidSupply).sub(totalSupply)
+            : prepaidSupply
+
+        return {
+            price: prepaidSupply.gt(0) ? ethers.BigNumber.from(0) : membershipPrice,
+            prepaidSupply,
+            remainingFreeSupply,
+        }
+    }
 
-        if (prepaidSupply.gt(0)) {
-            return ethers.BigNumber.from(0)
+    public async getMembershipFreeAllocation(spaceId: string) {
+        const space = this.getSpace(spaceId)
+        if (!space) {
+            throw new Error(`Space with spaceId "${spaceId}" is not found.`)
         }
-        return space.Membership.read.getMembershipPrice()
+        return space.Membership.read.getMembershipFreeAllocation()
     }
 
     public async joinSpace(
@@ -1365,26 +1388,36 @@ export class SpaceDapp implements ISpaceDapp {
         if (!space) {
             throw new Error(`Space with spaceId "${spaceId}" is not found.`)
         }
-        const [price, limit, currency, feeRecipient, duration, totalSupply, pricingModule] =
-            await Promise.all([
-                this.getJoinSpacePrice(spaceId),
-                space.Membership.read.getMembershipLimit(),
-                space.Membership.read.getMembershipCurrency(),
-                space.Ownable.read.owner(),
-                space.Membership.read.getMembershipDuration(),
-                space.ERC721A.read.totalSupply(),
-                space.Membership.read.getMembershipPricingModule(),
-            ])
+        const [
+            joinSpacePriceDetails,
+            limit,
+            currency,
+            feeRecipient,
+            duration,
+            totalSupply,
+            pricingModule,
+        ] = await Promise.all([
+            this.getJoinSpacePriceDetails(spaceId),
+            space.Membership.read.getMembershipLimit(),
+            space.Membership.read.getMembershipCurrency(),
+            space.Ownable.read.owner(),
+            space.Membership.read.getMembershipDuration(),
+            space.ERC721A.read.totalSupply(),
+            space.Membership.read.getMembershipPricingModule(),
+        ])
+        const { price, prepaidSupply, remainingFreeSupply } = joinSpacePriceDetails
 
         return {
-            price: price, // keep as BigNumber (wei)
+            price, // keep as BigNumber (wei)
             maxSupply: limit.toNumber(),
             currency: currency,
             feeRecipient: feeRecipient,
             duration: duration.toNumber(),
             totalSupply: totalSupply.toNumber(),
             pricingModule: pricingModule,
-        }
+            prepaidSupply: prepaidSupply.toNumber(),
+            remainingFreeSupply: remainingFreeSupply.toNumber(),
+        } satisfies MembershipInfo
     }
 
     public getWalletLink(): WalletLink {