1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
use crate::Program;
pub struct Flag {
vec: Vec<&'static str>,
}
impl From<&'static str> for Flag {
fn from(s: &'static str) -> Self {
Flag { vec: vec![s] }
}
}
impl From<&'static [&'static str]> for Flag {
fn from(slice: &'static [&'static str]) -> Self {
Flag {
vec: slice.to_vec(),
}
}
}
impl<const N: usize> From<[&'static str; N]> for Flag {
fn from(slice: [&'static str; N]) -> Self {
Flag {
vec: slice.to_vec(),
}
}
}
impl<const N: usize> From<&'static [&'static str; N]> for Flag {
fn from(slice: &'static [&'static str; N]) -> Self {
Flag {
vec: slice.to_vec(),
}
}
}
impl AsRef<[&'static str]> for Flag {
fn as_ref(&self) -> &[&'static str] {
&self.vec
}
}
impl std::ops::Deref for Flag {
type Target = [&'static str];
fn deref(&self) -> &Self::Target {
&self.vec
}
}
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_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
}};
}
impl Program {
/// Registers a global argument (with value) and its handler.
pub fn global_argument<F, A>(&mut self, arguments: A, do_fn: F)
where
F: Fn(&mut Program, String),
A: Into<Flag>,
{
let flag = arguments.into();
for argument in flag.iter() {
let value = special_argument!(self.args, argument);
if let Some(value) = value {
do_fn(self, value);
return;
}
}
}
/// Registers a global flag (boolean) and its handler.
pub fn global_flag<F, A>(&mut self, flag: A, do_fn: F)
where
F: Fn(&mut Program),
A: Into<Flag>,
{
let flag = flag.into();
for argument in flag.iter() {
let enabled = special_flag!(self.args, argument);
if enabled {
do_fn(self);
return;
}
}
}
}
|
