sol-parser-sdk-python 0.4.4__py3-none-any.whl

sol_parser/account_fillers/orca.py

@@ -0,0 +1,40 @@
+"""Orca Whirlpool 账户填充（对齐 ``account_fillers/orca.rs``）。"""
+
+from __future__ import annotations
+
+from typing import Callable
+
+from ..event_types import (
+    OrcaWhirlpoolLiquidityDecreasedEvent,
+    OrcaWhirlpoolLiquidityIncreasedEvent,
+    OrcaWhirlpoolSwapEvent,
+)
+
+Z = "11111111111111111111111111111111"
+
+
+def _empty(s: str) -> bool:
+    return not s or s == Z
+
+
+AccountGetter = Callable[[int], str]
+
+
+def fill_whirlpool_swap_accounts(_e: OrcaWhirlpoolSwapEvent, _get: AccountGetter) -> None:
+    pass
+
+
+def fill_whirlpool_liquidity_increased_accounts(
+    e: OrcaWhirlpoolLiquidityIncreasedEvent,
+    get: AccountGetter,
+) -> None:
+    if _empty(e.position):
+        e.position = get(3)
+
+
+def fill_whirlpool_liquidity_decreased_accounts(
+    e: OrcaWhirlpoolLiquidityDecreasedEvent,
+    get: AccountGetter,
+) -> None:
+    if _empty(e.position):
+        e.position = get(3)

sol_parser/account_fillers/pumpfun.py

@@ -0,0 +1,97 @@
+"""PumpFun 账户填充（对齐 ``account_fillers/pumpfun.rs``）。"""
+
+from __future__ import annotations
+
+from typing import Callable
+
+from ..event_types import (
+    PumpFunCreateEvent,
+    PumpFunCreateV2TokenEvent,
+    PumpFunMigrateEvent,
+    PumpFunTradeEvent,
+)
+
+Z = "11111111111111111111111111111111"
+
+
+def _empty(s: str) -> bool:
+    return not s or s == Z
+
+
+AccountGetter = Callable[[int], str]
+
+
+def fill_trade_accounts(e: PumpFunTradeEvent, get: AccountGetter) -> None:
+    if e.ix_name in ("buy_v2", "sell_v2", "buy_exact_quote_in_v2"):
+        if _empty(e.user):
+            e.user = get(13)
+        if _empty(e.bonding_curve):
+            e.bonding_curve = get(10)
+        if _empty(e.associated_bonding_curve):
+            e.associated_bonding_curve = get(11)
+        if _empty(e.creator_vault):
+            e.creator_vault = get(16)
+        if _empty(e.token_program):
+            e.token_program = get(3)
+        return
+    if _empty(e.user):
+        e.user = get(6)
+    if _empty(e.bonding_curve):
+        e.bonding_curve = get(3)
+    if _empty(e.associated_bonding_curve):
+        e.associated_bonding_curve = get(4)
+    if _empty(e.creator_vault):
+        e.creator_vault = get(9) if e.is_buy else get(8)
+    if _empty(e.token_program):
+        e.token_program = get(8) if e.is_buy else get(9)
+    a16 = get(16)
+    if not _empty(a16):
+        e.extra_instruction_account = a16
+
+
+def fill_create_accounts(e: PumpFunCreateEvent, get: AccountGetter) -> None:
+    if _empty(e.mint):
+        e.mint = get(0)
+    if _empty(e.bonding_curve):
+        e.bonding_curve = get(2)
+    if _empty(e.user):
+        e.user = get(7)
+
+
+def fill_create_v2_accounts(e: PumpFunCreateV2TokenEvent, get: AccountGetter) -> None:
+    if _empty(e.mint):
+        e.mint = get(0)
+    if _empty(e.bonding_curve):
+        e.bonding_curve = get(2)
+    if _empty(e.user):
+        e.user = get(5)
+    if _empty(e.mint_authority):
+        e.mint_authority = get(1)
+    if _empty(e.associated_bonding_curve):
+        e.associated_bonding_curve = get(3)
+    if _empty(e.global_account):
+        e.global_account = get(4)
+    if _empty(e.system_program):
+        e.system_program = get(6)
+    if _empty(e.token_program):
+        e.token_program = get(7)
+    if _empty(e.associated_token_program):
+        e.associated_token_program = get(8)
+    if _empty(e.mayhem_program_id):
+        e.mayhem_program_id = get(9)
+    if _empty(e.global_params):
+        e.global_params = get(10)
+    if _empty(e.sol_vault):
+        e.sol_vault = get(11)
+    if _empty(e.mayhem_state):
+        e.mayhem_state = get(12)
+    if _empty(e.mayhem_token_vault):
+        e.mayhem_token_vault = get(13)
+    if _empty(e.event_authority):
+        e.event_authority = get(14)
+    if _empty(e.program):
+        e.program = get(15)
+
+
+def fill_migrate_accounts(_e: PumpFunMigrateEvent, _get: AccountGetter) -> None:
+    pass

sol_parser/account_fillers/pumpswap.py

@@ -0,0 +1,93 @@
+"""PumpSwap 账户填充（对齐 ``account_fillers/pumpswap.rs`` 子集，字段与 Python dataclass 一致）。"""
+
+from __future__ import annotations
+
+from typing import Any, Callable
+
+from ..event_types import (
+    PumpSwapBuyEvent,
+    PumpSwapCreatePoolEvent,
+    PumpSwapLiquidityAddedEvent,
+    PumpSwapLiquidityRemovedEvent,
+    PumpSwapSellEvent,
+)
+
+Z = "11111111111111111111111111111111"
+
+
+def _empty(s: str) -> bool:
+    return not s or s == Z
+
+
+AccountGetter = Callable[[int], str]
+
+
+def _fill_trade_common(
+    e: PumpSwapBuyEvent | PumpSwapSellEvent,
+    get: AccountGetter,
+) -> None:
+    if _empty(e.pool):
+        e.pool = get(0)
+    if _empty(e.user):
+        e.user = get(1)
+    if _empty(e.base_mint):
+        e.base_mint = get(3)
+    if _empty(e.quote_mint):
+        e.quote_mint = get(4)
+    if _empty(e.user_base_token_account):
+        e.user_base_token_account = get(5)
+    if _empty(e.user_quote_token_account):
+        e.user_quote_token_account = get(6)
+    if _empty(e.pool_base_token_account):
+        e.pool_base_token_account = get(7)
+    if _empty(e.pool_quote_token_account):
+        e.pool_quote_token_account = get(8)
+    if _empty(e.protocol_fee_recipient):
+        e.protocol_fee_recipient = get(9)
+    if _empty(e.protocol_fee_recipient_token_account):
+        e.protocol_fee_recipient_token_account = get(10)
+    if _empty(e.base_token_program):
+        e.base_token_program = get(11)
+    if _empty(e.quote_token_program):
+        e.quote_token_program = get(12)
+    if _empty(e.coin_creator_vault_ata):
+        e.coin_creator_vault_ata = get(17)
+    if _empty(e.coin_creator_vault_authority):
+        e.coin_creator_vault_authority = get(18)
+
+
+def fill_buy_accounts(e: PumpSwapBuyEvent, get: AccountGetter) -> None:
+    _fill_trade_common(e, get)
+
+
+def fill_sell_accounts(e: PumpSwapSellEvent, get: AccountGetter) -> None:
+    _fill_trade_common(e, get)
+
+
+def fill_trade_accounts(_e: Any, _get: AccountGetter) -> None:
+    pass
+
+
+def fill_create_pool_accounts(e: PumpSwapCreatePoolEvent, get: AccountGetter) -> None:
+    if _empty(e.pool):
+        e.pool = get(0)
+    if _empty(e.creator):
+        e.creator = get(2)
+    if _empty(e.base_mint):
+        e.base_mint = get(3)
+    if _empty(e.quote_mint):
+        e.quote_mint = get(4)
+    if _empty(e.lp_mint):
+        e.lp_mint = get(5)
+    if _empty(e.user_base_token_account):
+        e.user_base_token_account = get(6)
+    if _empty(e.user_quote_token_account):
+        e.user_quote_token_account = get(7)
+
+
+def fill_liquidity_added_accounts(_e: PumpSwapLiquidityAddedEvent, _get: AccountGetter) -> None:
+    pass
+
+
+def fill_liquidity_removed_accounts(_e: PumpSwapLiquidityRemovedEvent, _get: AccountGetter) -> None:
+    pass

sol_parser/account_fillers/raydium.py

@@ -0,0 +1,119 @@
+"""Raydium 账户填充（对齐 ``account_fillers/raydium.rs``）。"""
+
+from __future__ import annotations
+
+from typing import Callable
+
+from ..event_types import (
+    RaydiumAmmV4DepositEvent,
+    RaydiumAmmV4SwapEvent,
+    RaydiumAmmV4WithdrawEvent,
+    RaydiumClmmCreatePoolEvent,
+    RaydiumClmmDecreaseLiquidityEvent,
+    RaydiumClmmIncreaseLiquidityEvent,
+    RaydiumClmmClosePositionEvent,
+    RaydiumClmmOpenPositionEvent,
+    RaydiumClmmOpenPositionWithTokenExtNftEvent,
+    RaydiumClmmSwapEvent,
+    RaydiumCpmmDepositEvent,
+    RaydiumCpmmInitializeEvent,
+    RaydiumCpmmSwapEvent,
+    RaydiumCpmmWithdrawEvent,
+)
+
+Z = "11111111111111111111111111111111"
+
+
+def _empty(s: str) -> bool:
+    return not s or s == Z
+
+
+AccountGetter = Callable[[int], str]
+
+
+def fill_clmm_swap_accounts(e: RaydiumClmmSwapEvent, get: AccountGetter) -> None:
+    if _empty(e.pool_state):
+        e.pool_state = get(2)
+    if _empty(e.sender):
+        e.sender = get(0)
+
+
+def fill_clmm_create_pool_accounts(e: RaydiumClmmCreatePoolEvent, get: AccountGetter) -> None:
+    if _empty(e.creator):
+        e.creator = get(0)
+
+
+def fill_clmm_increase_liquidity_accounts(e: RaydiumClmmIncreaseLiquidityEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+
+
+def fill_clmm_decrease_liquidity_accounts(e: RaydiumClmmDecreaseLiquidityEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+
+
+def fill_clmm_open_position_accounts(e: RaydiumClmmOpenPositionEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+    if _empty(e.position_nft_mint):
+        e.position_nft_mint = get(2)
+
+
+def fill_clmm_open_position_with_token_ext_nft_accounts(
+    e: RaydiumClmmOpenPositionWithTokenExtNftEvent,
+    get: AccountGetter,
+) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+    if _empty(e.position_nft_mint):
+        e.position_nft_mint = get(2)
+
+
+def fill_clmm_close_position_accounts(e: RaydiumClmmClosePositionEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+    if _empty(e.position_nft_mint):
+        e.position_nft_mint = get(1)
+
+
+def fill_cpmm_swap_accounts(_e: RaydiumCpmmSwapEvent, _get: AccountGetter) -> None:
+    pass
+
+
+def fill_cpmm_deposit_accounts(e: RaydiumCpmmDepositEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+
+
+def fill_cpmm_withdraw_accounts(e: RaydiumCpmmWithdrawEvent, get: AccountGetter) -> None:
+    if _empty(e.user):
+        e.user = get(0)
+
+
+def fill_cpmm_initialize_accounts(e: RaydiumCpmmInitializeEvent, get: AccountGetter) -> None:
+    if _empty(e.creator):
+        e.creator = get(0)
+    if _empty(e.pool):
+        e.pool = get(3)
+
+
+def fill_amm_v4_swap_accounts(e: RaydiumAmmV4SwapEvent, get: AccountGetter) -> None:
+    if _empty(e.amm):
+        e.amm = get(1)
+
+
+def fill_amm_v4_deposit_accounts(e: RaydiumAmmV4DepositEvent, get: AccountGetter) -> None:
+    if _empty(e.token_program):
+        e.token_program = get(0)
+    if _empty(e.amm_authority):
+        e.amm_authority = get(2)
+
+
+def fill_amm_v4_withdraw_accounts(e: RaydiumAmmV4WithdrawEvent, get: AccountGetter) -> None:
+    if _empty(e.token_program):
+        e.token_program = get(0)
+    if _empty(e.amm_authority):
+        e.amm_authority = get(2)
+    if _empty(e.amm_open_orders):
+        e.amm_open_orders = get(3)