Add AMM swap helpers and update documentation

Introduces AMM swap helper functions (`ammBuy`, `ammSell`) and supporting modules. Updates README to clarify curve vs AMM swap APIs and usage. Adds new tests and configuration for AMM swaps. Refactors code to separate curve and AMM logic for improved clarity and maintainability.

Author: henryoman

README.md

@@ -24,12 +24,17 @@ A TypeScript SDK for Pump.fun built on **Solana Kit 5.0**. Designed for high-per
 
 - **Solana Kit 5.0** – Built on the latest Solana development framework
 - **Modern transaction patterns** – Optimized for speed and reliability
-- **Human-readable API** – Work with SOL amounts and percentages instead of raw lamports
+- **Unified swaps** – `buy`/`sell` auto-route between bonding curves and AMM pools
+- **Curve helpers** – Direct access to bonding-curve instructions when you need them (`curveBuy`, `curveSell`)
+- **AMM helpers** – Deterministic AMM operations with percentage-aware selling (`ammBuy`, `ammSell`)
 - **Automatic slippage protection** – Built-in guards for buy/sell operations
-- **Launch capabilities** – Create and buy tokens in one atomic transaction
-- **Liquidity operations** – Add/remove liquidity with simple helpers
 - **Full TypeScript support** – Strongly typed throughout with complete type coverage
 
+### Coming Soon
+
+- **Launch & mint helpers** – Create and seed new Pump.fun tokens once integration is ready
+- **Liquidity tooling** – Add/remove liquidity with tested helpers
+
 ---
 
 ## Installation
@@ -50,12 +55,12 @@ import { createSolanaRpc } from "@solana/kit";
 const rpc = createSolanaRpc("https://api.mainnet-beta.solana.com");
 ```
 
-### Buy Tokens
+### Buy Tokens (Curve)
 
 ```ts
-import { buy } from "pump-kit";
+import { curveBuy } from "pump-kit";
 
-await buy({
+await curveBuy({
   user: myWallet,
   mint: "TokenMintAddress",
   solAmount: 0.5,        // 0.5 SOL
@@ -64,21 +69,21 @@ await buy({
 });
 ```
 
-### Sell Tokens
+### Sell Tokens (Curve)
 
 ```ts
-import { sell } from "pump-kit";
+import { curveSell } from "pump-kit";
 
 // Sell specific amount
-await sell({
+await curveSell({
   user: myWallet,
   mint: "TokenMintAddress",
   tokenAmount: 125_000,
   rpc,
 });
 
 // Sell percentage of wallet
-await sell({
+await curveSell({
   user: myWallet,
   mint: "TokenMintAddress",
   useWalletPercentage: true,
@@ -87,68 +92,78 @@ await sell({
 });
 ```
 
-### Create and Buy Token
+### Buy Tokens (AMM)
 
 ```ts
-import { mintWithFirstBuy } from "pump-kit";
-import { generateKeyPair } from "@solana/kit";
+import { ammBuy } from "pump-kit";
 
-const mintKeypair = await generateKeyPair();
-
-const { createInstruction, buyInstruction } = await mintWithFirstBuy({
+await ammBuy({
   user: myWallet,
-  mint: mintKeypair,
-  mintAuthority: myWallet.address,
-  name: "My Token",
-  symbol: "MTK",
-  uri: "https://arweave.net/metadata.json",
-  firstBuyTokenAmount: 1_000_000,
-  firstBuySolBudget: 1.0,
+  mint: "TokenMintAddress",
+  solAmount: 0.5,
+  poolCreator: "CreatorAddress", // optional if auto detection works
   rpc,
 });
 ```
 
-### Quick Helpers
+### Sell Tokens (AMM)
 
 ```ts
-import { quickBuy, quickSell } from "pump-kit";
+import { ammSell } from "pump-kit";
 
-// Get buy instruction
-const buyIx = await quickBuy(myWallet, "TokenMint", 0.25, { rpc });
-
-// Get sell instruction
-const sellIx = await quickSell(myWallet, "TokenMint", 100_000, { rpc });
+await ammSell({
+  user: myWallet,
+  mint: "TokenMintAddress",
+  useWalletPercentage: true,
+  walletPercentage: 100,
+  poolCreator: "CreatorAddress",
+  rpc,
+});
 ```
 
 ---
 
 ## API Reference
 
-### Core Functions
+### Curve Swap Helpers
 
 ```ts
-// Buy tokens
-buy({ user, mint, solAmount, slippageBps?, rpc, ... })
+// Buy tokens on the bonding curve
+curveBuy({ user, mint, solAmount, slippageBps?, rpc, ... })
 
-// Sell tokens
-sell({ user, mint, tokenAmount?, useWalletPercentage?, walletPercentage?, rpc, ... })
+// Sell tokens on the bonding curve
+curveSell({ user, mint, tokenAmount?, useWalletPercentage?, walletPercentage?, rpc, ... })
+```
 
-// Quick helpers (return instructions only)
-quickBuy(wallet, mint, solAmount, { rpc, ...options })
-quickSell(wallet, mint, tokenAmount, { rpc, ...options })
+### AMM Swap Helpers
 
-// Create token with initial buy
-mintWithFirstBuy({ user, mint, name, symbol, uri, firstBuyTokenAmount, firstBuySolBudget, rpc, ... })
+```ts
+// Buy tokens from the AMM pool using a SOL budget
+ammBuy({ user, mint, solAmount, rpc, quoteMint?, poolCreator?, poolAddress? })
+
+// Sell tokens into the AMM pool (supports percentage-based selling)
+ammSell({
+  user,
+  mint,
+  tokenAmount?,
+  useWalletPercentage?,
+  walletPercentage?,
+  rpc,
+  quoteMint?,
+  poolCreator?,
+  poolAddress?,
+})
 ```
 
-### Liquidity Management
+### Coming Soon
 
 ```ts
-// Add liquidity to pool
-addLiquidity({ user, baseMint, quoteMint?, maxBaseAmountIn, maxQuoteAmountIn, rpc, ... })
+// Create token with initial buy
+mintWithFirstBuy({ ... })
 
-// Remove liquidity from pool
-removeLiquidity({ user, baseMint, quoteMint?, lpAmountIn, rpc, ... })
+// Liquidity helpers
+addLiquidity({ ... })
+removeLiquidity({ ... })
 ```
 
 ### Transaction Utilities

idl/pumpfun.idl.json

@@ -1,6 +1,6 @@
 {
   "name": "pump-kit",
-  "version": "0.0.8",
+  "version": "0.1.0",
   "type": "module",
   "exports": {
     ".": {
@@ -29,8 +29,8 @@
     "build:types": "tsc -p tsconfig.build.json",
     "build": "rm -rf dist && bun run build:js && bun run build:types",
     "release:prepare": "bun run build && bun run test",
-    "release:publish": "npm publish --access public",
-    "release:dry-run": "npm publish --access public --dry-run",
+    "release:publish": "bun run npm publish --access public",
+    "release:dry-run": "bun run npm publish --access public --dry-run",
     "codegen": "bun run scripts/codama-generate.ts",
     "test": "bun test",
     "test:unit": "bun test tests/unit",

package.json

@@ -11,7 +11,12 @@ const BPS_DENOMINATOR = 10_000n;
 
 export type BondingCurveState = Pick<
   BondingCurve,
-  "virtualTokenReserves" | "virtualSolReserves" | "realTokenReserves" | "realSolReserves" | "creator"
+  | "virtualTokenReserves"
+  | "virtualSolReserves"
+  | "realTokenReserves"
+  | "realSolReserves"
+  | "creator"
+  | "complete"
 >;
 
 export type FeeStructure = Pick<Fees, "lpFeeBps" | "protocolFeeBps" | "creatorFeeBps">;

src/ammsdk/bondingCurveMath.ts

@@ -10,8 +10,23 @@ import "./config/polyfills";
 // Simple API (recommended - start here!)
 // ============================================================================
 
-export { buy, sell, quickBuy, quickSell } from "./swap";
-export type { BuyParams, SellParams } from "./swap";
+export {
+  buy,
+  sell,
+  curveBuy,
+  curveSell,
+  ammBuy,
+  ammSell,
+} from "./swap";
+
+export type {
+  BuyParams,
+  SellParams,
+  CurveBuyParams,
+  CurveSellParams,
+  AmmBuyParams,
+  AmmSellParams,
+} from "./swap";
 
 export {
   mintWithFirstBuy,

src/index.ts

@@ -138,4 +138,4 @@ export async function buySimple(params: SimpleBuyParams): Promise<Instruction> {
     rpc,
     commitment,
   });
-}
+}

src/recipes/buy.ts

@@ -122,4 +122,4 @@ export function validateMintParams(params: {
   if (!uri.startsWith("http://") && !uri.startsWith("https://") && !uri.startsWith("ipfs://")) {
     throw new Error("Metadata URI must be a valid URL or IPFS link");
   }
-}
+}

src/recipes/mintFirstBuy.ts

@@ -5,8 +5,19 @@
 
 import "./config/polyfills";
 
-export { buy, sell, quickBuy, quickSell } from "./swap";
-export type { BuyParams, SellParams } from "./swap";
+export {
+  buy,
+  sell,
+  curveBuy,
+  curveSell,
+} from "./swap";
+
+export type {
+  BuyParams,
+  SellParams,
+  CurveBuyParams,
+  CurveSellParams,
+} from "./swap";
 
 export { mintWithFirstBuy, validateMintParams } from "./recipes/mintFirstBuy";
 export type { MintWithFirstBuyParams } from "./recipes/mintFirstBuy";