aboutsummaryrefslogtreecommitdiffstats
path: root/checks/getopt.h
diff options
context:
space:
mode:
Diffstat (limited to 'checks/getopt.h')
-rw-r--r--checks/getopt.h91
1 files changed, 85 insertions, 6 deletions
diff --git a/checks/getopt.h b/checks/getopt.h
index 68e6bd874..67cf979f9 100644
--- a/checks/getopt.h
+++ b/checks/getopt.h
@@ -9,19 +9,92 @@
#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&) const;
- std::string value(const std::string&) const;
- std::string value_if_set(const std::string&) const;
+ 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]));
+ }
- void parse(char*[]);
- OptionParser(const std::string&);
private:
class OptionFlag
{
@@ -40,7 +113,13 @@ class OptionParser
bool opt_takes_arg;
};
- OptionFlag find_option(const std::string&) const;
+ 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;