diff options
Diffstat (limited to 'src/utils/cmd_macros.rs')
| -rw-r--r-- | src/utils/cmd_macros.rs | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/src/utils/cmd_macros.rs b/src/utils/cmd_macros.rs new file mode 100644 index 0000000..11b8da4 --- /dev/null +++ b/src/utils/cmd_macros.rs @@ -0,0 +1,47 @@ +#[macro_export] +macro_rules! special_flag { + ($args:expr, $($flag:expr),+) => {{ + let mut found = false; + $( + let flag = $flag; + if $args.iter().any(|arg| arg == flag) { + found = true; + } + $args.retain(|arg| arg != flag); + )+ + found + }}; +} + +#[macro_export] +macro_rules! special_argument { + ($args:expr, $($flag:expr),+) => {{ + let mut value: Option<String> = None; + let mut found = false; + $( + let flag = $flag; + if !found { + 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); + } + #[allow(unused_assignments)] + { + found = true; + } + break; + } + i += 1; + } + } + )+ + value + }}; +} |
