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
116
117
118
119
120
121
122
123
124
125
const OPTIONS_TEXT: &str = r"Options:
-a, --all all pid (include kernel threads)
--sort <order> sort by <order>: rss|swap|total
--pid <number> output only selected pid
--sleep <number> sleep <number> milli second
-H, --help display this help and exit
-V, --version display version information and exit
-X <x-options> x options. try -X help
";
#[repr(u8)]
#[derive(Debug, PartialEq)]
enum CmdOp {
All,
Sort,
Pid,
Sleep,
Help,
Version,
UcX,
}
impl std::convert::From<u8> for CmdOp {
fn from(value: u8) -> Self {
unsafe { std::mem::transmute(value) }
}
}
impl CmdOp {
pub const fn to(self) -> OptNum {
self as OptNum
}
}
#[rustfmt::skip]
const OPT_ARY: [Opt;7] = [
Opt { sho: b'X', lon: "", has: Arg::Yes, num: CmdOp::UcX.to(), },
Opt { sho: b'a', lon: "all", has: Arg::No, num: CmdOp::All.to(), },
Opt { sho: b'H', lon: "help", has: Arg::No, num: CmdOp::Help.to(), },
Opt { sho: 0u8, lon: "pid", has: Arg::Yes, num: CmdOp::Pid.to(), },
Opt { sho: 0u8, lon: "sleep", has: Arg::Yes, num: CmdOp::Sleep.to(), },
Opt { sho: 0u8, lon: "sort", has: Arg::Yes, num: CmdOp::Sort.to(), },
Opt { sho: b'V', lon: "version", has: Arg::No, num: CmdOp::Version.to(), },
];
#[rustfmt::skip]
const OPT_ARY_SHO_IDX: [(u8,usize);4] = [
(b'H',2),(b'V',6),(b'X',0),(b'a',1),];
#[derive(Debug, Default, PartialEq)]
pub struct CmdOptConf {
pub prog_name: String,
pub flg_all: bool,
pub opt_sort: OptSortOrder,
pub opt_pid: i32,
pub opt_sleep: u32,
pub flg_help: bool,
pub flg_version: bool,
pub opt_uc_x: Vec<OptUcXParam>,
pub arg_params: Vec<String>,
}
impl flood_tide::HelpVersion for CmdOptConf {
fn is_help(&self) -> bool {
self.flg_help
}
fn is_version(&self) -> bool {
self.flg_version
}
}
fn value_to_i32(nv: &NameVal<'_>) -> Result<i32, OptParseError> {
match nv.val {
Some(x) => match x.parse::<i32>() {
Ok(d) => Ok(d),
Err(err) => Err(OptParseError::invalid_option_argument(
&nv.opt.lon_or_sho(),
&err.to_string(),
)),
},
None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
}
}
fn value_to_u32(nv: &NameVal<'_>) -> Result<u32, OptParseError> {
match nv.val {
Some(x) => match x.parse::<u32>() {
Ok(d) => Ok(d),
Err(err) => Err(OptParseError::invalid_option_argument(
&nv.opt.lon_or_sho(),
&err.to_string(),
)),
},
None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
}
}
fn value_to_opt_sort_order(nv: &NameVal<'_>) -> Result<OptSortOrder, OptParseError> {
match nv.val {
Some(s) => match FromStr::from_str(s) {
Ok(x) => Ok(x),
Err(err) => Err(OptParseError::invalid_option_argument(
&nv.opt.lon_or_sho(),
&err.to_string(),
)),
},
None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
}
}
fn value_to_opt_uc_x_param(nv: &NameVal<'_>) -> Result<OptUcXParam, OptParseError> {
match nv.val {
Some(s) => match FromStr::from_str(s) {
Ok(x) => Ok(x),
Err(err) => Err(OptParseError::invalid_option_argument(
&nv.opt.lon_or_sho(),
&err.to_string(),
)),
},
None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
}
}