1use clap::Args;
8use scope::config::Config;
9use scope::market::{HealthOverrides, HealthThresholds, health};
10
11#[derive(Debug, Clone, Default, Args)]
16#[command(next_help_heading = "Health thresholds")]
17pub struct HealthArgs {
18 #[arg(long = "peg", value_name = "TARGET", help = with_default("Peg target (e.g., 1.0 for USD stablecoins).", health::DEFAULT_PEG_TARGET))]
19 pub peg_target: Option<f64>,
20
21 #[arg(long, value_name = "RANGE", help = with_default(&peg_range_help(), health::DEFAULT_PEG_RANGE))]
22 pub peg_range: Option<f64>,
23
24 #[arg(long, value_name = "N", help = with_default("Minimum valid order book levels (price > 0, quantity > 0) per side.", health::DEFAULT_MIN_LEVELS))]
25 pub min_levels: Option<usize>,
26
27 #[arg(long, value_name = "USDT", help = with_default("Minimum in-band depth per side in quote terms, e.g. USDT.", health::DEFAULT_MIN_DEPTH))]
28 pub min_depth: Option<f64>,
29
30 #[arg(long, value_name = "RATIO", help = with_default("Min bid/ask depth ratio (warn if ratio below this).", health::DEFAULT_MIN_BID_ASK_RATIO))]
31 pub min_bid_ask_ratio: Option<f64>,
32
33 #[arg(long, value_name = "RATIO", help = with_default("Max bid/ask depth ratio (warn if ratio above this).", health::DEFAULT_MAX_BID_ASK_RATIO))]
34 pub max_bid_ask_ratio: Option<f64>,
35
36 #[arg(long, value_name = "PCT", help = with_default("Max spread between best bid and best ask, in percent of mid price.", health::DEFAULT_MAX_SPREAD_PCT))]
37 pub max_spread_pct: Option<f64>,
38
39 #[arg(long, value_name = "USDT", help = with_default("Min depth of the top 3 valid levels per side, in quote terms.", health::DEFAULT_MIN_TOP3_DEPTH))]
40 pub min_top3_depth: Option<f64>,
41
42 #[arg(long, value_name = "USDT", help = with_default("Min depth of the top 10 valid levels per side, in quote terms.", health::DEFAULT_MIN_TOP10_DEPTH))]
43 pub min_top10_depth: Option<f64>,
44
45 #[arg(long, value_name = "PCT", value_parser = positive_pct, help = with_default("Price step between levels of a synthetic AMM book (DEX venues), in percent.", health::DEFAULT_AMM_STEP_PCT))]
46 pub amm_step_pct: Option<f64>,
47
48 #[arg(long, value_name = "PCT", value_parser = positive_pct, help = with_default("Half-width of the depth window for the depth/volume rule, in percent of the mid price.", health::DEFAULT_DEPTH_WINDOW_PCT))]
49 pub depth_window_pct: Option<f64>,
50
51 #[arg(long, value_name = "PCT", value_parser = positive_pct, help = with_default("Depth/volume floor: depth within the window below this percent of 24h volume fails.", health::DEFAULT_MIN_DEPTH_VOLUME_PCT))]
52 pub min_depth_volume_pct: Option<f64>,
53
54 #[arg(long, value_name = "PCT", value_parser = positive_pct, help = with_default("Depth/volume ceiling: depth within the window above this percent of 24h volume warns.", health::DEFAULT_MAX_DEPTH_VOLUME_PCT))]
55 pub max_depth_volume_pct: Option<f64>,
56}
57
58fn positive_pct(s: &str) -> Result<f64, String> {
60 let v: f64 = s.parse().map_err(|e| format!("{e}"))?;
61 if v.is_finite() && v > 0.0 {
62 Ok(v)
63 } else {
64 Err(format!("must be a number greater than 0, got {s}"))
65 }
66}
67
68fn peg_range_help() -> String {
70 let band_pct = health::PegBands::new(health::DEFAULT_PEG_TARGET, health::DEFAULT_PEG_RANGE)
71 .outlier_width()
72 / health::DEFAULT_PEG_TARGET
73 * health::PERCENT_PER_RATIO;
74 format!(
75 "Peg range. The outlier band is peg ± range×{factor}: levels outside it are left out of depth and level counts (best prices still include them). E.g., {range} = ±{band_pct}% around a peg of {target}.",
76 factor = health::PEG_OUTLIER_FACTOR,
77 range = health::DEFAULT_PEG_RANGE,
78 target = health::DEFAULT_PEG_TARGET,
79 )
80}
81
82fn with_default(desc: &str, default: impl std::fmt::Display) -> String {
84 format!("{desc} [default: config `market.health`, else {default}]")
85}
86
87impl HealthArgs {
88 pub fn resolve(&self, config: &Config) -> HealthThresholds {
90 config.market.health.with_overrides(&HealthOverrides {
91 peg_target: self.peg_target,
92 peg_range: self.peg_range,
93 min_levels: self.min_levels,
94 min_depth: self.min_depth,
95 min_bid_ask_ratio: self.min_bid_ask_ratio,
96 max_bid_ask_ratio: self.max_bid_ask_ratio,
97 max_spread_pct: self.max_spread_pct,
98 min_top3_depth: self.min_top3_depth,
99 min_top10_depth: self.min_top10_depth,
100 amm_step_pct: self.amm_step_pct,
101 depth_window_pct: self.depth_window_pct,
102 min_depth_volume_pct: self.min_depth_volume_pct,
103 max_depth_volume_pct: self.max_depth_volume_pct,
104 })
105 }
106}
107
108#[cfg(test)]
109mod tests {
110 use super::*;
111 use clap::Parser;
112
113 #[derive(Parser)]
114 struct Cli {
115 #[command(flatten)]
116 health: HealthArgs,
117 }
118
119 fn resolve(argv: &[&str], config: &Config) -> HealthThresholds {
120 let mut full = vec!["test"];
121 full.extend_from_slice(argv);
122 Cli::parse_from(full).health.resolve(config)
123 }
124
125 #[test]
126 fn test_no_flags_no_config_uses_builtin_defaults() {
127 assert_eq!(
128 resolve(&[], &Config::default()),
129 HealthThresholds::default()
130 );
131 }
132
133 #[test]
134 fn test_config_replaces_builtin_default() {
135 let mut config = Config::default();
136 config.market.health.min_levels = 20;
137 let t = resolve(&[], &config);
138 assert_eq!(t.min_levels, 20);
139 assert_eq!(t.min_top3_depth, HealthThresholds::default().min_top3_depth);
141 }
142
143 #[test]
144 fn test_flag_wins_over_config() {
145 let mut config = Config::default();
146 config.market.health.max_spread_pct = 5.0;
147 config.market.health.min_levels = 20;
148 let t = resolve(&["--max-spread-pct", "1.5"], &config);
149 assert_eq!(t.max_spread_pct, 1.5);
150 assert_eq!(t.min_levels, 20);
151 }
152
153 #[test]
154 fn test_peg_flag_sets_peg_target() {
155 let t = resolve(&["--peg", "0.98"], &Config::default());
156 assert_eq!(t.peg_target, 0.98);
157 }
158
159 #[test]
160 fn test_amm_step_flag_resolves_and_rejects_non_positive() {
161 #[derive(clap::Parser)]
162 struct W {
163 #[command(flatten)]
164 h: HealthArgs,
165 }
166 use clap::Parser;
167 let w = W::try_parse_from(["x", "--amm-step-pct", "0.05"]).unwrap();
168 assert_eq!(w.h.resolve(&Config::default()).amm_step_pct, 0.05);
169 let w = W::try_parse_from(["x"]).unwrap();
170 assert_eq!(
171 w.h.resolve(&Config::default()).amm_step_pct,
172 health::DEFAULT_AMM_STEP_PCT
173 );
174 for bad in ["0", "-1", "nan", "inf", "abc"] {
175 assert!(
176 W::try_parse_from(["x", "--amm-step-pct", bad]).is_err(),
177 "{}",
178 bad
179 );
180 }
181 }
182
183 #[test]
184 fn test_depth_volume_flags_override_config() {
185 let mut config = Config::default();
186 config.market.health.min_depth_volume_pct = 3.0;
187 let t = resolve(
188 &["--max-depth-volume-pct", "25", "--depth-window-pct", "1"],
189 &config,
190 );
191 assert_eq!(t.min_depth_volume_pct, 3.0);
192 assert_eq!(t.max_depth_volume_pct, 25.0);
193 assert_eq!(t.depth_window_pct, 1.0);
194 }
195}