clap/examples/13_enum_values.rs

55 lines
1.4 KiB
Rust
Raw Normal View History

// In the following example we will create an enum with 4 values, assign a positional argument
// that accepts only one of those values, and use clap to parse the argument.
//
// Start with bringing the trait into scope.
use std::str::FromStr;
// Add clap like normal
2020-04-12 03:24:49 +00:00
use clap::{App, Arg};
// Define your enum
enum Vals {
Foo,
Bar,
Baz,
2018-01-25 04:05:05 +00:00
Qux,
}
// Implement the trait
impl FromStr for Vals {
type Err = &'static str;
2015-10-01 01:45:35 +00:00
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"Foo" => Ok(Vals::Foo),
"Bar" => Ok(Vals::Bar),
"Baz" => Ok(Vals::Baz),
"Qux" => Ok(Vals::Qux),
2018-01-25 04:05:05 +00:00
_ => Err("no match"),
}
}
}
fn main() {
// Create the application like normal
let m = App::new("myapp")
2018-11-14 17:05:06 +00:00
// Use a single positional argument that is required
.arg(
Arg::from("<type> 'The type to use'")
// Define the list of possible values
.possible_values(["Foo", "Bar", "Baz", "Qux"]),
2018-11-14 17:05:06 +00:00
)
.get_matches();
2020-04-12 03:24:49 +00:00
// Note that you don't have to specify the type since rustc can infer it for you
let t = m.value_of_t("type").unwrap_or_else(|e| e.exit());
// Now we can use our enum like normal.
match t {
Vals::Foo => println!("Found a Foo"),
Vals::Bar => println!("Found a Bar"),
Vals::Baz => println!("Found a Baz"),
2018-01-25 04:05:05 +00:00
Vals::Qux => println!("Found a Qux"),
}
}