forked from johguse/profanity
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ArgParser.hpp
107 lines (87 loc) · 2.63 KB
/
ArgParser.hpp
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
#ifndef HPP_ARGPARSER
#define HPP_ARGPARSER
#include <type_traits>
#include <stdexcept>
#include <vector>
#include <map>
#include "lexical_cast.hpp"
class ArgParser {
private:
class IArgument {
public:
virtual ~IArgument() {}
virtual void parse(const std::string & s) = 0;
};
template <typename T>
class Argument : public IArgument {
public:
Argument(T & t) : m_t(t) {}
~Argument() {}
void parse(const std::string & s) {
m_t = fromString<T>(s);
}
private:
T & m_t;
};
template <typename T>
class MultiArgument : public IArgument {
public:
MultiArgument(std::vector<T> & t) : m_t(t) {}
MultiArgument() {}
void parse(const std::string & s) {
m_t.push_back(fromString<T>(s));
}
private:
std::vector<T> & m_t;
};
public:
ArgParser(int argc, char * * argv) {
for (int i = 1; i < argc; ++i) {
m_args.push_back(argv[i]);
}
}
~ArgParser() {
for (auto & i : m_mapArgs) {
delete i.second.second; // :)
}
}
template <typename T>
void addSwitch(const char switchShort, const std::string switchLong, T & t) {
const std::string strShort = std::string("-") + switchShort;
const std::string strLong = std::string("--") + switchLong;
// :)
IArgument * const pArgShort = new Argument<T>(t);
IArgument * const pArgLong = new Argument<T>(t);
m_mapArgs[strShort] = std::pair<bool, IArgument *>(std::is_same<bool, T>::value, pArgShort);
m_mapArgs[strLong] = std::pair<bool, IArgument *>(std::is_same<bool, T>::value, pArgLong);
}
template <typename T>
void addMultiSwitch(const char switchShort, const std::string switchLong, std::vector<T> & t) {
const std::string strShort = std::string("-") + switchShort;
const std::string strLong = std::string("--") + switchLong;
// :)
IArgument * const pArgShort = new MultiArgument<T>(t);
IArgument * const pArgLong = new MultiArgument<T>(t);
m_mapArgs[strShort] = std::pair<bool, IArgument *>(false, pArgShort);
m_mapArgs[strLong] = std::pair<bool, IArgument *>(false, pArgLong);
}
bool parse() const {
try {
std::vector<std::string>::size_type i = 0;
while (i < m_args.size()) {
auto p = m_mapArgs.at(m_args[i]);
const std::string s = p.first ? "1" : m_args.at(i + 1);
p.second->parse(s);
i += (p.first ? 1 : 2);
}
}
catch (std::out_of_range & e) {
return false;
}
return true;
}
private:
std::vector<std::string> m_args;
std::map<std::string, std::pair<bool, IArgument *>> m_mapArgs;
};
#endif /* HPP_ARGPARSER */