Add mayhem mode support and Token2022 integration (#149)

* 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

src/cleanup/manager.py

@@ -34,12 +34,19 @@ class AccountCleanupManager:
         self.use_priority_fee = use_priority_fee
         self.close_with_force_burn = force_burn
 
-    async def cleanup_ata(self, mint: Pubkey) -> None:
+    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
         """
-        ata = self.wallet.get_associated_token_address(mint)
+        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 = (
@@ -70,7 +77,7 @@ class AccountCleanupManager:
                         mint=mint,
                         owner=self.wallet.pubkey,
                         amount=balance,
-                        program_id=SystemAddresses.TOKEN_PROGRAM,
+                        program_id=token_program_id,
                     )
                 )
                 instructions.append(burn_ix)
@@ -89,7 +96,7 @@ class AccountCleanupManager:
                     account=ata,
                     dest=self.wallet.pubkey,
                     owner=self.wallet.pubkey,
-                    program_id=SystemAddresses.TOKEN_PROGRAM,
+                    program_id=token_program_id,
                 )
             )
             instructions.append(close_ix)