111 lines
2.7 KiB
Rust
111 lines
2.7 KiB
Rust
use rust_decimal::Decimal;
|
|
|
|
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
|
|
pub struct Symbol(pub String);
|
|
|
|
#[derive(Debug, Clone, Copy, serde::Serialize, serde::Deserialize)]
|
|
pub struct USD(pub Decimal);
|
|
|
|
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
|
|
pub enum Direction {
|
|
Buy,
|
|
Sell,
|
|
}
|
|
|
|
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
|
|
pub enum Volatility {
|
|
Low,
|
|
Medium,
|
|
High,
|
|
}
|
|
|
|
impl std::fmt::Display for Symbol {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
write!(f, "\x1b[35m{}\x1b[0m", self.0)
|
|
}
|
|
}
|
|
|
|
impl std::fmt::Display for USD {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
if self.0.is_sign_positive() {
|
|
write!(f, "\x1b[32m${}\x1b[0m", format_f64(self.0.as_f64()))
|
|
} else {
|
|
write!(f, "\x1b[31m${}\x1b[0m", format_f64(self.0.as_f64()))
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn format_f64(value: f64) -> String {
|
|
let abs = value.abs();
|
|
|
|
let (divisor, suffix) = if abs >= 1_000_000_000.0 {
|
|
(1_000_000_000.0, "B")
|
|
} else if abs >= 1_000_000.0 {
|
|
(1_000_000.0, "M")
|
|
} else if abs >= 1_000.0 {
|
|
(1_000.0, "K")
|
|
} else {
|
|
(1.0, "")
|
|
};
|
|
|
|
if divisor != 1.0 {
|
|
let formatted = value / divisor;
|
|
|
|
// Remove unnecessary trailing zeros
|
|
let s = format!("{:.2}", formatted)
|
|
.trim_end_matches('0')
|
|
.trim_end_matches('.')
|
|
.to_string();
|
|
|
|
return format!("{}{}", s, suffix);
|
|
}
|
|
|
|
let val = format!("{:.3}", value);
|
|
let parts: Vec<&str> = val.split('.').collect();
|
|
|
|
let int = parts[0].to_string();
|
|
let negative = int.starts_with('-');
|
|
|
|
let start = if negative { 1 } else { 0 };
|
|
let mut result = String::new();
|
|
|
|
for (i, c) in int[start..].chars().rev().enumerate() {
|
|
if i > 0 && i % 3 == 0 {
|
|
result.push(',');
|
|
}
|
|
result.push(c);
|
|
}
|
|
|
|
let mut formatted: String = result.chars().rev().collect();
|
|
|
|
if negative {
|
|
formatted.insert(0, '-');
|
|
}
|
|
|
|
if parts.len() > 1 {
|
|
formatted.push('.');
|
|
formatted.push_str(parts[1]);
|
|
}
|
|
|
|
formatted
|
|
}
|
|
|
|
impl std::fmt::Display for Direction {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
Self::Buy => write!(f, "BUY"),
|
|
Self::Sell => write!(f, "SELL"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl std::fmt::Display for Volatility {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
Self::Low => write!(f, "Low"),
|
|
Self::Medium => write!(f, "Medium"),
|
|
Self::High => write!(f, "High"),
|
|
}
|
|
}
|
|
}
|