mirror of
https://github.com/chainstacklabs/pumpfun-bonkfun-bot.git
synced 2026-07-27 23:37:45 +00:00
smypmsa
84 lines
3.4 KiB
Python
84 lines
3.4 KiB
Python
import asyncio
|
|
import hashlib
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
import websockets
|
|
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
from core.pubkeys import PumpAddresses
|
|
|
|
WSS_ENDPOINT = os.environ.get("SOLANA_NODE_WSS_ENDPOINT")
|
|
|
|
|
|
async def save_transaction(tx_data, tx_signature):
|
|
os.makedirs("blockSubscribe-transactions", exist_ok=True)
|
|
hashed_signature = hashlib.sha256(tx_signature.encode()).hexdigest()
|
|
file_path = os.path.join("blockSubscribe-transactions", f"{hashed_signature}.json")
|
|
with open(file_path, "w") as f:
|
|
json.dump(tx_data, f, indent=2)
|
|
print(f"Saved transaction: {hashed_signature[:8]}...")
|
|
|
|
|
|
async def listen_for_transactions():
|
|
async with websockets.connect(WSS_ENDPOINT) as websocket:
|
|
subscription_message = json.dumps(
|
|
{
|
|
"jsonrpc": "2.0",
|
|
"id": 1,
|
|
"method": "blockSubscribe",
|
|
"params": [
|
|
{"mentionsAccountOrProgram": str(PumpAddresses.PROGRAM)},
|
|
{
|
|
"commitment": "confirmed",
|
|
"encoding": "base64",
|
|
"showRewards": False,
|
|
"transactionDetails": "full",
|
|
"maxSupportedTransactionVersion": 0,
|
|
},
|
|
],
|
|
},
|
|
)
|
|
await websocket.send(subscription_message)
|
|
print(f"Subscribed to blocks mentioning program: {PumpAddresses.PROGRAM}")
|
|
|
|
while True:
|
|
try:
|
|
response = await websocket.recv()
|
|
data = json.loads(response)
|
|
|
|
if "method" in data and data["method"] == "blockNotification":
|
|
if "params" in data and "result" in data["params"]:
|
|
block_data = data["params"]["result"]
|
|
if "value" in block_data and "block" in block_data["value"]:
|
|
block = block_data["value"]["block"]
|
|
if "transactions" in block:
|
|
transactions = block["transactions"]
|
|
for tx in transactions:
|
|
if isinstance(tx, dict) and "transaction" in tx:
|
|
if (
|
|
isinstance(tx["transaction"], list)
|
|
and len(tx["transaction"]) > 0
|
|
):
|
|
tx_signature = tx["transaction"][0]
|
|
elif (
|
|
isinstance(tx["transaction"], dict)
|
|
and "signatures" in tx["transaction"]
|
|
):
|
|
tx_signature = tx["transaction"][
|
|
"signatures"
|
|
][0]
|
|
else:
|
|
continue
|
|
await save_transaction(tx, tx_signature)
|
|
elif "result" in data:
|
|
print("Subscription confirmed")
|
|
except Exception as e:
|
|
print(f"An error occurred: {e!s}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(listen_for_transactions())
|