1const OPTIONS_TEXT: &str = r"Options:
3 -a, --all all pid (include kernel threads)
4 --sort <order> sort by <order>: rss|swap|total
5 --pid <number> output only selected pid
6 --sleep <number> sleep <number> milli second
7 -l, --cmdline view command line
8
9 -H, --help display this help and exit
10 -V, --version display version information and exit
11 -X <x-options> x options. try -X help
12";
13
14#[repr(u8)]
15#[derive(Debug, PartialEq, Eq)]
16enum CmdOp {
17 All,
18 Sort,
19 Pid,
20 Sleep,
21 Cmdline,
22 Help,
23 Version,
24 UcX,
25}
26
27impl std::convert::From<u8> for CmdOp {
28 fn from(value: u8) -> Self {
29 unsafe { std::mem::transmute(value) }
30 }
31}
32impl CmdOp {
33 pub const fn to(self) -> OptNum {
34 self as OptNum
35 }
36}
37
38#[rustfmt::skip]
39const OPT_ARY: [Opt;8] = [
40 Opt { sho: b'X', lon: "", has: Arg::Yes, num: CmdOp::UcX.to(), },
41 Opt { sho: b'a', lon: "all", has: Arg::No, num: CmdOp::All.to(), },
42 Opt { sho: b'l', lon: "cmdline", has: Arg::No, num: CmdOp::Cmdline.to(), },
43 Opt { sho: b'H', lon: "help", has: Arg::No, num: CmdOp::Help.to(), },
44 Opt { sho: 0u8, lon: "pid", has: Arg::Yes, num: CmdOp::Pid.to(), },
45 Opt { sho: 0u8, lon: "sleep", has: Arg::Yes, num: CmdOp::Sleep.to(), },
46 Opt { sho: 0u8, lon: "sort", has: Arg::Yes, num: CmdOp::Sort.to(), },
47 Opt { sho: b'V', lon: "version", has: Arg::No, num: CmdOp::Version.to(), },
48];
49
50#[rustfmt::skip]
51const OPT_ARY_SHO_IDX: [(u8,usize);5] = [
52(b'H',3),(b'V',7),(b'X',0),(b'a',1),(b'l',2),];
53
54#[derive(Debug, Default, PartialEq, Eq)]
55pub struct CmdOptConf {
56 pub prog_name: String,
57 pub flg_all: bool,
59 pub opt_sort: OptSortOrder,
60 pub opt_pid: i32,
61 pub opt_sleep: u32,
62 pub flg_cmdline: bool,
63 pub flg_help: bool,
64 pub flg_version: bool,
65 pub opt_uc_x: Vec<OptUcXParam>,
66 pub arg_params: Vec<String>,
68}
69
70impl flood_tide::HelpVersion for CmdOptConf {
71 fn is_help(&self) -> bool {
72 self.flg_help
73 }
74 fn is_version(&self) -> bool {
75 self.flg_version
76 }
77}
78
79fn value_to_i32(nv: &NameVal<'_>) -> Result<i32, OptParseError> {
80 match nv.val {
81 Some(x) => match x.parse::<i32>() {
82 Ok(d) => Ok(d),
83 Err(err) => Err(OptParseError::invalid_option_argument(
84 &nv.opt.lon_or_sho(),
85 &err.to_string(),
86 )),
87 },
88 None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
89 }
90}
91
92fn value_to_u32(nv: &NameVal<'_>) -> Result<u32, OptParseError> {
93 match nv.val {
94 Some(x) => match x.parse::<u32>() {
95 Ok(d) => Ok(d),
96 Err(err) => Err(OptParseError::invalid_option_argument(
97 &nv.opt.lon_or_sho(),
98 &err.to_string(),
99 )),
100 },
101 None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
102 }
103}
104
105fn value_to_opt_sort_order(nv: &NameVal<'_>) -> Result<OptSortOrder, OptParseError> {
106 match nv.val {
107 Some(s) => match FromStr::from_str(s) {
108 Ok(x) => Ok(x),
109 Err(err) => Err(OptParseError::invalid_option_argument(
110 &nv.opt.lon_or_sho(),
111 &err.to_string(),
112 )),
113 },
114 None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
115 }
116}
117
118fn value_to_opt_uc_x_param(nv: &NameVal<'_>) -> Result<OptUcXParam, OptParseError> {
119 match nv.val {
120 Some(s) => match FromStr::from_str(s) {
121 Ok(x) => Ok(x),
122 Err(err) => Err(OptParseError::invalid_option_argument(
123 &nv.opt.lon_or_sho(),
124 &err.to_string(),
125 )),
126 },
127 None => Err(OptParseError::missing_option_argument(&nv.opt.lon_or_sho())),
128 }
129}