diff options
Diffstat (limited to 'src/macros/command_flag.rs')
| -rw-r--r-- | src/macros/command_flag.rs | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/macros/command_flag.rs b/src/macros/command_flag.rs new file mode 100644 index 0000000..894b3f4 --- /dev/null +++ b/src/macros/command_flag.rs @@ -0,0 +1,33 @@ +#[macro_export] +macro_rules! special_flag { + ($args:expr, $flag:expr) => {{ + let flag = $flag; + let found = $args.iter().any(|arg| arg == flag); + $args.retain(|arg| arg != flag); + found + }}; +} + +#[macro_export] +macro_rules! special_argument { + ($args:expr, $flag:expr) => {{ + let flag = $flag; + let mut value: Option<String> = None; + let mut i = 0; + while i < $args.len() { + if $args[i] == flag { + if i + 1 < $args.len() { + value = Some($args[i + 1].clone()); + $args.remove(i + 1); + $args.remove(i); + } else { + value = None; + $args.remove(i); + } + break; + } + i += 1; + } + value + }}; +} |
