CCats/cli/src/main.cpp

#include "../include/batchioman.h"
#include "../include/machineioman.h"
#include "../include/userioman.h"

#include <boost/program_options.hpp>
#include <iostream>
#include <string>

namespace bpo = boost::program_options;

/* this is provided for the readline callback in IoMan */
IoMan *gIOMAN = NULL;

void show_help(char *exec) {
	std::printf("ccats command line interface\n");
	std::printf("usage: %s IP [Options]\n", exec);
	std::printf("IP should be in the format \"xxx.xxx.xxx.xxx\"\n");
}

int main(int argc, char **argv) {
	bpo::options_description desc("Options");
	desc.add_options()("help", "show this help")("machine", "switch to machine mode for I/O (not designed for user interaction)")(
	    "batch", bpo::value<std::string>(), "run operations from arg as batch file")("usessl", "enable ssl for connection to server")("verbose",
	                                                                                                                                  "enable debug output");

	bpo::variables_map vm;
	bool machine = false, usessl = false, batch = false, verbose = false;
	const char *file = NULL;
	IoMan *ioman;

	if (argc < 2 || !std::strncmp(argv[1], "--help", 6)) {
		show_help(argv[0]);
		std::cout << desc;
		return 1;
	}

	try {
		store(parse_command_line(argc - 1, argv + 1, desc), vm);
		notify(vm);

		if (vm.count("help")) {
			show_help(argv[0]);
			std::cout << desc;
			return 1;
		}
		if (vm.count("machine")) {
			// enable machine/gui mode
			machine = true;
		}
		if (vm.count("batch")) {
			// handle batch file mode
			file = vm["batch"].as<std::string>().c_str();
			batch = true;
		}
		if (vm.count("usessl")) {
			usessl = true;
		}
		if (vm.count("verbose")) {
			verbose = true;
		}
	} catch (const bpo::error &ex) {
		std::fprintf(stderr, "%s\n", ex.what());
	}
	std::printf("ip %s machine mode is %d file is %s enablessl is %d verbose is %d\n", argv[1], machine, file ? file : "", usessl, verbose);
	if (batch) {
		ioman = new BatchIoMan(argv[1], usessl, verbose, file);
	} else if (machine) {
		ioman = new MachineIoMan(argv[1], usessl, verbose);
	} else {
		ioman = new UserIoMan(argv[1], usessl, verbose);
	}
	gIOMAN = ioman;
	if (ioman->init()) {
		ioman->run();
	}
	delete ioman;
	gIOMAN = NULL;
	std::printf("done\n");
}