2015-04-16 17:31:50 +00:00
|
|
|
// You can use clap's value_t! macro with a custom enum by implementing the std::str::FromStr
|
2015-10-01 01:45:35 +00:00
|
|
|
// trait which is very straight forward. There are three ways to do this, for simple enums
|
2016-10-30 19:03:05 +00:00
|
|
|
// meaning those that don't require 'pub' or any '#[derive()]' directives you can use clap's
|
2015-04-17 14:20:56 +00:00
|
|
|
// simple_enum! macro. For those that require 'pub' or any '#[derive()]'s you can use clap's
|
2015-10-01 01:45:35 +00:00
|
|
|
// arg_enum! macro. The third way is to implement std::str::FromStr manually.
|
2015-04-16 17:31:50 +00:00
|
|
|
//
|
2015-04-17 14:20:56 +00:00
|
|
|
// In most circumstances using either simple_enum! or arg_enum! is fine.
|
2015-04-16 17:31:50 +00:00
|
|
|
//
|
2015-04-17 14:20:56 +00:00
|
|
|
// In the following example we will create two enums using macros, assign a positional argument
|
|
|
|
// that accepts only one of those values, and use clap to parse the argument.
|
2015-04-16 17:31:50 +00:00
|
|
|
|
|
|
|
// Add clap like normal
|
|
|
|
#[macro_use]
|
|
|
|
extern crate clap;
|
|
|
|
|
|
|
|
use clap::{App, Arg};
|
|
|
|
|
2015-04-17 14:20:56 +00:00
|
|
|
// Using arg_enum! is more like traditional enum declarations
|
|
|
|
//
|
|
|
|
// **NOTE:** Only bare variants are supported
|
2019-04-05 19:51:22 +00:00
|
|
|
arg_enum! {
|
2015-04-17 14:20:56 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub enum Oof {
|
|
|
|
Rab,
|
|
|
|
Zab,
|
|
|
|
Xuq
|
|
|
|
}
|
|
|
|
}
|
2015-04-16 17:31:50 +00:00
|
|
|
|
2019-04-05 19:51:22 +00:00
|
|
|
arg_enum! {
|
2016-01-21 06:48:30 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
enum Foo {
|
|
|
|
Bar,
|
|
|
|
Baz,
|
|
|
|
Qux
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-16 17:31:50 +00:00
|
|
|
fn main() {
|
|
|
|
// Create the application like normal
|
2015-04-29 17:32:00 +00:00
|
|
|
let enum_vals = ["fast", "slow"];
|
2015-04-16 17:31:50 +00:00
|
|
|
let m = App::new("myapp")
|
2018-11-14 17:05:06 +00:00
|
|
|
// Use a single positional argument that is required
|
|
|
|
.arg(Arg::from("<foo> 'The Foo to use'").possible_values(&Foo::variants()))
|
|
|
|
.arg(
|
|
|
|
Arg::from("<speed> 'The speed to use'")
|
|
|
|
// You can define a list of possible values if you want the values to be
|
|
|
|
// displayed in the help information. Whether you use possible_values() or
|
|
|
|
// not, the valid values will ALWAYS be displayed on a failed parse.
|
|
|
|
.possible_values(&enum_vals),
|
|
|
|
)
|
|
|
|
// For the second positional, lets not use possible_values() just to show the difference
|
|
|
|
.arg("<oof> 'The Oof to use'")
|
|
|
|
.get_matches();
|
2015-04-16 17:31:50 +00:00
|
|
|
|
2016-01-21 06:48:30 +00:00
|
|
|
let t = value_t!(m.value_of("foo"), Foo).unwrap_or_else(|e| e.exit());
|
|
|
|
let t2 = value_t!(m.value_of("oof"), Oof).unwrap_or_else(|e| e.exit());
|
2015-04-17 14:20:56 +00:00
|
|
|
|
2015-04-16 17:31:50 +00:00
|
|
|
// Now we can use our enum like normal.
|
|
|
|
match t {
|
2015-04-17 14:20:56 +00:00
|
|
|
Foo::Bar => println!("Found a Bar"),
|
|
|
|
Foo::Baz => println!("Found a Baz"),
|
2018-01-25 04:05:05 +00:00
|
|
|
Foo::Qux => println!("Found a Qux"),
|
2015-04-16 17:31:50 +00:00
|
|
|
}
|
2015-04-17 14:20:56 +00:00
|
|
|
|
|
|
|
// Since our Oof derives Debug, we can do this:
|
|
|
|
println!("Oof: {:?}", t2);
|
2016-01-21 06:48:30 +00:00
|
|
|
}
|