ibapi 3.0.0

//! Stream Bars example
//!
//! # Usage
//!
//! ```bash
//! cargo run --features sync --example stream_bars
//! ```

use std::thread;
use std::time::Duration;

use clap::{arg, ArgMatches, Command};

use ibapi::client::blocking::Client;
use ibapi::contracts::Contract;
use ibapi::market_data::TradingHours;

fn main() -> anyhow::Result<()> {
    env_logger::init();

    let matches = Command::new("stream_bars")
        .version("1.0")
        .author("Wil Boayue <wil@wsbsolutions.com")
        .about("Streams realtime bars")
        .arg(arg!(--connection_string <VALUE>).default_value("localhost:4002"))
        .arg(arg!(--stock <SYMBOL>))
        .arg(arg!(--futures <SYMBOL>))
        .get_matches();

    let connection_string = matches.get_one::<String>("connection_string").expect("connection_string is required");
    let contract = extract_contract(&matches).expect("error parsing --stock or --future");

    println!("connection_string: {connection_string:?}");
    println!("contract: {contract:?}");

    let client = Client::connect("127.0.0.1:4002", 100).unwrap();

    println!("server_version: {}", client.server_version());
    println!("server_time: {:?}", client.connection_time());

    let bars = client.realtime_bars(&contract).trading_hours(TradingHours::Extended).subscribe()?;

    for (i, bar) in bars.iter_data().enumerate().take(60) {
        match bar {
            Ok(bar) => println!("bar: {i:?} {bar:?}"),
            Err(e) => {
                eprintln!("error: {e:?}");
                break;
            }
        }
    }

    thread::sleep(Duration::from_secs(5));

    Ok(())
}

fn extract_contract(matches: &ArgMatches) -> Option<Contract> {
    if let Some(symbol) = matches.get_one::<String>("stock") {
        Some(Contract::stock(symbol.to_uppercase()).build())
    } else {
        matches
            .get_one::<String>("futures")
            .map(|symbol| Contract::futures(symbol.to_uppercase()).front_month().build())
    }
}