sim-cli 0.1.0

CLI tool for running and comparing Solana simulator backtests
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use clap::Parser;
use eyre::Result;
use simulator_client::{BacktestClient, RangeBound, filter_ranges};

#[derive(Parser, Debug, Clone)]
pub struct RangesArgs {
    /// Simulator endpoint hostname (e.g. staging.simulator.termina.technology)
    #[arg(
        long,
        env = "SIMULATOR_URL",
        default_value = "simulator.termina.technology"
    )]
    pub url: String,

    /// Only show ranges that end at or after this point.
    /// Accepts a slot number or a timestamp (RFC 3339 or YYYY-MM-DD).
    #[arg(long)]
    pub after: Option<String>,

    /// Only show ranges that start at or before this point.
    /// Accepts a slot number or a timestamp (RFC 3339 or YYYY-MM-DD).
    #[arg(long)]
    pub before: Option<String>,

    /// Print output as JSON
    #[arg(long)]
    pub json: bool,
}

pub async fn ranges(args: RangesArgs) -> Result<()> {
    let after: Option<RangeBound> = args.after.as_deref().map(str::parse).transpose()?;
    let before: Option<RangeBound> = args.before.as_deref().map(str::parse).transpose()?;

    let client = BacktestClient::builder()
        .url(format!("wss://{}/backtest", args.url))
        .build();
    let available = client.available_ranges().await?;
    let available = filter_ranges(&available, after.as_ref(), before.as_ref());

    if available.is_empty() {
        if args.json {
            println!("[]");
        } else {
            println!("No available ranges matching the given filters.");
        }
        return Ok(());
    }

    if args.json {
        println!("{}", serde_json::to_string_pretty(&available)?);
        return Ok(());
    }

    println!(
        "{:<30} {:<30} {:<15} End Slot",
        "Start Time (UTC)", "End Time (UTC)", "Start Slot"
    );
    println!("{}", "-".repeat(90));
    for range in &available {
        let start_time = range.bundle_start_slot_utc.as_deref().unwrap_or("-");
        let end_time = range.max_bundle_end_slot_utc.as_deref().unwrap_or("-");
        let end_slot = range
            .max_bundle_end_slot
            .map(|s| s.to_string())
            .unwrap_or_else(|| "unbounded".to_string());
        println!(
            "{:<30} {:<30} {:<15} {}",
            start_time, end_time, range.bundle_start_slot, end_slot
        );
    }

    Ok(())
}