@pump-fun/pump-sdk 1.17.6 → 1.18.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pump-fun/pump-sdk",
-  "version": "1.17.6",
+  "version": "1.18.0",
   "description": "Pump Bonding Curve SDK",
   "keywords": [],
   "homepage": "https://github.com/pump-fun/pump-sdk#readme",

package/src/bondingCurve.ts

@@ -1,27 +1,7 @@
-import { PublicKey } from "@solana/web3.js";
-import { BondingCurve, Global } from "./state";
 import BN from "bn.js";
-
-function getFee(
-  global: Global,
-  bondingCurve: BondingCurve,
-  amount: BN,
-  isNewBondingCurve: boolean,
-): BN {
-  return computeFee(amount, global.feeBasisPoints).add(
-    isNewBondingCurve || !PublicKey.default.equals(bondingCurve.creator)
-      ? computeFee(amount, global.creatorFeeBasisPoints)
-      : new BN(0),
-  );
-}
-
-function computeFee(amount: BN, feeBasisPoints: BN): BN {
-  return ceilDiv(amount.mul(feeBasisPoints), new BN(10_000));
-}
-
-function ceilDiv(a: BN, b: BN): BN {
-  return a.add(b.subn(1)).div(b);
-}
+import { PublicKey } from "@solana/web3.js";
+import { BondingCurve, FeeConfig, Global } from "./state";
+import { computeFeesBps, getFee } from "./fees";
 
 export function newBondingCurve(global: Global): BondingCurve {
   return {
@@ -78,11 +58,19 @@ export function getSellSolAmountFromTokenAmountQuote({
     .div(virtualTokenReserves.add(inputAmount));
 }
 
-export function getBuyTokenAmountFromSolAmount(
-  global: Global,
-  bondingCurve: BondingCurve | null,
-  amount: BN,
-): BN {
+export function getBuyTokenAmountFromSolAmount({
+  global,
+  feeConfig,
+  mintSupply,
+  bondingCurve,
+  amount,
+}: {
+  global: Global;
+  feeConfig: FeeConfig | null;
+  mintSupply: BN | null;
+  bondingCurve: BondingCurve | null;
+  amount: BN;
+}): BN {
   if (amount.eq(new BN(0))) {
     return new BN(0);
   }
@@ -99,9 +87,18 @@ export function getBuyTokenAmountFromSolAmount(
     return new BN(0);
   }
 
-  const totalFeeBasisPoints = global.feeBasisPoints.add(
+  const { virtualSolReserves, virtualTokenReserves } = bondingCurve;
+  const { protocolFeeBps, creatorFeeBps } = computeFeesBps({
+    global,
+    feeConfig,
+    mintSupply,
+    virtualSolReserves,
+    virtualTokenReserves,
+  });
+
+  const totalFeeBasisPoints = protocolFeeBps.add(
     isNewBondingCurve || !PublicKey.default.equals(bondingCurve.creator)
-      ? global.creatorFeeBasisPoints
+      ? creatorFeeBps
       : new BN(0),
   );
 
@@ -116,11 +113,19 @@ export function getBuyTokenAmountFromSolAmount(
   return BN.min(tokensReceived, bondingCurve.realTokenReserves);
 }
 
-export function getBuySolAmountFromTokenAmount(
-  global: Global,
-  bondingCurve: BondingCurve | null,
-  amount: BN,
-): BN {
+export function getBuySolAmountFromTokenAmount({
+  global,
+  feeConfig,
+  mintSupply,
+  bondingCurve,
+  amount,
+}: {
+  global: Global;
+  feeConfig: FeeConfig | null;
+  mintSupply: BN | null;
+  bondingCurve: BondingCurve | null;
+  amount: BN;
+}): BN {
   if (amount.eq(new BN(0))) {
     return new BN(0);
   }
@@ -145,14 +150,29 @@ export function getBuySolAmountFromTokenAmount(
     virtualSolReserves: bondingCurve.virtualSolReserves,
   });
 
-  return solCost.add(getFee(global, bondingCurve, solCost, isNewBondingCurve));
+  return solCost.add(getFee({
+    global,
+    feeConfig,
+    mintSupply,
+    bondingCurve,
+    amount: solCost,
+    isNewBondingCurve,
+  }));
 }
 
-export function getSellSolAmountFromTokenAmount(
-  global: Global,
-  bondingCurve: BondingCurve,
-  amount: BN,
-): BN {
+export function getSellSolAmountFromTokenAmount({
+  global,
+  feeConfig,
+  mintSupply,
+  bondingCurve,
+  amount,
+}: {
+  global: Global;
+  feeConfig: FeeConfig | null;
+  mintSupply: BN | null;
+  bondingCurve: BondingCurve;
+  amount: BN;
+}): BN {
   if (amount.eq(new BN(0))) {
     return new BN(0);
   }
@@ -168,7 +188,14 @@ export function getSellSolAmountFromTokenAmount(
     virtualSolReserves: bondingCurve.virtualSolReserves,
   });
 
-  return solCost.sub(getFee(global, bondingCurve, solCost, false));
+  return solCost.sub(getFee({
+    global,
+    feeConfig,
+    mintSupply,
+    bondingCurve,
+    amount: solCost,
+    isNewBondingCurve: false,
+  }));
 }
 
 export function getStaticRandomFeeRecipient(): PublicKey {
@@ -186,3 +213,18 @@ const CURRENT_FEE_RECIPIENTS = [
   "FWsW1xNtWscwNmKv6wVsU1iTzRN6wmmk3MjxRP5tT7hz",
   "G5UZAVbAf46s7cKWoyKu8kYTip9DGTpbLZ2qa9Aq69dP",
 ];
+
+export function bondingCurveMarketCap({
+  mintSupply,
+  virtualSolReserves,
+  virtualTokenReserves,
+}: {
+  mintSupply: BN;
+  virtualSolReserves: BN;
+  virtualTokenReserves: BN;
+}): BN {
+  if (virtualTokenReserves.isZero()) {
+    throw new Error("Division by zero: virtual token reserves cannot be zero");
+  }
+  return virtualSolReserves.mul(mintSupply).div(virtualTokenReserves);
+}

package/src/fees.ts

@@ -0,0 +1,118 @@
+import BN from "bn.js";
+import { PublicKey } from "@solana/web3.js";
+import { FeeConfig, Global, Fees, BondingCurve, FeeTier } from "./state";
+import { bondingCurveMarketCap } from "./bondingCurve";
+
+export interface CalculatedFeesBps {
+    protocolFeeBps: BN;
+    creatorFeeBps: BN;
+}
+
+export interface CalculatedFees {
+    protocolFee: BN;
+    creatorFee: BN;
+}
+
+export function createFeeConfigFromGlobalConfig(globalConfig: Global): FeeConfig {
+    let fees: Fees = {
+        lpFeeBps: new BN(0), // unused for pump
+        protocolFeeBps: globalConfig.feeBasisPoints,
+        creatorFeeBps: globalConfig.creatorFeeBasisPoints,
+    };
+    return {
+        admin: globalConfig.authority,
+        flatFees: fees,
+        feeTiers: [
+            {
+                marketCapLamportsThreshold: new BN(0), // unused for pump
+                fees,
+            },
+        ],
+    };
+}
+
+export function getFee({
+    global,
+    feeConfig,
+    mintSupply,
+    bondingCurve,
+    amount,
+    isNewBondingCurve,
+}: {
+    global: Global;
+    feeConfig: FeeConfig | null;
+    mintSupply: BN | null;
+    bondingCurve: BondingCurve;
+    amount: BN;
+    isNewBondingCurve: boolean;
+}) {
+    const { virtualSolReserves, virtualTokenReserves } = bondingCurve;
+    const { protocolFeeBps, creatorFeeBps } = computeFeesBps({
+        global,
+        feeConfig,
+        mintSupply,
+        virtualSolReserves,
+        virtualTokenReserves,
+    });
+
+    return fee(amount, protocolFeeBps).add(
+        isNewBondingCurve || !PublicKey.default.equals(bondingCurve.creator) ? fee(amount, creatorFeeBps) : new BN(0)
+    );
+}
+
+export function computeFeesBps({
+    global,
+    feeConfig,
+    mintSupply,
+    virtualSolReserves,
+    virtualTokenReserves,
+}: {
+    global: Global;
+    feeConfig: FeeConfig | null;
+    mintSupply: BN | null;
+    virtualSolReserves: BN;
+    virtualTokenReserves: BN;
+}): CalculatedFeesBps {
+    if (feeConfig != null && mintSupply != null) {
+        const marketCap = bondingCurveMarketCap({
+            mintSupply,
+            virtualSolReserves,
+            virtualTokenReserves,
+        });
+
+        return calculateFeeTier({
+            feeTiers: feeConfig.feeTiers,
+            marketCap,
+        });
+    }
+
+    return {
+        protocolFeeBps: global.feeBasisPoints,
+        creatorFeeBps: global.creatorFeeBasisPoints,
+    };
+}
+
+/// rust reference: pump-fees-math::calculate_fee_tier()
+export function calculateFeeTier({ feeTiers, marketCap }: { feeTiers: FeeTier[]; marketCap: BN }): Fees {
+    const firstTier = feeTiers[0];
+
+    if (marketCap.lt(firstTier.marketCapLamportsThreshold)) {
+        return firstTier.fees;
+    }
+
+    for (const tier of feeTiers.slice().reverse()) {
+        if (marketCap.gte(tier.marketCapLamportsThreshold)) {
+            return tier.fees;
+        }
+    }
+
+    return firstTier.fees;
+}
+
+function fee(amount: BN, feeBasisPoints: BN): BN {
+    return ceilDiv(amount.mul(feeBasisPoints), new BN(10_000));
+}
+
+function ceilDiv(a: BN, b: BN): BN {
+    return a.add(b.subn(1)).div(b);
+}

package/src/idl/pump.json

@@ -656,7 +656,8 @@
           }
         },
         {
-          "name": "program"
+          "name": "program",
+          "address": "6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P"
         },
         {
           "name": "global_volume_accumulator",
@@ -735,6 +736,75 @@
               }
             ]
           }
+        },
+        {
+          "name": "fee_config",
+          "optional": true,
+          "pda": {
+            "seeds": [
+              {
+                "kind": "const",
+                "value": [
+                  102,
+                  101,
+                  101,
+                  95,
+                  99,
+                  111,
+                  110,
+                  102,
+                  105,
+                  103
+                ]
+              },
+              {
+                "kind": "const",
+                "value": [
+                  1,
+                  86,
+                  224,
+                  246,
+                  147,
+                  102,
+                  90,
+                  207,
+                  68,
+                  219,
+                  21,
+                  104,
+                  191,
+                  23,
+                  91,
+                  170,
+                  81,
+                  137,
+                  203,
+                  151,
+                  245,
+                  210,
+                  255,
+                  59,
+                  101,
+                  93,
+                  43,
+                  182,
+                  253,
+                  109,
+                  24,
+                  176
+                ]
+              }
+            ],
+            "program": {
+              "kind": "account",
+              "path": "fee_program"
+            }
+          }
+        },
+        {
+          "name": "fee_program",
+          "optional": true,
+          "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
         }
       ],
       "args": [
@@ -1010,7 +1080,8 @@
           }
         },
         {
-          "name": "program"
+          "name": "program",
+          "address": "6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P"
         },
         {
           "name": "payer",
@@ -2492,7 +2563,77 @@
           }
         },
         {
-          "name": "program"
+          "name": "program",
+          "address": "6EF8rrecthR5Dkzon8Nwu78hRvfCKubJ14M5uBEwF6P"
+        },
+        {
+          "name": "fee_config",
+          "optional": true,
+          "pda": {
+            "seeds": [
+              {
+                "kind": "const",
+                "value": [
+                  102,
+                  101,
+                  101,
+                  95,
+                  99,
+                  111,
+                  110,
+                  102,
+                  105,
+                  103
+                ]
+              },
+              {
+                "kind": "const",
+                "value": [
+                  1,
+                  86,
+                  224,
+                  246,
+                  147,
+                  102,
+                  90,
+                  207,
+                  68,
+                  219,
+                  21,
+                  104,
+                  191,
+                  23,
+                  91,
+                  170,
+                  81,
+                  137,
+                  203,
+                  151,
+                  245,
+                  210,
+                  255,
+                  59,
+                  101,
+                  93,
+                  43,
+                  182,
+                  253,
+                  109,
+                  24,
+                  176
+                ]
+              }
+            ],
+            "program": {
+              "kind": "account",
+              "path": "fee_program"
+            }
+          }
+        },
+        {
+          "name": "fee_program",
+          "optional": true,
+          "address": "pfeeUxB6jkeY1Hxd7CsFCAjcbHA9rWtchMGdZ6VojVZ"
         }
       ],
       "args": [
@@ -3243,6 +3384,19 @@
         96
       ]
     },
+    {
+      "name": "FeeConfig",
+      "discriminator": [
+        143,
+        52,
+        146,
+        187,
+        219,
+        123,
+        76,
+        155
+      ]
+    },
     {
       "name": "Global",
       "discriminator": [
@@ -4048,6 +4202,80 @@
         ]
       }
     },
+    {
+      "name": "FeeConfig",
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "bump",
+            "type": "u8"
+          },
+          {
+            "name": "admin",
+            "type": "pubkey"
+          },
+          {
+            "name": "flat_fees",
+            "type": {
+              "defined": {
+                "name": "Fees"
+              }
+            }
+          },
+          {
+            "name": "fee_tiers",
+            "type": {
+              "vec": {
+                "defined": {
+                  "name": "FeeTier"
+                }
+              }
+            }
+          }
+        ]
+      }
+    },
+    {
+      "name": "FeeTier",
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "market_cap_lamports_threshold",
+            "type": "u128"
+          },
+          {
+            "name": "fees",
+            "type": {
+              "defined": {
+                "name": "Fees"
+              }
+            }
+          }
+        ]
+      }
+    },
+    {
+      "name": "Fees",
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "lp_fee_bps",
+            "type": "u64"
+          },
+          {
+            "name": "protocol_fee_bps",
+            "type": "u64"
+          },
+          {
+            "name": "creator_fee_bps",
+            "type": "u64"
+          }
+        ]
+      }
+    },
     {
       "name": "Global",
       "type": {