quantoxide 0.5.3

//! Example demonstrating direct interaction with the parallel backtest process.
//!
//! This example runs multiple operators in parallel over the same time period and starting balance,
//! while maintaining isolated trade execution state per operator. Candle management overhead is
//! shared across all operators, making this significantly more efficient than running separate
//! backtests. Useful for comparing strategies or running parameter sweeps.

use std::{collections::HashMap, env};

use dotenvy::dotenv;
use serde_json::json;
use tokio::time::{self, Duration};

use quantoxide::{
    Database,
    error::Result,
    models::SATS_PER_BTC,
    sync::PriceHistoryState,
    trade::{
        BacktestConfig, BacktestParallelEngine, BacktestParallelUpdate, BacktestStatus,
        TradingState,
    },
};

#[path = "operators/mod.rs"]
mod operators;
#[path = "util/mod.rs"]
mod util;

use operators::{
    raw::RawOperatorTemplate,
    signal::{MultiSignalOperatorTemplate, SupportedSignal, evaluator::SignalEvaluatorTemplate},
};
use util::{input, metrics};

/// Prints usage information and exits.
fn print_usage() {
    eprintln!(
        "Usage: cargo run --example backtest_direct_parallel -- --start <DATE> --end <DATE> [OPTIONS]"
    );
    eprintln!();
    eprintln!("Required:");
    eprintln!("  --start <DATE>       Start date in YYYY-MM-DD format");
    eprintln!("  --end <DATE>         End date in YYYY-MM-DD format");
    eprintln!();
    eprintln!("Options:");
    eprintln!("  --balance <SATS>     Starting balance in sats (default: 10000000)");
    eprintln!("  --rfr-sats <RATE>    Annual risk-free rate for sats as decimal (default: 0.0)");
    eprintln!("  --rfr-usd <RATE>     Annual risk-free rate for USD as decimal (default: 0.0)");
    eprintln!();
    eprintln!("Example:");
    eprintln!(
        "  cargo run --example backtest_direct_parallel -- --start 2025-09-01 --end 2025-12-01 --balance 10000000"
    );
}

/// Tracks state for a single operator during backtest.
#[derive(Default)]
struct OperatorStats {
    last_trading_state: Option<TradingState>,
    start_market_price: Option<f64>,
    daily_net_values_sats: Vec<f64>,
    daily_net_values_usd: Vec<f64>,
}

#[tokio::main]
async fn main() -> Result<()> {
    dotenv().ok();

    let pg_url = env::var("POSTGRES_DB_URL").expect("POSTGRES_DB_URL must be set");

    println!("Initializing database...");

    let db = Database::new(&pg_url).await?;

    println!("Database ready. Evaluating `PriceHistoryState`...");

    let price_history_state = PriceHistoryState::evaluate(&db).await?;

    println!("\n{price_history_state}");

    if price_history_state.bound_end().is_none() {
        println!(
            "\nSome price history must be available in the local database to run the backtest."
        );
        println!("Run a synchronization example first to fetch historical data.");

        return Ok(());
    }

    let args = input::parse_args();

    let Some(start_str) = args.get("start") else {
        print_usage();
        return Err("Missing required argument: --start".into());
    };

    let Some(end_str) = args.get("end") else {
        print_usage();
        return Err("Missing required argument: --end".into());
    };

    let start_time = input::parse_date(start_str).map_err(|e| {
        eprintln!("Error parsing --start: {}", e);
        print_usage();
        e
    })?;

    let end_time = input::parse_date(end_str).map_err(|e| {
        eprintln!("Error parsing --end: {}", e);
        print_usage();
        e
    })?;

    let start_balance = match args.get("balance") {
        Some(v) => v.parse::<u64>().map_err(|e| {
            eprintln!("Error parsing --balance: {}", e);
            print_usage();
            e
        })?,
        None => 10_000_000,
    };

    let rfr_sats = match args.get("rfr-sats") {
        Some(v) => v.parse::<f64>().map_err(|e| {
            eprintln!("Error parsing --rfr-sats: {}", e);
            print_usage();
            e
        })?,
        None => 0.0,
    };

    let rfr_usd = match args.get("rfr-usd") {
        Some(v) => v.parse::<f64>().map_err(|e| {
            eprintln!("Error parsing --rfr-usd: {}", e);
            print_usage();
            e
        })?,
        None => 0.0,
    };

    println!("\nParallel Backtest Configuration:");
    println!("Start date: {}", start_time.format("%Y-%m-%d %H:%M %Z"));
    println!("Start balance: {} (per operator)", start_balance);
    println!("Risk-free rate (sats): {:.2}%", rfr_sats * 100.0);
    println!("Risk-free rate (USD): {:.2}%", rfr_usd * 100.0);
    println!("End date: {}\n", end_time.format("%Y-%m-%d %H:%M %Z"));

    println!("Initializing `BacktestParallelEngine`...");

    // Create operators of different types to demonstrate compatibility
    // RawOperator - directly processes candles
    let raw_operator = RawOperatorTemplate::new();

    // SignalOperator - processes signals from evaluators
    let signal_evaluators = vec![SignalEvaluatorTemplate::new::<SupportedSignal>()];
    let signal_operator = MultiSignalOperatorTemplate::new();

    let backtest_engine = BacktestParallelEngine::new(
        BacktestConfig::default(),
        db,
        start_time,
        end_time,
        start_balance,
    )
    .await?
    .add_raw_operator("Raw Strategy", raw_operator)?
    .add_signal_operator("Signal Strategy", signal_evaluators, signal_operator)?;

    let mut backtest_rx = backtest_engine.receiver();

    // Clone values for the spawned task
    let start_balance_for_task = start_balance;
    let rfr_sats_for_task = rfr_sats;
    let rfr_usd_for_task = rfr_usd;

    tokio::spawn(async move {
        let mut operator_stats: HashMap<String, OperatorStats> = HashMap::new();

        loop {
            match backtest_rx.recv().await {
                Ok(backtest_update) => match backtest_update {
                    BacktestParallelUpdate::Status(backtest_status) => match backtest_status {
                        BacktestStatus::Finished => {
                            println!("\n========== BACKTEST COMPLETE ==========\n");

                            // Print summary for each operator
                            let mut summaries = Vec::new();

                            for (name, stats) in &operator_stats {
                                if let Some(state) = &stats.last_trading_state {
                                    let trade_count =
                                        state.closed_history().len() + state.running_map().len();

                                    let final_net_value_sats = state.total_net_value();
                                    let final_market_price = state.market_price().as_f64();

                                    let start_price =
                                        stats.start_market_price.unwrap_or(final_market_price);
                                    let start_balance_usd =
                                        start_balance_for_task as f64 * start_price / SATS_PER_BTC;
                                    let final_net_value_usd = final_net_value_sats as f64
                                        * final_market_price
                                        / SATS_PER_BTC;

                                    let pnl_sats =
                                        final_net_value_sats as i64 - start_balance_for_task as i64;
                                    let pnl_sats_pct =
                                        (pnl_sats as f64 / start_balance_for_task as f64) * 100.0;
                                    let pnl_usd = final_net_value_usd - start_balance_usd;
                                    let pnl_usd_pct = (pnl_usd / start_balance_usd) * 100.0;

                                    let sharpe_sats = metrics::calculate_sharpe_ratio(
                                        &stats.daily_net_values_sats,
                                        rfr_sats_for_task,
                                    );
                                    let sharpe_usd = metrics::calculate_sharpe_ratio(
                                        &stats.daily_net_values_usd,
                                        rfr_usd_for_task,
                                    );

                                    let max_dd_sats = metrics::calculate_max_drawdown(
                                        &stats.daily_net_values_sats,
                                    );
                                    let max_dd_usd = metrics::calculate_max_drawdown(
                                        &stats.daily_net_values_usd,
                                    );

                                    let format_sharpe = |s: Option<f64>| {
                                        s.map_or("N/A".to_string(), |v| format!("{:.4}", v))
                                    };

                                    let format_dd = |dd: Option<f64>| {
                                        dd.map_or("N/A".to_string(), |v| {
                                            format!("{:.2}", v * 100.0)
                                        })
                                    };

                                    let summary = json!({
                                        "operator": name,
                                        "start_market_price": format!("{:.2}", start_price),
                                        "final_market_price": format!("{:.2}", final_market_price),
                                        "trade_count": trade_count,
                                        "start_balance_sats": start_balance_for_task,
                                        "start_balance_usd": format!("{:.2}", start_balance_usd),
                                        "final_net_value_sats": final_net_value_sats,
                                        "final_net_value_usd": format!("{:.2}", final_net_value_usd),
                                        "pnl_sats": pnl_sats,
                                        "pnl_sats_percent": format!("{:.2}", pnl_sats_pct),
                                        "pnl_usd": format!("{:.2}", pnl_usd),
                                        "pnl_usd_percent": format!("{:.2}", pnl_usd_pct),
                                        "daily_data_points": stats.daily_net_values_sats.len(),
                                        "sharpe_sats": format_sharpe(sharpe_sats),
                                        "sharpe_usd": format_sharpe(sharpe_usd),
                                        "max_drawdown_sats_percent": format_dd(max_dd_sats),
                                        "max_drawdown_usd_percent": format_dd(max_dd_usd),
                                    });

                                    summaries.push(summary);
                                }
                            }

                            let combined = json!({ "backtest_summaries": summaries });
                            let output = serde_json::to_string_pretty(&combined).unwrap();
                            println!("{output}");

                            return;
                        }
                        BacktestStatus::Failed(error) => {
                            eprintln!("\nBacktest failed: {error}");
                            return;
                        }
                        BacktestStatus::Aborted => {
                            println!("\nBacktest aborted.");
                            return;
                        }
                        BacktestStatus::NotInitiated
                        | BacktestStatus::Starting
                        | BacktestStatus::Running => {}
                    },
                    BacktestParallelUpdate::TradingState {
                        operator_name,
                        state,
                    } => {
                        // Backtest updates correspond to midnight (UTC) of each day of the period
                        let stats = operator_stats.entry(operator_name).or_default();

                        if stats.start_market_price.is_none() {
                            stats.start_market_price = Some(state.market_price().as_f64());
                        }

                        let net_value_sats = state.total_net_value() as f64;
                        let market_price = state.market_price().as_f64();
                        let net_value_usd = net_value_sats * market_price / SATS_PER_BTC;

                        stats.daily_net_values_sats.push(net_value_sats);
                        stats.daily_net_values_usd.push(net_value_usd);

                        stats.last_trading_state = Some(*state);
                    }
                },
                Err(e) => {
                    eprintln!("{:?}", e);
                    return;
                }
            }
        }
    });

    println!("Initialization OK. Starting `BacktestParallelEngine`...");
    println!("Running operators: Raw Strategy (RawOperator), Signal Strategy (SignalOperator)");

    let backtest_controller = backtest_engine.start();

    let final_status = backtest_controller.until_stopped().await;

    // Delay for printing all `backtest_rx` updates
    time::sleep(Duration::from_millis(100)).await;

    println!("\nBacktest status: {final_status}");

    Ok(())
}