rasul
5 years ago
parent
4dfa3dd745
commit
4e36fb32e6
@ -0,0 +1,74 @@
|
||||
use std::env;
|
||||
use std::path::PathBuf;
|
||||
|
||||
use dirs::config_dir;
|
||||
use getopts::Options;
|
||||
use log::LevelFilter;
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct Config {
|
||||
pub config_directory: PathBuf,
|
||||
pub log_level: LevelFilter,
|
||||
}
|
||||
|
||||
impl Config {
|
||||
pub fn from_args() -> Config {
|
||||
let args: Vec<String> = env::args().collect();
|
||||
let program = args[0].clone();
|
||||
|
||||
let mut opts = Options::new();
|
||||
opts.optopt("c", "config", "config directory", "CONFIG");
|
||||
opts.optopt(
|
||||
"l",
|
||||
"level",
|
||||
"log level (error, warn, info, debug, trace)",
|
||||
"LEVEL",
|
||||
);
|
||||
opts.optflag("h", "help", "print this help menu");
|
||||
|
||||
let matches = match opts.parse(&args[1..]) {
|
||||
Ok(m) => m,
|
||||
Err(e) => {
|
||||
println!("FATAL: Error parsing arguments: {:?}", e);
|
||||
std::process::exit(1);
|
||||
}
|
||||
};
|
||||
|
||||
if matches.opt_present("h") {
|
||||
let brief = format!("Usage: {} [options]", program);
|
||||
print!("{}", opts.usage(&brief));
|
||||
std::process::exit(0);
|
||||
}
|
||||
|
||||
let config_directory = if let Some(dir) = matches.opt_str("c") {
|
||||
PathBuf::from(dir)
|
||||
} else if let Some(mut dir) = config_dir() {
|
||||
dir.push("sup");
|
||||
dir
|
||||
} else {
|
||||
PathBuf::from(".")
|
||||
};
|
||||
|
||||
let mut log_level: LevelFilter = LevelFilter::Info;
|
||||
|
||||
if let Some(raw_level) = matches.opt_str("l") {
|
||||
log_level = match raw_level.as_str() {
|
||||
"error" => LevelFilter::Error,
|
||||
"warn" => LevelFilter::Warn,
|
||||
"info" => LevelFilter::Info,
|
||||
"debug" => LevelFilter::Debug,
|
||||
"trace" => LevelFilter::Trace,
|
||||
_ => {
|
||||
println!("FATAL: unknown log level: {}", raw_level);
|
||||
println!("Options are error, warn, info, debug, or trace");
|
||||
std::process::exit(1);
|
||||
}
|
||||
};
|
||||
};
|
||||
|
||||
Config {
|
||||
config_directory,
|
||||
log_level,
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue