@zemyth/raise-sdk 0.1.1 → 0.1.3

package/src/pdas/index.ts

@@ -402,3 +402,349 @@ export function getFounderVestingPDA(projectPda: PublicKey, programId: PublicKey
   );
   return pda;
 }
+
+// =============================================================================
+// Dynamic Tokenomics PDAs
+// =============================================================================
+
+/**
+ * Derive Allocation Proposal PDA
+ *
+ * @param projectPda - Project account PDA
+ * @param proposalIndex - Proposal index (from tokenomics.proposal_count)
+ * @param programId - Raise program ID
+ * @returns Allocation proposal PDA
+ */
+export function getAllocationProposalPDA(
+  projectPda: PublicKey,
+  proposalIndex: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from('allocation_proposal'),
+      projectPda.toBuffer(),
+      Buffer.from([proposalIndex]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Allocation Vote PDA
+ *
+ * @param proposalPda - Allocation proposal PDA
+ * @param nftMint - NFT mint used for voting
+ * @param programId - Raise program ID
+ * @returns Allocation vote PDA
+ */
+export function getAllocationVotePDA(
+  proposalPda: PublicKey,
+  nftMint: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from('allocation_vote'),
+      proposalPda.toBuffer(),
+      nftMint.toBuffer(),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Sub-Allocation Vesting PDA
+ *
+ * @param projectPda - Project account PDA
+ * @param subAllocationId - Sub-allocation ID (0-9)
+ * @param programId - Raise program ID
+ * @returns Sub-allocation vesting PDA
+ */
+export function getSubAllocationVestingPDA(
+  projectPda: PublicKey,
+  subAllocationId: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from('sub_allocation_vesting'),
+      projectPda.toBuffer(),
+      Buffer.from([subAllocationId]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+// =============================================================================
+// Per-Milestone Vesting PDAs (add-per-milestone-vesting)
+// =============================================================================
+
+/**
+ * Derive InvestorMilestoneVesting PDA
+ * Tracks investor vesting for a specific milestone with configurable schedules
+ *
+ * @param projectPda - Project account PDA
+ * @param milestoneIndex - Milestone index
+ * @param investmentPda - Investment account PDA
+ * @param programId - Raise program ID
+ * @returns Investor milestone vesting PDA
+ */
+export function getInvestorMilestoneVestingPDA(
+  projectPda: PublicKey,
+  milestoneIndex: number,
+  investmentPda: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from('investor_ms_vesting'),
+      projectPda.toBuffer(),
+      Buffer.from([milestoneIndex]),
+      investmentPda.toBuffer(),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive FounderMilestoneVesting PDA
+ * Tracks founder milestone-based vesting with configurable schedules
+ *
+ * @param projectPda - Project account PDA
+ * @param milestoneIndex - Milestone index
+ * @param programId - Raise program ID
+ * @returns Founder milestone vesting PDA
+ */
+export function getFounderMilestoneVestingPDA(
+  projectPda: PublicKey,
+  milestoneIndex: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from('founder_ms_vesting'),
+      projectPda.toBuffer(),
+      Buffer.from([milestoneIndex]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+// =============================================================================
+// Future Round Allocation PDAs (add-future-round-allocation)
+// =============================================================================
+
+/**
+ * Derive FutureRoundTokenVault PDA - Token account holding reserved tokens
+ *
+ * @param projectPda - Project account PDA
+ * @param programId - Raise program ID
+ * @returns Future round token vault PDA
+ */
+export function getFutureRoundTokenVaultPDA(
+  projectPda: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [Buffer.from(SEEDS.FUTURE_ROUND_VAULT), projectPda.toBuffer()],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive FutureRoundVault state PDA - Tracks future round token usage
+ *
+ * @param projectPda - Project account PDA
+ * @param programId - Raise program ID
+ * @returns Future round vault state PDA
+ */
+export function getFutureRoundVaultPDA(
+  projectPda: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [Buffer.from(SEEDS.FUTURE_ROUND_STATE), projectPda.toBuffer()],
+    programId
+  );
+  return pda;
+}
+
+// =============================================================================
+// Multi-Round Fundraising PDAs (add-second-round-fundraising)
+// =============================================================================
+
+/**
+ * Derive FundingRound PDA
+ * Each round (R2, R3, R4...) has its own FundingRound account
+ *
+ * @param projectPda - Project account PDA
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param programId - Raise program ID
+ * @returns Funding round PDA
+ */
+export function getFundingRoundPDA(
+  projectPda: PublicKey,
+  roundNumber: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.FUNDING_ROUND),
+      projectPda.toBuffer(),
+      Buffer.from([roundNumber]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Round Escrow PDA
+ * Each round has its own USDC escrow account
+ *
+ * @param projectPda - Project account PDA
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param programId - Raise program ID
+ * @returns Round escrow PDA
+ */
+export function getRoundEscrowPDA(
+  projectPda: PublicKey,
+  roundNumber: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.ROUND_ESCROW),
+      projectPda.toBuffer(),
+      Buffer.from([roundNumber]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Round Milestone PDA
+ * Each round has its own milestones (R2M1, R2M2, etc.)
+ *
+ * @param projectPda - Project account PDA
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param milestoneIndex - Milestone index (0-based)
+ * @param programId - Raise program ID
+ * @returns Round milestone PDA
+ */
+export function getRoundMilestonePDA(
+  projectPda: PublicKey,
+  roundNumber: number,
+  milestoneIndex: number,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.MILESTONE),
+      projectPda.toBuffer(),
+      Buffer.from([roundNumber]),
+      Buffer.from([milestoneIndex]),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Round NFT Mint PDA
+ * NFT mints for R2+ investments include round_number in seeds
+ *
+ * @param projectId - Project identifier
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param investor - Investor's public key
+ * @param investmentCount - Investment count
+ * @param programId - Raise program ID
+ * @returns Round NFT mint PDA and bump
+ */
+export function getRoundNftMintPDA(
+  projectId: BN | number | string,
+  roundNumber: number,
+  investor: PublicKey,
+  investmentCount: BN | number,
+  programId: PublicKey
+): [PublicKey, number] {
+  const projectIdBN = ensureBN(projectId);
+  const countBN = ensureBN(investmentCount);
+  return PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.NFT_MINT),
+      projectIdBN.toArrayLike(Buffer, 'le', 8),
+      Buffer.from([roundNumber]),
+      investor.toBuffer(),
+      countBN.toArrayLike(Buffer, 'le', 8),
+    ],
+    programId
+  );
+}
+
+/**
+ * Derive Round Investment PDA
+ * R2+ investments include round_number in seeds
+ *
+ * @param projectPda - Project account PDA
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param nftMint - Investment NFT mint address
+ * @param programId - Raise program ID
+ * @returns Round investment PDA
+ */
+export function getRoundInvestmentPDA(
+  projectPda: PublicKey,
+  roundNumber: number,
+  nftMint: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.INVESTMENT),
+      projectPda.toBuffer(),
+      Buffer.from([roundNumber]),
+      nftMint.toBuffer(),
+    ],
+    programId
+  );
+  return pda;
+}
+
+/**
+ * Derive Round Investor Milestone Vesting PDA
+ * R2+ investor vesting includes round_number in seeds
+ *
+ * @param projectPda - Project account PDA
+ * @param roundNumber - Round number (2, 3, 4...)
+ * @param milestoneIndex - Milestone index
+ * @param investmentPda - Investment account PDA
+ * @param programId - Raise program ID
+ * @returns Round investor milestone vesting PDA
+ */
+export function getRoundInvestorMilestoneVestingPDA(
+  projectPda: PublicKey,
+  roundNumber: number,
+  milestoneIndex: number,
+  investmentPda: PublicKey,
+  programId: PublicKey
+): PublicKey {
+  const [pda] = PublicKey.findProgramAddressSync(
+    [
+      Buffer.from(SEEDS.INVESTOR_MS_VESTING),
+      projectPda.toBuffer(),
+      Buffer.from([roundNumber]),
+      Buffer.from([milestoneIndex]),
+      investmentPda.toBuffer(),
+    ],
+    programId
+  );
+  return pda;
+}