blob: c83f9e6931dd5effaf56ffc8617a0e11f573e7f9 (
plain)
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
|
#include "core/pch.hh"
#include "core/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)
;
return std::string(string.cbegin() + i, string.cend());
}
void 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 cmdline::insert(const char* option, const char* argument)
{
if(argument == nullptr) {
options.insert_or_assign(option, std::string());
} else {
options.insert_or_assign(option, argument);
}
}
const char* cmdline::get(const char* option, const char* fallback)
{
auto it = options.find(option);
if(it == options.cend()) {
return fallback;
}
return it->second.c_str();
}
bool cmdline::contains(const char* option)
{
return options.count(option);
}
|
