Rust PN1 Orderbook

use polynode::{EngineOptions, OrderbookEngine};

#[tokio::main]
async fn main() -> polynode::Result<()> {
    let api_key = std::env::var("POLYNODE_API_KEY")
        .expect("Set POLYNODE_API_KEY");
    let token_id = std::env::var("POLYMARKET_TOKEN_ID")
        .expect("Set POLYMARKET_TOKEN_ID");

    let engine = OrderbookEngine::connect(&api_key, EngineOptions {
        integrity: true,
        allow_stale_reads: false,
        ..Default::default()
    }).await?;

    let mut integrity_errors = engine.integrity_errors();
    tokio::spawn(async move {
        while let Ok(error) = integrity_errors.recv().await {
            eprintln!("{:?} {} {}", error.code, error.token, error.message);
        }
    });

    engine.subscribe(vec![token_id.clone()]).await?;

    while engine.book(&token_id).await.is_none() {
        tokio::time::sleep(std::time::Duration::from_millis(10)).await;
    }

    println!("best bid: {:?}", engine.best_bid(&token_id).await);
    println!("best ask: {:?}", engine.best_ask(&token_id).await);
    println!("midpoint: {:?}", engine.midpoint(&token_id).await);
    println!("spread: {:?}", engine.spread(&token_id).await);

    engine.close().await?;
    Ok(())
}