freja 0.0.1

Command-line toolbox integrating multiple utilities; includes a WebSocket tester
Documentation 
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

use clap::Args;

mod client;
mod server;

use crate::error::Result;
use std::future::Future;

/// Sub command for `freja ws`
#[derive(Args, Debug)]
pub struct WebsocketOpts {
    /// Server mode (`-l` listen)
    #[arg(short, long, help = "Start in server mode")]
    pub listen: bool,

    /// Host name or address to connect to
    #[arg(value_name = "URL", help = "Host name or IP address")]
    pub url: String,

    /// Port to connect to (used only in client mode)
    #[arg(value_name = "PORT", help = "Port number", default_value_t = 4444)]
    pub port: u16,
}

pub async fn run_with<FC, FCfut, FS, FSfut>(
    opts: WebsocketOpts,
    client_fn: FC,
    server_fn: FS,
) -> Result<()>
where
    FC: Fn(String, u16) -> FCfut,
    FCfut: Future<Output = Result<()>> + Send,
    FS: Fn(WebsocketOpts) -> FSfut,
    FSfut: Future<Output = Result<()>> + Send,
{
    if opts.listen {
        println!("Listening on {}", opts.url);
        server_fn(opts).await
    } else {
        let port = opts.port;
        let url = opts.url;
        client_fn(url, port).await
    }
}

pub async fn run(opts: WebsocketOpts) -> Result<()> {
    run_with(opts, client::run, server::run).await
}

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

    #[test]
    fn parse_client_mode() {
        #[derive(Parser)]
        struct Wrapper {
            #[command(flatten)]
            opts: WebsocketOpts,
        }
        let wrapper = Wrapper::try_parse_from(["cmd", "example.com", "5555"]).unwrap();
        assert!(!wrapper.opts.listen);
        assert_eq!(wrapper.opts.url, "example.com");
        assert_eq!(wrapper.opts.port, 5555);
    }

    #[test]
    fn parse_server_mode() {
        #[derive(Parser)]
        struct Wrapper {
            #[command(flatten)]
            opts: WebsocketOpts,
        }
        let wrapper = Wrapper::try_parse_from(["cmd", "-l", "127.0.0.1", "8888"]).unwrap();
        assert!(wrapper.opts.listen);
        assert_eq!(wrapper.opts.url, "127.0.0.1");
        assert_eq!(wrapper.opts.port, 8888);
    }
}