typed Syntax and improved formatting

pulse-wire/src/general.rs

@@ -1,6 +1,9 @@
 use pulse_macros::pwp;
 
-use crate::PulseWire;
+use crate::{
+    PulseWire,
+    units::{Symbol, USD},
+};
 
 #[pwp]
 pub enum MarketTrend {
@@ -28,9 +31,9 @@ pub struct Signal {
     pub kind: Direction,
     pub confidence: f32,
     pub size: f64,
-    pub price: f64,
-    pub take_profit: f64,
-    pub take_stop_loss: f64,
+    pub price: USD,
+    pub take_profit: USD,
+    pub stop_loss: USD,
 }
 
 #[pwp]
@@ -50,10 +53,10 @@ pub struct EventLog {
 
 #[pwp]
 pub struct Position {
-    pub symbol: String,
+    pub symbol: Symbol,
     pub size: f64,
-    pub entry_price: f64,
-    pub profit: f64,
+    pub entry_price: USD,
+    pub profit: USD,
 }
 
 impl std::fmt::Display for Direction {
@@ -65,7 +68,6 @@ impl std::fmt::Display for Direction {
     }
 }
 
-
 impl std::fmt::Display for MarketTrend {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {

pulse-wire/src/lib.rs

@@ -1,8 +1,15 @@
 #[cfg(target_os = "macos")]
 use std::path::PathBuf;
 
-pub mod terminal;
 pub mod general;
+pub mod terminal;
+pub mod units;
+
+pub mod prelude {
+    pub use crate::general::*;
+    pub use crate::terminal::*;
+    pub use crate::units::*;
+}
 
 pub fn server_path() -> PathBuf {
     PathBuf::from("/tmp/pulse-engine.sock")

pulse-wire/src/terminal.rs

@@ -1,6 +1,7 @@
 use crate::{
     PulseWire,
     general::{EventLog, MarketTrend, Position, Signal, Volatility},
+    units::{Symbol, USD},
 };
 use pulse_macros::pwp;
 
@@ -36,10 +37,10 @@ pub enum TerminalClientMessage {
 
 #[pwp]
 pub struct MarketItem {
-    pub symbol: String,
-    pub price: f64,
+    pub symbol: Symbol,
+    pub price: USD,
     pub trend: f64,
-    pub volume_24h: f64,
+    pub volume_24h: USD,
 }
 
 #[pwp]

pulse-wire/src/units.rs

@@ -0,0 +1,98 @@
+use crate::PulseWire;
+
+#[derive(Debug, Clone)]
+pub struct Symbol(pub String);
+
+#[derive(Debug, Clone, Copy)]
+pub struct USD(pub f64);
+
+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 > 0.0 {
+            write!(f, "\x1b[32m${}\x1b[0m", format_f64(self.0))
+        } else {
+            write!(f, "\x1b[31m${}\x1b[0m", format_f64(self.0))
+        }
+    }
+}
+
+impl PulseWire for Symbol {
+    fn from_com(com: &mut Vec<u8>) -> Self {
+        Self(String::from_com(com))
+    }
+
+    fn to_com(&self) -> Vec<u8> {
+        self.0.to_com()
+    }
+}
+
+impl PulseWire for USD {
+    fn from_com(com: &mut Vec<u8>) -> Self {
+        Self(f64::from_com(com))
+    }
+
+    fn to_com(&self) -> Vec<u8> {
+        self.0.to_com()
+    }
+}
+
+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
+}