summaryrefslogtreecommitdiff
path: root/getoptpp.hpp
blob: 8788ddebd96c527076c367ec476ae8cdcc1d4363 (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
#pragma once
#include <iostream>
#include <cassert>
#include <functional>
#include <vector>
#include <unordered_map>
#include <string>
#include <getopt.h>

namespace dg {

using Handle = std::function<int()>;

class Options {
	public:
		/// @param name		name of the option
		/// @param has_arg	kind of arguments that are used (no_argument, required_argument, optional_argument)
		/// @param val		identifies the option (see getopt documentation)
		/// @param handle	lambda that is invoked when the option occures
		void add(std::string const & name, int has_arg, int val, Handle handle) {
			add(name, has_arg, nullptr, val, handle);
		}

		/// @param name		name of the option
		/// @param has_arg	kind of arguments that are used (no_argument, required_argument, optional_argument)
		/// @param flag		pointer that is set after the option occured
		/// @param val		value for the flag to be set
		/// @param handle	lambda that is invoked when the option occures
		void add(std::string const & name, int has_arg, int* flag, int val, Handle handle) {
			// create a new option from the args
			options.push_back({name.data(), has_arg, flag, val});

			int index = val; // let val be the option's unique identifier
			if (flag != nullptr) {
				// flag is not null, so the val is not used as identifier
				// so pick another one
				index = num_flags++;
			}
			// store the handle
			handles[index] = handle;
		}

		int process(int argc, char* argv[]) {
			int ret = 0;
			std::string shortopts;
			for (auto const & option: options) {
				if (option.flag != nullptr) {
					continue;
				}
				shortopts += static_cast<char>(option.val);

				switch (option.has_arg) {
				case no_argument:
					break;
				case required_argument:
					shortopts += ":";
					break;
				case optional_argument:
					shortopts += "::";
					break;
				}
			}

			// add termination option
			options.push_back({0, 0, 0, 0});

			// handle arguments
			while (true) {
				int index = 0;
				int key = getopt_long(argc, argv, shortopts.data(), options.data(), &index);

				if (key == -1) {
					break;
				} else if (key == '?') {
					return 1;
				} else if (key == ':') {
					return 1;
				} else if (key == 0) {
					// call flag's handle
					ret = handles.at(index)();
				} else {
					// call option's handle
					ret = handles.at(key)();
				}
			}

			if(ret) {
				return ret;
			}

			for (int i = optind; i < argc; ++i) {
				args.push_back(argv[i]);
			}

			// remove terminating option
			options.pop_back();
			assert(options.size() == handles.size());
			return 0;
		}

		const std::vector<std::string> arguments() const { return args; }

	private:
		std::size_t num_flags{};
		std::vector<option> options{};
		std::vector<std::string> args;
		std::unordered_map<int, Handle> handles{};
};

}