core/rustybot/src/main.rs

83 lines
2.3 KiB
Rust
Raw Normal View History

2021-01-14 12:42:23 +00:00
#![feature(drain_filter)]
2021-01-15 11:10:00 +00:00
#![feature(bool_to_option)]
2021-01-14 12:42:23 +00:00
2021-01-14 18:56:31 +00:00
use fern::colors::{Color, ColoredLevelConfig};
2021-01-26 11:15:04 +00:00
use log::LevelFilter::{Debug, Trace};
2021-01-14 18:56:31 +00:00
use tokio::time::Duration;
2021-01-04 10:45:54 +00:00
use crate::bot::BfxBot;
2021-01-26 11:15:04 +00:00
use crate::connectors::ExchangeDetails;
2021-01-14 18:56:31 +00:00
use crate::currency::Symbol;
2021-01-02 14:10:16 +00:00
2021-01-05 12:58:47 +00:00
mod bot;
mod connectors;
mod currency;
2021-01-02 12:15:19 +00:00
mod events;
2021-01-13 09:03:24 +00:00
mod managers;
mod models;
2021-01-02 14:10:16 +00:00
mod strategy;
2021-01-05 12:58:47 +00:00
mod ticker;
2021-01-01 14:07:16 +00:00
pub type BoxError = Box<dyn std::error::Error + Send + Sync>;
#[tokio::main]
async fn main() -> Result<(), BoxError> {
2021-01-14 18:56:31 +00:00
setup_logger()?;
2021-01-26 11:15:04 +00:00
// TEST
2021-01-05 19:50:14 +00:00
let test_api_key = "P1EVE68DJByDAkGQvpIkTwfrbYXd2Vo2ZaIhTYb9vx2";
let test_api_secret = "1nicg8z0zKVEt5Rb7ZDpIYjVYVTgvCaCPMZqB0niFli";
2021-01-14 12:42:23 +00:00
2021-01-26 11:15:04 +00:00
// REAL
// let orders_api_key = "hc5nDvYbFYJZMKdnzYq8P4AzCSwjxfQHnMyrg69Sf4c";
// let orders_api_secret = "53x9goIOpbOtBoPi7dmigK5Cq5e0282EUO2qRIMEXlh";
// let prices_api_key = "gTfFZUCwRBE0Z9FZjyk9HNe4lZ7XuiZY9rrW71SyUr9";
// let prices_api_secret = "zWbxvoFZad3BPIiXK4DKfEvC0YsAuaApbeAyI8OBXgN";
// let positions_api_key = "PfR7BadPZPNdVZnkHFBfAjsg7gjt8pAecMj5B8eRPFi";
// let positions_api_secret = "izzvxtE3XsBBRpVCHGJ8f60UA56SmPNbBvJGVd67aqD";
let bitfinex = ExchangeDetails::Bitfinex {
prices_api_key: test_api_key.into(),
prices_api_secret: test_api_secret.into(),
orders_api_key: test_api_key.into(),
orders_api_secret: test_api_secret.into(),
positions_api_key: test_api_key.into(),
positions_api_secret: test_api_secret.into(),
2021-01-14 12:42:23 +00:00
};
2021-01-05 19:50:14 +00:00
let mut bot = BfxBot::new(
2021-01-14 12:42:23 +00:00
vec![bitfinex],
2021-01-26 11:15:04 +00:00
vec![Symbol::BTC, Symbol::ETH, Symbol::XMR],
Symbol::USD,
2021-01-14 12:42:23 +00:00
Duration::new(1, 0),
);
2021-01-02 14:10:16 +00:00
2021-01-05 19:50:14 +00:00
Ok(bot.start_loop().await?)
2021-01-01 14:07:16 +00:00
}
2021-01-14 18:56:31 +00:00
fn setup_logger() -> Result<(), fern::InitError> {
let colors = ColoredLevelConfig::new()
.info(Color::Green)
.error(Color::Red)
.trace(Color::Blue)
.debug(Color::Cyan)
.warn(Color::Yellow);
fern::Dispatch::new()
.format(move |out, message, record| {
out.finish(format_args!(
"[{}][{}] {}",
record.target(),
colors.color(record.level()),
message
))
})
2021-01-26 11:15:04 +00:00
.level(Trace)
2021-01-14 18:56:31 +00:00
.filter(|metadata| metadata.target().contains("rustybot"))
.chain(std::io::stdout())
2021-01-15 10:40:48 +00:00
// .chain(fern::log_file("rustico.log")?)
2021-01-14 18:56:31 +00:00
.apply()?;
Ok(())
}