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
|
use getopts::Options;
type Milliseconds = u16;
#[repr(C)]
#[derive(Default)]
struct Args {
reload: bool,
daemon: bool,
}
#[repr(C)]
#[derive(Deserialize)]
pub struct Config {
#[serde(skip)]
args: Args,
timeout: Milliseconds,
}
impl Default for Config {
fn default() -> Self {
Config {
args: Args::default(),
timeout: 500,
}
}
}
fn print_usage(opts: Options) {
let brief = "Usage: dome-key [options]";
print!("{}", opts.usage(&brief));
}
pub fn parse_args(args: &[String]) -> Config {
let mut opts = Options::new();
opts.optflag("d", "daemon", "run the daemon in the current shell");
opts.optflag("r", "reload-mappings", "reload the mappings file");
opts.optflag("h", "help", "print this help menu");
let matches = match opts.parse(&args[1..]) {
Ok(m) => m,
Err(e) => panic!(e),
};
let mut config = Config::default();
if matches.opt_present("h") {
print_usage(opts);
return config;
}
if matches.opt_present("r") {
config.args.reload = true;
} else if matches.opt_present("d") {
config.args.daemon = true;
} else {
print_usage(opts);
}
config
}
|
