2015-04-16 13:31:50 -04:00
|
|
|
// You can use clap's value_t! macro with a custom enum by implementing the std::str::FromStr
|
2015-09-30 21:45:35 -04:00
|
|
|
// trait which is very straight forward. There are three ways to do this, for simple enums
|
2016-10-30 20:03:05 +01:00
|
|
|
// meaning those that don't require 'pub' or any '#[derive()]' directives you can use clap's
|
2015-04-17 10:20:56 -04:00
|
|
|
// simple_enum! macro. For those that require 'pub' or any '#[derive()]'s you can use clap's
|
2015-09-30 21:45:35 -04:00
|
|
|
// arg_enum! macro. The third way is to implement std::str::FromStr manually.
|
2015-04-16 13:31:50 -04:00
|
|
|
//
|
2015-04-17 10:20:56 -04:00
|
|
|
// In most circumstances using either simple_enum! or arg_enum! is fine.
|
2015-04-16 13:31:50 -04:00
|
|
|
//
|
2015-04-17 10:20:56 -04: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 13:31:50 -04:00
|
|
|
|
|
|
|
// Add clap like normal
|
|
|
|
#[macro_use]
|
|
|
|
extern crate clap;
|
|
|
|
|
|
|
|
use clap::{App, Arg};
|
|
|
|
|
2015-04-17 10:20:56 -04:00
|
|
|
// Using arg_enum! is more like traditional enum declarations
|
|
|
|
//
|
|
|
|
// **NOTE:** Only bare variants are supported
|
2019-04-05 15:51:22 -04:00
|
|
|
arg_enum! {
|
2015-04-17 10:20:56 -04:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub enum Oof {
|
|
|
|
Rab,
|
|
|
|
Zab,
|
|
|
|
Xuq
|
|
|
|
}
|
|
|
|
}
|
2015-04-16 13:31:50 -04:00
|
|
|
|
2019-04-05 15:51:22 -04:00
|
|
|
arg_enum! {
|
2016-01-21 01:48:30 -05:00
|
|
|
#[derive(Debug)]
|
|
|
|
enum Foo {
|
|
|
|
Bar,
|
|
|
|
Baz,
|
|
|
|
Qux
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-16 13:31:50 -04:00
|
|
|
fn main() {
|
|
|
|
// Create the application like normal
|
2015-04-29 13:32:00 -04:00
|
|
|
let enum_vals = ["fast", "slow"];
|
2015-04-16 13:31:50 -04:00
|
|
|
let m = App::new("myapp")
|
2018-11-14 12:05:06 -05: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 13:31:50 -04:00
|
|
|
|
2016-01-21 01:48:30 -05: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 10:20:56 -04:00
|
|
|
|
2015-04-16 13:31:50 -04:00
|
|
|
// Now we can use our enum like normal.
|
|
|
|
match t {
|
2015-04-17 10:20:56 -04:00
|
|
|
Foo::Bar => println!("Found a Bar"),
|
|
|
|
Foo::Baz => println!("Found a Baz"),
|
2018-01-24 23:05:05 -05:00
|
|
|
Foo::Qux => println!("Found a Qux"),
|
2015-04-16 13:31:50 -04:00
|
|
|
}
|
2015-04-17 10:20:56 -04:00
|
|
|
|
|
|
|
// Since our Oof derives Debug, we can do this:
|
|
|
|
println!("Oof: {:?}", t2);
|
2016-01-21 01:48:30 -05:00
|
|
|
}
|