Com protocol

src/daemon/ptc.rs

@@ -1 +1,2 @@
+include!("../pc.rs");
 include!("../ptc.rs");

src/pc.rs

@@ -0,0 +1,105 @@
+pub trait PulseCom {
+    fn to_com(&self) -> Vec<u8>;
+    fn from_com(_com: &mut Vec<u8>) -> Self;
+}
+
+impl<T: PulseCom> PulseCom for Vec<T> {
+    fn to_com(&self) -> Vec<u8> {
+        let mut vec = Vec::new();
+
+        vec.extend_from_slice(&(self.len() as u32).to_le_bytes());
+
+        for item in self {
+            vec.extend(item.to_com());
+        }
+
+        vec
+    }
+
+    fn from_com(com: &mut Vec<u8>) -> Self {
+        let len_bytes: [u8; 4] = com.drain(..4).collect::<Vec<_>>().try_into().unwrap();
+
+        let len = u32::from_le_bytes(len_bytes) as usize;
+
+        let mut result = Vec::with_capacity(len);
+
+        for _ in 0..len {
+            result.push(T::from_com(com));
+        }
+
+        result
+    }
+}
+
+impl PulseCom for String {
+    fn to_com(&self) -> Vec<u8> {
+        let bytes = self.as_bytes();
+        let mut out = Vec::with_capacity(4 + bytes.len());
+
+        out.extend_from_slice(&(bytes.len() as u32).to_le_bytes());
+        out.extend_from_slice(bytes);
+
+        out
+    }
+
+    fn from_com(com: &mut Vec<u8>) -> Self {
+        let len = u32::from_le_bytes(com[..4].try_into().unwrap()) as usize;
+        com.drain(..4);
+
+        let bytes: Vec<u8> = com.drain(..len).collect();
+        String::from_utf8(bytes).unwrap()
+    }
+}
+
+macro_rules! int_com {
+    ($t:ty) => {
+        impl PulseCom for $t {
+            fn to_com(&self) -> Vec<u8> {
+                self.to_le_bytes().to_vec()
+            }
+
+            fn from_com(com: &mut Vec<u8>) -> Self {
+                const N: usize = std::mem::size_of::<$t>();
+                let bytes: [u8; N] = com.drain(..N).collect::<Vec<_>>().try_into().unwrap();
+                <$t>::from_le_bytes(bytes)
+            }
+        }
+    };
+}
+
+int_com!(i8);
+int_com!(i16);
+int_com!(i32);
+int_com!(i64);
+int_com!(isize);
+
+int_com!(u8);
+int_com!(u16);
+int_com!(u32);
+int_com!(u64);
+int_com!(usize);
+
+int_com!(f64);
+int_com!(f32);
+
+#[macro_export]
+macro_rules! p_com {
+    (struct $name:ident { $($n:ident: $v:ty),* $(,)? }) => {
+        #[derive(Debug, Clone)]
+        pub struct $name { $(pub $n: $v),* }
+
+        impl PulseCom for $name {
+            fn to_com(&self) -> Vec<u8> {
+                let mut vec = Vec::new();
+                $(vec.extend(self.$n.to_com());)*
+                vec
+            }
+
+            fn from_com(com: &mut Vec<u8>) -> Self {
+                Self {
+                    $($n: <$v>::from_com(com),)*
+                }
+            }
+        }
+    };
+}

src/ptc.rs

@@ -1,24 +1,117 @@
-#[derive(Debug, Clone)]
-pub struct WatchListItem {
-    pub symbol: String,
-    pub price: f64,
-    pub trend: f64,
+use pulse_macros::p_com;
+
+#[p_com]
+struct WatchListItem {
+    symbol: String,
+    price: f64,
+    trend: f64,
 }
 
-#[derive(Debug, Clone)]
-pub struct ActivePosition {
-    pub symbol: String,
-    pub profit: f64,
-    pub amount: f64,
+#[p_com]
+struct ActivePosition {
+    symbol: String,
+    profit: f64,
+    amount: f64,
 }
 
-#[derive(Debug, Clone, Copy)]
-pub enum MarketTrend {
+#[p_com]
+enum MarketTrend {
     Bullish,
     Bearish,
     Neutral,
 }
 
+#[p_com]
+enum Volatility {
+    Low,
+    Medium,
+    High,
+}
+
+#[p_com]
+struct MarketOverview {
+    trend: MarketTrend,
+    volatility: Volatility,
+    pressure: f64,
+
+    alerts: Vec<Alert>,
+}
+
+#[p_com]
+enum Feed {
+    Connected,
+    Disconnected,
+    Connecting,
+    Failed,
+}
+
+#[p_com]
+struct Status {
+    feed: Feed,
+    exchange: String,
+    dex: String,
+    latency: u16,
+}
+
+#[p_com]
+enum SignalKind {
+    Buy,
+    Sell,
+}
+
+#[p_com]
+enum SignalParameter {
+    Lim,
+    Stl,
+    Tap,
+    Chk,
+}
+
+#[p_com]
+struct Signal {
+    kind: SignalKind,
+    symbol: String,
+    param: SignalParameter,
+    price: f64,
+}
+
+#[p_com]
+enum LogKind {
+    Info,
+    Warn,
+    Err,
+    Debug,
+}
+
+#[p_com]
+struct EventLog {
+    kind: LogKind,
+    name: String,
+    message: String,
+}
+
+#[p_com]
+enum AlertLevel {
+    High,
+    Medium,
+    Low,
+}
+
+#[p_com]
+struct Alert {
+    level: AlertLevel,
+    message: String,
+}
+
+#[p_com]
+enum InspectTarget {
+    None,
+    Symbol(WatchListItem),
+    Position(ActivePosition),
+    Signal(Signal),
+    Alert(Alert),
+}
+
 impl std::fmt::Display for MarketTrend {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
@@ -29,11 +122,14 @@ impl std::fmt::Display for MarketTrend {
     }
 }
 
-#[derive(Debug, Clone, Copy)]
-pub enum Volatility {
-    Low,
-    Medium,
-    High,
+impl std::fmt::Display for AlertLevel {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Self::High => write!(f, "H"),
+            Self::Medium => write!(f, "M"),
+            Self::Low => write!(f, "L"),
+        }
+    }
 }
 
 impl std::fmt::Display for Volatility {
@@ -46,80 +142,6 @@ impl std::fmt::Display for Volatility {
     }
 }
 
-#[derive(Debug, Clone)]
-pub struct MarketOverview {
-    pub trend: MarketTrend,
-    pub volatility: Volatility,
-    pub pressure: f64,
-
-    pub alerts: Vec<Alert>,
-}
-
-#[derive(Debug, Clone, Copy)]
-pub enum Feed {
-    Connected,
-    Disconnected,
-    Connecting,
-    Failed,
-}
-
-pub struct Status {
-    pub feed: Feed,
-    pub exchange: String,
-    pub dex: String,
-    pub latency: u16,
-}
-
-#[derive(Debug, Clone, Copy)]
-pub enum SignalKind {
-    Buy,
-    Sell,
-}
-
-impl std::fmt::Display for SignalKind {
-    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
-        match self {
-            Self::Buy => write!(f, "BUY"),
-            Self::Sell => write!(f, "SELL"),
-        }
-    }
-}
-
-#[derive(Debug, Clone, Copy)]
-pub enum SignalParameter {
-    Lim,
-    Stl,
-    Tap,
-    Chk,
-}
-
-impl std::fmt::Display for SignalParameter {
-    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
-        match self {
-            Self::Lim => write!(f, "LIM"),
-            Self::Stl => write!(f, "STL"),
-            Self::Tap => write!(f, "TAP"),
-            Self::Chk => write!(f, "CHK"),
-        }
-    }
-}
-
-#[derive(Debug, Clone)]
-pub struct Signal {
-    pub kind: SignalKind,
-    pub symbol: String,
-    pub param: SignalParameter,
-    pub price: f64,
-}
-
-#[derive(Debug, Clone, Copy)]
-pub enum LogKind {
-    Info,
-    Warn,
-    Err,
-    Debug,
-}
-
 impl std::fmt::Display for LogKind {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
@@ -131,41 +153,22 @@ impl std::fmt::Display for LogKind {
     }
 }
 
-#[derive(Debug, Clone)]
-pub struct EventLog {
-    pub kind: LogKind,
-    pub name: &'static str,
-    pub message: String,
-}
-
-#[derive(Debug, Clone, Copy)]
-pub enum AlertLevel {
-    High,
-    Medium,
-    Low,
-}
-
-impl std::fmt::Display for AlertLevel {
+impl std::fmt::Display for SignalKind {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         match self {
-            Self::High => write!(f, "H"),
-            Self::Medium => write!(f, "M"),
-            Self::Low => write!(f, "L"),
+            Self::Buy => write!(f, "BUY"),
+            Self::Sell => write!(f, "SELL"),
         }
     }
 }
 
-#[derive(Debug, Clone)]
-pub struct Alert {
-    pub level: AlertLevel,
-    pub message: String,
-}
-
-#[derive(Debug, Clone)]
-pub enum InspectTarget {
-    None,
-    Symbol(Box<(WatchListItem, MarketOverview)>),
-    Position(ActivePosition),
-    Signal(Signal),
-    Alert(Alert),
+impl std::fmt::Display for SignalParameter {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Self::Lim => write!(f, "LIM"),
+            Self::Stl => write!(f, "STL"),
+            Self::Tap => write!(f, "TAP"),
+            Self::Chk => write!(f, "CHK"),
+        }
+    }
 }

src/terminal/command.rs

@@ -20,7 +20,7 @@ impl PulseTradeApp {
             _ => {
                 self.logs.lock().await.push(EventLog {
                     kind: crate::ptc::LogKind::Err,
-                    name: "cmd",
+                    name: "cmd".to_string(),
                     message: format!("Command '{}' not found", command),
                 });
             }

src/terminal/formatting.rs

@@ -12,22 +12,15 @@ impl Formatted for InspectTarget {
         match self {
             Self::None => vec!["\x1b[2mnothing to inspect\x1b[0m".to_string()],
 
-            Self::Symbol(boxed) => {
-                let (watch, overview) = &**boxed;
-                vec![
-                    Property("Symbol", format!("\x1b[35m{}\x1b[0m", watch.symbol)),
-                    Property("Price", format!("\x1b[96m{}\x1b[0m", format_f64(watch.price))),
-                    Property("Trend", format!("{}", format_f64(watch.trend))),
-                    Property("Market", format!("{}", overview.trend)),
-                    Property("Volatility", format!("{}", overview.volatility)),
-                    Property(
-                        "Pressure",
-                        format!("{:+.2}%", overview.pressure * 100.0),
-                    ),
-                    Property("Alerts", format!("{}", overview.alerts.len())),
-                ]
-                .get_formatted()
-            }
+            Self::Symbol(watch) => vec![
+                Property("Symbol", format!("\x1b[35m{}\x1b[0m", watch.symbol)),
+                Property(
+                    "Price",
+                    format!("\x1b[96m{}\x1b[0m", format_f64(watch.price)),
+                ),
+                Property("Trend", format!("{}", format_f64(watch.trend))),
+            ]
+            .get_formatted(),
 
             Self::Position(pos) => vec![
                 Property("Symbol", format!("\x1b[35m{}\x1b[0m", pos.symbol)),

src/terminal/main.rs

@@ -112,7 +112,7 @@ impl App for PulseTradeApp {
 
         logs.push(EventLog {
             kind: ptc::LogKind::Warn,
-            name: "pulse.init",
+            name: "pulse.init".to_string(),
             message: "We're still not done yet ;)".to_string(),
         });
 

src/terminal/ptc.rs

@@ -1 +1,2 @@
+include!("../pc.rs");
 include!("../ptc.rs");