@symmetry-hq/sdk 1.0.1

package/src/instructions/management/claimFees.ts

@@ -0,0 +1,129 @@
+import {
+    ASSOCIATED_TOKEN_PROGRAM_ID, TOKEN_2022_PROGRAM_ID, TOKEN_PROGRAM_ID
+} from '@solana/spl-token';
+import {  PublicKey, SystemProgram, TransactionInstruction } from '@solana/web3.js';
+
+import { BASKETS_V3_PROGRAM_ID } from '../../constants';
+import {
+    getAta, getBasketFeesPda, getBasketState, getGlobalConfigPda, getWithdrawBasketFeesPda
+} from '../pda';
+
+const WITHDRAW_FEES_DISCRIMINATOR = Buffer.from([159, 222, 146, 66, 254, 108, 55, 51]);
+
+const CLAIM_FEE_TOKENS_FROM_BASKET_DISCRIMINATOR = Buffer.from([53, 109, 220, 141, 193, 193, 13, 32]);
+
+export function withdrawFeesIx(params: {
+    claimer: PublicKey,
+    basketTokenMint: PublicKey,
+    feeType: number,
+}): TransactionInstruction {
+    let { claimer, basketTokenMint, feeType } = params;
+    let basket = getBasketState(basketTokenMint);
+    let withdrawBasketFees = getWithdrawBasketFeesPda(basket, feeType);
+    let basketFeesWallet = getBasketFeesPda(basket);
+    let basketFeesTokenAccount = getAta(basketFeesWallet, basketTokenMint);
+    let globalConfig = getGlobalConfigPda();
+    
+    const discriminator = WITHDRAW_FEES_DISCRIMINATOR;
+    const args = Buffer.from([feeType]);
+    const data = Buffer.concat([discriminator, args]);
+
+    const keys = [
+        { pubkey: claimer, isWritable: true, isSigner: true },
+        { pubkey: basket, isWritable: true, isSigner: false },
+        { pubkey: withdrawBasketFees, isWritable: true, isSigner: false },
+        { pubkey: basketTokenMint, isWritable: true, isSigner: false },
+        { pubkey: basketFeesWallet, isWritable: true, isSigner: false },
+        { pubkey: basketFeesTokenAccount, isWritable: true, isSigner: false },
+        { pubkey: globalConfig, isWritable: false, isSigner: false },
+        { pubkey: SystemProgram.programId, isWritable: false, isSigner: false },
+        { pubkey: TOKEN_PROGRAM_ID, isWritable: false, isSigner: false },
+        { pubkey: ASSOCIATED_TOKEN_PROGRAM_ID, isWritable: false, isSigner: false },
+    ];
+
+    return new TransactionInstruction({
+        keys,
+        programId: BASKETS_V3_PROGRAM_ID,
+        data,
+    });
+}
+
+export function claimFeeTokensFromBasketIx(params: {
+    signer: PublicKey,
+    basket: PublicKey,
+    withdrawBasketFees: PublicKey,
+    rentPayer: PublicKey,
+    owners: PublicKey[],
+    tokenMints: PublicKey[],
+}): TransactionInstruction {
+    let { signer, basket, withdrawBasketFees, rentPayer, owners, tokenMints } = params;
+    let numOwners = owners.length;
+    
+    const discriminator = CLAIM_FEE_TOKENS_FROM_BASKET_DISCRIMINATOR;
+    const args = Buffer.from([numOwners]);
+    const data = Buffer.concat([discriminator, args]);
+
+    const keys = [
+        { pubkey: signer, isWritable: true, isSigner: true },
+        { pubkey: basket, isWritable: true, isSigner: false },
+        { pubkey: withdrawBasketFees, isWritable: true, isSigner: false },
+        { pubkey: rentPayer, isWritable: true, isSigner: false },
+        { pubkey: SystemProgram.programId, isWritable: false, isSigner: false },
+        { pubkey: TOKEN_PROGRAM_ID, isWritable: false, isSigner: false },
+        { pubkey: TOKEN_2022_PROGRAM_ID, isWritable: false, isSigner: false },
+        { pubkey: ASSOCIATED_TOKEN_PROGRAM_ID, isWritable: false, isSigner: false },
+    ];
+
+    // Remaining accounts: for each token: (mint, basketAta, claimer1Ata, claimer2Ata, ...)
+    for (let i = 0; i < tokenMints.length; i++) {
+        const mint = tokenMints[i];
+        const basketAta = getAta(basket, mint);
+        const claimerATAs = owners.map(owner => getAta(owner, mint));
+        keys.push({ pubkey: mint, isWritable: false, isSigner: false });
+        keys.push({ pubkey: basketAta, isWritable: true, isSigner: false });
+        
+        for (let j = 0; j < numOwners; j++) {
+            keys.push({ pubkey: claimerATAs[j], isWritable: true, isSigner: false });
+        }
+    }
+
+    return new TransactionInstruction({
+        keys,
+        programId: BASKETS_V3_PROGRAM_ID,
+        data,
+    });
+}
+
+export function claimFeeTokensFromBasketIxs(
+    signer: PublicKey,
+    basket: PublicKey,
+    withdrawBasketFees: PublicKey,
+    rentPayer: PublicKey,
+    owners: PublicKey[],
+    tokenMints: PublicKey[],
+): TransactionInstruction[] {
+    if (owners.length === 0 || tokenMints.length === 0) return [];
+    
+    // Account limit: 8 fixed accounts + (owners + 2) * number_of_mints <= 30
+    // So: (owners + 2) * number_of_mints <= 24
+    // Max tokens per tx: Math.floor(24 / (owners.length + 2))
+    const accountsPerToken = owners.length + 2; // mint, basketAta, and one account per owner
+    const maxRemainingAccounts = 24; // 30 - 8 fixed accounts
+    const maxTokensPerTx = Math.floor(maxRemainingAccounts / accountsPerToken);
+    
+    let ixs: TransactionInstruction[] = [];
+
+    for (let i = 0; i < tokenMints.length; i += maxTokensPerTx) {
+        const batchTokenMints = tokenMints.slice(i, i + maxTokensPerTx);
+        ixs.push(claimFeeTokensFromBasketIx({
+            signer: signer,
+            basket: basket,
+            withdrawBasketFees: withdrawBasketFees,
+            rentPayer: rentPayer,
+            owners: owners,
+            tokenMints: batchTokenMints,
+        }));
+    }
+    
+    return ixs;
+}

package/src/instructions/management/createBasket.ts

@@ -0,0 +1,138 @@
+import BN from 'bn.js';
+import {
+    ASSOCIATED_TOKEN_PROGRAM_ID,
+    TOKEN_PROGRAM_ID
+} from '@solana/spl-token';
+import {
+    PublicKey, SYSVAR_RENT_PUBKEY, TransactionInstruction, SystemProgram
+} from '@solana/web3.js';
+
+import {
+    ADDRESS_LOOKUP_TABLE_PROGRAM_ID,
+    BASKETS_V3_PROGRAM_ID,
+    METADATA_PROGRAM_ID,
+    MINTS,
+    PYTHNET_CUSTODY_PRICE_USDC_ACCOUNT,
+    PYTHNET_CUSTODY_PRICE_WSOL_ACCOUNT,
+} from '../../constants';
+import { Fraction } from '../../layouts/fraction';
+import { getAta, getGlobalConfigPda, getLookupTableAccount, getMetadataAccount, getRentPayerPda, serializeString } from '../pda';
+import { HostFees, HostFeesLayout, MetadataParams } from '../../layouts/config';
+
+
+const CREATE_BASKET_STATE_ACCOUNT_DISCRIMINATOR = Buffer.from([119, 105, 241, 40, 193, 2, 32, 12]);
+const RESIZE_BASKET_STATE_DISCRIMINATOR = Uint8Array.from([103, 216, 111, 212, 45, 172, 125, 84]);
+const CREATE_BASKET_DISCRIMINATOR = Buffer.from([175, 138, 181, 18, 46, 58, 146, 159]);
+    
+
+export function createBasketStateAccountIx(params: {
+    creator: PublicKey;
+    basket: PublicKey;
+}): TransactionInstruction {
+    
+    let { creator, basket } = params;
+    let rentPayerPda = getRentPayerPda();
+
+    const data = Buffer.concat([
+        CREATE_BASKET_STATE_ACCOUNT_DISCRIMINATOR,
+    ]);
+
+    const keys = [
+        { pubkey: creator, isSigner: true, isWritable: true },
+        { pubkey: basket, isSigner: false, isWritable: true },
+        { pubkey: rentPayerPda, isSigner: false, isWritable: true },
+        { pubkey: getGlobalConfigPda(), isSigner: false, isWritable: false },
+        { pubkey: SYSVAR_RENT_PUBKEY, isSigner: false, isWritable: false },
+        { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
+    ];
+
+    return new TransactionInstruction({
+        keys,
+        programId: BASKETS_V3_PROGRAM_ID,
+        data,
+    });
+}
+
+export function resizeBasketStateIx(params: {
+        basket: PublicKey;
+}): TransactionInstruction {
+    let { basket } = params;
+    const data = Buffer.concat([
+        RESIZE_BASKET_STATE_DISCRIMINATOR,
+    ]);
+
+    const keys = [
+        { pubkey: basket, isSigner: false, isWritable: true },
+    ];
+
+    return new TransactionInstruction({
+        keys,
+        programId: BASKETS_V3_PROGRAM_ID,
+        data,
+    });
+}
+
+export function createBasketIx(params: {
+    basket: PublicKey,
+    mint: PublicKey,
+    slot: number,
+    creator: PublicKey,
+    host: PublicKey,
+    startPrice: Fraction,
+    hostFees: HostFees, // [u16; 4]
+    metadataParams: MetadataParams,
+    network: "devnet" | "mainnet",
+}): TransactionInstruction {
+    let { basket, mint, slot, creator, host, startPrice, hostFees, metadataParams } = params;
+
+    let metadataAccount = getMetadataAccount(mint);
+    
+    let lookupTable0 = getLookupTableAccount(basket, slot);
+    let lookupTable1 = getLookupTableAccount(basket, slot - 1);
+
+    let wsolBasketTokenAccount = getAta(basket, MINTS[params.network].WSOL);
+    let usdcBasketTokenAccount = getAta(basket, MINTS[params.network].USDC);
+
+    const hostFeesBuffer = Buffer.alloc(HostFeesLayout.span);
+    HostFeesLayout.encode(hostFees, hostFeesBuffer);
+
+    const data = Buffer.concat([
+        CREATE_BASKET_DISCRIMINATOR,
+        Buffer.from(new BN(slot).toArray("le", 8)),
+        host.toBuffer(),
+        Buffer.from(startPrice.high.toArray("le", 8)),
+        Buffer.from(startPrice.low.toArray("le", 8)),
+        hostFeesBuffer,
+        serializeString(metadataParams.name),
+        serializeString(metadataParams.symbol),
+        serializeString(metadataParams.uri),
+    ]);
+
+    const keys = [
+        { pubkey: creator, isSigner: true, isWritable: true },
+        { pubkey: basket, isSigner: false, isWritable: true },
+        { pubkey: mint, isSigner: false, isWritable: true },
+        { pubkey: metadataAccount, isSigner: false, isWritable: true },
+        { pubkey: lookupTable0, isSigner: false, isWritable: true },
+        { pubkey: lookupTable1, isSigner: false, isWritable: true },
+        { pubkey: wsolBasketTokenAccount, isSigner: false, isWritable: true },
+        { pubkey: MINTS[params.network].WSOL, isSigner: false, isWritable: true },
+        { pubkey: PYTHNET_CUSTODY_PRICE_WSOL_ACCOUNT, isSigner: false, isWritable: true },
+        { pubkey: usdcBasketTokenAccount, isSigner: false, isWritable: true },
+        { pubkey: MINTS[params.network].USDC, isSigner: false, isWritable: true },
+        { pubkey: PYTHNET_CUSTODY_PRICE_USDC_ACCOUNT, isSigner: false, isWritable: true },
+        { pubkey: getGlobalConfigPda(), isSigner: false, isWritable: true },
+        { pubkey: ADDRESS_LOOKUP_TABLE_PROGRAM_ID, isSigner: false, isWritable: false },
+        { pubkey: METADATA_PROGRAM_ID, isSigner: false, isWritable: false },
+        { pubkey: SYSVAR_RENT_PUBKEY, isSigner: false, isWritable: false },
+        { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
+        { pubkey: TOKEN_PROGRAM_ID, isSigner: false, isWritable: false },
+        { pubkey: ASSOCIATED_TOKEN_PROGRAM_ID, isSigner: false, isWritable: false },
+    ];
+
+    return new TransactionInstruction({
+        keys,
+        programId: BASKETS_V3_PROGRAM_ID,
+        data,
+    });
+}