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
|
#include "core/pch.hh"
#include "core/io/cmdline.hh"
// Valid options always start with OPTION_PREFIX, can contain
// a bunch of OPTION_PREFIX'es inside and never end with one
constexpr static char OPTION_PREFIX = '-';
static std::unordered_map<std::string, std::string> options;
static inline bool is_option_string(const std::string& string)
{
if(string.find_last_of(OPTION_PREFIX) >= (string.size() - 1)) {
return false;
}
return string[0] == OPTION_PREFIX;
}
static inline std::string get_option(const std::string& string)
{
std::size_t i;
for(i = 0; string[i] == OPTION_PREFIX; ++i) {
// empty
}
return std::string(string.cbegin() + i, string.cend());
}
void io::cmdline::create(int argc, char** argv)
{
for(int idx = 1; idx < argc; ++idx) {
std::string string = argv[idx];
if(!is_option_string(string)) {
spdlog::warn("cmdline: non-argument at {}: {}", idx, string);
continue;
}
auto option_string = get_option(string);
auto next_idx = idx + 1;
if(next_idx < argc) {
std::string argument = argv[next_idx];
if(!is_option_string(argument)) {
options.insert_or_assign(option_string, argument);
idx = next_idx;
continue;
}
}
// The option is either last or has no
// argument (happens when there is a valid
// option right next to the one we're parsing)
options.insert_or_assign(option_string, std::string());
}
}
void io::cmdline::insert(std::string_view option)
{
options.insert_or_assign(std::string(option), std::string());
}
void io::cmdline::insert(std::string_view option, std::string_view argument)
{
options.insert_or_assign(std::string(option), std::string(argument));
}
std::string_view io::cmdline::get(std::string_view option, std::string_view fallback)
{
auto it = options.find(std::string(option));
if(it == options.cend()) {
return fallback;
}
return it->second;
}
const char* io::cmdline::get_cstr(std::string_view option, const char* fallback)
{
auto it = options.find(std::string(option));
if(it == options.cend()) {
return fallback;
}
return it->second.c_str();
}
bool io::cmdline::contains(std::string_view option)
{
return options.count(std::string(option));
}
|
