Pump Fun

const WebSocket = require("ws");

class PumpFunMonitor {
  constructor(apiKey) {
    this.apiKey = apiKey;
    this.ws = null;
    this.tokenStats = {};
    this.newTokens = [];
  }

  connect() {
    this.ws = new WebSocket(
      `wss://api.nolimitnodes.com/socket?apikey=${this.apiKey}`
    );
    this.setupHandlers();
  }

  setupHandlers() {
    this.ws.on("open", () => {
      console.log("Connected to Pump Fun streams");
      this.subscribeToAll();
    });

    this.ws.on("message", (data) => {
      const message = JSON.parse(data);

      if (message.method === "pumpFunTradesSubscribe") {
        this.processTrade(message.result);
      } else if (message.method === "pumpFunCreateTokensSubscribe") {
        this.processTokenCreation(message.result);
      } else {
        console.log("Info:", JSON.stringify(message));
      }
    });

    this.ws.on("error", (error) => {
      console.error("WebSocket error:", error.message);
    });

    this.ws.on("close", () => {
      console.log("Connection closed. Reconnecting...");
      this.reconnect();
    });
  }

  subscribeToAll() {
    // Subscribe to all trades
    this.ws.send(
      JSON.stringify({
        method: "dataSubscribe",
        params: {
          referenceId: "PUMP_TRADES",
          streams: [
            { stream: "pumpFunTradesSubscribe", params: {} },
          ],
        },
        id: 1,
      })
    );

    // Subscribe to token creations
    this.ws.send(
      JSON.stringify({
        method: "dataSubscribe",
        params: {
          referenceId: "PUMP_CREATES",
          streams: [
            { stream: "pumpFunCreateTokensSubscribe", params: {} },
          ],
        },
        id: 2,
      })
    );
  }

  processTrade(trade) {
    const mint = trade.mint;

    // Initialize token stats if not tracked
    if (!this.tokenStats[mint]) {
      this.tokenStats[mint] = {
        buys: 0,
        sells: 0,
        totalVolumeSol: 0,
        lastMarketCap: 0,
      };
    }

    const stats = this.tokenStats[mint];
    if (trade.txType === "buy") {
      stats.buys++;
    } else {
      stats.sells++;
    }
    stats.totalVolumeSol += trade.solAmount;
    stats.lastMarketCap = trade.marketCapSol;

    console.log(
      `${trade.txType.toUpperCase()} ${mint.slice(0, 8)}... | ` +
      `${trade.solAmount} SOL | MCap: ${trade.marketCapSol.toFixed(2)} SOL | ` +
      `B/S: ${stats.buys}/${stats.sells}`
    );
  }

  processTokenCreation(token) {
    this.newTokens.push({
      mint: token.mint,
      name: token.name,
      symbol: token.symbol,
      creator: token.creator,
      timestamp: token.createdTimestamp,
    });

    console.log(
      `\nNEW TOKEN: ${token.name} (${token.symbol})`
    );
    console.log(`  Mint: ${token.mint}`);
    console.log(`  Creator: ${token.creator.slice(0, 12)}...`);
    console.log(`  Twitter: ${token.twitter || "N/A"}`);
    console.log(`  Website: ${token.website || "N/A"}`);
    console.log(`  Total Supply: ${token.totalSupply}`);
  }

  reconnect() {
    setTimeout(() => this.connect(), 5000);
  }

  getTopTokens(limit = 10) {
    return Object.entries(this.tokenStats)
      .sort((a, b) => b[1].totalVolumeSol - a[1].totalVolumeSol)
      .slice(0, limit)
      .map(([mint, stats]) => ({ mint, ...stats }));
  }
}

// Usage
const monitor = new PumpFunMonitor("YOUR_API_KEY");
monitor.connect();