netget-tool 0.3.5

Network protocol stubbing toolkit: ~280 protocol servers and clients whose responses come from an LLM, a script, a static rule or a person.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

//! The one-line instance summary: status glyph, id, protocol, address, live
//! peers, sparkline, driver badge. A card's header row is one of these.

use crate::tui::app::UiKey;
use crate::tui::driver::{driver_of, Driver};
use crate::tui::projection::{ClientRow, ServerRow};

/// How a piece of text should be coloured, resolved by the renderer.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Tone {
    Normal,
    Dim,
    Good,
    Warn,
    Bad,
    Accent,
    Server,
    Client,
    Title,
    Reasoning,
}

impl Driver {
    pub fn tone(&self) -> Tone {
        match self {
            Driver::Manual => Tone::Warn,
            Driver::Llm => Tone::Reasoning,
            Driver::Silent => Tone::Dim,
            Driver::Rules => Tone::Good,
        }
    }
}

/// Everything one instance line shows, before it is fitted to a width.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct InstanceLine {
    pub key: UiKey,
    pub glyph: &'static str,
    pub glyph_tone: Tone,
    pub id: u32,
    pub protocol: String,
    /// `:8080` for a server, `→ host:port` for a client.
    pub target: String,
    /// `2⇄` live peers for a server; empty for a client.
    pub peers: String,
    pub driver: Driver,
    /// Requests parked for the human.
    pub waiting: usize,
    /// The error message when the instance is in an error state.
    pub error: Option<String>,
}

pub fn server_line(row: &ServerRow) -> InstanceLine {
    use crate::state::server::ServerStatus;
    let (glyph, glyph_tone, error) = match &row.status {
        ServerStatus::Running => ("●", Tone::Good, None),
        ServerStatus::Starting => ("◐", Tone::Warn, None),
        ServerStatus::Stopped => ("○", Tone::Dim, None),
        ServerStatus::Error(e) => ("✗", Tone::Bad, Some(e.clone())),
    };
    let live = row.conns.iter().filter(|c| c.active).count();
    InstanceLine {
        key: UiKey::Server(row.id),
        glyph,
        glyph_tone,
        id: row.id.as_u32(),
        protocol: row.protocol.to_lowercase(),
        target: row
            .local_addr
            .as_deref()
            .and_then(|a| a.rsplit_once(':').map(|(_, p)| format!(":{p}")))
            .unwrap_or_else(|| format!(":{}", row.port)),
        peers: format!("{live}⇄"),
        driver: driver_of(row.routing.as_ref()),
        waiting: row.intercepts.len(),
        error,
    }
}

pub fn client_line(row: &ClientRow) -> InstanceLine {
    use crate::state::client::ClientStatus;
    let (glyph, glyph_tone, error) = match &row.status {
        ClientStatus::Connected => ("●", Tone::Good, None),
        ClientStatus::Connecting => ("◐", Tone::Warn, None),
        ClientStatus::Disconnected => ("○", Tone::Dim, None),
        ClientStatus::Error(e) => ("✗", Tone::Bad, Some(e.clone())),
    };
    InstanceLine {
        key: UiKey::Client(row.id),
        glyph,
        glyph_tone,
        id: row.id.as_u32(),
        protocol: row.protocol.to_lowercase(),
        target: format!("→{}", row.remote_addr),
        peers: String::new(),
        driver: driver_of(row.routing.as_ref()),
        waiting: row.intercepts.len(),
        error,
    }
}

/// Truncate to `width` columns with an ellipsis, counting chars.
pub fn fit(text: &str, width: usize) -> String {
    let count = text.chars().count();
    if count <= width {
        return text.to_string();
    }
    if width == 0 {
        return String::new();
    }
    let mut out: String = text.chars().take(width - 1).collect();
    out.push('…');
    out
}