Rust Short-Form and TWAP

use polynode::{
    Coin, PolyNodeClient, ShortFormInterval, ShortFormMessage,
};

#[tokio::main]
async fn main() -> polynode::Result<()> {
    let client = PolyNodeClient::new(
        std::env::var("POLYNODE_API_KEY").expect("Set POLYNODE_API_KEY")
    )?;
    let mut stream = client
        .short_form(ShortFormInterval::FiveMin)
        .coins(&[Coin::Btc, Coin::Eth, Coin::Sol])
        .start()
        .await?;

    while let Some(message) = stream.next().await {
        match message {
            ShortFormMessage::Rotation(rotation) => {
                println!("TWAP: {:?}", rotation.twap_window_seconds);
                for market in rotation.markets {
                    println!(
                        "{} {} {:?} {:?}",
                        market.coin.id(),
                        market.slug,
                        market.chainlink_feed,
                        market.price_to_beat,
                    );
                }
            }
            ShortFormMessage::PriceFeed(price) => println!("{price:?}"),
            ShortFormMessage::Event(event) => println!("{event:?}"),
            ShortFormMessage::Error(error) => eprintln!("{error}"),
        }
    }

    stream.stop().await;
    Ok(())
}