use simple_logger::SimpleLogger;
use tonapi::SseApi;

async fn subscribe_to_transactions(sse: &SseApi) -> Result<(), Box<dyn std::error::Error>> {
    let mut stream = sse.transactions_stream(
        Some(vec![
            "-1:5555555555555555555555555555555555555555555555555555555555555555".to_string(),
        ]),
        None,
    );

    while let Ok(evt) = stream.next().await {
        if let Some(evt) = evt {
            println!("Event: {}", evt.tx_hash);
        } else {
            // Stream ended
            break;
        }
    }

    Ok(())
}

async fn subscribe_to_traces(sse: &SseApi) -> Result<(), Box<dyn std::error::Error>> {
    let mut stream = sse.traces_stream(Some(vec![
        "-1:5555555555555555555555555555555555555555555555555555555555555555".to_string(),
    ]));

    while let Ok(evt) = stream.next().await {
        if let Some(evt) = evt {
            println!("Event: {}", evt.hash);
        } else {
            // Stream ended
            break;
        }
    }

    Ok(())
}

async fn subscribe_to_mempool(sse: &SseApi) -> Result<(), Box<dyn std::error::Error>> {
    let mut stream = sse.mempool_stream(Some(vec![
        "-1:5555555555555555555555555555555555555555555555555555555555555555".to_string(),
    ]));

    while let Ok(evt) = stream.next().await {
        if let Some(evt) = evt {
            println!("Boc: {}", evt.boc);
        } else {
            // Stream ended
            break;
        }
    }

    Ok(())
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    SimpleLogger::new().init().expect("logging init");

    let sse_api = SseApi::new(tonapi::Network::Mainnet, None);

    let _ = subscribe_to_transactions(&sse_api).await;
    let _ = subscribe_to_traces(&sse_api).await;
    let _ = subscribe_to_mempool(&sse_api).await;

    Ok(())
}