proxmox-backup/src/getopts.rs

163 lines
5.2 KiB
Rust
Raw Normal View History

2018-11-16 12:14:11 +00:00
use crate::api::schema::*;
use failure::*;
2018-11-17 10:28:26 +00:00
use std::collections::HashMap;
2018-11-16 12:14:11 +00:00
use serde_json::{json, Value};
#[derive(Debug)]
enum RawArgument {
Separator,
Argument { value: String },
Option { name: String, value: Option<String> },
}
fn parse_argument(arg: &str) -> RawArgument {
let chars: Vec<char> = arg.chars().collect();
let length = chars.len();
if length >= 2 {
2018-11-16 12:14:11 +00:00
if length == 2 { return RawArgument::Separator; }
if chars[0] == '-' {
let first = if chars[1] == '-' { 2 } else { 1 };
2018-11-16 12:14:11 +00:00
for start in first..length {
if chars[start] == '=' {
let name: String = chars[first..start].iter().collect();
let value: String = chars[start+1..length].iter().collect();
return RawArgument::Option { name, value: Some(value) }
}
}
2018-11-16 12:14:11 +00:00
let name: String = chars[first..].iter().collect();
return RawArgument::Option { name: name, value: None }
}
2018-11-16 12:14:11 +00:00
}
RawArgument::Argument { value: arg.to_string() }
}
2018-11-17 08:57:26 +00:00
pub fn parse_arguments(
args: &Vec<String>,
schema: &ObjectSchema,
2018-11-17 08:57:26 +00:00
) -> Result<(Value,Vec<String>), ParameterError> {
2018-11-16 12:14:11 +00:00
2018-11-17 08:57:26 +00:00
let mut errors = ParameterError::new();
2018-11-16 12:14:11 +00:00
let properties = &schema.properties;
2018-11-16 12:14:11 +00:00
let mut data: Vec<(String, String)> = vec![];
let mut rest: Vec<String> = vec![];
let mut pos = 0;
let mut skip = false;
loop {
if skip {
rest.push(args[pos].clone());
} else {
match parse_argument(&args[pos]) {
RawArgument::Separator => {
skip = true;
}
RawArgument::Option { name, value } => {
match value {
None => {
2018-11-17 10:28:26 +00:00
let param_schema = properties.get::<str>(&name);
let (want_bool, can_default) = match param_schema {
Some(Schema::Boolean(boolean_schema)) => {
if let Some(default) = boolean_schema.default {
if default == true { (true, false); }
}
(true, true)
}
_ => (false, false),
};
2018-11-18 07:55:21 +00:00
let mut next_is_argument = false;
let mut next_is_bool = false;
2018-11-17 10:28:26 +00:00
2018-11-18 07:55:21 +00:00
if (pos + 1) < args.len() {
let next = &args[pos+1];
if let RawArgument::Argument { value: _} = parse_argument(next) {
next_is_argument = true;
2018-11-17 10:28:26 +00:00
if let Ok(_) = parse_boolean(next) { next_is_bool = true; }
}
2018-11-18 07:55:21 +00:00
}
2018-11-17 10:28:26 +00:00
2018-11-18 07:55:21 +00:00
if want_bool {
2018-11-17 10:28:26 +00:00
if next_is_bool {
2018-11-16 12:14:11 +00:00
pos += 1;
2018-11-17 10:28:26 +00:00
data.push((name, args[pos].clone()));
} else if can_default {
data.push((name, "true".to_string()));
2018-11-16 12:14:11 +00:00
} else {
2018-11-17 10:28:26 +00:00
errors.push(format_err!("parameter '{}': {}", name,
"missing boolean value."));
}
} else {
2018-11-18 07:55:21 +00:00
if next_is_argument {
2018-11-17 10:28:26 +00:00
pos += 1;
data.push((name, args[pos].clone()));
} else {
errors.push(format_err!("parameter '{}': {}", name,
"missing parameter value."));
2018-11-16 12:14:11 +00:00
}
}
}
Some(v) => {
data.push((name, v));
}
}
}
RawArgument::Argument { value } => {
rest.push(value);
}
}
}
pos += 1;
if pos >= args.len() { break; }
}
2018-11-17 08:57:26 +00:00
if errors.len() > 0 { return Err(errors); }
2018-11-16 12:14:11 +00:00
2018-11-17 08:57:26 +00:00
let options = parse_parameter_strings(&data, schema, true)?;
Ok((options,rest))
2018-11-16 12:14:11 +00:00
}
2018-11-17 10:28:26 +00:00
#[test]
fn test_boolean_arg() {
let schema = parameter!{enable => Boolean!{ optional => false }};
let mut variants: Vec<Vec<&str>> = vec![];
variants.push(vec!["-enable"]);
variants.push(vec!["-enable=1"]);
variants.push(vec!["-enable", "yes"]);
variants.push(vec!["-enable", "Yes"]);
2018-11-17 10:28:26 +00:00
variants.push(vec!["--enable", "1"]);
for args in variants {
let string_args = args.iter().map(|s| s.to_string()).collect();
let res = parse_arguments(&string_args, &schema);
println!("RES: {:?}", res);
assert!(res.is_ok());
if let Ok((options, rest)) = res {
assert!(options["enable"] == true);
assert!(rest.len() == 0);
}
}
//Ok((options, rest)) => {
}