-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathparse.cppm
More file actions
82 lines (69 loc) · 3.33 KB
/
parse.cppm
File metadata and controls
82 lines (69 loc) · 3.33 KB
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
module;
export module mcpplibs.cmdline:parse;
import std;
namespace mcpplibs::cmdline::detail {
/// 选项解析结果：同时表示 flag（用 count）和带值/多值选项（用 values）。
/// 不用 std::optional<std::string> 的原因：flag 无值且需出现次数；multiple 需多个值。
export struct OptionValue {
int count = 0; // flag 出现次数（如 -v -v -v）
std::vector<std::string> values; // 带值选项的值，multiple 时多个
[[nodiscard]] std::string value() const {
return values.empty() ? std::string{} : values.back();
}
[[nodiscard]] std::string value_or(std::string_view default_value) const {
return values.empty() ? std::string{default_value} : values.back();
}
[[nodiscard]] bool is_set() const { return count > 0 || !values.empty(); }
};
export struct ParseError {
enum Kind : std::uint8_t { help, version, error };
Kind kind = error;
std::string message;
bool is_error() const { return kind == error; }
static ParseError make_help() { return {.kind = help}; }
static ParseError make_version() { return {.kind = version}; }
friend std::ostream& operator<<(std::ostream& os, const ParseError& e) {
return os << e.message;
}
};
export struct ParsedArgs {
std::map<std::string, OptionValue, std::less<>> opts;
std::vector<std::string> positionals;
std::vector<std::string> positional_names_;
std::string subcommand_name_;
std::unique_ptr<ParsedArgs> subcommand_matches;
[[nodiscard]] std::optional<std::reference_wrapper<const OptionValue>> option(std::string_view name) const {
auto it = opts.find(name);
if (it == opts.end()) return std::nullopt;
return std::cref(it->second);
}
[[nodiscard]] OptionValue option_or_empty(std::string_view name) const {
auto opt = option(name);
return opt ? opt->get() : OptionValue{};
}
[[nodiscard]] bool is_flag_set(std::string_view name) const {
return option_or_empty(name).is_set();
}
[[nodiscard]] std::optional<std::string> value(std::string_view name) const {
auto opt = option(name);
if (opt && opt->get().is_set()) return opt->get().value();
for (std::size_t idx = 0; idx < positional_names_.size() && idx < positionals.size(); ++idx)
if (positional_names_[idx] == name) return positionals[idx];
return std::nullopt;
}
[[nodiscard]] std::string positional(std::size_t index) const {
return index < positionals.size() ? positionals[index] : std::string{};
}
[[nodiscard]] std::string positional_or(std::size_t index, std::string_view default_value) const {
return index < positionals.size() ? positionals[index] : std::string{default_value};
}
[[nodiscard]] std::size_t positional_count() const { return positionals.size(); }
[[nodiscard]] bool has_subcommand() const { return !subcommand_name_.empty(); }
[[nodiscard]] std::string_view subcommand_name() const { return std::string_view(subcommand_name_); }
[[nodiscard]] std::optional<std::reference_wrapper<const ParsedArgs>> subcommand() const {
if (subcommand_matches) return std::cref(*subcommand_matches);
return std::nullopt;
}
};
export using ParseResult = std::expected<ParsedArgs, ParseError>;
} // namespace mcpplibs::cmdline::detail