Anton Sauchyk
03a4e7bcbc
* feat: mayhem update in idl * feat(examples): update bonding curve scripts * feat(example): update listen_blocksubscribe * feat(examples): update geyser listener * feat(examples): update all new token listeners * feat(examples): add comments, fix printing, formatting * feat(examples): pumpswap buy and sell update with mayhem mode * fix(examples): sell pump amm fee recipient * feat(examples): update decode scripts * feat(examples): update fetch price * feat(examples): buy and sell bonding curve scripts * feat(examples): add mint with mayhem mode enabled * feat(examples): improve listening to wallet txs * feat(examples): migration listener improvements * feat(examples): global vol accumulator is not writable * feat(examples): support token/token2022 programs in buy instructions * feat(examples): token/token2022 for pumpswap buy * feat(examples): token/token2022 supprot for sell instructions * feat(bot): support create_v2 with token2022, mayhem mode, other fixes * fix(bot): support only token2022 in logs and pumportal listeners * feat(bot): token2022 support in cleanup flow * fix(bot): update token program handling and improve price validation in trading logic * feat(bot): enhance token program handling for LetsBonk integration
117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
import asyncio
|
|
|
|
from solders.pubkey import Pubkey
|
|
from spl.token.instructions import BurnParams, CloseAccountParams, burn, close_account
|
|
|
|
from core.client import SolanaClient
|
|
from core.priority_fee.manager import PriorityFeeManager
|
|
from core.pubkeys import SystemAddresses
|
|
from core.wallet import Wallet
|
|
from utils.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class AccountCleanupManager:
|
|
"""Handles safe cleanup of token accounts (ATA) after trading sessions."""
|
|
|
|
def __init__(
|
|
self,
|
|
client: SolanaClient,
|
|
wallet: Wallet,
|
|
priority_fee_manager: PriorityFeeManager,
|
|
use_priority_fee: bool = False,
|
|
force_burn: bool = False,
|
|
):
|
|
"""
|
|
Args:
|
|
client: Solana RPC client
|
|
wallet: Wallet for signing transactions
|
|
"""
|
|
self.client = client
|
|
self.wallet = wallet
|
|
self.priority_fee_manager = priority_fee_manager
|
|
self.use_priority_fee = use_priority_fee
|
|
self.close_with_force_burn = force_burn
|
|
|
|
async def cleanup_ata(self, mint: Pubkey, token_program_id: Pubkey | None = None) -> None:
|
|
"""
|
|
Attempt to burn any remaining tokens and close the ATA.
|
|
Skips if account doesn't exist or is already empty/closed.
|
|
|
|
Args:
|
|
mint: Token mint address
|
|
token_program_id: Token program (TOKEN or TOKEN_2022). Defaults to TOKEN_2022_PROGRAM
|
|
"""
|
|
if token_program_id is None:
|
|
token_program_id = SystemAddresses.TOKEN_2022_PROGRAM
|
|
|
|
ata = self.wallet.get_associated_token_address(mint, token_program_id)
|
|
solana_client = await self.client.get_client()
|
|
|
|
priority_fee = (
|
|
await self.priority_fee_manager.calculate_priority_fee([ata])
|
|
if self.use_priority_fee
|
|
else None
|
|
)
|
|
|
|
logger.info("Waiting for 15 seconds for RPC node to synchronize...")
|
|
await asyncio.sleep(15)
|
|
|
|
try:
|
|
info = await solana_client.get_account_info(ata, encoding="base64")
|
|
if not info.value:
|
|
logger.info(f"ATA {ata} does not exist or already closed.")
|
|
return
|
|
|
|
balance = await self.client.get_token_account_balance(ata)
|
|
instructions = []
|
|
|
|
if balance > 0 and self.close_with_force_burn:
|
|
logger.info(
|
|
f"Burning {balance} tokens from ATA {ata} (mint: {mint})..."
|
|
)
|
|
burn_ix = burn(
|
|
BurnParams(
|
|
account=ata,
|
|
mint=mint,
|
|
owner=self.wallet.pubkey,
|
|
amount=balance,
|
|
program_id=token_program_id,
|
|
)
|
|
)
|
|
instructions.append(burn_ix)
|
|
|
|
elif balance > 0:
|
|
logger.info(
|
|
f"Skipping ATA {ata} with non-zero balance ({balance} tokens) "
|
|
f"because CLEANUP_FORCE_CLOSE_WITH_BURN is disabled."
|
|
)
|
|
return
|
|
|
|
# Include close account instruction
|
|
logger.info(f"Closing ATA: {ata}")
|
|
close_ix = close_account(
|
|
CloseAccountParams(
|
|
account=ata,
|
|
dest=self.wallet.pubkey,
|
|
owner=self.wallet.pubkey,
|
|
program_id=token_program_id,
|
|
)
|
|
)
|
|
instructions.append(close_ix)
|
|
|
|
# Send both burn and close instructions in the same transaction
|
|
if instructions:
|
|
tx_sig = await self.client.build_and_send_transaction(
|
|
instructions,
|
|
self.wallet.keypair,
|
|
skip_preflight=True,
|
|
priority_fee=priority_fee,
|
|
)
|
|
await self.client.confirm_transaction(tx_sig)
|
|
logger.info(f"Closed successfully: {ata}")
|
|
|
|
except Exception as e:
|
|
logger.warning(f"Cleanup failed for ATA {ata}: {e!s}")
|