#include "Compass.h"

Compass::Compass() {
}

Compass::~Compass() {
}

void Compass::perform(int argc, char* argv[]) {
	// Figure out action
	if (argc < 2) {
		std::cerr << ERR_NO_ACTION << std::endl;
		std::cout << "Run 'compass help' for more information." << std::endl;
		return;
	}

	try {
		std::string action(argv[1]);

		// Need Help?
		if (action == "help") {
			initNoArgs(argc, argv);
			help();
		}

		// Supported Actions
		else if (action == "initialize") {
			initNoArgs(argc, argv);
			initialize();
		}

		else if (action == "create") {
			std::string service = initOneArg(argc, argv);
			create(service);
		}

		else if (action == "walk" && ALLOW_WALK) {
			initNoArgs(argc, argv);
			walk();
		}

		else if (action == "rekey") {
			initNoArgs(argc, argv);
			rekey();
		}

		else if (action == "get") {
			std::string service = initOneArg(argc, argv);
			get(service);
		}

		else if (action == "delete") {
			std::string service = initOneArg(argc, argv);
			_delete(service);
		}

		else if (action == "show") {
			initNoArgs(argc, argv);
			show("", false);
		}

		else if (action == "show1") {
			std::string service = initOneArg(argc, argv);
			show(service, false);
		}

		else if (action == "showlike") {
			std::string service = initOneArg(argc, argv);
			show(service, true);
		}

		else if (action == "random") {
			initNoArgs(argc, argv);
			random();
		}

		else if (action == "markreset") {
			initNoArgs(argc, argv);
			markReset("");
		}

		else if (action == "markreset1") {
			std::string service = initOneArg(argc, argv);
			markReset(service);
		}

		else if (action == "checkreset") {
			initNoArgs(argc, argv);
			checkReset("");
		}

		else if (action == "checkreset1") {
			std::string service = initOneArg(argc, argv);
			checkReset(service);
		}

		else if (action == "ruser") {
			std::string service = initOneArg(argc, argv);
			ruser(service);
		}

		else if (action == "rpass") {
			std::string service = initOneArg(argc, argv);
			rpass(service);
		}

		else {
			std::cerr << ERR_INVALID_ACTION << std::endl;
			std::cout << "Run 'compass help' for more information." << std::endl;
		}
	}
	catch (const int e) {
		std::cerr << ERR_GENERIC << std::endl;
	}
	catch (const double e) {
		std::cerr << ERR_INPUT << std::endl;
	}
	catch (const bool e) {
		// do nothing (message was produced by called function)
	}
}

void Compass::initNoArgs(int argc, char* argv[]) {
	opt.parseArgv(2, argc, argv);
}

std::string Compass::initOneArg(int argc, char* argv[]) {
	if (argc < 3) {
		std::cerr << ERR_REQ_PARAMS_MISSING << std::endl;
		throw false;
	}

	std::string arg(argv[2]);
	opt.checkString(arg);
	opt.parseArgv(3, argc, argv);
	return arg;
}

void Compass::showResult(Credential c) {
	std::cout << "Username is: " << c.username << std::endl;

	if (opt.pp)
		std::cout << "Password is: " << c.password << std::endl;
	else
		pasteClipboard(c.password, !opt.cc, false);
}

void Compass::pasteClipboard(std::string text, bool clean, bool recur) {
	#ifdef WIN32

		if (OpenClipboard(NULL)) {
			HGLOBAL clipbuffer;
			char* buffer;
			EmptyClipboard();
			const char* source = text.c_str();
			clipbuffer = GlobalAlloc(GMEM_DDESHARE, strlen(source) + 1);
			buffer = (char*)GlobalLock(clipbuffer);
			strcpy(buffer, LPCSTR(source));
			GlobalUnlock(clipbuffer);
			SetClipboardData(CF_TEXT, clipbuffer);
			CloseClipboard();
			std::cout << MSG_CLIPBOARD << std::endl;
		}
		else {
			std::cerr << ERR_CLIPBOARD << std::endl;
		}

	#else
	#ifdef __APPLE__

		std::string sink = "pbcopy";

	#else

		std::string sink = "xclip -selection c";

	#endif // __APPLE__

		// Mac/Linux procedure
		std::string cmd = "printf %s '" + text + "' | " + sink;
		system(cmd.c_str());
		if (!recur) std::cout << MSG_CLIPBOARD << std::endl;
		else exit(EXIT_SUCCESS);

		// Cleanup
		if (clean) {
			pid_t pid = fork();
			if (pid == 0) {
				usleep(SECONDS_PSWD_CLEANUP * 1000000);
				pasteClipboard(CLEAN_STRING, clean, true);
			}
			else if (pid < 0) {
				std::cerr << WARN_CANT_CLEANUP << std::endl;
			}
		}

	#endif // WIN32

	if (!clean) std::cout << WARN_WONT_CLEANUP << std::endl;
}

void Compass::help() {
	Help h;
	h.dispGeneral();
}

void Compass::initialize() {
	std::cout << "WARNING: This will destroy the keychain (if one exists) at '" << opt.dr << "'" << std::endl;
	std::cout << "You may still abort this action by pressing CTRL-C now. If you wish to procede, create a new master password for your keychain below..." << std::endl;
	std::cout << std::endl;

	Keychain* kc = Keychain::newKeychain(opt.dr);
	kc->saveKeychain();
	delete kc;
}

void Compass::create(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	Credential c;

	if (opt.user != "" && opt.pass != "")
		c = kc->create(service, opt.user, opt.pass);

	else if (opt.user != "")
		c = kc->create(service, opt.user, opt.getPasswordSpec());

	else if (opt.pass != "")
		c = kc->create(service, opt.pass);

	else
		c = kc->create(service, opt.getPasswordSpec());

	std::cout << "Created new credential for " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;
	showResult(c);

	kc->saveKeychain();
	delete kc;
}

void Compass::walk() {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	kc->walk();
	delete kc;
}

void Compass::rekey() {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	Cryptor::rekey();

	std::cout << "Enter a new master password for your keychain... (CTRL-C to abort)" << std::endl;
	kc->saveKeychain();
	delete kc;
}

void Compass::get(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	Credential c;

	if (opt.cn != -1)
		c = kc->get(service, opt.cn);

	else if (opt.user != "")
		c = kc->get(service, opt.user);

	else
		c = kc->get(service);

	std::cout << "Found from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;
	showResult(c);

	if (c.reset) {
		std::cout << "************************************************" << std::endl;
		std::cout << "* Notice: this credential is marked for reset! *" << std::endl;
		std::cout << "************************************************" << std::endl;
	}

	delete kc;
}

void Compass::_delete(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	bool b;

	if (opt.cn != -1)
		b = kc->_delete(service, opt.cn);

	else if (opt.user != "")
		b = kc->_delete(service, opt.user);

	else
		b = kc->_delete(service);

	if (b)
		std::cout << service << "(" << kc->getServiceCount(service) << ") has been fully removed from the keychain." << std::endl;
	else
		std::cout << service << "(" << kc->getServiceCount(service) << ") has been partially removed from the keychain." << std::endl;

	kc->saveKeychain();
	delete kc;
}

void Compass::show(std::string service, bool like) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);

	if (service == "")
		kc->show();

	else if (like)
		kc->showlike(service);

	else
		kc->show1(service);

	delete kc;
}

void Compass::random() {
	std::cout << Cryptor::createRandomPassword(opt.getPasswordSpec()) << std::endl;
}

void Compass::markReset(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	int i;

	if (service == "")
		i = kc->markResetAll();

	else
		i = kc->markResetOne(service);

	std::cout << "Marked " << i << " credentials for reset." << std::endl;

	kc->saveKeychain();
	delete kc;
}

void Compass::checkReset(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	int i;

	if (service == "")
		i = kc->checkResetAll();

	else
		i = kc->checkResetOne(service);

	std::cout << i << " credentials are marked for reset." << std::endl;
	delete kc;
}

void Compass::ruser(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	Credential c;

	if (opt.user == "") {
		if (opt.cn == -1)
			c = kc->ruser(service);

		else
			c = kc->ruser(service, opt.cn);
	}

	else {
		if (opt.cn == -1)
			c = kc->ruser(service, opt.user);

		else
			c = kc->ruser(service, opt.user, opt.cn);
	}

	std::cout << "Altered from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;
	showResult(c);

	kc->saveKeychain();
	delete kc;
}

void Compass::rpass(std::string service) {
	Keychain* kc = Keychain::loadKeychain(opt.dr);
	Credential c;

	if (opt.pass == "") {
		if (opt.user == "" && opt.cn == -1)
			c = kc->rpass(service, opt.getPasswordSpec());

		else if (opt.cn != -1)
			c = kc->rpass(service, opt.getPasswordSpec(), opt.cn);

		else
			c = kc->rpass(service, opt.getPasswordSpec(), opt.user);
	}

	else {
		if (opt.user == "" && opt.cn == -1)
			c = kc->rpass(service, opt.pass);

		else if (opt.cn != -1)
			c = kc->rpass(service, opt.pass, opt.cn);

		else
			c = kc->rpass(service, opt.pass, opt.user);
	}

	std::cout << "Altered from " << service << "(" << kc->getServiceCount(service) << ")" << std::endl;
	showResult(c);

	kc->saveKeychain();
	delete kc;
}