add sdk integration

me-foundation · Apr 12, 2024 · 92c7347 · 92c7347
1 parent 5df0354
commit 92c7347
Show file tree

Hide file tree

Showing 5 changed files with 293 additions and 10 deletions.
diff --git a/sdk/src/index.ts b/sdk/src/index.ts
@@ -1,6 +1,7 @@
-export * from './idl/mmm';
 export * from './constants';
-export * from './pda';
+export * from './idl/mmm';
 export * from './metadataProvider';
 export * from './mmmClient';
+export * from './pda';
 export * from './price';
+export * from './transferHookProvider';
diff --git a/sdk/src/mmmClient.ts b/sdk/src/mmmClient.ts
@@ -32,13 +32,18 @@ import {
   getMMMPoolPDA,
   getMMMSellStatePDA,
   getTokenRecordPDA,
+  M2_PROGRAM,
 } from './pda';
 import {
   doesTokenExtensionExist,
   MetadataProvider,
   MintStateWithAddress,
   RpcMetadataProvider,
 } from './metadataProvider';
+import {
+  LibreplexRoyaltyProvider,
+  TransferHookProvider,
+} from './transferHookProvider';
 
 export const getEmptyAllowLists = (num: number) => {
   const emptyAllowList = {
@@ -68,6 +73,10 @@ export class MMMClient {
     mint: PublicKey,
     conn: Connection,
   ) => Promise<MetadataProvider>;
+  private readonly transferHookProviderGenerator: (
+    mint: PublicKey,
+    conn: Connection,
+  ) => Promise<TransferHookProvider>;
 
   poolData: (anchor.IdlAccounts<Mmm>['pool'] & { pool: PublicKey }) | undefined;
 
@@ -78,6 +87,10 @@ export class MMMClient {
       mint: PublicKey,
       conn: Connection,
     ) => Promise<MetadataProvider>,
+    transferHookProviderGenerator?: (
+      mint: PublicKey,
+      conn: Connection,
+    ) => Promise<TransferHookProvider>,
   ) {
     this.conn = conn;
     this.provider = new anchor.AnchorProvider(this.conn, dummyKeypair, {
@@ -89,6 +102,8 @@ export class MMMClient {
     this.mpl = new Metaplex(conn);
     this.metadataProviderGenerator =
       metadataProviderGenerator ?? RpcMetadataProvider.loadFromRpc;
+    this.transferHookProviderGenerator =
+      transferHookProviderGenerator ?? LibreplexRoyaltyProvider.loadFromRpc;
   }
 
   signTx(insArr: TransactionInstruction[]): Transaction {
@@ -255,11 +270,16 @@ export class MMMClient {
     assetTokenAccount: PublicKey,
     allowlistAuxAccount?: PublicKey,
     metadataProvider?: MetadataProvider,
+    transferHookProvider?: TransferHookProvider,
   ): Promise<TransactionInstruction> {
     if (!this.poolData) throw MMMClient.ErrPoolDataEmpty;
     const mintContext =
       metadataProvider ??
       (await this.metadataProviderGenerator(assetMint, this.conn));
+    const transferHookContext =
+      transferHookProvider ??
+      (await this.transferHookProviderGenerator(assetMint, this.conn));
+
     let { key: buysideSolEscrowAccount } = getMMMBuysideSolEscrowPDA(
       MMMProgramID,
       this.poolData.pool,
@@ -401,10 +421,27 @@ export class MMMClient {
       }
     }
 
+    // Add m2 and shared escrow account here as remaining accounts for shared escrow pool.
+    if (!this.poolData.sharedEscrowAccount.equals(PublicKey.default)) {
+      builder = builder.remainingAccounts([
+        {
+          pubkey: M2_PROGRAM,
+          isSigner: false,
+          isWritable: false,
+        },
+        {
+          pubkey: this.poolData.sharedEscrowAccount,
+          isWritable: true,
+          isSigner: false,
+        },
+      ]);
+    }
+
     if (
-      this.poolData.buysideCreatorRoyaltyBp > 0 ||
-      ocpMintState ||
-      tokenStandard === TokenStandard.ProgrammableNonFungible
+      (this.poolData.buysideCreatorRoyaltyBp > 0 ||
+        ocpMintState ||
+        tokenStandard === TokenStandard.ProgrammableNonFungible) &&
+      !transferHookContext.isTransferHookProgramAllowed()
     ) {
       const creators = mintContext.creators;
       if (creators.length > 0) {
@@ -417,6 +454,12 @@ export class MMMClient {
         );
       }
     }
+
+    if (transferHookContext.isTransferHookProgramAllowed()) {
+      builder = builder.remainingAccounts(
+        await transferHookContext.getRemainingAccounts(true),
+      );
+    }
     return await builder.instruction();
   }
 
@@ -426,6 +469,7 @@ export class MMMClient {
     assetMint: PublicKey,
     allowlistAuxAccount?: PublicKey,
     metadataProvider?: MetadataProvider,
+    transferHookProvider?: TransferHookProvider,
   ): Promise<TransactionInstruction> {
     if (!this.poolData) throw MMMClient.ErrPoolDataEmpty;
     let { key: buysideSolEscrowAccount } = getMMMBuysideSolEscrowPDA(
@@ -436,6 +480,9 @@ export class MMMClient {
     const mintContext =
       metadataProvider ??
       (await this.metadataProviderGenerator(assetMint, this.conn));
+    const transferHookContext =
+      transferHookProvider ??
+      (await this.transferHookProviderGenerator(assetMint, this.conn));
 
     const { key: sellState } = getMMMSellStatePDA(
       MMMProgramID,
@@ -572,9 +619,10 @@ export class MMMClient {
     }
 
     if (
-      args.buysideCreatorRoyaltyBp > 0 ||
-      ocpMintState ||
-      tokenStandard === TokenStandard.ProgrammableNonFungible
+      (args.buysideCreatorRoyaltyBp > 0 ||
+        ocpMintState ||
+        tokenStandard === TokenStandard.ProgrammableNonFungible) &&
+      !transferHookContext.isTransferHookProgramAllowed()
     ) {
       const creators = mintContext.creators;
       if (creators.length > 0) {
@@ -587,6 +635,12 @@ export class MMMClient {
         );
       }
     }
+
+    if (transferHookContext.isTransferHookProgramAllowed()) {
+      builder = builder.remainingAccounts(
+        await transferHookContext.getRemainingAccounts(true),
+      );
+    }
     return await builder.instruction();
   }
 
@@ -595,12 +649,16 @@ export class MMMClient {
     assetMint: PublicKey,
     allowlistAuxAccount?: PublicKey,
     metadataProvider?: MetadataProvider,
+    transferHookProvider?: TransferHookProvider,
   ): Promise<TransactionInstruction> {
     if (!this.poolData) throw MMMClient.ErrPoolDataEmpty;
     const assetMetadata = this.mpl.nfts().pdas().metadata({ mint: assetMint });
     const mintContext =
       metadataProvider ??
       (await this.metadataProviderGenerator(assetMint, this.conn));
+    const transferHookContext =
+      transferHookProvider ??
+      (await this.transferHookProviderGenerator(assetMint, this.conn));
 
     const { key: sellState } = getMMMSellStatePDA(
       MMMProgramID,
@@ -708,6 +766,12 @@ export class MMMClient {
         });
       }
     }
+
+    if (transferHookContext.isTransferHookProgramAllowed()) {
+      builder = builder.remainingAccounts(
+        await transferHookContext.getRemainingAccounts(false),
+      );
+    }
     return await builder.instruction();
   }
 
@@ -716,11 +780,15 @@ export class MMMClient {
     assetMint: PublicKey,
     allowlistAuxAccount?: PublicKey,
     metadataProvider?: MetadataProvider,
+    transferHookProvider?: TransferHookProvider,
   ): Promise<TransactionInstruction> {
     if (!this.poolData) throw MMMClient.ErrPoolDataEmpty;
     const mintContext =
       metadataProvider ??
       (await this.metadataProviderGenerator(assetMint, this.conn));
+    const transferHookContext =
+      transferHookProvider ??
+      (await this.transferHookProviderGenerator(assetMint, this.conn));
 
     const { key: sellState } = getMMMSellStatePDA(
       MMMProgramID,
@@ -837,6 +905,12 @@ export class MMMClient {
         });
       }
     }
+
+    if (transferHookContext.isTransferHookProgramAllowed()) {
+      builder = builder.remainingAccounts(
+        await transferHookContext.getRemainingAccounts(false),
+      );
+    }
     return await builder.instruction();
   }