blob: 67cf979f917034dddc33b870cce242cdca2b9428 (
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
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* (C) 2009 Jack Lloyd
*
* Distributed under the terms of the Botan license
*/
#ifndef BOTAN_CHECK_GETOPT_H__
#define BOTAN_CHECK_GETOPT_H__
#include <string>
#include <vector>
#include <stdexcept>
#include <map>
#include <botan/parsing.h>
class OptionParser
{
public:
std::vector<std::string> leftovers() const { return leftover; }
bool is_set(const std::string& key) const
{
return (options.find(key) != options.end());
}
std::string value(const std::string& key) const
{
std::map<std::string, std::string>::const_iterator i = options.find(key);
if(i == options.end())
throw std::runtime_error("Option " + key + " not found");
return i->second;
}
std::string value_if_set(const std::string& key) const
{
return value_or_else(key, "");
}
std::string value_or_else(const std::string& key,
const std::string& or_else) const
{
return is_set(key) ? value(key) : or_else;
}
void parse(char* argv[])
{
std::vector<std::string> args;
for(int j = 1; argv[j]; j++)
args.push_back(argv[j]);
for(size_t j = 0; j != args.size(); j++)
{
std::string arg = args[j];
if(arg.size() > 2 && arg[0] == '-' && arg[1] == '-')
{
const std::string opt_name = arg.substr(0, arg.find('='));
arg = arg.substr(2);
std::string::size_type mark = arg.find('=');
OptionFlag opt = find_option(arg.substr(0, mark));
if(opt.takes_arg())
{
if(mark == std::string::npos)
throw std::runtime_error("Option " + opt_name +
" requires an argument");
std::string name = arg.substr(0, mark);
std::string value = arg.substr(mark+1);
options[name] = value;
}
else
{
if(mark != std::string::npos)
throw std::runtime_error("Option " + opt_name +
" does not take an argument");
options[arg] = "";
}
}
else
leftover.push_back(arg);
}
}
OptionParser(const std::string& opt_string)
{
std::vector<std::string> opts = Botan::split_on(opt_string, '|');
for(size_t j = 0; j != opts.size(); j++)
flags.push_back(OptionFlag(opts[j]));
}
private:
class OptionFlag
{
public:
std::string name() const { return opt_name; }
bool takes_arg() const { return opt_takes_arg; }
OptionFlag(const std::string& opt_string)
{
std::string::size_type mark = opt_string.find('=');
opt_name = opt_string.substr(0, mark);
opt_takes_arg = (mark != std::string::npos);
}
private:
std::string opt_name;
bool opt_takes_arg;
};
OptionFlag find_option(const std::string& name) const
{
for(size_t j = 0; j != flags.size(); j++)
if(flags[j].name() == name)
return flags[j];
throw std::runtime_error("Unknown option " + name);
}
std::vector<OptionFlag> flags;
std::map<std::string, std::string> options;
std::vector<std::string> leftover;
};
#endif
|
