Support equal signs in options, e.g. --log-file=a.log

This commit is contained in:
Fangrui Song 2017-12-29 14:00:02 -08:00
parent 43ea2fcc53
commit 720266e97a

View File

@ -8,22 +8,16 @@ std::unordered_map<std::string, std::string> ParseOptions(int argc,
char** argv) {
std::unordered_map<std::string, std::string> output;
std::string previous_arg;
for (int i = 1; i < argc; ++i) {
std::string arg = argv[i];
if (arg[0] != '-') {
if (previous_arg.size() == 0) {
LOG_S(FATAL) << "Invalid arguments; switches must start with -";
exit(1);
}
output[previous_arg] = arg;
previous_arg = "";
} else {
output[arg] = "";
previous_arg = arg;
if (arg[0] == '-') {
auto equal = arg.find('=');
if (equal != std::string::npos) {
output[arg.substr(0, equal)] = arg.substr(equal + 1);
} else if (i + 1 < argc && argv[i + 1][0] != '-')
output[arg] = argv[++i];
else
output[arg] = "";
}
}
@ -33,4 +27,4 @@ std::unordered_map<std::string, std::string> ParseOptions(int argc,
bool HasOption(const std::unordered_map<std::string, std::string>& options,
const std::string& option) {
return options.find(option) != options.end();
}
}