confab 0.1.0-alpha

Asynchronous line-oriented interactive TCP client
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156

use crate::util::{chomp, display_vis, JsonStrMap};
use chrono::{DateTime, Local};
use crossterm::style::{StyledContent, Stylize};
use std::net::SocketAddr;

pub(crate) enum Event {
    ConnectStart {
        timestamp: DateTime<Local>,
        host: String,
        port: u16,
    },
    ConnectFinish {
        timestamp: DateTime<Local>,
        peer: SocketAddr,
    },
    TlsStart {
        timestamp: DateTime<Local>,
    },
    TlsFinish {
        timestamp: DateTime<Local>,
    },
    Recv {
        timestamp: DateTime<Local>,
        data: String,
    },
    Send {
        timestamp: DateTime<Local>,
        data: String,
    },
    Disconnect {
        timestamp: DateTime<Local>,
    },
    Error {
        timestamp: DateTime<Local>,
        data: anyhow::Error,
    },
}

impl Event {
    pub(crate) fn connect_start(host: &str, port: u16) -> Self {
        Event::ConnectStart {
            timestamp: Local::now(),
            host: String::from(host),
            port,
        }
    }

    pub(crate) fn connect_finish(peer: SocketAddr) -> Self {
        Event::ConnectFinish {
            timestamp: Local::now(),
            peer,
        }
    }

    pub(crate) fn tls_start() -> Self {
        Event::TlsStart {
            timestamp: Local::now(),
        }
    }

    pub(crate) fn tls_finish() -> Self {
        Event::TlsFinish {
            timestamp: Local::now(),
        }
    }

    pub(crate) fn recv(data: String) -> Self {
        Event::Recv {
            timestamp: Local::now(),
            data,
        }
    }

    pub(crate) fn send(data: String) -> Self {
        Event::Send {
            timestamp: Local::now(),
            data,
        }
    }

    pub(crate) fn disconnect() -> Self {
        Event::Disconnect {
            timestamp: Local::now(),
        }
    }

    pub(crate) fn error(data: anyhow::Error) -> Self {
        Event::Error {
            timestamp: Local::now(),
            data,
        }
    }

    pub(crate) fn timestamp(&self) -> &DateTime<Local> {
        match self {
            Event::ConnectStart { timestamp, .. } => timestamp,
            Event::ConnectFinish { timestamp, .. } => timestamp,
            Event::TlsStart { timestamp } => timestamp,
            Event::TlsFinish { timestamp } => timestamp,
            Event::Recv { timestamp, .. } => timestamp,
            Event::Send { timestamp, .. } => timestamp,
            Event::Disconnect { timestamp } => timestamp,
            Event::Error { timestamp, .. } => timestamp,
        }
    }

    pub(crate) fn display_time(&self) -> String {
        self.timestamp().format("%H:%M:%S").to_string()
    }

    pub(crate) fn sigil(&self) -> char {
        match self {
            Event::Recv { .. } => '<',
            Event::Send { .. } => '>',
            Event::Error { .. } => '!',
            _ => '*',
        }
    }

    pub(crate) fn message(&self) -> Vec<StyledContent<String>> {
        match self {
            Event::ConnectStart { .. } => vec![String::from("Connecting ...").stylize()],
            Event::ConnectFinish { peer, .. } => vec![format!("Connected to {peer}").stylize()],
            Event::TlsStart { .. } => vec![String::from("Initializing TLS ...").stylize()],
            Event::TlsFinish { .. } => vec![String::from("TLS established").stylize()],
            Event::Recv { data, .. } => display_vis(chomp(data)),
            Event::Send { data, .. } => display_vis(chomp(data)),
            Event::Disconnect { .. } => vec![String::from("Disconnected").stylize()],
            Event::Error { data, .. } => vec![format!("{data:#}").stylize()],
        }
    }

    pub(crate) fn to_json(&self) -> String {
        let json = JsonStrMap::new().field("timestamp", &self.timestamp().to_rfc3339());
        match self {
            Event::ConnectStart { host, port, .. } => json
                .field("event", "connection-start")
                .field("host", host)
                .raw_field("port", &port.to_string())
                .finish(),
            Event::ConnectFinish { peer, .. } => json
                .field("event", "connection-complete")
                .field("peer_ip", &peer.ip())
                .finish(),
            Event::TlsStart { .. } => json.field("event", "tls-start").finish(),
            Event::TlsFinish { .. } => json.field("event", "tls-complete").finish(),
            Event::Recv { data, .. } => json.field("event", "recv").field("data", data).finish(),
            Event::Send { data, .. } => json.field("event", "send").field("data", data).finish(),
            Event::Disconnect { .. } => json.field("event", "disconnect").finish(),
            Event::Error { data, .. } => json
                .field("event", "error")
                .field("data", &format!("{data:#}"))
                .finish(),
        }
    }
}