100x-sdk 1.0.1

package/src/modules/token.js

@@ -0,0 +1,455 @@
+const { ComputeBudgetProgram, PublicKey, Transaction, Keypair, SystemProgram, SYSVAR_RENT_PUBKEY } = require('@solana/web3.js');
+const { TOKEN_PROGRAM_ID, getAssociatedTokenAddress, createAssociatedTokenAccountInstruction, ASSOCIATED_TOKEN_PROGRAM_ID } = require('@solana/spl-token');
+const anchor = require('@coral-xyz/anchor');
+// 统一使用 buffer 包，所有平台一致
+const { Buffer } = require('buffer');
+
+// Metaplex Token Metadata 程序ID
+const METADATA_PROGRAM_ID = new PublicKey("metaqbxxUerdq28cj1RbAWkYQm3ybzjb6a8bt518x1s");
+
+/**
+ * Token Module
+ * Handles token creation, queries, balance and other operations
+ */
+class TokenModule {
+  constructor(sdk) {
+    this.sdk = sdk;
+  }
+
+  /**
+   * 创建新代币
+   * @param {Object} params - 创建参数
+   * @param {Keypair} params.mint - 代币mint keypair
+   * @param {string} params.name - 代币名称
+   * @param {string} params.symbol - 代币符号
+   * @param {string} params.uri - 元数据URI
+   * @param {PublicKey} params.payer - 创建者公钥（付款方）
+   *
+   * === 第二阶段新增：高级版池子可选参数（5个参数必须全部提供或全部不提供）===
+   * @param {anchor.BN} [params.customLpSol] - 自定义流动池SOL数量（lamports，9位精度）
+   *   - 不传或传null：创建普通版池子，使用默认值30 SOL
+   *   - 传入值：创建高级版池子，需在 10 SOL ~ 1000 SOL 范围内
+   *   - ⚠️ 必须与其他4个 custom 参数同时提供
+   *
+   * @param {anchor.BN} [params.customLpToken] - 自定义流动池Token数量（最小单位，9位精度）
+   *   - 不传或传null：使用默认值10.73亿 Token
+   *   - 传入值：需在 1亿 ~ 100亿 Token 范围内
+   *
+   * @param {number} [params.customBorrowRatio] - 自定义借贷池代币占比（5-30表示5%-30%）
+   *   - 不传或传null：使用默认值20%
+   *   - 传入值：需在 5 ~ 30 范围内
+   *
+   * @param {number} [params.customBorrowDuration] - 自定义借贷时长（秒）
+   *   - 不传或传null：使用Params账户配置的默认值
+   *   - 传入值：需在 3天(259200秒) ~ 6个月(15552000秒) 范围内
+   *
+   * @param {number} [params.customFee] - 自定义手续费率（单位：基点 basis points）
+   *   - 不传或传null：使用Params账户配置的默认手续费
+   *   - 传入值：需在 1000 ~ 5000 范围内（表示 1% ~ 5%）
+   *   - 示例：2000 表示 2%
+   *   - 借贷手续费会自动上浮20%（如设置2000，则 swap_fee=2000，borrow_fee=2400）
+   *
+   * @returns {Promise<Object>} 包含transaction、signers和账户信息的对象
+   *
+   * @example
+   * // 创建普通版代币（使用默认参数）
+   * const result = await sdk.token.create({
+   *   mint: mintKeypair,
+   *   name: "My Token",
+   *   symbol: "MTK",
+   *   uri: "https://example.com/metadata.json",
+   *   payer: wallet.publicKey
+   * });
+   *
+   * @example
+   * // 创建高级版代币（自定义流动池参数，必须提供全部5个参数）
+   * const anchor = require('@coral-xyz/anchor');
+   * const result = await sdk.token.create({
+   *   mint: mintKeypair,
+   *   name: "Advanced Token",
+   *   symbol: "ADV",
+   *   uri: "https://example.com/metadata.json",
+   *   payer: wallet.publicKey,
+   *   customLpSol: new anchor.BN('60000000000'),              // 60 SOL
+   *   customLpToken: new anchor.BN('1073000000000000000'),    // 10.73亿 Token
+   *   customBorrowRatio: 15,                                   // 15%
+   *   customBorrowDuration: 7 * 24 * 3600,                     // 7天
+   *   customFee: 2000                                          // 2% (2000 basis points)
+   * });
+   */
+  async create({
+    mint,
+    name,
+    symbol,
+    uri,
+    payer,
+    // 高级版池子可选参数（5个参数必须全部提供或全部不提供）
+    customLpSol = null,
+    customLpToken = null,
+    customBorrowRatio = null,
+    customBorrowDuration = null,
+    customFee = null
+  }) {
+    console.log('Token Module - Create:', {
+      mint: mint.publicKey.toString(),
+      name,
+      symbol,
+      uri,
+      payer: payer.toString(),
+      // 高级版池子参数日志
+      poolType: (customLpSol === null && customLpToken === null &&
+                 customBorrowRatio === null && customBorrowDuration === null &&
+                 customFee === null)
+                ? 'Standard (default params)'
+                : 'Advanced (custom params)',
+      customLpSol: customLpSol?.toString() || 'null',
+      customLpToken: customLpToken?.toString() || 'null',
+      customBorrowRatio: customBorrowRatio ?? 'null',
+      customBorrowDuration: customBorrowDuration ?? 'null',
+      customFee: customFee ?? 'null'
+    });
+
+    // Calculate borrowing liquidity pool account address (borrowing_curve)
+    const [curveAccount] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("borrowing_curve"),
+        mint.publicKey.toBuffer(),
+      ],
+      this.sdk.programId
+    );
+
+    // Calculate liquidity pool token account address (pool_token)
+    const [poolTokenAccount] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("pool_token"),
+        mint.publicKey.toBuffer(),
+      ],
+      this.sdk.programId
+    );
+
+    // Calculate liquidity pool SOL account address (pool_sol)
+    const [poolSolAccount] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("pool_sol"),
+        mint.publicKey.toBuffer(),
+      ],
+      this.sdk.programId
+    );
+
+    // Calculate order book accounts (new)
+    const [upOrderbook] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("up_orderbook"),
+        mint.publicKey.toBuffer(),
+      ],
+      this.sdk.programId
+    );
+
+    const [downOrderbook] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("down_orderbook"),
+        mint.publicKey.toBuffer(),
+      ],
+      this.sdk.programId
+    );
+
+    // Calculate Metaplex metadata account address
+    const [metadataAccount] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from("metadata"),
+        METADATA_PROGRAM_ID.toBuffer(),
+        mint.publicKey.toBuffer(),
+      ],
+      METADATA_PROGRAM_ID
+    );
+
+    console.log('Calculated account addresses:');
+    console.log('  Borrowing liquidity pool account:', curveAccount.toString());
+    console.log('  Liquidity pool token account:', poolTokenAccount.toString());
+    console.log('  Liquidity pool SOL account:', poolSolAccount.toString());
+    console.log('  Up orderbook:', upOrderbook.toString());
+    console.log('  Down orderbook:', downOrderbook.toString());
+    console.log('  Metadata account:', metadataAccount.toString());
+    console.log('  Params account:', this.sdk.paramsAccount?.toString() || 'Not set');
+
+    // Validate required configuration
+    if (!this.sdk.paramsAccount) {
+      throw new Error('SDK paramsAccount not configured, please provide paramsAccount configuration during initialization');
+    }
+
+    // Create compute budget instruction
+    const modifyComputeUnits = ComputeBudgetProgram.setComputeUnitLimit({
+      units: 400000
+    });
+
+    // 构建创建代币指令（传入5个自定义参数）
+    const createIx = await this.sdk.program.methods
+      .createToken(
+        name,
+        symbol,
+        uri,
+        customLpSol,           // Option<u64>
+        customLpToken,         // Option<u64>
+        customBorrowRatio,     // Option<u8>
+        customBorrowDuration,  // Option<u32>
+        customFee              // Option<u16> - 新增参数
+      )
+      .accounts({
+        payer: payer,
+        mintAccount: mint.publicKey,
+        curveAccount: curveAccount,
+        poolTokenAccount: poolTokenAccount,
+        poolSolAccount: poolSolAccount,
+        upOrderbook: upOrderbook,
+        downOrderbook: downOrderbook,
+        metadata: metadataAccount,
+        metadataProgram: METADATA_PROGRAM_ID,
+        params: this.sdk.paramsAccount,
+        tokenProgram: TOKEN_PROGRAM_ID,
+        systemProgram: SystemProgram.programId,
+        rent: SYSVAR_RENT_PUBKEY,
+      })
+      .instruction();
+
+    // Create transaction and add instructions
+    const transaction = new Transaction();
+    transaction.add(modifyComputeUnits);
+    transaction.add(createIx);
+
+    console.log('Token creation transaction built, signers required:', [payer.toString(), mint.publicKey.toString()]);
+
+    return {
+      transaction,
+      signers: [mint], // mint keypair needs to be a signer
+      accounts: {
+        mint: mint.publicKey,
+        curveAccount,
+        poolTokenAccount,
+        poolSolAccount,
+        upOrderbook,
+        downOrderbook,
+        metadataAccount,
+        payer
+      }
+    };
+  }
+
+  /**
+   * Create token and buy in one transaction
+   * 将 create 和 buy 两个指令合并到一个交易中，一次签名提交
+   *
+   * @param {Object} params - Creation and buy parameters
+   * @param {Keypair} params.mint - Token mint keypair
+   * @param {string} params.name - Token name
+   * @param {string} params.symbol - Token symbol
+   * @param {string} params.uri - Metadata URI
+   * @param {PublicKey} params.payer - Creator public key (payer)
+   * @param {anchor.BN} params.buyTokenAmount - Amount of tokens to buy
+   * @param {anchor.BN} params.maxSolAmount - Maximum SOL to spend
+   *
+   * === 第二阶段新增：高级版池子可选参数（5个参数必须全部提供或全部不提供）===
+   * @param {anchor.BN} [params.customLpSol] - 自定义流动池SOL数量（lamports，9位精度）
+   * @param {anchor.BN} [params.customLpToken] - 自定义流动池Token数量（最小单位，9位精度）
+   * @param {number} [params.customBorrowRatio] - 自定义借贷池代币占比（5-30表示5%-30%）
+   * @param {number} [params.customBorrowDuration] - 自定义借贷时长（秒）
+   * @param {number} [params.customFee] - 自定义手续费率（1000-5000 basis points，表示1%-5%）
+   *
+   * @param {Object} options - Optional parameters
+   * @param {number} options.computeUnits - Compute units limit, default 1800000
+   * @returns {Promise<Object>} Object containing transaction, signers and account info
+   */
+  async createAndBuy({
+    mint,
+    name,
+    symbol,
+    uri,
+    payer,
+    buyTokenAmount,
+    maxSolAmount,
+    // 高级版池子可选参数（5个参数必须全部提供或全部不提供）
+    customLpSol = null,
+    customLpToken = null,
+    customBorrowRatio = null,
+    customBorrowDuration = null,
+    customFee = null
+  }, options = {}) {
+    const { computeUnits = 1800000 } = options;
+
+    console.log('Token Module - CreateAndBuy:', {
+      mint: mint.publicKey.toString(),
+      name,
+      symbol,
+      uri,
+      payer: payer.toString(),
+      buyTokenAmount: buyTokenAmount.toString(),
+      maxSolAmount: maxSolAmount.toString()
+    });
+
+    // 1. 参数验证
+    if (!anchor.BN.isBN(buyTokenAmount) || !anchor.BN.isBN(maxSolAmount)) {
+      throw new Error('buyTokenAmount and maxSolAmount must be anchor.BN type');
+    }
+
+    // 2. 调用 create 方法获取 create 交易（传递所有自定义参数）
+    console.log('Step 1: Building create transaction...');
+    const createResult = await this.create({
+      mint,
+      name,
+      symbol,
+      uri,
+      payer,
+      // 传递自定义流动池参数（5个参数）
+      customLpSol,
+      customLpToken,
+      customBorrowRatio,
+      customBorrowDuration,
+      customFee
+    });
+
+    // 3. 从 params 账户读取手续费接收地址
+    // 因为此时 curve_account 还未创建，无法从链上读取
+    console.log('Step 2: Fetching fee recipient accounts from params...');
+
+    // 直接从 SDK 配置中获取手续费接收账户（这些在 SDK 初始化时已经设置）
+    // 避免使用 program.account.params.fetch() 因为可能有 provider 配置问题
+    const feeRecipientAccount = this.sdk.feeRecipient;
+    const baseFeeRecipientAccount = this.sdk.baseFeeRecipient;
+
+    // 验证这些账户已配置
+    if (!feeRecipientAccount || !baseFeeRecipientAccount) {
+      throw new Error('Fee recipient accounts not configured in SDK options');
+    }
+
+    console.log('Fee recipient accounts:');
+    console.log('  Partner fee recipient:', feeRecipientAccount.toString());
+    console.log('  Base fee recipient:', baseFeeRecipientAccount.toString());
+
+    // 4. 准备 buy 所需的额外账户
+    console.log('Step 3: Calculating buy-related accounts...');
+    const mintPubkey = mint.publicKey;
+
+    // 计算用户代币账户
+    const userTokenAccount = await getAssociatedTokenAddress(
+      mintPubkey,
+      payer
+    );
+
+    // 计算 cooldown PDA
+    const [cooldownPDA] = PublicKey.findProgramAddressSync(
+      [
+        Buffer.from('trade_cooldown'),
+        mintPubkey.toBuffer(),
+        payer.toBuffer()
+      ],
+      this.sdk.programId
+    );
+
+    // 计算 orderbook PDAs (复用 create 中计算的值)
+    const [upOrderbook] = PublicKey.findProgramAddressSync(
+      [Buffer.from('up_orderbook'), mintPubkey.toBuffer()],
+      this.sdk.programId
+    );
+
+    const [downOrderbook] = PublicKey.findProgramAddressSync(
+      [Buffer.from('down_orderbook'), mintPubkey.toBuffer()],
+      this.sdk.programId
+    );
+
+    console.log('Buy-related accounts:');
+    console.log('  User token account:', userTokenAccount.toString());
+    console.log('  Cooldown PDA:', cooldownPDA.toString());
+
+    // 5. 检查用户代币账户是否存在，创建 ATA 指令
+    console.log('Step 4: Checking if user token account exists...');
+    const userTokenAccountInfo = await this.sdk.connection.getAccountInfo(userTokenAccount);
+    const createAtaIx = userTokenAccountInfo === null
+      ? createAssociatedTokenAccountInstruction(
+          payer,
+          userTokenAccount,
+          payer,
+          mintPubkey,
+          TOKEN_PROGRAM_ID,
+          ASSOCIATED_TOKEN_PROGRAM_ID
+        )
+      : null;
+
+    if (createAtaIx) {
+      console.log('  User token account does not exist, will create it');
+    } else {
+      console.log('  User token account already exists');
+    }
+
+    // 6. 构建 buy 指令
+    console.log('Step 5: Building buy instruction...');
+    const buyIx = await this.sdk.program.methods
+      .buy(buyTokenAmount, maxSolAmount)
+      .accounts({
+        payer: payer,
+        mintAccount: mintPubkey,
+        curveAccount: createResult.accounts.curveAccount,
+        poolTokenAccount: createResult.accounts.poolTokenAccount,
+        poolSolAccount: createResult.accounts.poolSolAccount,
+        upOrderbook: upOrderbook,
+        downOrderbook: downOrderbook,
+        userTokenAccount: userTokenAccount,
+        tokenProgram: TOKEN_PROGRAM_ID,
+        systemProgram: SystemProgram.programId,
+        rent: SYSVAR_RENT_PUBKEY,
+        associatedTokenProgram: ASSOCIATED_TOKEN_PROGRAM_ID,
+        feeRecipientAccount: feeRecipientAccount,
+        baseFeeRecipientAccount: baseFeeRecipientAccount,
+        cooldown: cooldownPDA
+      })
+      .instruction();
+
+    // 7. 合并交易：create + buy
+    console.log('Step 6: Merging create and buy transactions...');
+    const transaction = new Transaction();
+
+    // 设置计算单元限制
+    const modifyComputeUnits = ComputeBudgetProgram.setComputeUnitLimit({
+      units: computeUnits
+    });
+    transaction.add(modifyComputeUnits);
+
+    // 添加 create 交易的所有指令（跳过 create 中的计算单元指令）
+    createResult.transaction.instructions.forEach(ix => {
+      // 跳过 create 交易中的计算单元指令（我们已经添加了）
+      if (ix.programId.equals(ComputeBudgetProgram.programId)) {
+        return;
+      }
+      transaction.add(ix);
+    });
+
+    // 添加 ATA 创建指令（如果需要）
+    if (createAtaIx) {
+      transaction.add(createAtaIx);
+    }
+
+    // 添加 buy 指令 
+    transaction.add(buyIx);
+
+    console.log('CreateAndBuy transaction built successfully:');
+    console.log('  Total instructions:', transaction.instructions.length);
+    console.log('  Compute units:', computeUnits);
+    console.log('  Signers required:', [payer.toString(), mint.publicKey.toString()]);
+
+    // 8. 返回合并后的交易
+    return {
+      transaction,
+      signers: [mint],  // mint keypair 需要签名
+      accounts: {
+        // create 的账户
+        ...createResult.accounts,
+        // buy 的账户
+        userTokenAccount,
+        cooldown: cooldownPDA,
+        feeRecipientAccount,
+        baseFeeRecipientAccount
+      }
+    };
+  }
+
+}
+
+module.exports = TokenModule;