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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include "Flag.h"
#include "StringPiece.h"
#include <functional>
#include <iomanip>
#include <iostream>
#include <string>
#include <vector>
namespace aapt {
namespace flag {
struct Flag {
std::string name;
std::string description;
std::function<void(const StringPiece&)> action;
bool required;
bool* flagResult;
bool parsed;
};
static std::vector<Flag> sFlags;
static std::vector<std::string> sArgs;
void optionalFlag(const StringPiece& name, const StringPiece& description,
std::function<void(const StringPiece&)> action) {
sFlags.push_back(
Flag{ name.toString(), description.toString(), action, false, nullptr, false });
}
void requiredFlag(const StringPiece& name, const StringPiece& description,
std::function<void(const StringPiece&)> action) {
sFlags.push_back(
Flag{ name.toString(), description.toString(), action, true, nullptr, false });
}
void optionalSwitch(const StringPiece& name, const StringPiece& description, bool* result) {
sFlags.push_back(
Flag{ name.toString(), description.toString(), {}, false, result, false });
}
void usageAndDie(const StringPiece& command) {
std::cerr << command << " [options]";
for (const Flag& flag : sFlags) {
if (flag.required) {
std::cerr << " " << flag.name << " arg";
}
}
std::cerr << " files..." << std::endl << std::endl << "Options:" << std::endl;
for (const Flag& flag : sFlags) {
std::string command = flag.name;
if (!flag.flagResult) {
command += " arg ";
}
std::cerr << " " << std::setw(30) << std::left << command
<< flag.description << std::endl;
}
exit(1);
}
void parse(int argc, char** argv, const StringPiece& command) {
for (int i = 0; i < argc; i++) {
const StringPiece arg(argv[i]);
if (*arg.data() != '-') {
sArgs.emplace_back(arg.toString());
continue;
}
bool match = false;
for (Flag& flag : sFlags) {
if (arg == flag.name) {
match = true;
flag.parsed = true;
if (flag.flagResult) {
*flag.flagResult = true;
} else {
i++;
if (i >= argc) {
std::cerr << flag.name << " missing argument." << std::endl
<< std::endl;
usageAndDie(command);
}
flag.action(argv[i]);
}
break;
}
}
if (!match) {
std::cerr << "unknown option '" << arg << "'." << std::endl << std::endl;
usageAndDie(command);
}
}
for (const Flag& flag : sFlags) {
if (flag.required && !flag.parsed) {
std::cerr << "missing required flag " << flag.name << std::endl << std::endl;
usageAndDie(command);
}
}
}
const std::vector<std::string>& getArgs() {
return sArgs;
}
} // namespace flag
} // namespace aapt
|
