use std::thread;
use std::time::Duration;
use clap::{arg, ArgMatches, Command};
use ibapi::client::blocking::Client;
use ibapi::contracts::{Contract, Currency, Exchange, Symbol};
use ibapi::market_data::realtime::{BarSize, WhatToShow};
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());
println!("next_order_id: {}", client.next_order_id());
let bars = client.realtime_bars(&contract, BarSize::Sec5, WhatToShow::Trades, TradingHours::Extended)?;
for (i, bar) in bars.iter().enumerate().take(60) {
println!("bar: {i:?} {bar:?}");
}
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 {
symbol: Symbol::from(symbol.to_uppercase()),
security_type: ibapi::contracts::SecurityType::Future,
exchange: Exchange::from("GLOBEX"),
currency: Currency::from("USD"),
..Default::default()
}
})
}
}