#include <iostream>
#include <iomanip>

#include <vector>
#include <array>
#include <string>
#include <tuple>
#include <cmath>

#include <png++/png.hpp>

#include "FRC.h"
#include "input.h"
#include "map.h"

extern const char* vnum;

//HelpStr, iHelpStr
#include "strings.h"

using std::string;
using std::vector;
using std::cin;
using std::cout;
using std::cerr;
using std::endl;
using namespace std::string_literals;

//main interface, generates 1024x1024 map with all default settings
// void genMap(vector<string> inList);

bool Continue = true;
#ifdef _DEBUG
bool verbose = true;
#else
bool verbose = false;
#endif
input_trie commands;

void runCommand(input_trie commands, string in);
void drawMap(vector<string> inList)
{
	std::string file{inList[2]};
	unsigned cmapID{lookupMap(inList[1])},
		size{pFromStr(unsigned short, inList[6])};
	float x{pFromStr(float, inList[3])},
		y{pFromStr(float, inList[4])},
		width{pFromStr(float, inList[5])};
	bool archimedian{!(inList[0].find_first_of('l') != string::npos)};
	
	if (current_cmap[cmapID].second.size() > 256) {
		auto image = genMap_impl<png::rgb_pixel>(cmapID, archimedian, x, y, width, size);
		image.write(file);
	} else {
		auto image = genMap_impl<png::index_pixel>(cmapID, archimedian, x, y, width, size);
		image.write(file);
	}
}

int main(int argc, char** argv)
{
	string in;
	commands.insert(vector<std::pair<string, functdata*>>({{
		std::pair<string, functdata*>("h"s, new functdata(
			[](vector<string> inList){
				if (inList.size() > 1) {
					int len = 0;
					functdata *found = commands.longestPrefix(toLower(inList[1]), &len);
					if (found) {
						cout<<found->help<<endl;
					} else {
						cout<<"Nothing found for "<<toLower(inList[1])<<endl;
					}
				} else {
					//from strings.h
					cout<<iHelpStr;
			}},
			0, "Get help"s)),
		std::pair<string, functdata*>("q"s, new functdata(
			[]([[maybe_unused]] vector<string> inList){
				Continue = false;
			},
			0, "Quit"s)),
		std::pair<string, functdata*>("e"s, new functdata(
			[](vector<string> inList){
				cout<<'[';
				for (unsigned i = 0; i != inList.size(); i++) {
					cout<<inList[i]<<':';
				}
				cout<<']'<<endl;
			},
			0, "Echo inputs"s)),
		std::pair<string, functdata*>("x"s, new functdata(
			[](vector<string> inList){
				inList.erase(inList.begin());
				int len = 0;
				functdata *found = commands.longestPrefix(toLower(inList[0]), &len);
				if (found) {
					if (inList.size() > (unsigned)found->argC) {
						(*found)(inList);
					} else {
						cout<<"Please specify all "<<found->argC<<" arguments for "<<inList[0]<<endl;
						cout<<"Tokens received (e notation): "<<endl<<'[';
						for (auto i : inList) {
							cout<<i<<':';
						}
						cout<<']'<<endl;
					}
				} else {
					cout<<"Nothing found for "<<toLower(inList[0])<<endl;
			}},
			1, "Execute Command (Very sophisticated no-op)"s)),
		std::pair<string, functdata*>("p"s, new functdata(
			[](vector<string> inList){
				if (inList.size() == 1) {
				auto dump = commands.getIndex();
					for (unsigned i = 0; i != dump.size(); i++) {
						cout<<i<<": ";
						for (unsigned char j = 0; j != 128; j++) {
							if (dump[i][j].first || dump[i][j].second) {
								cout<<(char)(j?j:'/')<<dump[i][j].first<<(dump[i][j].second?'*':' ')<<' '<<std::flush;
							}
						}
						cout<<endl;
						if (inList[0][1] == 'l') {
							for (unsigned char j = 0; j != 128; j++) {
								cout<<dump[i][j].first<<(dump[i][j].second?'*':' ')<<' '<<std::flush;
								if (j%16 == 15)
									cout<<endl;
							}
							cout<<endl;
						}
					}
				} else {
					int len = 0;
					functdata *found = commands.longestPrefix(toLower(inList[1]), &len);
					if (found)
						cout<<"Matched \""<<inList[1].substr(0,len)<<'"'<<endl
							<<found->help<<" [takes "<<found->argC<<']'<<(found->call?' ':'!')<<endl;
					else
						cout<<len<<";Nothing found for "<<toLower(inList[1])<<endl;
			}},
			0, "Print/Probe input trie"s)),
		std::pair<string, functdata*>("m"s, new functdata(
			[](vector<string> inList){
				bool archimedian{true};
				unsigned cmapID{1};
				
				auto split = inList[0].find_first_of("0123456789abcdef");
				if (split == string::npos) {
					split = inList[0].size();
				}
				string fname{inList[0].substr(0,split)},
					cmapnum{inList[0].substr(split)};
				if (fname.find_first_of('l') != string::npos) {
					archimedian = false;
				}
				inList.insert(inList.begin()+1, cmapnum);
				inList[0] = "d"s + fname;
				drawMap(inList);
			},
			5, "Draw map (filename left upper width resolution)"s)),
		std::pair<string, functdata*>("d"s, new functdata(
			drawMap,
			6, "Draw map <cmap> <output.png> <left> <upper> <width> <res>"s)),
		std::pair<string, functdata*>("c"s, new functdata(
			[](vector<string> inList){
				int maxlen{0};
				for (auto& m : current_cmap) {
					(m.second.size() > maxlen)
						? maxlen = m.second.size()
						: false;
				}
				png::image<png::rgb_pixel> image(maxlen,current_cmap.size());
				for (int i = 0; i < (signed)current_cmap.size(); ++i) {
					auto cmap = current_cmap[i].second;
					for (int x = 0; x < maxlen; ++x) {
						auto v = x*cmap.size()/maxlen;
						image[i][x] = {cmap[v].red,cmap[v].green,cmap[v].blue};
					}
				}
				image.write(inList[1]);
			},
			1, "Print colormaps (<output.png>)"s)),
		std::pair<string, functdata*>("cr"s, new functdata(
			[](vector<string> inList){
				auto cmaps = read_cmap(inList[1]);
				int maxlen{0};
				for (auto& m : cmaps) {
					(m.second.size() > maxlen)
						? maxlen = m.second.size()
						: false;
				}
				png::image<png::rgb_pixel> image(maxlen,cmaps.size());
				for (int i = 0; i < (signed)cmaps.size(); ++i) {
					auto& cmap = cmaps[i].second;
					for (int x = 0; x < maxlen; ++x) {
						auto v = x*cmap.size()/maxlen;
						image[i][x] = {cmap[v].red,cmap[v].green,cmap[v].blue};
					}
				}
				image.write(inList[2]);
			},
			2, "Print colormaps from file <cmapfile> <output.png>"s)),
		std::pair<string, functdata*>("pc"s, new functdata(
			[]([[maybe_unused]] vector<string> inList){
				for (int i = 0; i < (signed)current_cmap.size(); ++i) {
					std::string name;
					png::palette cmap;
					std::tie(name,cmap) = current_cmap[i];
					cout<<":"<<name<<":"<<std::dec<<cmap.size()<<std::hex;
					for (int x = 0; x < (signed)cmap.size(); ++x) {
						if (x%8==0) {
							cout<<endl<<"\t";
						}
						cout<<std::setw(2)<<std::setfill('0')<<+cmap[x].red
							<<std::setw(2)<<std::setfill('0')<<+cmap[x].green
							<<std::setw(2)<<std::setfill('0')<<+cmap[x].blue
							<<" ";
					}
					cout<<endl;
				}
				cout<<std::dec;
			},
			0, "Formats colormaps"s)),
		std::pair<string, functdata*>("pcr"s, new functdata(
			[]([[maybe_unused]] vector<string> inList){
				auto cmaps = read_cmap(inList[1]);
				for (int i = 0; i < (signed)cmaps.size(); ++i) {
					std::string name;
					png::palette cmap;
					std::tie(name,cmap) = cmaps[i];
					for (int c = 0; c != (signed)name.size(); ++c) {
						if (name[c] == ':' || name[c] == '\\') {
							name.insert(c, "\\");
							++c;
						}
					}
					cout<<":"<<name<<":"<<std::dec<<cmap.size()<<std::hex;
					for (int x = 0; x < (signed)cmap.size(); ++x) {
						if (x%8==0) {
							cout<<endl<<"\t";
						}
						cout<<std::setw(2)<<std::setfill('0')<<+cmap[x].red
							<<std::setw(2)<<std::setfill('0')<<+cmap[x].green
							<<std::setw(2)<<std::setfill('0')<<+cmap[x].blue
							<<" ";
					}
					cout<<endl;
				}
				cout<<std::dec;
			},
			0, "Formats colormaps"s)),
		std::pair<string, functdata*>("rc"s, new functdata(
			[](vector<string> inList){
				try {
					current_cmap = read_cmap(inList[1]);
					cout<<"Read "<<current_cmap.size()<<" cmaps"<<endl;
					for (auto& map : current_cmap) {
						cout<<map.first<<": "<<map.second.size()<<" colors"<<endl;
					}
				} catch (malformed_file_error& e) {
					cout<<e.what()<<endl;
				}
			},
			1, "Read colormap <cmapfile>"s)),
		}}));
	
	vector<string> scriptedCommands;
	
	bool display_promt = true, display_promt_s = false, cont_after = false;
	
	for (int i = 1; i < argc; i++) {
		//cout<<"["<<i<<"] "<<argv[i]<<endl;
		if (argv[i][0] == '-') {
			for (int j = 1; argv[i][j] != '\0'; ++j) {
				if (argv[i][j] == 'q') { //quiet (Don't display prompts)
					display_promt = false;
				} else if (argv[i][j] == 'h') { //help
					//from strings.h
					cout<<HelpStr;
					Continue = false;
				//verbose (print timing info)
				} else if (argv[i][j] == 't') {
					verbose = true;
				//verbose (Simulate interactive display for scripted commands)
				} else if (argv[i][j] == 'v') {
					display_promt_s = true;
				} else if (argv[i][j] == 'V') {
					cout<<"Version "<<vnum<<endl;
					Continue = false;
				} else if (argv[i][j] == 'c') { //continue after scripted commands
					cont_after = true;
				} else if (argv[i][j] == '-') { //continue after scripted commands
					continue;
				} else {
					cerr<<"Note: ignored option "<<argv[i][j]<<std::endl;
				}
			}
		} else { //interpret other strings as commands to be run
			scriptedCommands.emplace_back(argv[i]);
		}
	}
	
	Continue = Continue & !scriptedCommands.size();
	
	for (auto i : scriptedCommands) {
		if (display_promt_s && display_promt) cout<<"> "<<i<<endl;
		runCommand(commands, i);
	}
	
	//Since both are simple variables short-circuiting doesn't really help, and
		//bools are guaranteed to only be 1 or 0
	Continue = Continue | cont_after;
	
	//Don't end when cin fails, clear and move on
	while (Continue) {
		//Actual input loop, large-scope try because all exceptions are outside my code
		while (cin && Continue) {
			try {
				if (display_promt)
					cout<<"> ";
				getline(cin, in);
				if (in.empty()) {
					continue;
				}
				runCommand(commands, in);
			} catch (const std::runtime_error& err) {
				cout<<"Failed to read input. "<<err.what()<<endl;
			} catch (const std::exception& err) {
				cout<<"Unknown error: "<<err.what()<<endl;
				throw err;
			} catch (...) {
				cout<<"Unknown error"<<endl;
				throw;
			}
		}
		if (!cin.eof())
			cin.clear();
		else {
			cout<<endl;
			break;
		}
	}
	return 0;
}

void runCommand(input_trie commands, string in)
{
	if (in.empty()) {
		return;
	}
	
	//Split input on word boundaries
	vector<string> inList = tokenizeInput(in);
	//inList[0] = toLower(inList[0]);
	std::transform(inList[0].begin(), inList[0].end(), inList[0].begin(), ::tolower);
	
	int len = 0;
	functdata *found = commands.longestPrefix(toLower(inList[0]), &len);
	if (found) {
		if (inList.size() > (unsigned)found->argC) {
			(*found)(inList);
		} else {
			cout<<"Please specify all "<<found->argC<<" arguments for "<<inList[0]<<endl;
			cout<<"arguments recieved: [";
			for (auto i : inList) {
				cout<<i<<':';
			}
			cout<<']'<<endl;
		}
	} else {
		cout<<"Nothing found for "<<toLower(inList[0])<<endl;
	}
}