pulse-trader/pulse-wire/src/terminal.rs

use crate::PulseWire;
use pulse_macros::pwp;

#[pwp]
pub enum TerminalServerMessage {
    // WatchList
    WatchListUpdated(Vec<WatchListItem>),

    // Positions
    PositionsUpdated(Vec<ActivePosition>),

    // Overview
    OverviewUpdated(MarketOverview),

    // Signals
    SignalsUpdated(Vec<Signal>),

    // Inspector
    Inspect(InspectTarget),

    // Status
    SetStatus(Status),

    // Logs
    SetLogs(Vec<EventLog>),
    AddLog(EventLog),
}

#[pwp]
pub enum TerminalClientMessage {
    ExecuteCommand(String),
}

#[pwp]
pub struct WatchListItem {
    symbol: String,
    price: f64,
    trend: f64,
    volume_24h: f64,
}

#[pwp]
pub struct ActivePosition {
    pub symbol: String,
    pub size: f64,
    pub entry_price: f64,
    pub mark_price: f64,
    pub profit: f64,
}

#[pwp]
pub enum MarketTrend {
    Bullish,
    Bearish,
    Neutral,
}

#[pwp]
pub enum Volatility {
    Low,
    Medium,
    High,
}

#[pwp]
pub struct MarketOverview {
    trend: MarketTrend,
    volatility: Volatility,
    pressure: f64,

    alerts: Vec<Alert>,
}

#[pwp]
pub enum Feed {
    Connected,
    Disconnected,
    Connecting,
    Failed,
}

#[pwp]
pub struct Status {
    feed: Feed,
    exchange: String,
    dex: String,
    latency: u16,
}

#[pwp]
pub enum SignalKind {
    Buy,
    Sell,
}

#[pwp]
pub enum SignalParameter {
    Lim,
    Stl,
    Tap,
    Chk,
}

#[pwp]
pub struct Signal {
    kind: SignalKind,
    symbol: String,
    param: SignalParameter,
    price: f64,
}

#[pwp]
pub enum LogKind {
    Info,
    Warn,
    Err,
    Debug,
}

#[pwp]
pub struct EventLog {
    kind: LogKind,
    name: String,
    message: String,
}

#[pwp]
pub enum AlertLevel {
    High,
    Medium,
    Low,
}

#[pwp]
pub struct Alert {
    level: AlertLevel,
    message: String,
}

#[pwp]
pub enum InspectTarget {
    None,
    Some(Vec<InspectItem>),
}

#[pwp]
pub enum InspectItem {
    String(String),
    Symbol(String),
    USD(f64),
    F64(f64),
}

impl std::fmt::Display for MarketTrend {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Bullish => write!(f, "Bullish"),
            Self::Bearish => write!(f, "Bearish"),
            Self::Neutral => write!(f, "Neutral"),
        }
    }
}

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 {
    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"),
        }
    }
}

impl std::fmt::Display for LogKind {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Info => write!(f, "INFO"),
            Self::Warn => write!(f, "WARN"),
            Self::Err => write!(f, "ERR"),
            Self::Debug => write!(f, "DEBUG"),
        }
    }
}

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"),
        }
    }
}

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"),
        }
    }
}