Mercurial > irccd
annotate common/options.cpp @ 61:adfc2be5ee30 release-2.0
Added signature for changeset 223487a685b1
| author | David Demelier <markand@malikania.fr> |
|---|---|
| date | Tue, 01 Mar 2016 08:53:16 +0100 |
| parents | 03068f5ed79d |
| children |
| rev | line source |
|---|---|
| 0 | 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 irccd { | |
| 24 | |
| 25 namespace parser { | |
| 26 | |
| 27 namespace { | |
| 28 | |
| 29 using Iterator = std::vector<std::string>::iterator; | |
| 30 using Args = std::vector<std::string>; | |
| 31 | |
| 32 inline bool isOption(const std::string &arg) noexcept | |
| 33 { | |
| 34 return arg.size() >= 2 && arg[0] == '-'; | |
| 35 } | |
| 36 | |
| 37 inline bool isLongOption(const std::string &arg) noexcept | |
| 38 { | |
| 39 assert(isOption(arg)); | |
| 40 | |
| 41 return arg.size() >= 3 && arg[1] == '-'; | |
| 42 } | |
| 43 | |
| 44 inline bool isShortSimple(const std::string &arg) noexcept | |
| 45 { | |
| 46 assert(isOption(arg)); | |
| 47 assert(!isLongOption(arg)); | |
| 48 | |
| 49 return arg.size() == 2; | |
| 50 } | |
| 51 | |
| 52 void parseLongOption(Result &result, Args &args, Iterator &it, Iterator &end, const Options &definition) | |
| 53 { | |
| 54 auto arg = *it++; | |
| 55 auto opt = definition.find(arg); | |
| 56 | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
57 if (opt == definition.end()) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
58 throw InvalidOption(arg); |
| 0 | 59 |
| 60 /* Need argument? */ | |
| 61 if (opt->second) { | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
62 if (it == end || isOption(*it)) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
63 throw MissingValue(arg); |
| 0 | 64 |
| 65 result.insert(std::make_pair(arg, *it++)); | |
| 66 it = args.erase(args.begin(), it); | |
| 67 end = args.end(); | |
| 68 } else { | |
| 69 result.insert(std::make_pair(arg, "")); | |
| 70 it = args.erase(args.begin()); | |
| 71 end = args.end(); | |
| 72 } | |
| 73 } | |
| 74 | |
| 75 void parseShortOption(Result &result, Args &args, Iterator &it, Iterator &end, const Options &definition) | |
| 76 { | |
| 77 if (isShortSimple(*it)) { | |
| 78 /* | |
| 79 * Here two cases: | |
| 80 * | |
| 81 * -v (no option) | |
| 82 * -c value | |
| 83 */ | |
| 84 auto arg = *it++; | |
| 85 auto opt = definition.find(arg); | |
| 86 | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
87 if (opt == definition.end()) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
88 throw InvalidOption(arg); |
| 0 | 89 |
| 90 /* Need argument? */ | |
| 91 if (opt->second) { | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
92 if (it == end || isOption(*it)) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
93 throw MissingValue(arg); |
| 0 | 94 |
| 95 result.insert(std::make_pair(arg, *it++)); | |
| 96 it = args.erase(args.begin(), it); | |
| 97 end = args.end(); | |
| 98 } else { | |
| 99 result.insert(std::make_pair(arg, "")); | |
| 100 it = args.erase(args.begin()); | |
| 101 end = args.end(); | |
| 102 } | |
| 103 } else { | |
| 104 /* | |
| 105 * Here multiple scenarios: | |
| 106 * | |
| 107 * 1. -abc (-a -b -c if all are simple boolean arguments) | |
| 108 * 2. -vc foo.conf (-v -c foo.conf if -c is argument dependant) | |
| 109 * 3. -vcfoo.conf (-v -c foo.conf also) | |
| 110 */ | |
| 111 auto value = it->substr(1); | |
| 112 auto len = value.length(); | |
| 113 int toremove = 1; | |
| 114 | |
| 115 for (decltype(len) i = 0; i < len; ++i) { | |
| 116 auto arg = std::string{'-'} + value[i]; | |
| 117 auto opt = definition.find(arg); | |
| 118 | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
119 if (opt == definition.end()) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
120 throw InvalidOption(arg); |
| 0 | 121 |
| 122 if (opt->second) { | |
| 123 if (i == (len - 1)) { | |
| 124 /* End of string, get the next argument (see 2.) */ | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
125 if (++it == end || isOption(*it)) |
|
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
126 throw MissingValue(arg); |
| 0 | 127 |
| 128 result.insert(std::make_pair(arg, *it)); | |
| 129 toremove += 1; | |
| 130 } else { | |
| 131 result.insert(std::make_pair(arg, value.substr(i + 1))); | |
| 132 i = len; | |
| 133 } | |
| 134 } else { | |
| 135 result.insert(std::make_pair(arg, "")); | |
| 136 } | |
| 137 } | |
| 138 | |
| 139 it = args.erase(args.begin(), args.begin() + toremove); | |
| 140 end = args.end(); | |
| 141 } | |
| 142 } | |
| 143 | |
| 144 } // !namespace | |
| 145 | |
| 146 Result read(std::vector<std::string> &args, const Options &definition) | |
| 147 { | |
| 148 Result result; | |
| 149 | |
| 150 auto it = args.begin(); | |
| 151 auto end = args.end(); | |
| 152 | |
| 153 while (it != end) { | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
154 if (!isOption(*it)) |
| 0 | 155 break; |
| 156 | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
157 if (isLongOption(*it)) |
| 0 | 158 parseLongOption(result, args, it, end, definition); |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
159 else |
| 0 | 160 parseShortOption(result, args, it, end, definition); |
| 161 } | |
| 162 | |
| 163 return result; | |
| 164 } | |
| 165 | |
| 166 Result read(int &argc, char **&argv, const Options &definition) | |
| 167 { | |
| 168 std::vector<std::string> args; | |
| 169 | |
|
23
03068f5ed79d
Misc: various style issues, #419
David Demelier <markand@malikania.fr>
parents:
0
diff
changeset
|
170 for (int i = 0; i < argc; ++i) |
| 0 | 171 args.push_back(argv[i]); |
| 172 | |
| 173 auto before = args.size(); | |
| 174 auto result = read(args, definition); | |
| 175 | |
| 176 argc -= before - args.size(); | |
| 177 argv += before - args.size(); | |
| 178 | |
| 179 return result; | |
| 180 } | |
| 181 | |
| 182 } // !parser | |
| 183 | |
| 184 } // !irccd |