Mercurial > code
annotate modules/options/options.cpp @ 508:8fc7fe1ec915
Sockets: pass Condition by reference, fix Tls
author | David Demelier <markand@malikania.fr> |
---|---|
date | Tue, 23 Feb 2016 12:04:40 +0100 |
parents | d8ed4da7688c |
children | f48bb09bccc7 |
rev | line source |
---|---|
486 | 1 /* |
2 * options.cpp -- parse Unix command line options | |
3 * | |
4 * Copyright (c) 2015 David Demelier <markand@malikania.fr> | |
5 * | |
6 * Permission to use, copy, modify, and/or distribute this software for any | |
7 * purpose with or without fee is hereby granted, provided that the above | |
8 * copyright notice and this permission notice appear in all copies. | |
9 * | |
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES | |
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF | |
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR | |
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF | |
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |
17 */ | |
18 | |
19 #include <cassert> | |
20 | |
21 #include "options.h" | |
22 | |
23 namespace option { | |
24 | |
25 namespace { | |
26 | |
27 using Iterator = std::vector<std::string>::iterator; | |
28 using Args = std::vector<std::string>; | |
29 | |
30 inline bool isOption(const std::string &arg) noexcept | |
31 { | |
32 return arg.size() >= 2 && arg[0] == '-'; | |
33 } | |
34 | |
35 inline bool isLongOption(const std::string &arg) noexcept | |
36 { | |
37 assert(isOption(arg)); | |
38 | |
39 return arg.size() >= 3 && arg[1] == '-'; | |
40 } | |
41 | |
42 inline bool isShortSimple(const std::string &arg) noexcept | |
43 { | |
44 assert(isOption(arg)); | |
45 assert(!isLongOption(arg)); | |
46 | |
47 return arg.size() == 2; | |
48 } | |
49 | |
50 void parseLongOption(Result &result, Args &args, Iterator &it, Iterator &end, const Options &definition) | |
51 { | |
52 auto arg = *it++; | |
53 auto opt = definition.find(arg); | |
54 | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
55 if (opt == definition.end()) |
486 | 56 throw InvalidOption{arg}; |
57 | |
58 /* Need argument? */ | |
59 if (opt->second) { | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
60 if (it == end || isOption(*it)) |
486 | 61 throw MissingValue{arg}; |
62 | |
63 result.insert(std::make_pair(arg, *it++)); | |
64 it = args.erase(args.begin(), it); | |
65 end = args.end(); | |
66 } else { | |
67 result.insert(std::make_pair(arg, "")); | |
68 it = args.erase(args.begin()); | |
69 end = args.end(); | |
70 } | |
71 } | |
72 | |
73 void parseShortOption(Result &result, Args &args, Iterator &it, Iterator &end, const Options &definition) | |
74 { | |
75 if (isShortSimple(*it)) { | |
76 /* | |
77 * Here two cases: | |
78 * | |
79 * -v (no option) | |
80 * -c value | |
81 */ | |
82 auto arg = *it++; | |
83 auto opt = definition.find(arg); | |
84 | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
85 if (opt == definition.end()) |
486 | 86 throw InvalidOption{arg}; |
87 | |
88 /* Need argument? */ | |
89 if (opt->second) { | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
90 if (it == end || isOption(*it)) |
486 | 91 throw MissingValue{arg}; |
92 | |
93 result.insert(std::make_pair(arg, *it++)); | |
94 it = args.erase(args.begin(), it); | |
95 end = args.end(); | |
96 } else { | |
97 result.insert(std::make_pair(arg, "")); | |
98 it = args.erase(args.begin()); | |
99 end = args.end(); | |
100 } | |
101 } else { | |
102 /* | |
103 * Here multiple scenarios: | |
104 * | |
105 * 1. -abc (-a -b -c if all are simple boolean arguments) | |
106 * 2. -vc foo.conf (-v -c foo.conf if -c is argument dependant) | |
107 * 3. -vcfoo.conf (-v -c foo.conf also) | |
108 */ | |
109 auto value = it->substr(1); | |
110 auto len = value.length(); | |
111 int toremove = 1; | |
112 | |
113 for (decltype(len) i = 0; i < len; ++i) { | |
114 auto arg = std::string{'-'} + value[i]; | |
115 auto opt = definition.find(arg); | |
116 | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
117 if (opt == definition.end()) |
486 | 118 throw InvalidOption{arg}; |
119 | |
120 if (opt->second) { | |
121 if (i == (len - 1)) { | |
122 /* End of string, get the next argument (see 2.) */ | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
123 if (++it == end || isOption(*it)) |
486 | 124 throw MissingValue{arg}; |
125 | |
126 result.insert(std::make_pair(arg, *it)); | |
127 toremove += 1; | |
128 } else { | |
129 result.insert(std::make_pair(arg, value.substr(i + 1))); | |
130 i = len; | |
131 } | |
132 } else { | |
133 result.insert(std::make_pair(arg, "")); | |
134 } | |
135 } | |
136 | |
137 it = args.erase(args.begin(), args.begin() + toremove); | |
138 end = args.end(); | |
139 } | |
140 } | |
141 | |
142 } // !namespace | |
143 | |
144 Result read(std::vector<std::string> &args, const Options &definition) | |
145 { | |
146 Result result; | |
147 | |
148 auto it = args.begin(); | |
149 auto end = args.end(); | |
150 | |
151 while (it != end) { | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
152 if (!isOption(*it)) |
486 | 153 break; |
154 | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
155 if (isLongOption(*it)) |
486 | 156 parseLongOption(result, args, it, end, definition); |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
157 else |
486 | 158 parseShortOption(result, args, it, end, definition); |
159 } | |
160 | |
161 return result; | |
162 } | |
163 | |
164 Result read(int &argc, char **&argv, const Options &definition) | |
165 { | |
166 std::vector<std::string> args; | |
167 | |
500
d8ed4da7688c
Options: new braces style
David Demelier <markand@malikania.fr>
parents:
486
diff
changeset
|
168 for (int i = 0; i < argc; ++i) |
486 | 169 args.push_back(argv[i]); |
170 | |
171 auto before = args.size(); | |
172 auto result = read(args, definition); | |
173 | |
174 argc -= before - args.size(); | |
175 argv += before - args.size(); | |
176 | |
177 return result; | |
178 } | |
179 | |
180 } // !option |