swaybar_info/src/args.rs

use std::collections::HashMap;
use std::io;
use std::io::Write;

pub struct ArgsResult {
    pub map: HashMap<String, String>,
    pub regex_cmds: Vec<String>,
}

pub fn get_args() -> ArgsResult {
    let mut map = HashMap::new();
    let mut regex_cmds = Vec::new();

    let mut first = true;
    for arg in std::env::args() {
        if first {
            first = false;
            continue;
        } else if arg.starts_with("--netdev=") {
            let (_, back) = arg.split_at(9);
            map.insert("netdev".into(), back.into());
        } else if arg.starts_with("--interval-sec=") {
            let (_, back) = arg.split_at(15);
            map.insert("interval-sec".into(), back.into());
        } else if arg.starts_with("--regex-cmd=") {
            let (_, back) = arg.split_at(12);
            regex_cmds.push(back.to_owned());
        } else if arg == "--help" || arg == "-h" {
            map.insert("help".into(), "".into());
        } else {
            let mut stderr_handle = io::stderr().lock();
            stderr_handle
                .write_all(format!("WARNING: Got invalid arg \"{}\"!\n", arg).as_bytes())
                .ok();
        }
    }

    ArgsResult { map, regex_cmds }
}

pub fn print_usage() {
    let mut stderr_handle = io::stderr().lock();
    stderr_handle.write_all(b"Usage:\n").ok();
    stderr_handle
        .write_all(b"  -h | --help\t\t\t\tPrints help\n")
        .ok();
    stderr_handle
        .write_all(b"  --netdev=<device_name>\t\tCheck network traffic on specified device\n")
        .ok();
    stderr_handle
        .write_all(b"  --interval-sec=<seconds>\t\tOutput at intervals of <seconds> (default 5)\n")
        .ok();
    stderr_handle
        .write_all(
            b"  --regex-cmd=<cmd>,<args...>,<regex>\tUse an output of a command as a metric\n",
        )
        .ok();
}