mirror of
https://github.com/chainstacklabs/pumpfun-bonkfun-bot.git
synced 2026-08-06 20:17:44 +00:00
feat(pumpfun): migrate to buy_v2/sell_v2 and support non-SOL quote assets (#176)
Refresh the vendored IDLs from pump-fun/pump-public-docs @ 9c82f61 and move all pump.fun trading onto the v2 instruction interface. This is required, not optional: legacy buy/sell cannot trade coins paired against a quote asset other than SOL, and USDC is already whitelisted in the on-chain Global account. Protocol changes absorbed: - buy_v2 (27 accounts) / sell_v2 (26 accounts) replace the legacy instructions. Every account is mandatory and the order is identical for all coins, so the conditional cashback/mayhem account lists are gone. Legacy remains available via PumpFunInstructionBuilder(use_legacy_instructions=True). - BondingCurve is 151 bytes: virtual_sol_reserves -> virtual_quote_reserves, real_sol_reserves -> real_quote_reserves, plus quote_mint at offset 83. Old field names are kept as aliases so existing callers keep working. - v2 instruction data drops the track_volume OptionBool; amounts are in the quote mint's raw units rather than always lamports. - create_v2 carries a non-SOL quote mint as optional remaining accounts 17-19, and CreateEvent gained quote_mint, so extreme_fast_mode can resolve the quote asset without an extra fetch. USDC support: new trade.quote_amounts and filters.allowed_quote_mints config, accepting "sol"/"usdc" aliases or raw mints. Amounts are per-quote-mint because 1 USDC and 1 SOL are not interchangeable. A coin whose quote mint has no configured amount is skipped rather than traded at the wrong size, so SOL-only configs are unaffected. Bug fixes found while verifying: - The logs and blocks listeners set no websocket max_size, so any frame over 1 MiB closed the connection with 1009 and the token in it was lost. Raised to 32 MiB. - PumpSwap priced against the raw quote vault balance, ignoring the new Pool.virtual_quote_reserves (i128 at offset 245; live pools are 301 bytes). Upstream's note that this field is 0 everywhere is out of date: a live pool carries 17.58 SOL against a 148 SOL vault, a 10.15% price error. - The seller read curve state once at confirmed commitment and silently fell back to create-time values, risking a stale creator_vault and ConstraintSeeds. It now retries at processed, matching the buyer. - Account cleanup would burn wrapped SOL when force_burn was set, destroying value that closing the account returns. WSOL is now closed without burning. - The mint scripts treated a landed transaction as a successful one, so a reverted buy printed as success. They now assert the on-chain result. Compute unit limits retuned from mainnet measurements: buy 100k -> 180k, sell 60k -> 120k. Mint-and-buy is no longer atomic, because create_v2 plus buy_v2 exceeds the 1232-byte transaction limit; both mint scripts send two transactions. Adds learning-examples/pump_v2.py as one shared, standalone v2 toolkit for the example scripts, and three verification scripts: an offline layout check against the IDL, a no-funds mainnet simulation, and a live listener matrix that buys, sells and closes the ATA per listener. Verified on mainnet: all four listeners (geyser, logs, blocks, pumpportal) and all eight example scripts completed a real buy, sell and ATA close, each confirmed by reading the transaction result back rather than trusting confirmation alone. The USDC path is verified structurally only; no USDC-paired coin could be found on-chain to exercise it. Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
Anton Sauchyk
@@ -0,0 +1,230 @@
|
||||
"""Dry-run the bot's real buy path against a freshly created coin.
|
||||
|
||||
Detects a new pump.fun coin with the bot's own listener, runs the same
|
||||
PlatformAwareBuyer code the bot uses (including the extreme_fast_mode curve
|
||||
refresh), but intercepts the transaction just before submission and simulates it
|
||||
instead. This exercises the listener -> event parser -> curve manager ->
|
||||
address provider -> instruction builder chain as a unit.
|
||||
|
||||
No funds move: `build_and_send_transaction` is monkeypatched to simulate.
|
||||
|
||||
Usage:
|
||||
uv run learning-examples/simulate_bot_buy_path.py
|
||||
uv run learning-examples/simulate_bot_buy_path.py --no-extreme-fast
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
from base64 import b64encode
|
||||
from pathlib import Path
|
||||
|
||||
PROJECT_ROOT = Path(__file__).resolve().parent.parent
|
||||
sys.path.insert(0, str(PROJECT_ROOT / "src"))
|
||||
|
||||
from dotenv import load_dotenv # noqa: E402
|
||||
from solders.compute_budget import set_compute_unit_limit, set_compute_unit_price # noqa: E402
|
||||
from solders.message import Message # noqa: E402
|
||||
from solders.transaction import Transaction # noqa: E402
|
||||
|
||||
from core.client import SolanaClient # noqa: E402
|
||||
from core.priority_fee.manager import PriorityFeeManager # noqa: E402
|
||||
from core.wallet import Wallet # noqa: E402
|
||||
from interfaces.core import Platform, TokenInfo # noqa: E402
|
||||
from monitoring.listener_factory import ListenerFactory # noqa: E402
|
||||
from trading.platform_aware import PlatformAwareBuyer # noqa: E402
|
||||
|
||||
load_dotenv(PROJECT_ROOT / ".env")
|
||||
|
||||
BUY_AMOUNT_SOL = 0.0001
|
||||
EXTREME_FAST_TOKEN_AMOUNT = 20
|
||||
# Matches retries.wait_after_creation in the bot configs. Only used when
|
||||
# extreme_fast_mode is off, where the buyer reads the curve at `confirmed`.
|
||||
CURVE_STABILIZE_SECONDS = 15
|
||||
|
||||
|
||||
async def wait_for_token(timeout_seconds: float = 90.0) -> TokenInfo | None:
|
||||
"""Wait for the bot's geyser listener to report a new coin.
|
||||
|
||||
Args:
|
||||
timeout_seconds: How long to wait
|
||||
|
||||
Returns:
|
||||
The first TokenInfo seen, or None on timeout
|
||||
"""
|
||||
listener = ListenerFactory.create_listener(
|
||||
listener_type="geyser",
|
||||
geyser_endpoint=os.environ["GEYSER_ENDPOINT"],
|
||||
geyser_api_token=os.environ["GEYSER_API_TOKEN"],
|
||||
geyser_auth_type=os.environ.get("GEYSER_AUTH_TYPE", "x-token"),
|
||||
platforms=[Platform.PUMP_FUN],
|
||||
)
|
||||
|
||||
seen: list[TokenInfo] = []
|
||||
|
||||
async def on_token(token_info: TokenInfo) -> None:
|
||||
seen.append(token_info)
|
||||
|
||||
task = asyncio.create_task(listener.listen_for_tokens(on_token))
|
||||
try:
|
||||
for _ in range(int(timeout_seconds / 0.5)):
|
||||
if seen:
|
||||
break
|
||||
await asyncio.sleep(0.5)
|
||||
finally:
|
||||
task.cancel()
|
||||
|
||||
return seen[0] if seen else None
|
||||
|
||||
|
||||
def install_simulation_hook(client: SolanaClient) -> dict:
|
||||
"""Replace transaction submission with simulation.
|
||||
|
||||
Args:
|
||||
client: Client whose send path should be intercepted
|
||||
|
||||
Returns:
|
||||
Dict that will be populated with the simulation outcome
|
||||
"""
|
||||
outcome: dict = {}
|
||||
|
||||
async def simulate_instead(
|
||||
instructions,
|
||||
signer_keypair,
|
||||
skip_preflight=True,
|
||||
max_retries=3,
|
||||
priority_fee=None,
|
||||
compute_unit_limit=None,
|
||||
account_data_size_limit=None,
|
||||
):
|
||||
preamble = []
|
||||
if compute_unit_limit:
|
||||
preamble.append(set_compute_unit_limit(compute_unit_limit))
|
||||
if priority_fee:
|
||||
preamble.append(set_compute_unit_price(priority_fee))
|
||||
|
||||
blockhash = await client.get_latest_blockhash()
|
||||
message = Message.new_with_blockhash(
|
||||
[*preamble, *instructions], signer_keypair.pubkey(), blockhash
|
||||
)
|
||||
transaction = Transaction([signer_keypair], message, blockhash)
|
||||
|
||||
response = await client.post_rpc(
|
||||
{
|
||||
"jsonrpc": "2.0",
|
||||
"id": 1,
|
||||
"method": "simulateTransaction",
|
||||
"params": [
|
||||
b64encode(bytes(transaction)).decode(),
|
||||
{
|
||||
"encoding": "base64",
|
||||
"sigVerify": False,
|
||||
"replaceRecentBlockhash": True,
|
||||
"commitment": "processed",
|
||||
},
|
||||
],
|
||||
}
|
||||
)
|
||||
value = (response or {}).get("result", {}).get("value", {})
|
||||
outcome.update(
|
||||
{
|
||||
"err": value.get("err"),
|
||||
"units": value.get("unitsConsumed"),
|
||||
"logs": value.get("logs") or [],
|
||||
"cu_limit": compute_unit_limit,
|
||||
"priority_fee": priority_fee,
|
||||
"instruction_count": len(instructions),
|
||||
"account_count": len(instructions[-1].accounts),
|
||||
}
|
||||
)
|
||||
# Returning a sentinel signature: confirm_transaction is stubbed below.
|
||||
return "SIMULATED"
|
||||
|
||||
async def never_confirm(_signature, **_kwargs):
|
||||
return False
|
||||
|
||||
client.build_and_send_transaction = simulate_instead
|
||||
client.confirm_transaction = never_confirm
|
||||
return outcome
|
||||
|
||||
|
||||
async def main() -> int:
|
||||
"""Run the bot's buy path in simulation mode.
|
||||
|
||||
Returns:
|
||||
Process exit code (0 if the simulated buy had no program error)
|
||||
"""
|
||||
extreme_fast = "--no-extreme-fast" not in sys.argv
|
||||
|
||||
print("Waiting for a fresh pump.fun coin via the bot's geyser listener...")
|
||||
token_info = await wait_for_token()
|
||||
if token_info is None:
|
||||
print("No coin detected before timeout.")
|
||||
return 2
|
||||
|
||||
print(f"\ndetected: {token_info.symbol} ({token_info.mint})")
|
||||
print(f"quote_mint (from CreateEvent): {token_info.quote_mint}")
|
||||
print(f"token program: {token_info.token_program_id}")
|
||||
print(f"mayhem={token_info.is_mayhem_mode} cashback={token_info.is_cashback_coin}")
|
||||
print(f"extreme_fast_mode={extreme_fast}\n")
|
||||
|
||||
client = SolanaClient(os.environ["SOLANA_NODE_RPC_ENDPOINT"])
|
||||
wallet = Wallet(os.environ["SOLANA_PRIVATE_KEY"])
|
||||
priority_fee_manager = PriorityFeeManager(
|
||||
client=client,
|
||||
enable_dynamic_fee=False,
|
||||
enable_fixed_fee=True,
|
||||
fixed_fee=1_000_000,
|
||||
extra_fee=0.0,
|
||||
hard_cap=1_000_000,
|
||||
)
|
||||
|
||||
outcome = install_simulation_hook(client)
|
||||
buyer = PlatformAwareBuyer(
|
||||
client,
|
||||
wallet,
|
||||
priority_fee_manager,
|
||||
BUY_AMOUNT_SOL,
|
||||
slippage=0.3,
|
||||
max_retries=1,
|
||||
extreme_fast_token_amount=EXTREME_FAST_TOKEN_AMOUNT,
|
||||
extreme_fast_mode=extreme_fast,
|
||||
)
|
||||
|
||||
if not extreme_fast:
|
||||
# Mirror the bot's retries.wait_after_creation pause. Without it the
|
||||
# curve read races the account's confirmation and fails before any
|
||||
# instruction is built.
|
||||
print(f"Waiting {CURVE_STABILIZE_SECONDS}s for the curve to stabilize...")
|
||||
await asyncio.sleep(CURVE_STABILIZE_SECONDS)
|
||||
|
||||
try:
|
||||
result = await buyer.execute(token_info)
|
||||
finally:
|
||||
await client.close()
|
||||
|
||||
if not outcome:
|
||||
print(f"Buy path never reached transaction submission: {result.error_message}")
|
||||
return 1
|
||||
|
||||
print("simulated buy:")
|
||||
print(f" instructions: {outcome['instruction_count']}")
|
||||
print(f" trade accounts: {outcome['account_count']}")
|
||||
print(f" cu_limit: {outcome['cu_limit']}")
|
||||
print(f" unitsConsumed: {outcome['units']}")
|
||||
print(f" err: {outcome['err']}")
|
||||
|
||||
if outcome["err"]:
|
||||
for line in outcome["logs"]:
|
||||
if "Error" in line or "failed" in line or "Instruction:" in line:
|
||||
print(f" {line}")
|
||||
return 1
|
||||
|
||||
headroom = outcome["cu_limit"] - (outcome["units"] or 0)
|
||||
print(f"\nCU headroom: {headroom} ({headroom / outcome['cu_limit']:.0%})")
|
||||
print("Buy path validated end to end against live mainnet state.")
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(asyncio.run(main()))
|
||||
Reference in New Issue
Block a user