rns-server 0.1.2

Batteries-included Reticulum node server
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
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

use std::collections::HashMap;

#[derive(Clone)]
pub struct Args {
    pub flags: HashMap<String, String>,
    pub positional: Vec<String>,
    pub verbosity: u8,
    pub quiet: u8,
}

impl Args {
    pub fn parse() -> Self {
        Self::parse_from(std::env::args().skip(1).collect())
    }

    pub fn parse_from(args: Vec<String>) -> Self {
        let mut flags = HashMap::new();
        let mut positional = Vec::new();
        let mut verbosity: u8 = 0;
        let mut quiet: u8 = 0;
        let mut iter = args.into_iter();

        while let Some(arg) = iter.next() {
            if arg == "--" {
                positional.extend(iter);
                break;
            } else if let Some(key) = arg.strip_prefix("--") {
                if let Some(eq_pos) = key.find('=') {
                    let (k, v) = key.split_at(eq_pos);
                    flags.insert(k.to_string(), v[1..].to_string());
                    continue;
                }
                match key {
                    "help" | "version" | "dry-run" | "disable-auth" | "no-http" => {
                        flags.insert(key.to_string(), "true".into());
                    }
                    _ => {
                        if let Some(value) = iter.next() {
                            flags.insert(key.to_string(), value);
                        } else {
                            flags.insert(key.to_string(), "true".into());
                        }
                    }
                }
            } else if arg.starts_with('-') && arg.len() > 1 {
                let chars: Vec<char> = arg[1..].chars().collect();
                for &c in &chars {
                    match c {
                        'v' => verbosity = verbosity.saturating_add(1),
                        'q' => quiet = quiet.saturating_add(1),
                        'h' => {
                            flags.insert("help".into(), "true".into());
                        }
                        'c' => {
                            if chars.len() == 1 {
                                if let Some(value) = iter.next() {
                                    flags.insert("config".into(), value);
                                } else {
                                    flags.insert("config".into(), "true".into());
                                }
                            } else {
                                flags.insert("config".into(), "true".into());
                            }
                        }
                        _ => {
                            flags.insert(c.to_string(), "true".into());
                        }
                    }
                }
            } else {
                positional.push(arg);
            }
        }

        Self {
            flags,
            positional,
            verbosity,
            quiet,
        }
    }

    pub fn get(&self, key: &str) -> Option<&str> {
        self.flags.get(key).map(|s| s.as_str())
    }

    pub fn has(&self, key: &str) -> bool {
        self.flags.contains_key(key)
    }

    pub fn config_path(&self) -> Option<&str> {
        self.get("config")
    }
}

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

    #[test]
    fn parse_start_with_config() {
        let args = Args::parse_from(vec![
            "start".into(),
            "--config".into(),
            "/tmp/rns".into(),
            "-vv".into(),
        ]);
        assert_eq!(args.positional, vec!["start"]);
        assert_eq!(args.config_path(), Some("/tmp/rns"));
        assert_eq!(args.verbosity, 2);
    }

    #[test]
    fn parse_short_config() {
        let args = Args::parse_from(vec!["start".into(), "-c".into(), "/tmp/rns".into()]);
        assert_eq!(args.config_path(), Some("/tmp/rns"));
    }
}