mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2024-09-17 20:44:55 +00:00
173 lines
4.5 KiB
Rust
173 lines
4.5 KiB
Rust
|
// Copyright © 2020 Intel Corporation
|
||
|
//
|
||
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
//
|
||
|
|
||
|
use std::collections::HashMap;
|
||
|
use std::fmt;
|
||
|
use std::str::FromStr;
|
||
|
|
||
|
#[derive(Default)]
|
||
|
pub struct OptionParser {
|
||
|
options: HashMap<String, OptionParserValue>,
|
||
|
}
|
||
|
|
||
|
struct OptionParserValue {
|
||
|
value: Option<String>,
|
||
|
requires_value: bool,
|
||
|
}
|
||
|
|
||
|
#[derive(Debug)]
|
||
|
pub enum OptionParserError {
|
||
|
UnknownOption(String),
|
||
|
InvalidSyntax(String),
|
||
|
Conversion(String, String),
|
||
|
}
|
||
|
|
||
|
impl fmt::Display for OptionParserError {
|
||
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||
|
match self {
|
||
|
OptionParserError::UnknownOption(s) => write!(f, "unknown option: {}", s),
|
||
|
OptionParserError::InvalidSyntax(s) => write!(f, "invalid syntax:{}", s),
|
||
|
OptionParserError::Conversion(field, value) => {
|
||
|
write!(f, "unable to parse {} for {}", value, field)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
type OptionParserResult<T> = std::result::Result<T, OptionParserError>;
|
||
|
|
||
|
impl OptionParser {
|
||
|
pub fn new() -> Self {
|
||
|
Self {
|
||
|
options: HashMap::new(),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn parse(&mut self, input: &str) -> OptionParserResult<()> {
|
||
|
if input.trim().is_empty() {
|
||
|
return Ok(());
|
||
|
}
|
||
|
|
||
|
let options_list: Vec<&str> = input.trim().split(',').collect();
|
||
|
|
||
|
for option in options_list.iter() {
|
||
|
let parts: Vec<&str> = option.split('=').collect();
|
||
|
|
||
|
match self.options.get_mut(parts[0]) {
|
||
|
None => return Err(OptionParserError::UnknownOption(parts[0].to_owned())),
|
||
|
Some(value) => {
|
||
|
if value.requires_value {
|
||
|
if parts.len() != 2 {
|
||
|
return Err(OptionParserError::InvalidSyntax((*option).to_owned()));
|
||
|
}
|
||
|
value.value = Some(parts[1].trim().to_owned());
|
||
|
} else {
|
||
|
value.value = Some(String::new());
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
pub fn add(&mut self, option: &str) -> &mut Self {
|
||
|
self.options.insert(
|
||
|
option.to_owned(),
|
||
|
OptionParserValue {
|
||
|
value: None,
|
||
|
requires_value: true,
|
||
|
},
|
||
|
);
|
||
|
|
||
|
self
|
||
|
}
|
||
|
|
||
|
pub fn add_valueless(&mut self, option: &str) -> &mut Self {
|
||
|
self.options.insert(
|
||
|
option.to_owned(),
|
||
|
OptionParserValue {
|
||
|
value: None,
|
||
|
requires_value: false,
|
||
|
},
|
||
|
);
|
||
|
|
||
|
self
|
||
|
}
|
||
|
|
||
|
pub fn get(&self, option: &str) -> Option<String> {
|
||
|
self.options
|
||
|
.get(option)
|
||
|
.and_then(|v| v.value.clone())
|
||
|
.and_then(|s| if s.is_empty() { None } else { Some(s) })
|
||
|
}
|
||
|
|
||
|
pub fn is_set(&self, option: &str) -> bool {
|
||
|
self.options
|
||
|
.get(option)
|
||
|
.and_then(|v| v.value.as_ref())
|
||
|
.is_some()
|
||
|
}
|
||
|
|
||
|
pub fn convert<T: FromStr>(&self, option: &str) -> OptionParserResult<Option<T>> {
|
||
|
match self.get(option) {
|
||
|
None => Ok(None),
|
||
|
Some(v) => Ok(Some(v.parse().map_err(|_| {
|
||
|
OptionParserError::Conversion(option.to_owned(), v.to_owned())
|
||
|
})?)),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub struct Toggle(pub bool);
|
||
|
|
||
|
pub enum ToggleParseError {
|
||
|
InvalidValue(String),
|
||
|
}
|
||
|
|
||
|
impl FromStr for Toggle {
|
||
|
type Err = ToggleParseError;
|
||
|
|
||
|
fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
|
||
|
match s.to_lowercase().as_str() {
|
||
|
"" => Ok(Toggle(false)),
|
||
|
"on" => Ok(Toggle(true)),
|
||
|
"off" => Ok(Toggle(false)),
|
||
|
"true" => Ok(Toggle(true)),
|
||
|
"false" => Ok(Toggle(false)),
|
||
|
_ => Err(ToggleParseError::InvalidValue(s.to_owned())),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub struct ByteSized(pub u64);
|
||
|
|
||
|
pub enum ByteSizedParseError {
|
||
|
InvalidValue(String),
|
||
|
}
|
||
|
|
||
|
impl FromStr for ByteSized {
|
||
|
type Err = ByteSizedParseError;
|
||
|
|
||
|
fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
|
||
|
Ok(ByteSized({
|
||
|
let s = s.trim();
|
||
|
let shift = if s.ends_with('K') {
|
||
|
10
|
||
|
} else if s.ends_with('M') {
|
||
|
20
|
||
|
} else if s.ends_with('G') {
|
||
|
30
|
||
|
} else {
|
||
|
0
|
||
|
};
|
||
|
|
||
|
let s = s.trim_end_matches(|c| c == 'K' || c == 'M' || c == 'G');
|
||
|
s.parse::<u64>()
|
||
|
.map_err(|_| ByteSizedParseError::InvalidValue(s.to_owned()))?
|
||
|
<< shift
|
||
|
}))
|
||
|
}
|
||
|
}
|