// Convenience for writing to stderr thanks to https://github.com/BurntSushi macro_rules! wlnerr( ($($arg:tt)*) => ({ use std::io::{Write, stderr}; writeln!(&mut stderr(), $($arg)*).ok(); }) ); macro_rules! werr( ($($arg:tt)*) => ({ use std::io::{Write, stderr}; write!(&mut stderr(), $($arg)*).ok(); }) ); #[cfg(feature = "debug")] macro_rules! debugln { ($fmt:expr) => (println!(concat!("**DEBUG** ", $fmt))); ($fmt:expr, $($arg:tt)*) => (println!(concat!("**DEBUG** ",$fmt), $($arg)*)); } #[cfg(feature = "debug")] macro_rules! debug { ($fmt:expr) => (print!(concat!("**DEBUG** ", $fmt))); ($fmt:expr, $($arg:tt)*) => (println!(concat!("**DEBUG** ",$fmt), $($arg)*)); } #[cfg(not(feature = "debug"))] macro_rules! debugln { ($fmt:expr) => (); ($fmt:expr, $($arg:tt)*) => (); } #[cfg(not(feature = "debug"))] macro_rules! debug { ($fmt:expr) => (); ($fmt:expr, $($arg:tt)*) => (); } #[cfg(feature = "yaml")] #[macro_export] macro_rules! load_yaml { ($yml:expr) => ( &::clap::YamlLoader::load_from_str(include_str!($yml)).ok().expect("failed to load YAML file")[0] ); } // used in src/args/arg_builder/option.rs macro_rules! print_opt_help { ($opt:ident, $spc:expr, $w:ident) => { if let Some(h) = $opt.help { if h.contains("{n}") { let mut hel = h.split("{n}"); if let Some(part) = hel.next() { try!(write!($w, "{}", part)); } for part in hel { try!(write!($w, "\n")); write_spaces!($spc, $w); try!(write!($w, "{}", part)); } } else { try!(write!($w, "{}", h)); } if let Some(ref pv) = $opt.possible_vals { try!(write!($w, " [values:")); for pv_s in pv.iter() { try!(write!($w, " {}", pv_s)); } try!(write!($w, "]")); } } }; } // Helper/deduplication macro for printing the correct number of spaces in help messages // used in: // src/args/arg_builder/*.rs // src/app/mod.rs macro_rules! write_spaces { ($num:expr, $w:ident) => ({ for _ in 0..$num { try!(write!($w, " ")); } }) } // convenience macro for remove an item from a vec macro_rules! vec_remove { ($vec:expr, $to_rem:ident) => { { let mut ix = None; $vec.dedup(); for (i, val) in $vec.iter().enumerate() { if &val == &$to_rem { ix = Some(i); break; } } if let Some(i) = ix { $vec.remove(i); } } } } // Thanks to bluss and flan3002 in #rust IRC // // Helps with rightward drift when iterating over something and matching each // item. macro_rules! for_match { ($it:ident, $($p:pat => $($e:expr);+),*) => { for i in $it { match i { $( $p => { $($e)+ } )* } } }; } /// Convenience macro getting a typed value `T` where `T` implements `std::str::FromStr` /// This macro returns a `Result` which allows you as the developer to decide /// what you'd like to do on a failed parse. There are two types of errors, parse failures /// and those where the argument wasn't present (such as a non-required argument). /// /// You can use it to get a single value, or a `Vec` with the `values_of()` /// /// **NOTE:** Be cautious, as since this a macro invocation it's not exactly like /// standard syntax. /// /// /// # Examples single value /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::App; /// # fn main() { /// let matches = App::new("myapp") /// .arg_from_usage("[length] 'Set the length to use as a pos whole num, i.e. 20'") /// .get_matches(); /// let len = value_t!(matches.value_of("length"), u32) /// .unwrap_or_else(|e|{ /// println!("{}",e); /// std::process::exit(1) /// }); /// /// println!("{} + 2: {}", len, len + 2); /// # } /// ``` /// /// /// # Examples multiple values /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::App; /// # fn main() { /// let matches = App::new("myapp") /// .arg_from_usage("[seq]... 'A sequence of pos whole nums, i.e. 20 45'") /// .get_matches(); /// for v in value_t!(matches.values_of("seq"), u32) /// .unwrap_or_else(|e|{ /// println!("{}",e); /// std::process::exit(1) /// }) { /// println!("{} + 2: {}", v, v + 2); /// } /// # } /// ``` #[macro_export] macro_rules! value_t { ($m:ident.value_of($v:expr), $t:ty) => { match $m.value_of($v) { Some(v) => { match v.parse::<$t>() { Ok(val) => Ok(val), Err(_) => Err(format!("'{}' isn't a valid value", ::clap::Format::Warning(v))), } }, None => Err(format!("The argument '{}' not found", ::clap::Format::Warning($v))) } }; ($m:ident.values_of($v:expr), $t:ty) => { match $m.values_of($v) { Some(ref v) => { let mut tmp = Vec::with_capacity(v.len()); let mut err = None; for pv in v { match pv.parse::<$t>() { Ok(rv) => tmp.push(rv), Err(e) => { err = Some(format!("'{}' isn't a valid value\n\t{}", ::clap::Format::Warning(pv),e)); break } } } match err { Some(e) => Err(e), None => Ok(tmp) } }, None => Err(format!("The argument '{}' was not found", ::clap::Format::Warning($v))) } }; } /// Convenience macro getting a typed value `T` where `T` implements `std::str::FromStr` /// This macro returns a `T` or `Vec` or exits with a usage string upon failure. This /// removes some of the boiler plate to handle failures from value_t! above. /// /// You can use it to get a single value `T`, or a `Vec` with the `values_of()` /// /// **NOTE:** This should only be used on required arguments, as it can be confusing to the user /// why they are getting error messages when it appears they're entering all required arguments. /// /// **NOTE:** Be cautious, as since this a macro invocation it's not exactly like /// standard syntax. /// /// /// # Examples single value /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::App; /// # fn main() { /// let matches = App::new("myapp") /// .arg_from_usage("[length] 'Set the length to use as a pos whole num, i.e. 20'") /// .get_matches(); /// let len = value_t_or_exit!(matches.value_of("length"), u32); /// /// println!("{} + 2: {}", len, len + 2); /// # } /// ``` /// /// /// # Examples multiple values /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::App; /// # fn main() { /// let matches = App::new("myapp") /// .arg_from_usage("[seq]... 'A sequence of pos whole nums, i.e. 20 45'") /// .get_matches(); /// for v in value_t_or_exit!(matches.values_of("seq"), u32) { /// println!("{} + 2: {}", v, v + 2); /// } /// # } /// ``` #[macro_export] macro_rules! value_t_or_exit { ($m:ident.value_of($v:expr), $t:ty) => { match $m.value_of($v) { Some(v) => { match v.parse::<$t>() { Ok(val) => val, Err(..) => { println!("{} '{}' isn't a valid value\n\n{}\n\nPlease re-run with {} for \ more information", ::clap::Format::Error("error:"), ::clap::Format::Warning(v.to_string()), $m.usage(), ::clap::Format::Good("--help")); ::std::process::exit(1); } } }, None => { println!("{} The argument '{}' was not found or is not valid\n\n{}\n\nPlease re-run with \ {} for more information", ::clap::Format::Error("error:"), ::clap::Format::Warning($v.to_string()), $m.usage(), ::clap::Format::Good("--help")); ::std::process::exit(1); } } }; ($m:ident.values_of($v:expr), $t:ty) => { match $m.values_of($v) { Some(ref v) => { let mut tmp = Vec::with_capacity(v.len()); for pv in v { match pv.parse::<$t>() { Ok(rv) => tmp.push(rv), Err(_) => { println!("{} '{}' isn't a valid value\n\n{}\n\nPlease re-run with {} for more \ information", ::clap::Format::Error("error:"), ::clap::Format::Warning(pv), $m.usage(), ::clap::Format::Good("--help")); ::std::process::exit(1); } } } tmp }, None => { println!("{} The argument '{}' not found or is not valid\n\n{}\n\nPlease re-run with \ {} for more information", ::clap::Format::Error("error:"), ::clap::Format::Warning($v.to_string()), $m.usage(), ::clap::Format::Good("--help")); ::std::process::exit(1); } } }; } /// Convenience macro generated a simple enum with variants to be used as a type when parsing /// arguments. This enum also provides a `variants()` function which can be used to retrieve a /// `Vec<&'static str>` of the variant names. /// /// **NOTE:** This macro automatically implements std::str::FromStr and std::fmt::Display /// /// # Examples /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::{App, Arg}; /// simple_enum!{Foo => Bar, Baz, Qux} /// // Foo enum can now be used via Foo::Bar, or Foo::Baz, etc /// // and implements std::str::FromStr to use with the value_t! macros /// fn main() { /// let enum_vals = ["Bar", "Baz", "Qux"]; /// let m = App::new("app") /// .arg(Arg::from_usage(" 'the foo'") /// .possible_values(&enum_vals)) /// .get_matches(); /// let f = value_t_or_exit!(m.value_of("foo"), Foo); /// /// // Use f like any other Foo variant... /// } /// ``` #[macro_export] macro_rules! simple_enum { ($e:ident => $($v:ident),+) => { enum $e { $($v),+ } impl ::std::str::FromStr for $e { type Err = String; fn from_str(s: &str) -> Result { match s { $(stringify!($v) => Ok($e::$v),)+ _ => Err({ let v = vec![ $(stringify!($v),)+ ]; format!("valid values:{}", v.iter().fold(String::new(), |a, i| { a + &format!(" {}", i)[..] })) }) } } } impl ::std::fmt::Display for $e { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { $($e::$v => write!(f, stringify!($v)),)+ } } } impl $e { #[allow(dead_code)] pub fn variants() -> Vec<&'static str> { vec![ $(stringify!($v),)+ ] } } }; } /// Convenience macro to generate more complete enums with variants to be used as a type when /// parsing arguments. This enum also provides a `variants()` function which can be used to retrieve a /// `Vec<&'static str>` of the variant names. /// /// **NOTE:** Case insensitivity is supported for ASCII characters /// /// **NOTE:** This macro automatically implements std::str::FromStr and std::fmt::Display /// /// These enums support pub (or not) and use of the #[derive()] traits /// /// /// # Examples /// /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::{App, Arg}; /// arg_enum!{ /// #[derive(Debug)] /// pub enum Foo { /// Bar, /// Baz, /// Qux /// } /// } /// // Foo enum can now be used via Foo::Bar, or Foo::Baz, etc /// // and implements std::str::FromStr to use with the value_t! macros /// fn main() { /// let m = App::new("app") /// .arg_from_usage(" 'the foo'") /// .get_matches(); /// let f = value_t_or_exit!(m.value_of("foo"), Foo); /// /// // Use f like any other Foo variant... /// } /// ``` #[macro_export] macro_rules! arg_enum { (enum $e:ident { $($v:ident),+ } ) => { enum $e { $($v),+ } impl ::std::str::FromStr for $e { type Err = String; fn from_str(s: &str) -> Result { use ::std::ascii::AsciiExt; match s { $(stringify!($v) | _ if s.eq_ignore_ascii_case(stringify!($v)) => Ok($e::$v),)+ _ => Err({ let v = vec![ $(stringify!($v),)+ ]; format!("valid values:{}", v.iter().fold(String::new(), |a, i| { a + &format!(" {}", i)[..] })) }) } } } impl ::std::fmt::Display for $e { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { $($e::$v => write!(f, stringify!($v)),)+ } } } impl $e { #[allow(dead_code)] fn variants() -> Vec<&'static str> { vec![ $(stringify!($v),)+ ] } } }; (pub enum $e:ident { $($v:ident),+ } ) => { pub enum $e { $($v),+ } impl ::std::str::FromStr for $e { type Err = String; fn from_str(s: &str) -> Result { use ::std::ascii::AsciiExt; match s { $(stringify!($v) | _ if s.eq_ignore_ascii_case(stringify!($v)) => Ok($e::$v),)+ _ => Err({ let v = vec![ $(stringify!($v),)+ ]; format!("valid values:{}", v.iter().fold(String::new(), |a, i| { a + &format!(" {}", i)[..] })) }) } } } impl ::std::fmt::Display for $e { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { $($e::$v => write!(f, stringify!($v)),)+ } } } impl $e { #[allow(dead_code)] pub fn variants() -> Vec<&'static str> { vec![ $(stringify!($v),)+ ] } } }; (#[derive($($d:ident),+)] enum $e:ident { $($v:ident),+ } ) => { #[derive($($d,)+)] enum $e { $($v),+ } impl ::std::str::FromStr for $e { type Err = String; fn from_str(s: &str) -> Result { use ::std::ascii::AsciiExt; match s { $(stringify!($v) | _ if s.eq_ignore_ascii_case(stringify!($v)) => Ok($e::$v),)+ _ => Err({ let v = vec![ $(stringify!($v),)+ ]; format!("valid values:{}", v.iter().fold(String::new(), |a, i| { a + &format!(" {}", i)[..] })) }) } } } impl ::std::fmt::Display for $e { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { $($e::$v => write!(f, stringify!($v)),)+ } } } impl $e { #[allow(dead_code)] pub fn variants() -> Vec<&'static str> { vec![ $(stringify!($v),)+ ] } } }; (#[derive($($d:ident),+)] pub enum $e:ident { $($v:ident),+ } ) => { #[derive($($d,)+)] pub enum $e { $($v),+ } impl ::std::str::FromStr for $e { type Err = String; fn from_str(s: &str) -> Result { use ::std::ascii::AsciiExt; match s { $(stringify!($v) | _ if s.eq_ignore_ascii_case(stringify!($v)) => Ok($e::$v),)+ _ => Err({ let v = vec![ $(stringify!($v),)+ ]; format!("valid values:{}", v.iter().fold(String::new(), |a, i| { a + &format!(" {}", i)[..] })) }) } } } impl ::std::fmt::Display for $e { fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result { match *self { $($e::$v => write!(f, stringify!($v)),)+ } } } impl $e { #[allow(dead_code)] pub fn variants() -> Vec<&'static str> { vec![ $(stringify!($v),)+ ] } } }; } /// Allows you pull the version for an from your Cargo.toml as MAJOR.MINOR.PATCH_PKGVERSION_PRE /// /// # Examples /// ```no_run /// # #[macro_use] /// # extern crate clap; /// # use clap::App; /// # fn main() { /// let m = App::new("app") /// .version(&crate_version!()[..]) /// .get_matches(); /// # } /// ``` #[macro_export] macro_rules! crate_version { () => { env!("CARGO_PKG_VERSION").to_owned() } } /// App, Arg, SubCommand and Group builder macro (Usage-string like input) #[macro_export] macro_rules! clap_app { (@app ($builder:expr)) => { $builder }; (@app ($builder:expr) (@arg $name:ident: $($tail:tt)*) $($tt:tt)*) => { clap_app!{ @app ($builder.arg(clap_app!{ @arg ($crate::Arg::with_name(stringify!($name))) (-) $($tail)* })) $($tt)* } }; (@app ($builder:expr) (@setting $setting:ident) $($tt:tt)*) => { clap_app!{ @app ($builder.setting($crate::AppSettings::$setting)) $($tt)* } }; // Treat the application builder as an argument to set it's attributes (@app ($builder:expr) (@attributes $($attr:tt)*) $($tt:tt)*) => { clap_app!{ @app (clap_app!{ @arg ($builder) $($attr)* }) $($tt)* } }; (@app ($builder:expr) (@group $name:ident => $($tail:tt)*) $($tt:tt)*) => { clap_app!{ @app (clap_app!{ @group ($builder, $crate::ArgGroup::with_name(stringify!($name))) $($tail)* }) $($tt)* } }; // Handle subcommand creation (@app ($builder:expr) (@subcommand $name:ident => $($tail:tt)*) $($tt:tt)*) => { clap_app!{ @app ($builder.subcommand( clap_app!{ @app ($crate::SubCommand::with_name(stringify!($name))) $($tail)* } )) $($tt)* } }; // Yaml like function calls - used for setting various meta directly against the app (@app ($builder:expr) ($ident:ident: $($v:expr),*) $($tt:tt)*) => { // clap_app!{ @app ($builder.$ident($($v),*)) $($tt)* } clap_app!{ @app ($builder.$ident($($v),*)) $($tt)* } }; // Add members to group and continue argument handling with the parent builder (@group ($builder:expr, $group:expr)) => { $builder.arg_group($group) }; (@group ($builder:expr, $group:expr) (@attributes $($attr:tt)*) $($tt:tt)*) => { clap_app!{ @group ($builder, clap_app!{ @arg ($group) (-) $($attr)* }) $($tt)* } }; (@group ($builder:expr, $group:expr) (@arg $name:ident: $($tail:tt)*) $($tt:tt)*) => { clap_app!{ @group (clap_app!{ @app ($builder) (@arg $name: $($tail)*) }, $group.add(stringify!($name))) $($tt)* } }; // No more tokens to munch (@arg ($arg:expr) $modes:tt) => { $arg }; // Shorthand tokens influenced by the usage_string (@arg ($arg:expr) $modes:tt --$long:ident $($tail:tt)*) => { clap_app!{ @arg ($arg.long(stringify!($long))) $modes $($tail)* } }; (@arg ($arg:expr) $modes:tt -$short:ident $($tail:tt)*) => { clap_app!{ @arg ($arg.short(stringify!($short))) $modes $($tail)* } }; (@arg ($arg:expr) (-) <$var:ident> $($tail:tt)*) => { clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) +takes_value +required $($tail)* } }; (@arg ($arg:expr) (+) <$var:ident> $($tail:tt)*) => { clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) $($tail)* } }; (@arg ($arg:expr) (-) [$var:ident] $($tail:tt)*) => { clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) +takes_value $($tail)* } }; (@arg ($arg:expr) (+) [$var:ident] $($tail:tt)*) => { clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) $($tail)* } }; (@arg ($arg:expr) $modes:tt ... $($tail:tt)*) => { clap_app!{ @arg ($arg) $modes +multiple $($tail)* } }; // Shorthand magic (@arg ($arg:expr) $modes:tt #{$n:expr, $m:expr} $($tail:tt)*) => { clap_app!{ @arg ($arg) $modes min_values($n) max_values($m) $($tail)* } }; (@arg ($arg:expr) $modes:tt * $($tail:tt)*) => { clap_app!{ @arg ($arg) $modes +required $($tail)* } }; // !foo -> .foo(false) (@arg ($arg:expr) $modes:tt !$ident $($tail:tt)*) => { clap_app!{ @arg ($arg.$ident(false)) $modes $($tail)* } }; // foo -> .foo(true) (@arg ($arg:expr) $modes:tt +$ident:ident $($tail:tt)*) => { clap_app!{ @arg ($arg.$ident(true)) $modes $($tail)* } }; // Validator (@arg ($arg:expr) $modes:tt {$fn_:expr} $($tail:tt)*) => { clap_app!{ @arg ($arg.validator($fn_)) $modes $($tail)* } }; (@as_expr $expr:expr) => { $expr }; // Help (@arg ($arg:expr) $modes:tt $desc:tt) => { $arg.help(clap_app!{ @as_expr $desc }) }; // Handle functions that need to be called multiple times for each argument (@arg ($arg:expr) $modes:tt $ident:ident[$($target:ident)*] $($tail:tt)*) => { clap_app!{ @arg ($arg $( .$ident(stringify!($target)) )*) $modes $($tail)* } }; // Inherit builder's functions (@arg ($arg:expr) $modes:tt $ident:ident($($expr:expr)*) $($tail:tt)*) => { clap_app!{ @arg ($arg.$ident($($expr)*)) $modes $($tail)* } }; // Build a subcommand outside of an app. (@subcommand $name:ident => $($tail:tt)*) => { clap_app!{ @app ($crate::SubCommand::with_name(stringify!($name))) $($tail)* } }; // Start the magic ($name:ident => $($tail:tt)*) => {{ clap_app!{ @app ($crate::App::new(stringify!($name))) $($tail)*} }}; }