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

use std::{env, ops::Index};

#[allow(dead_code)]
pub fn collect_arguments() -> Vec<String> {
    return env::args().collect::<Vec<String>>();
}

#[allow(dead_code)]
pub fn get_named_argument(args: &Vec<String>, name: &str) -> Option<String> {
    for (i, v) in args.iter().enumerate() {
        // if name does not match, continue
        if v != &format!("--{}", name) {
            continue;
        };

        // return value
        let val: &String = args.index(i + 1);

        // ...make sure val exists (return None if it doesn't!)
        if val.is_empty() {
            return Option::None;
        }

        return Option::Some(String::from(val));
    }

    return Option::None;
}

pub fn get_var(var: &str) -> Option<String> {
    let res = env::var(var);

    if res.is_ok() {
        Option::Some(res.unwrap())
    } else {
        Option::None
    }
}