proxelar 0.3.0

MITM proxy for HTTP/HTTPS traffic
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

use crossterm::style::{Color, Stylize};
use proxyapi::ProxyEvent;
use tokio::sync::mpsc;
use tokio_util::sync::CancellationToken;

use crate::interface::format_size;

pub async fn run(mut event_rx: mpsc::Receiver<ProxyEvent>, cancel: CancellationToken) {
    println!("{}", "Proxelar proxy running. Press Ctrl+C to stop.".bold());
    println!();

    loop {
        let event = tokio::select! {
            event = event_rx.recv() => match event {
                Some(e) => e,
                None => break,
            },
            () = cancel.cancelled() => break,
        };

        match &event {
            ProxyEvent::RequestComplete {
                id,
                request,
                response,
            } => {
                let now = chrono::Local::now().format("%H:%M:%S");
                let method = request.method().as_str();
                let status = response.status().as_u16();
                let uri = request.uri().to_string();
                let body_len = response.body().len();

                let method_colored = match method {
                    "GET" => method.with(Color::Green),
                    "POST" => method.with(Color::Yellow),
                    "PUT" => method.with(Color::Blue),
                    "DELETE" => method.with(Color::Red),
                    "PATCH" => method.with(Color::Magenta),
                    _ => method.with(Color::White),
                };

                let status_colored = match status {
                    200..=299 => status.to_string().with(Color::Green),
                    300..=399 => status.to_string().with(Color::Cyan),
                    400..=499 => status.to_string().with(Color::Yellow),
                    500..=599 => status.to_string().with(Color::Red),
                    _ => status.to_string().with(Color::White),
                };

                let size = format_size(body_len);

                println!("[{now}] #{id} {method_colored} {status_colored} {uri} ({size})");
            }
            ProxyEvent::Error { message } => {
                eprintln!(
                    "[{}] {}",
                    chrono::Local::now().format("%H:%M:%S"),
                    message.as_str().with(Color::Red)
                );
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_format_size() {
        assert_eq!(format_size(512), "512B");
        assert_eq!(format_size(2048), "2.0KB");
        assert_eq!(format_size(1048576), "1.0MB");
    }
}