router: cli: move freestanding new parse loop into its method

Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
This commit is contained in:
Wolfgang Bumiller 2024-07-04 09:54:45 +02:00
parent 95c0614ccd
commit 9288c00372

View File

@ -315,18 +315,6 @@ impl<'t, 'o> ParseOptions<'t, 'o> {
/// If `stop_at_positional` is set, non-positional parameters after the first positional one /// If `stop_at_positional` is set, non-positional parameters after the first positional one
/// are also included in the returned list. /// are also included in the returned list.
pub fn parse<A, AI>(self, args: A) -> Result<Vec<AI>, ParameterError> pub fn parse<A, AI>(self, args: A) -> Result<Vec<AI>, ParameterError>
where
A: IntoIterator<Item = AI>,
AI: AsRef<str>,
{
parse_parameters(args, self)
}
}
pub(crate) fn parse_parameters<A, AI>(
args: A,
parse_opts: ParseOptions<'_, '_>,
) -> Result<Vec<AI>, ParameterError>
where where
A: IntoIterator<Item = AI>, A: IntoIterator<Item = AI>,
AI: AsRef<str>, AI: AsRef<str>,
@ -339,7 +327,7 @@ where
let arg = orig_arg.as_ref(); let arg = orig_arg.as_ref();
if arg == "--" { if arg == "--" {
if parse_opts.retain_separator { if self.retain_separator {
positional.push(orig_arg); positional.push(orig_arg);
} }
break; break;
@ -349,7 +337,7 @@ where
Some(opt) => opt, Some(opt) => opt,
None => { None => {
positional.push(orig_arg); positional.push(orig_arg);
if parse_opts.stop_at_positional { if self.stop_at_positional {
break; break;
} }
continue; continue;
@ -358,32 +346,29 @@ where
if let Some(eq) = option.find('=') { if let Some(eq) = option.find('=') {
let (option, argument) = (&option[..eq], &option[(eq + 1)..]); let (option, argument) = (&option[..eq], &option[(eq + 1)..]);
if parse_opts.deny_unknown && !parse_opts.option_schemas.contains_key(option) { if self.deny_unknown && !self.option_schemas.contains_key(option) {
if parse_opts.stop_at_unknown { if self.stop_at_unknown {
positional.push(orig_arg); positional.push(orig_arg);
break; break;
} }
errors.push(option.to_string(), format_err!("unknown option {option:?}")); errors.push(option.to_string(), format_err!("unknown option {option:?}"));
} }
parse_opts self.target.push((option.to_string(), argument.to_string()));
.target
.push((option.to_string(), argument.to_string()));
continue; continue;
} }
if parse_opts.deny_unknown && !parse_opts.option_schemas.contains_key(option) { if self.deny_unknown && !self.option_schemas.contains_key(option) {
if parse_opts.stop_at_unknown { if self.stop_at_unknown {
positional.push(orig_arg); positional.push(orig_arg);
break; break;
} }
errors.push(option.to_string(), format_err!("unknown option {option:?}")); errors.push(option.to_string(), format_err!("unknown option {option:?}"));
} }
match parse_opts.option_schemas.get(option) { match self.option_schemas.get(option) {
Some(Schema::Boolean(schema)) => { Some(Schema::Boolean(schema)) => {
if let Some(value) = args.next_if(|v| parse_boolean(v.as_ref()).is_ok()) { if let Some(value) = args.next_if(|v| parse_boolean(v.as_ref()).is_ok()) {
parse_opts self.target
.target
.push((option.to_string(), value.as_ref().to_string())); .push((option.to_string(), value.as_ref().to_string()));
} else { } else {
// next parameter is not a boolean value // next parameter is not a boolean value
@ -391,9 +376,7 @@ where
// default-true booleans cannot be passed without values: // default-true booleans cannot be passed without values:
errors.push(option.to_string(), format_err!("missing boolean value.")); errors.push(option.to_string(), format_err!("missing boolean value."));
} }
parse_opts self.target.push((option.to_string(), "true".to_string()))
.target
.push((option.to_string(), "true".to_string()))
} }
} }
_ => { _ => {
@ -401,13 +384,12 @@ where
let next = match args.next() { let next = match args.next() {
Some(next) => next.as_ref().to_string(), Some(next) => next.as_ref().to_string(),
None => { None => {
errors.push(option.to_string(), format_err!("missing parameter value.")); errors
.push(option.to_string(), format_err!("missing parameter value."));
break; break;
} }
}; };
parse_opts self.target.push((option.to_string(), next.to_string()));
.target
.push((option.to_string(), next.to_string()));
continue; continue;
} }
} }
@ -420,3 +402,4 @@ where
positional.extend(args); positional.extend(args);
Ok(positional) Ok(positional)
} }
}