From 6b2ae66c73cd7f3204373c33ae71cc7ad97492b0 Mon Sep 17 00:00:00 2001
From: smypmsa <antonsauchyk@gmail.com>
Date: Tue, 25 Mar 2025 15:58:50 +0000
Subject: [PATCH] feat: add script to close ATA and reclaim rent

---
 learning-examples/cleanup_accounts.py | 87 +++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 learning-examples/cleanup_accounts.py

diff --git a/learning-examples/cleanup_accounts.py b/learning-examples/cleanup_accounts.py
new file mode 100644
index 0000000..d75d4fb
--- /dev/null
+++ b/learning-examples/cleanup_accounts.py
@@ -0,0 +1,87 @@
+import asyncio
+import os
+
+from dotenv import load_dotenv
+from solders.pubkey import Pubkey
+from spl.token.instructions import BurnParams, CloseAccountParams, burn, close_account
+
+from core.client import SolanaClient
+from core.pubkeys import SystemAddresses
+from core.wallet import Wallet
+from utils.logger import get_logger
+
+load_dotenv()
+logger = get_logger(__name__)
+
+RPC_ENDPOINT = os.getenv("SOLANA_NODE_RPC_ENDPOINT")
+PRIVATE_KEY = os.getenv("SOLANA_PRIVATE_KEY")
+
+# Update this address to MINT address of a token you want to close
+MINT_ADDRESS = Pubkey.from_string("9WHpYbqG6LJvfCYfMjvGbyo1wHXgroCrixPb33s2pump")
+
+
+async def close_account_if_exists(client: SolanaClient, wallet: Wallet, account: Pubkey, mint: Pubkey):
+    """Safely close a token account if it exists and reclaim rent."""
+    try:
+        solana_client = await client.get_client()
+        info = await solana_client.get_account_info(account)
+
+        # WARNING: This will permanently burn all tokens in the account before closing it
+        # Closing account is impossible if balance is positive
+        balance = await client.get_token_account_balance(account)
+        if balance > 0:
+            logger.info(f"Burning {balance} tokens from account {account}...")
+            burn_ix = burn(
+                BurnParams(
+                    account=account,
+                    mint=mint,
+                    owner=wallet.pubkey,
+                    amount=balance,
+                    program_id=SystemAddresses.TOKEN_PROGRAM,
+                )
+            )
+            await client.build_and_send_transaction([burn_ix], wallet.keypair)
+            logger.info(f"Burned tokens from {account}")
+
+        # If account exists, attempt to close it
+        if info.value:
+            logger.info(f"Closing account: {account}")
+            close_params = CloseAccountParams(
+                account=account,
+                dest=wallet.pubkey,
+                owner=wallet.pubkey,
+                program_id=SystemAddresses.TOKEN_PROGRAM,
+            )
+            ix = close_account(close_params)
+
+            tx_sig = await client.build_and_send_transaction(
+                [ix],
+                wallet.keypair,
+                skip_preflight=True,
+            )
+            await client.confirm_transaction(tx_sig)
+            logger.info(f"Closed successfully: {account}")
+        else:
+            logger.info(f"Account does not exist or already closed: {account}")
+
+    except Exception as e:
+        logger.error(f"Error while processing account {account}: {e}")
+
+
+async def main():
+    try:
+        client = SolanaClient(RPC_ENDPOINT)
+        wallet = Wallet(PRIVATE_KEY)
+
+        # Get user's ATA for the token
+        ata = wallet.get_associated_token_address(MINT_ADDRESS)
+        await close_account_if_exists(client, wallet, ata, MINT_ADDRESS)
+
+    except Exception as e:
+        logger.error(f"Unexpected error: {e}")
+    finally:
+        await client.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())