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
126
127
128
129
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
-l, --cmdline view command line
-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, Eq)]
enum CmdOp {
All,
Sort,
Pid,
Sleep,
Cmdline,
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;8] = [
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'l', lon: "cmdline", has: Arg::No, num: CmdOp::Cmdline.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);5] = [
(b'H',3),(b'V',7),(b'X',0),(b'a',1),(b'l',2),];
#[derive(Debug, Default, PartialEq, Eq)]
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_cmdline: bool,
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())),
}
}