Track Pump.fun launches and trades

use {
    carbon_core::{
        error::CarbonResult, instruction::InstructionProcessorInputType, pipeline::Pipeline,
        processor::Processor,
    },
    carbon_log_metrics::LogMetrics,
    carbon_pumpfun_decoder::{
        PROGRAM_ID, PumpfunDecoder,
        instructions::{CpiEvent, PumpfunInstruction},
    },
    carbon_rpc_transaction_crawler_datasource::{ConnectionConfig, Filters, RpcTransactionCrawler},
    std::sync::Arc,
};

#[tokio::main]
async fn main() -> CarbonResult<()> {
    env_logger::init();

    let rpc_url = std::env::var("RPC_URL")
        .unwrap_or_else(|_| "https://api.mainnet-beta.solana.com".to_string());

    // Every transaction that calls Pump.fun, newest first, then new ones as they land.
    let datasource = RpcTransactionCrawler::new(
        rpc_url,
        PROGRAM_ID,
        ConnectionConfig::default(),
        Filters::new(None, None, None),
        None,
    );

    Pipeline::builder()
        .datasource(datasource)
        .metrics(Arc::new(LogMetrics::new()))
        .instruction(PumpfunDecoder, LaunchTracker)
        .build()?
        .run()
        .await
}

/// SOL has 9 decimals; Pump.fun tokens have 6.
const LAMPORTS_PER_SOL: f64 = 1e9;
const TOKEN_UNITS: f64 = 1e6;

struct LaunchTracker;

impl Processor<InstructionProcessorInputType<'_, PumpfunInstruction>> for LaunchTracker {
    async fn process(
        &mut self,
        input: &InstructionProcessorInputType<'_, PumpfunInstruction>,
    ) -> CarbonResult<()> {
        // Pump.fun reports what happened in events, emitted as a self-CPI after each action.
        let PumpfunInstruction::CpiEvent { data, .. } = input.decoded_instruction else {
            return Ok(());
        };
        match data {
            CpiEvent::CreateEvent(e) => {
                println!(
                    "launch   {} ({}) mint={} creator={}",
                    e.name, e.symbol, e.mint, e.creator
                )
            }
            CpiEvent::TradeEvent(e) => println!(
                "{} {:>12.4} SOL {:>16.2} tokens mint={}",
                if e.is_buy { "buy     " } else { "sell    " },
                e.sol_amount as f64 / LAMPORTS_PER_SOL,
                e.token_amount as f64 / TOKEN_UNITS,
                e.mint,
            ),
            CpiEvent::CompleteEvent(e) => println!("complete mint={} bonding curve filled", e.mint),
            _ => {}
        }
        Ok(())
    }
}