#include <iostream>
#include <vector>
#include <string>
#include <optional>
#include <utility>
#include <cstring>
#include "xutil.h"
#include "keysym_table.h"


namespace {
	void usage(const char *argv0) {
		std::cerr
			<< "Usage:\n"
			<< "  " << argv0 << " <key>\n"
			<< "  " << argv0 << " [<key> <output>]*\n"
			<< "\n"
			<< "Keys are specified with the X11 keysym name or ID (in hexadecimal with \"0x\").\n"
			<< "Both are given by xev(1).\n"
			<< "Examples:\n"
			<< "- KP_1 and 0xffb1 are valid key values for the '1' key on the numpad.\n"
			<< "- XF86Launch7 and 0x1008ff47 are valid key values for F16 on an Apple keyboard.\n"
			<< "The values can also be found in /usr/include/xkbcommon/xkbcommon-keysyms.h .\n"
			<< "\n"
			<< "If a single key is given, the program grabs that key and exits when the key is\n"
			<< "pressed. If multiple keys are given, the output string corresponding to the\n"
			<< "pressed key will be printed before exiting.\n"
			;
	}

	std::vector<std::pair<x::Keysym, std::string>> parse_watchlist(int argc, char **argv) {
		std::vector<std::pair<x::Keysym, std::string>> watchlist;

		if (argc <= 1) {
			usage(argv[0]);
			exit(1);
		}

		for (int i = 1; i < argc; i++) {
			if (strcmp(argv[i], "--help") == 0 || strcmp(argv[i], "-h") == 0) {
				usage(argv[0]);
				exit(0);
			}

			std::optional<x::Keysym> msym;

			if (argv[i][0] == '0' && argv[i][1] == 'x') {
				char *endp;
				unsigned code = strtol(argv[i] + 2, &endp, 16);
				if (argv[i][2] == '\0' || *endp != '\0') {
					std::cerr << "Invalid hexadecimal number in '" << argv[i] << "'\n";
					exit(1);
				}
				msym = x::Keysym{code};
			}

			if (!msym) {
				for (size_t j = 0; j < keysym_table.size(); j++) {
					if (strcmp(argv[i], keysym_table[j].first) == 0) {
						msym = keysym_table[j].second;
						break;
					}
				}
			}

			if (!msym) {
				std::cerr << "Unknown keysym '" << argv[i] << "'\n";
				exit(1);
			}

			std::string output;
			if (i == 1 && argc == 2) {
				// empty output
			} else if (i >= argc - 1) {
				std::cerr << "Missing output for keysym '" << argv[i] << "'\n";
				exit(1);
			} else {
				output = argv[i+1];
				i++;  // skip output
			}

			watchlist.emplace_back(*msym, std::move(output));
		}

		return watchlist;
	}
}

int main(int argc, char **argv) {
	std::vector<std::pair<x::Keysym, std::string>> watchlist = parse_watchlist(argc, argv);

	auto dpypair = x::XOpenDisplayRAII(nullptr);
	Display *dpy = dpypair.first;

	Window root = DefaultRootWindow(dpy);

	std::vector<x::UponExit<std::function<void()>>> cleaners;
	cleaners.reserve(watchlist.size());
	std::vector<x::Keycode> keycodes;
	keycodes.reserve(watchlist.size());

	for (const auto &p : watchlist) {
		x::Keycode code = p.first.toCode(dpy);
		keycodes.push_back(code);
		cleaners.push_back(x::XGrabKeyRAII(dpy, code, AnyModifier, root));
	}

	XSelectInput(dpy, root, KeyPressMask);
	while (true) {
		XEvent ev;
		XNextEvent(dpy, &ev);
		if (ev.type != KeyPress) continue;
		for (size_t i = 0; i < watchlist.size(); i++) {
			if (ev.xkey.keycode == (unsigned int)keycodes[i]) {
				if (watchlist[i].second.size() > 0) std::cout << watchlist[i].second << std::endl;
				return 0;
			}
		}
	}
}