#include <iostream>
#include <string>
#include <getopt.h>

#include "accumulation.h"

/*
Notes:
	Datapins for same var/symbol are randomized in order of their level, from bottom of nest to top
	The CU uid symbol helps differentiating between e.g. "file.c" and "sub/file.c" (symbtab has no idea)
	Between CUs, types with the same name HAVE TO HAVE the same layout -> randomized together
	To begin with, anonymous types are not allowed for randomization (later solved with hash(type) instead of name)!
*/

int main(int argc, char** argv) {
	static option long_options[] = {
		{ "help", no_argument, 0, 0 },
		{ "spslr", required_argument, 0, 0 },
		{ "bin", required_argument, 0, 0 },
		{ "out", required_argument, 0, 0 },
		{ "strip", no_argument, 0, 0 },
		{ 0, 0, 0, 0 }
	};

	int option_index = 0;
	int c;

	std::string spslr_dir, bin_file, out_file;

	while ((c = getopt_long(argc, argv, "", long_options, &option_index)) == 0) {
		const option& opt = long_options[option_index];
		std::string optname { opt.name };

		if (optname == "help") {
			std::cout << "To use spslr_finalize, supply these 3 arguments:" << std::endl;
			std::cout << "    --spslr=<dir> (the directory of .spslr files produced by spslr_pinpoint)" << std::endl;
			std::cout << "    --bin=<file> (the binary compiled with spslr_pinpoint to be finalized)" << std::endl;
			std::cout << "    --out=<file> (the finalized binary file to be written)" << std::endl;
			return 0;
		} else if (optname == "spslr") {
			spslr_dir = std::string{ optarg };
		} else if (optname == "bin") {
			bin_file = std::string{ optarg };
		} else if (optname == "out") {
			out_file = std::string{ optarg };
		} else if (optname == "strip") {
			std::cerr << "Symbol stripping (--strip) is not yet implemented!" << std::endl;
			return 1;
		} else {
			std::cerr << "Invalid option, try \"--help\"!" << std::endl;
			return 1;
		}
	}

	if (spslr_dir.empty()) {
		std::cerr << "Missing spslr directory, supply it via --spslr=<dir>!" << std::endl;
		return 1;
	}

	if (bin_file.empty()) {
		std::cerr << "Missing input file path, supply it via --bin=<file>!" << std::endl;
		return 1;
	}

	if (out_file.empty()) {
		std::cerr << "Missing output file path, supply it via --out=<file>!" << std::endl;
		return 1;
	}

	if (!accumulate(spslr_dir)) {
		std::cerr << "Failed to accumulate data from spslr directory!" << std::endl;
		return 1;
	}

	std::cout << "Gathered a total of " << targets.size() << " distinct targets from "
		<< units.size() << " compilation units!" << std::endl;

	// TODO
	/*
	1. Loop over all symbols of the binary
		-> associate blocks via CU uid symbol
		-> find __spslr_program symbol (spslr vaddr pivot)
	2. Find virtual address and file address for all pins
	3. Emit patcher program into final executable and set __spslr_program
		-> make all patcher addresses relative to &__spslr_program
		-> dpatch entries ordered by level (high levels/nest depths first)
	*/
}