#![feature(drain_filter)]
#![feature(bool_to_option)]

use std::env;

use fern::colors::{Color, ColoredLevelConfig};
use log::LevelFilter::{Debug, Trace};
use tokio::time::Duration;

use crate::bot::BfxBot;
use crate::connectors::ExchangeDetails;
use crate::currency::Symbol;

mod bot;
mod connectors;
mod currency;
mod events;
mod managers;
mod models;
mod strategy;
mod ticker;

pub type BoxError = Box<dyn std::error::Error + Send + Sync>;

#[tokio::main]
async fn main() -> Result<(), BoxError> {
    setup_logger()?;
    dotenv::dotenv()?;

    let api_key = env::vars()
        .find(|(k, v)| k == "API_KEY")
        .map(|(k, v)| v)
        .ok_or("API_KEY not set!")?;
    let api_secret = env::vars()
        .find(|(k, v)| k == "API_SECRET")
        .map(|(k, v)| v)
        .ok_or("API_SECRET not set!")?;

    let bitfinex = ExchangeDetails::Bitfinex {
        api_key: api_key.into(),
        api_secret: api_secret.into(),
    };

    let mut bot = BfxBot::new(
        vec![bitfinex],
        vec![Symbol::DERIV_ETH, Symbol::DERIV_BTC],
        Symbol::DERIV_USDT,
        Duration::new(10, 0),
    );

    Ok(bot.start_loop().await?)
}

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
            ))
        })
        .level(Trace)
        .filter(|metadata| metadata.target().contains("rustybot"))
        .chain(std::io::stdout())
        // .chain(fern::log_file("rustico.log")?)
        .apply()?;

    Ok(())
}