clap/src/macros.rs

709 lines
25 KiB
Rust
Raw Normal View History

2016-06-24 10:23:08 +00:00
/// A convenience macro for loading the YAML file at compile time (relative to the current file,
2016-01-26 19:02:10 +00:00
/// like modules work). That YAML object can then be passed to this function.
///
/// # Panics
///
/// The YAML file must be properly formatted or this function will panic!(). A good way to
/// ensure this doesn't happen is to run your program with the `--help` switch. If this passes
/// without error, you needn't worry because the YAML is properly formatted.
///
/// # Examples
///
/// The following example shows how to load a properly formatted YAML file to build an instance
2016-01-26 19:02:10 +00:00
/// of an `App` struct.
///
/// ```ignore
/// # #[macro_use]
/// # extern crate clap;
2016-01-26 19:02:10 +00:00
/// # use clap::App;
/// # fn main() {
2016-01-26 19:02:10 +00:00
/// let yml = load_yaml!("app.yml");
/// let app = App::from(yml);
2016-01-26 19:02:10 +00:00
///
/// // continued logic goes here, such as `app.get_matches()` etc.
/// # }
2016-01-26 19:02:10 +00:00
/// ```
#[cfg(feature = "yaml")]
#[macro_export]
macro_rules! load_yaml {
2018-08-02 03:13:51 +00:00
($yml:expr) => {
&$crate::YamlLoader::load_from_str(include_str!($yml)).expect("failed to load YAML file")[0]
2018-08-02 03:13:51 +00:00
};
}
// _clap_count_exprs! is derived from https://github.com/DanielKeep/rust-grabbag
// commit: 82a35ca5d9a04c3b920622d542104e3310ee5b07
// License: MIT
// Copyright ⓒ 2015 grabbag contributors.
// Licensed under the MIT license (see LICENSE or <http://opensource.org
// /licenses/MIT>) or the Apache License, Version 2.0 (see LICENSE of
// <http://www.apache.org/licenses/LICENSE-2.0>), at your option. All
// files in the project carrying such notice may not be copied, modified,
// or distributed except according to those terms.
//
/// Counts the number of comma-delimited expressions passed to it. The result is a compile-time
/// evaluable expression, suitable for use as a static array size, or the value of a `const`.
///
/// # Examples
///
/// ```
/// # #[macro_use] extern crate clap;
/// # fn main() {
/// const COUNT: usize = _clap_count_exprs!(a, 5+1, "hi there!".into_string());
/// assert_eq!(COUNT, 3);
/// # }
/// ```
#[macro_export]
macro_rules! _clap_count_exprs {
() => { 0 };
($e:expr) => { 1 };
($e:expr, $($es:expr),+) => { 1 + $crate::_clap_count_exprs!($($es),*) };
}
2016-04-12 05:37:36 +00:00
/// Allows you to pull the version from your Cargo.toml at compile time as
2017-06-08 18:42:10 +00:00
/// `MAJOR.MINOR.PATCH_PKGVERSION_PRE`
///
/// # Examples
2016-01-26 19:02:10 +00:00
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # use clap::App;
/// # fn main() {
/// let m = App::new("app")
/// .version(crate_version!())
/// .get_matches();
/// # }
/// ```
2020-03-01 21:55:57 +00:00
#[cfg(feature = "cargo")]
#[macro_export]
macro_rules! crate_version {
() => {
env!("CARGO_PKG_VERSION")
};
}
2016-05-06 21:35:53 +00:00
/// Allows you to pull the authors for the app from your Cargo.toml at
/// compile time in the form:
/// `"author1 lastname <author1@example.com>:author2 lastname <author2@example.com>"`
///
/// You can replace the colons with a custom separator by supplying a
/// replacement string, so, for example,
/// `crate_authors!(",\n")` would become
/// `"author1 lastname <author1@example.com>,\nauthor2 lastname <author2@example.com>,\nauthor3 lastname <author3@example.com>"`
///
/// # Examples
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # use clap::App;
/// # fn main() {
/// let m = App::new("app")
/// .author(crate_authors!("\n"))
/// .get_matches();
/// # }
/// ```
2020-03-01 21:55:57 +00:00
#[cfg(feature = "cargo")]
#[macro_export]
macro_rules! crate_authors {
($sep:expr) => {{
use std::ops::Deref;
2019-11-02 18:48:16 +00:00
use std::boxed::Box;
use std::cell::Cell;
#[allow(missing_copy_implementations)]
2020-05-02 20:33:43 +00:00
#[allow(unused)]
2018-08-02 03:13:51 +00:00
struct CargoAuthors {
authors: Cell<Option<&'static str>>,
2018-08-02 03:13:51 +00:00
__private_field: (),
};
2018-11-14 17:05:06 +00:00
2017-10-04 22:09:18 +00:00
impl Deref for CargoAuthors {
type Target = str;
2017-10-04 22:09:18 +00:00
fn deref(&self) -> &'static str {
let authors = self.authors.take();
if authors.is_some() {
let unwrapped_authors = authors.unwrap();
self.authors.replace(Some(unwrapped_authors));
unwrapped_authors
} else {
2019-11-02 20:03:35 +00:00
// This caches the result for subsequent invocations of the same instance of the macro
// to avoid performing one memory allocation per call.
// If performance ever becomes a problem for this code, it should be moved to build.rs
let s: Box<String> = Box::new(env!("CARGO_PKG_AUTHORS").replace(':', $sep));
let static_string = Box::leak(s);
self.authors.replace(Some(&*static_string));
&*static_string // weird but compiler-suggested way to turn a String into &str
}
}
}
2018-08-02 03:13:51 +00:00
&*CargoAuthors {
authors: std::cell::Cell::new(Option::None),
2018-08-02 03:13:51 +00:00
__private_field: (),
}
}};
() => {
env!("CARGO_PKG_AUTHORS")
};
}
/// Allows you to pull the description from your Cargo.toml at compile time.
///
/// # Examples
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # use clap::App;
/// # fn main() {
/// let m = App::new("app")
/// .about(crate_description!())
/// .get_matches();
/// # }
/// ```
2020-03-01 21:55:57 +00:00
#[cfg(feature = "cargo")]
#[macro_export]
macro_rules! crate_description {
() => {
env!("CARGO_PKG_DESCRIPTION")
};
}
/// Allows you to pull the name from your Cargo.toml at compile time.
///
/// # Examples
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # use clap::App;
/// # fn main() {
/// let m = App::new(crate_name!())
/// .get_matches();
/// # }
/// ```
2020-03-01 21:55:57 +00:00
#[cfg(feature = "cargo")]
#[macro_export]
macro_rules! crate_name {
() => {
env!("CARGO_PKG_NAME")
};
}
/// Allows you to build the `App` instance from your Cargo.toml at compile time.
///
/// Equivalent to using the `crate_*!` macros with their respective fields.
///
/// Provided separator is for the [`crate_authors!`](macro.crate_authors.html) macro,
/// refer to the documentation therefor.
///
/// **NOTE:** Changing the values in your `Cargo.toml` does not trigger a re-build automatically,
/// and therefore won't change the generated output until you recompile.
///
/// **Pro Tip:** In some cases you can "trick" the compiler into triggering a rebuild when your
/// `Cargo.toml` is changed by including this in your `src/main.rs` file
/// `include_str!("../Cargo.toml");`
///
/// # Examples
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # fn main() {
/// let m = app_from_crate!().get_matches();
/// # }
/// ```
2020-03-01 21:55:57 +00:00
#[cfg(feature = "cargo")]
#[macro_export]
macro_rules! app_from_crate {
() => {
$crate::App::new($crate::crate_name!())
.version($crate::crate_version!())
.author($crate::crate_authors!())
.about($crate::crate_description!())
};
($sep:expr) => {
$crate::App::new($crate::crate_name!())
.version($crate::crate_version!())
.author($crate::crate_authors!($sep))
.about($crate::crate_description!())
};
}
2020-04-12 10:00:10 +00:00
/// Build `App`, `Arg` and `Group` with Usage-string like input
/// but without the associated parsing runtime cost.
///
/// `clap_app!` also supports several shorthand syntaxes.
///
/// # Examples
///
/// ```no_run
/// # #[macro_use]
/// # extern crate clap;
/// # fn main() {
/// let matches = clap_app!(myapp =>
/// (version: "1.0")
/// (author: "Kevin K. <kbknapp@gmail.com>")
/// (about: "Does awesome things")
/// (@arg CONFIG: -c --config +takes_value "Sets a custom config file")
/// (@arg INPUT: +required "Sets the input file to use")
/// (@arg debug: -d ... "Sets the level of debugging information")
/// (@group difficulty =>
/// (@arg hard: -h --hard "Sets hard mode")
/// (@arg normal: -n --normal "Sets normal mode")
/// (@arg easy: -e --easy "Sets easy mode")
/// )
/// (@subcommand test =>
/// (about: "controls testing features")
/// (version: "1.3")
/// (author: "Someone E. <someone_else@other.com>")
/// (@arg verbose: -v --verbose "Print test information verbosely")
/// )
/// )
/// .get_matches();
/// # }
/// ```
2020-04-12 10:00:10 +00:00
///
/// # Shorthand Syntax for Args
///
/// * A single hyphen followed by a character (such as `-c`) sets the [`Arg::short`]
/// * A double hyphen followed by a character or word (such as `--config`) sets [`Arg::long`]
/// * Three dots (`...`) sets [`Arg::multiple(true)`]
/// * Angled brackets after either a short or long will set [`Arg::value_name`] and
/// `Arg::required(true)` such as `--config <FILE>` = `Arg::value_name("FILE")` and
/// `Arg::required(true)`
/// * Square brackets after either a short or long will set [`Arg::value_name`] and
/// `Arg::required(false)` such as `--config [FILE]` = `Arg::value_name("FILE")` and
/// `Arg::required(false)`
/// * There are short hand syntaxes for Arg methods that accept booleans
/// * A plus sign will set that method to `true` such as `+required` = `Arg::required(true)`
/// * An exclamation will set that method to `false` such as `!required` = `Arg::required(false)`
/// * A `#{min, max}` will set [`Arg::min_values(min)`] and [`Arg::max_values(max)`]
/// * An asterisk (`*`) will set `Arg::required(true)`
/// * Curly brackets around a `fn` will set [`Arg::validator`] as in `{fn}` = `Arg::validator(fn)`
/// * An Arg method that accepts a string followed by square brackets will set that method such as
/// `conflicts_with[FOO]` will set `Arg::conflicts_with("FOO")` (note the lack of quotes around
/// `FOO` in the macro)
/// * An Arg method that takes a string and can be set multiple times (such as
/// [`Arg::conflicts_with`]) followed by square brackets and a list of values separated by spaces
/// will set that method such as `conflicts_with[FOO BAR BAZ]` will set
/// `Arg::conflicts_with("FOO")`, `Arg::conflicts_with("BAR")`, and `Arg::conflicts_with("BAZ")`
/// (note the lack of quotes around the values in the macro)
///
/// # Shorthand Syntax for Groups
///
/// * There are short hand syntaxes for `ArgGroup` methods that accept booleans
/// * A plus sign will set that method to `true` such as `+required` = `ArgGroup::required(true)`
/// * An exclamation will set that method to `false` such as `!required` = `ArgGroup::required(false)`
///
2019-07-20 23:39:30 +00:00
/// # Alternative form for non-ident values
///
2019-07-20 23:39:30 +00:00
/// Certain places that normally accept an `ident`, will optionally accept an alternative of `("expr enclosed by parens")`
/// * `(@arg something: --something)` could also be `(@arg ("something-else"): --("something-else"))`
/// * `(@subcommand something => ...)` could also be `(@subcommand ("something-else") => ...)`
///
2020-04-12 10:00:10 +00:00
/// Or it can be even simpler by using the literal directly
/// * `(@arg "something-else": --"something-else")`
/// * `(@subcommand "something-else" => ...)`
///
/// [`Arg::short`]: ./struct.Arg.html#method.short
/// [`Arg::long`]: ./struct.Arg.html#method.long
/// [`Arg::multiple(true)`]: ./struct.Arg.html#method.multiple
/// [`Arg::value_name`]: ./struct.Arg.html#method.value_name
/// [`Arg::min_values(min)`]: ./struct.Arg.html#method.min_values
/// [`Arg::max_values(max)`]: ./struct.Arg.html#method.max_values
/// [`Arg::validator`]: ./struct.Arg.html#method.validator
/// [`Arg::conflicts_with`]: ./struct.Arg.html#method.conflicts_with
#[macro_export]
macro_rules! clap_app {
(@app ($builder:expr)) => { $builder };
(@app ($builder:expr) (@arg ($name:expr): $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($builder.arg(
$crate::clap_app!{ @arg ($crate::Arg::with_name($name)) (-) $($tail)* }))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
(@app ($builder:expr) (@arg $name:literal: $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($builder.arg(
$crate::clap_app!{ @arg ($crate::Arg::with_name(stringify!($name).trim_matches('"'))) (-) $($tail)* }))
$($tt)*
}
};
(@app ($builder:expr) (@arg $name:ident: $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
2016-05-06 21:35:53 +00:00
($builder.arg(
$crate::clap_app!{ @arg ($crate::Arg::with_name(stringify!($name))) (-) $($tail)* }))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// Settings
(@app ($builder:expr) (@setting $setting:ident) $($tt:tt)*) => {
$crate::clap_app!{ @app
($builder.setting($crate::AppSettings::$setting))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// Treat the application builder as an argument to set its attributes
(@app ($builder:expr) (@attributes $($attr:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app ($crate::clap_app!{ @arg ($builder) $($attr)* }) $($tt)* }
};
2020-04-12 10:00:10 +00:00
// ArgGroup
(@app ($builder:expr) (@group $name:ident => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($crate::clap_app!{ @group ($builder, $crate::ArgGroup::with_name(stringify!($name))) $($tail)* })
$($tt)*
}
};
(@app ($builder:expr) (@group $name:ident !$ident:ident => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($crate::clap_app!{ @group ($builder, $crate::ArgGroup::with_name(stringify!($name)).$ident(false)) $($tail)* })
$($tt)*
}
};
(@app ($builder:expr) (@group $name:ident +$ident:ident => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($crate::clap_app!{ @group ($builder, $crate::ArgGroup::with_name(stringify!($name)).$ident(true)) $($tail)* })
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// Handle subcommand creation
(@app ($builder:expr) (@subcommand ($name:expr) => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($builder.subcommand(
2020-04-12 10:00:10 +00:00
$crate::clap_app!{ @app ($crate::App::new($name)) $($tail)* }
))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
(@app ($builder:expr) (@subcommand $name:literal => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
2019-07-20 23:39:30 +00:00
($builder.subcommand(
2020-04-12 10:00:10 +00:00
$crate::clap_app!{ @app ($crate::App::new(stringify!($name).trim_matches('"'))) $($tail)* }
))
$($tt)*
}
};
(@app ($builder:expr) (@subcommand $name:ident => $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @app
($builder.subcommand(
$crate::clap_app!{ @app ($crate::App::new(stringify!($name))) $($tail)* }
2019-07-20 23:39:30 +00:00
))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// Yaml like function calls - used for setting various meta directly against the app
(@app ($builder:expr) ($ident:ident: $($v:expr),*) $($tt:tt)*) => {
$crate::clap_app!{ @app
2015-09-09 04:00:17 +00:00
($builder.$ident($($v),*))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// Add members to group and continue argument handling with the parent builder
(@group ($builder:expr, $group:expr)) => { $builder.group($group) };
// Treat the group builder as an argument to set its attributes
(@group ($builder:expr, $group:expr) (@attributes $($attr:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @group ($builder, $crate::clap_app!{ @arg ($group) (-) $($attr)* }) $($tt)* }
};
2020-04-12 10:00:10 +00:00
(@group ($builder:expr, $group:expr) (@arg ($name:expr): $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @group
($crate::clap_app!{ @app ($builder) (@arg ($name): $($tail)*) },
$group.arg($name))
$($tt)*
}
};
(@group ($builder:expr, $group:expr) (@arg $name:literal: $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @group
($crate::clap_app!{ @app ($builder) (@arg $name: $($tail)*) },
$group.arg(stringify!($name).trim_matches('"')))
$($tt)*
}
};
(@group ($builder:expr, $group:expr) (@arg $name:ident: $($tail:tt)*) $($tt:tt)*) => {
$crate::clap_app!{ @group
($crate::clap_app!{ @app ($builder) (@arg $name: $($tail)*) },
$group.arg(stringify!($name)))
$($tt)*
}
};
2020-04-12 10:00:10 +00:00
// No more tokens to munch
(@arg ($arg:expr) $modes:tt) => { $arg };
2020-04-12 10:00:10 +00:00
// Shorthand tokens influenced by the usage_string
(@arg ($arg:expr) $modes:tt --($long:expr) $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.long($long)) $modes $($tail)* }
};
2020-04-12 10:00:10 +00:00
(@arg ($arg:expr) $modes:tt --$long:literal $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.long(stringify!($long).trim_matches('"'))) $modes $($tail)* }
};
(@arg ($arg:expr) $modes:tt --$long:ident $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.long(stringify!($long))) $modes $($tail)* }
};
2020-04-12 10:00:10 +00:00
(@arg ($arg:expr) $modes:tt -($short:expr) $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.short($short)) $modes $($tail)* }
};
(@arg ($arg:expr) $modes:tt -$short:literal $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.short($short.to_string().chars().next().expect(r#""" is not allowed here"#))) $modes $($tail)* }
};
(@arg ($arg:expr) $modes:tt -$short:ident $($tail:tt)*) => {
2020-03-13 18:26:45 +00:00
$crate::clap_app!{ @arg ($arg.short(stringify!($short).chars().next().unwrap())) $modes $($tail)* }
};
(@arg ($arg:expr) (-) <$var:ident> $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) +takes_value +required $($tail)* }
};
(@arg ($arg:expr) (+) <$var:ident> $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) $($tail)* }
};
(@arg ($arg:expr) (-) [$var:ident] $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) +takes_value $($tail)* }
};
(@arg ($arg:expr) (+) [$var:ident] $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.value_name(stringify!($var))) (+) $($tail)* }
};
(@arg ($arg:expr) $modes:tt ... $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg) $modes +multiple $($tail)* }
};
2020-04-12 10:00:10 +00:00
// Shorthand magic
(@arg ($arg:expr) $modes:tt #{$n:expr, $m:expr} $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg) $modes min_values($n) max_values($m) $($tail)* }
};
(@arg ($arg:expr) $modes:tt * $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg) $modes +required $($tail)* }
};
2020-04-12 10:00:10 +00:00
// !foo -> .foo(false)
(@arg ($arg:expr) $modes:tt !$ident:ident $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.$ident(false)) $modes $($tail)* }
};
2020-04-12 10:00:10 +00:00
// +foo -> .foo(true)
(@arg ($arg:expr) $modes:tt +$ident:ident $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.$ident(true)) $modes $($tail)* }
};
2020-04-12 10:00:10 +00:00
// Validator
(@arg ($arg:expr) $modes:tt {$fn_:expr} $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.validator($fn_)) $modes $($tail)* }
};
(@as_expr $expr:expr) => { $expr };
2020-04-12 10:00:10 +00:00
// Help
(@arg ($arg:expr) $modes:tt $desc:tt) => { $arg.about(clap_app!{ @as_expr $desc }) };
2020-04-12 10:00:10 +00:00
// Handle functions that need to be called multiple times for each argument
(@arg ($arg:expr) $modes:tt $ident:ident[$($target:literal)*] $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg $( .$ident(stringify!($target).trim_matches('"')) )*) $modes $($tail)* }
};
(@arg ($arg:expr) $modes:tt $ident:ident[$($target:ident)*] $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg $( .$ident(stringify!($target)) )*) $modes $($tail)* }
};
// Inherit builder's functions, e.g. `index(2)`, `requires_if("val", "arg")`
(@arg ($arg:expr) $modes:tt $ident:ident($($expr:expr),*) $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.$ident($($expr),*)) $modes $($tail)* }
};
// Inherit builder's functions with trailing comma, e.g. `index(2,)`, `requires_if("val", "arg",)`
(@arg ($arg:expr) $modes:tt $ident:ident($($expr:expr,)*) $($tail:tt)*) => {
$crate::clap_app!{ @arg ($arg.$ident($($expr),*)) $modes $($tail)* }
};
2020-04-12 10:00:10 +00:00
// Build a subcommand outside of an app.
2019-07-20 23:39:30 +00:00
(@subcommand ($name:expr) => $($tail:tt)*) => {
$crate::clap_app!{ @app ($crate::App::new($name)) $($tail)* }
};
2020-04-12 10:00:10 +00:00
(@subcommand $name:literal => $($tail:tt)*) => {
$crate::clap_app!{ @app ($crate::App::new(stringify!($name).trim_matches('"'))) $($tail)* }
};
(@subcommand $name:ident => $($tail:tt)*) => {
$crate::clap_app!{ @app ($crate::App::new(stringify!($name))) $($tail)* }
};
// Start the magic
(($name:expr) => $($tail:tt)*) => {{
$crate::clap_app!{ @app ($crate::App::new($name)) $($tail)*}
}};
2020-04-12 10:00:10 +00:00
($name:literal => $($tail:tt)*) => {{
$crate::clap_app!{ @app ($crate::App::new(stringify!($name).trim_matches('"'))) $($tail)*}
}};
($name:ident => $($tail:tt)*) => {{
$crate::clap_app!{ @app ($crate::App::new(stringify!($name))) $($tail)*}
}};
}
2016-01-26 19:02:10 +00:00
macro_rules! impl_settings {
2020-02-04 16:02:08 +00:00
($settings:ident, $flags:ident,
$( $setting:ident($str:expr) => $flag:path ),+
) => {
impl $flags {
2020-03-19 07:17:52 +00:00
pub(crate) fn set(&mut self, s: $settings) {
2020-02-04 16:02:08 +00:00
match s {
$($settings::$setting => self.0.insert($flag)),*
}
}
2020-03-19 07:17:52 +00:00
pub(crate) fn unset(&mut self, s: $settings) {
2020-02-04 16:02:08 +00:00
match s {
$($settings::$setting => self.0.remove($flag)),*
}
2016-01-26 19:02:10 +00:00
}
2020-03-19 07:17:52 +00:00
pub(crate) fn is_set(&self, s: $settings) -> bool {
2020-02-04 16:02:08 +00:00
match s {
$($settings::$setting => self.0.contains($flag)),*
}
2016-01-26 19:02:10 +00:00
}
}
2020-02-04 16:02:08 +00:00
impl FromStr for $settings {
type Err = String;
fn from_str(s: &str) -> Result<Self, <Self as FromStr>::Err> {
2020-02-04 16:02:08 +00:00
match &*s.to_ascii_lowercase() {
2020-02-05 06:56:56 +00:00
$( $str => Ok($settings::$setting), )*
2020-02-04 16:02:08 +00:00
_ => Err(format!("unknown AppSetting: `{}`", s)),
}
2016-01-26 19:02:10 +00:00
}
}
2020-02-04 16:02:08 +00:00
}
2016-01-26 19:02:10 +00:00
}
// Convenience for writing to stderr thanks to https://github.com/BurntSushi
2020-04-22 18:14:47 +00:00
macro_rules! wlnerr {
2016-01-26 19:02:10 +00:00
($($arg:tt)*) => ({
use std::io::{Write, stderr};
writeln!(&mut stderr(), $($arg)*).ok();
})
2020-04-22 18:14:47 +00:00
}
2016-01-26 19:02:10 +00:00
#[cfg(feature = "debug")]
2020-04-22 18:14:47 +00:00
macro_rules! debug {
($($arg:tt)*) => {
print!("[{:>w$}] \t", module_path!(), w = 28);
println!($($arg)*)
2016-01-26 19:02:10 +00:00
}
}
#[cfg(not(feature = "debug"))]
2020-04-22 18:14:47 +00:00
macro_rules! debug {
($($arg:tt)*) => {};
2016-01-26 19:02:10 +00:00
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! flags {
2018-07-24 16:48:24 +00:00
($app:expr, $how:ident) => {{
$app.get_arguments()
2018-08-02 03:13:51 +00:00
.$how()
.filter(|a| !a.is_set($crate::ArgSettings::TakesValue) && a.get_index().is_none())
.filter(|a| !a.get_help_heading().is_some())
2018-07-24 16:48:24 +00:00
}};
($app:expr) => {
2020-02-05 10:04:59 +00:00
$crate::flags!($app, iter)
2018-08-02 03:13:51 +00:00
};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! opts {
2018-07-24 16:48:24 +00:00
($app:expr, $how:ident) => {{
$app.get_arguments()
2018-08-02 03:13:51 +00:00
.$how()
.filter(|a| a.is_set($crate::ArgSettings::TakesValue) && a.get_index().is_none())
.filter(|a| !a.get_help_heading().is_some())
2018-07-24 16:48:24 +00:00
}};
($app:expr) => {
opts!($app, iter)
2018-08-02 03:13:51 +00:00
};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! positionals {
($app:expr) => {{
$app.get_arguments()
.iter()
.filter(|a| !(a.get_short().is_some() || a.get_long().is_some()))
2020-02-14 16:22:01 +00:00
}};
}
macro_rules! groups_for_arg {
($app:expr, $grp:expr) => {{
2020-04-22 18:14:47 +00:00
debug!("groups_for_arg: name={:?}", $grp);
$app.groups
.iter()
2020-04-25 11:50:17 +00:00
.filter(|grp| grp.args.iter().any(|a| a == $grp))
2020-04-11 16:07:06 +00:00
.map(|grp| grp.id.clone())
2018-08-02 03:13:51 +00:00
}};
}
macro_rules! find_subcmd_cloned {
2020-02-14 16:22:01 +00:00
($app:expr, $sc:expr) => {{
$app.get_subcommands()
.iter()
2020-02-14 16:22:01 +00:00
.cloned()
.find(|a| match_alias!(a, $sc, a.get_name()))
}};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! find_subcmd {
($app:expr, $sc:expr) => {{
$app.get_subcommands()
.iter()
.find(|a| match_alias!(a, $sc, a.get_name()))
}};
}
#[macro_export]
#[doc(hidden)]
macro_rules! find_subcmd_mut {
($app:expr, $sc:expr) => {{
$app.get_subcommands_mut()
2020-04-12 01:39:13 +00:00
.iter_mut()
.find(|a| match_alias!(a, $sc, a.get_name()))
}};
}
macro_rules! longs {
2020-02-14 16:22:01 +00:00
($app:expr, $how:ident) => {{
use crate::mkeymap::KeyType;
2018-11-14 00:44:43 +00:00
$app.args.keys.iter().map(|x| &x.key).filter_map(|a| {
2018-07-24 16:48:24 +00:00
if let KeyType::Long(v) = a {
Some(v)
} else {
None
}
})
}};
2020-02-14 16:22:01 +00:00
($app:expr) => {
longs!($app, iter)
};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! names {
(@args $app:expr) => {{
$app.get_arguments().iter().map(|a| &*a.get_name())
}};
(@sc $app:expr) => {{
$app.get_subcommands().iter().map(|s| &*s.get_name()).chain(
$app.get_subcommands()
2018-07-24 16:48:24 +00:00
.iter()
2020-05-04 18:01:11 +00:00
.filter(|s| !s.aliases.is_empty()) // REFACTOR
.flat_map(|s| s.aliases.iter().map(|&(n, _)| n)),
2018-07-24 16:48:24 +00:00
)
}};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! sc_names {
($app:expr) => {{
2020-02-05 10:04:59 +00:00
names!(@sc $app)
}};
}
2020-02-05 10:04:59 +00:00
#[macro_export]
#[doc(hidden)]
macro_rules! match_alias {
($a:expr, $to:expr, $what:expr) => {{
$what == $to || $a.get_all_aliases().any(|alias| alias == $to)
2018-08-02 03:13:51 +00:00
}};
2018-01-25 04:05:05 +00:00
}