/*
This file is part of Telegram Desktop,
the official desktop version of Telegram messaging app, see https://telegram.org

Telegram Desktop is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

It is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.

In addition, as a special exception, the copyright holders give permission
to link the code of portions of this program with the OpenSSL library.

Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
Copyright (c) 2014-2016 John Preston, https://desktop.telegram.org
*/
#include "stdafx.h"

#include "shortcuts.h"

#include "window.h"
#include "passcodewidget.h"
#include "mainwidget.h"

namespace ShortcutCommands {
	typedef void(*Handler)();

	void lock_telegram() {
		if (Window *w = App::wnd()) {
			if (App::passcoded()) {
				w->passcodeWidget()->onSubmit();
			} else if (cHasPasscode()) {
				w->setupPasscode(true);
			}
		}
	}

	void minimize_telegram() {
		if (Window *w = App::wnd()) {
			if (cWorkMode() == dbiwmTrayOnly) {
				w->minimizeToTray();
			} else {
				w->setWindowState(Qt::WindowMinimized);
			}
		}
	}

	void close_telegram() {
		if (!Ui::hideWindowNoQuit()) {
			if (Window *w = App::wnd()) {
				App::wnd()->close();
			}
		}
	}

	void quit_telegram() {
		App::quit();
	}

	//void start_stop_recording() {

	//}

	//void cancel_recording() {

	//}

	void media_play() {
		if (MainWidget *m = App::main()) {
			m->player()->playPressed();
		}
	}

	void media_pause() {
		if (MainWidget *m = App::main()) {
			m->player()->pausePressed();
		}
	}

	void media_playpause() {
		if (MainWidget *m = App::main()) {
			m->player()->playPausePressed();
		}
	}

	void media_stop() {
		if (MainWidget *m = App::main()) {
			m->player()->stopPressed();
		}
	}

	void media_previous() {
		if (MainWidget *m = App::main()) {
			m->player()->prevPressed();
		}
	}

	void media_next() {
		if (MainWidget *m = App::main()) {
			m->player()->nextPressed();
		}
	}

	void search() {
		if (MainWidget *m = App::main()) {
			m->cmd_search();
		}
	}

	void previous_chat() {
		if (MainWidget *m = App::main()) {
			m->cmd_previous_chat();
		}
	}

	void next_chat() {
		if (MainWidget *m = App::main()) {
			m->cmd_next_chat();
		}
	}

	// other commands here

}

inline bool qMapLessThanKey(const ShortcutCommands::Handler &a, const ShortcutCommands::Handler &b) {
	return a < b;
}

namespace Shortcuts {

	struct DataStruct;
	DataStruct *DataPtr = nullptr;

	void _createCommand(const QString &command, ShortcutCommands::Handler handler);
	QKeySequence _setShortcut(const QString &keys, const QString &command);
	struct DataStruct {
		DataStruct() {
			t_assert(DataPtr == nullptr);
			DataPtr = this;

#define DeclareAlias(keys, command) _setShortcut(qsl(keys), qsl(#command))
#define DeclareCommand(keys, command) _createCommand(qsl(#command), ShortcutCommands::command); DeclareAlias(keys, command)

			DeclareCommand("ctrl+w", close_telegram);
			DeclareAlias("ctrl+f4", close_telegram);
			DeclareCommand("ctrl+l", lock_telegram);
			DeclareCommand("ctrl+m", minimize_telegram);
			DeclareCommand("ctrl+q", quit_telegram);

			//DeclareCommand("ctrl+r", start_stop_recording);
			//DeclareCommand("ctrl+shift+r", cancel_recording);
			//DeclareCommand("media record", start_stop_recording);

			DeclareCommand("media play", media_play);
			DeclareCommand("media pause", media_pause);
			DeclareCommand("toggle media play/pause", media_playpause);
			DeclareCommand("media stop", media_stop);
			DeclareCommand("media previous", media_previous);
			DeclareCommand("media next", media_next);

			DeclareCommand("ctrl+f", search);
			DeclareAlias("search", search);
			DeclareAlias("find", search);

			DeclareCommand("ctrl+pgdown", next_chat);
			DeclareAlias("alt+down", next_chat);
			DeclareCommand("ctrl+pgup", previous_chat);
			DeclareAlias("alt+up", previous_chat);
			if (cPlatform() == dbipMac || cPlatform() == dbipMacOld) {
				DeclareAlias("meta+tab", next_chat);
				DeclareAlias("meta+shift+tab", previous_chat);
				DeclareAlias("meta+backtab", previous_chat);
			} else {
				DeclareAlias("ctrl+tab", next_chat);
				DeclareAlias("ctrl+shift+tab", previous_chat);
				DeclareAlias("ctrl+backtab", previous_chat);
			}

			// other commands here

#undef DeclareCommand
#undef DeclareAlias
		}
		QStringList errors;

		QMap<QString, ShortcutCommands::Handler> commands;
		QMap<ShortcutCommands::Handler, QString> commandnames;

		QMap<QKeySequence, QShortcut*> sequences;
		QMap<int, ShortcutCommands::Handler> handlers;
	};

	void _createCommand(const QString &command, ShortcutCommands::Handler handler) {
		t_assert(DataPtr != nullptr);
		t_assert(!command.isEmpty());

		DataPtr->commands.insert(command, handler);
		DataPtr->commandnames.insert(handler, command);
	}

	QKeySequence _setShortcut(const QString &keys, const QString &command) {
		t_assert(DataPtr != nullptr);
		t_assert(!command.isEmpty());
		if (keys.isEmpty()) return QKeySequence();

		QKeySequence seq(keys, QKeySequence::PortableText);
		if (seq.isEmpty()) {
			DataPtr->errors.push_back(qsl("Could not derive key sequence '%1'!").arg(keys));
		} else {
			QMap<QString, ShortcutCommands::Handler>::const_iterator it = DataPtr->commands.constFind(command);
			if (it == DataPtr->commands.cend()) {
				LOG(("Warning: could not find shortcut command handler '%1'").arg(command));
			} else {
				QShortcut *shortcut(new QShortcut(seq, App::wnd(), nullptr, nullptr, Qt::ApplicationShortcut));
				int shortcutId = shortcut->id();
				if (!shortcutId) {
					DataPtr->errors.push_back(qsl("Could not create shortcut '%1'!").arg(keys));
				} else {
					QMap<QKeySequence, QShortcut*>::iterator seqIt = DataPtr->sequences.find(seq);
					if (seqIt == DataPtr->sequences.cend()) {
						seqIt = DataPtr->sequences.insert(seq, shortcut);
					} else {
						DataPtr->handlers.remove(seqIt.value()->id());
						delete seqIt.value();
						seqIt.value() = shortcut;
					}
					DataPtr->handlers.insert(shortcutId, it.value());
				}
			}
		}
		return seq;
	}

	QKeySequence _removeShortcut(const QString &keys) {
		t_assert(DataPtr != nullptr);
		if (keys.isEmpty()) return QKeySequence();

		QKeySequence seq(keys, QKeySequence::PortableText);
		if (seq.isEmpty()) {
			DataPtr->errors.push_back(qsl("Could not derive key sequence '%1'!").arg(keys));
		} else {
			QMap<QKeySequence, QShortcut*>::iterator seqIt = DataPtr->sequences.find(seq);
			if (seqIt != DataPtr->sequences.cend()) {
				DataPtr->handlers.remove(seqIt.value()->id());
				delete seqIt.value();
				DataPtr->sequences.erase(seqIt);
			}
		}
		return seq;
	}

	void start() {
		t_assert(Global::started());

		new DataStruct();

		QJsonArray shortcuts;
		OrderedSet<QKeySequence> notfound;
		QFile f(cWorkingDir() + qsl("tdata/shortcuts.json"));
		if (f.exists()) {
			if (f.open(QIODevice::ReadOnly)) {
				QJsonParseError error = { 0, QJsonParseError::NoError };
				QJsonDocument doc = QJsonDocument::fromJson(f.readAll(), &error);
				if (error.error != QJsonParseError::NoError) {
					DataPtr->errors.push_back(qsl("Failed to parse '%1'! Error: %2").arg(f.fileName()).arg(error.errorString()));
				} else if (!doc.isArray()) {
					DataPtr->errors.push_back(qsl("Failed to parse '%1'! Error: array expected").arg(f.fileName()));
				} else {
					for (QMap<QKeySequence, QShortcut*>::const_iterator i = DataPtr->sequences.cbegin(), e = DataPtr->sequences.cend(); i != e; ++i) {
						notfound.insert(i.key());
					}

					shortcuts = doc.array();
					int limit = ShortcutsCountLimit;
					for (QJsonArray::const_iterator i = shortcuts.constBegin(), e = shortcuts.constEnd(); i != e; ++i) {
						if (!i->isObject()) {
							DataPtr->errors.push_back(qsl("Bad entry in '%1'! Error: object expected").arg(f.fileName()));
						} else {
							QKeySequence seq;
							QJsonObject entry(i->toObject());
							QJsonObject::const_iterator keys = entry.constFind(qsl("keys")), command = entry.constFind(qsl("command"));
							if (keys == entry.constEnd() || command == entry.constEnd() || !keys->isString() || (!command->isString() && !command->isNull())) {
								DataPtr->errors.push_back(qsl("Bad entry in '%1'! Error: {\"keys\": \"..\", \"command\": [ \"..\" | null ]} expected").arg(f.fileName()));
							} else if (command->isNull()) {
								seq = _removeShortcut(keys->toString());
							} else {
								seq = _setShortcut(keys->toString(), command->toString());
							}
							if (!--limit) {
								DataPtr->errors.push_back(qsl("Too many entries in '%1'!").arg(f.fileName()));
								break;
							} else if (DataPtr->errors.isEmpty()) {
								notfound.remove(seq);
							}
						}
					}
				}
				f.close();
			} else {
				DataPtr->errors.push_back(qsl("Could not read '") + f.fileName() + qsl("'!"));
			}
		}
		if (DataPtr->errors.isEmpty() && (shortcuts.isEmpty() || !notfound.isEmpty()) && f.open(QIODevice::WriteOnly)) {
			for (OrderedSet<QKeySequence>::const_iterator i = notfound.cbegin(), e = notfound.cend(); i != e; ++i) {
				QMap<QKeySequence, QShortcut*>::const_iterator s = DataPtr->sequences.constFind(i.key());
				if (s != DataPtr->sequences.cend()) {
					QMap<int, ShortcutCommands::Handler>::const_iterator h = DataPtr->handlers.constFind(s.value()->id());
					if (h != DataPtr->handlers.cend()) {
						QMap<ShortcutCommands::Handler, QString>::const_iterator n = DataPtr->commandnames.constFind(h.value());
						if (n != DataPtr->commandnames.cend()) {
							QJsonObject entry;
							entry.insert(qsl("keys"), i.key().toString().toLower());
							entry.insert(qsl("command"), n.value());
							shortcuts.append(entry);
						}
					}
				}
			}

			QJsonDocument doc;
			doc.setArray(shortcuts);
			f.write(doc.toJson(QJsonDocument::Indented));
			f.close();
		}
	}

	const QStringList &errors() {
		t_assert(DataPtr != nullptr);
		return DataPtr->errors;
	}

	void launch(int shortcutId) {
		t_assert(DataPtr != nullptr);

		QMap<int, ShortcutCommands::Handler>::const_iterator it = DataPtr->handlers.constFind(shortcutId);
		if (it != DataPtr->handlers.cend()) {
			(*it.value())();
		}
	}

	void launch(const QString &command) {
		t_assert(DataPtr != nullptr);

		QMap<QString, ShortcutCommands::Handler>::const_iterator it = DataPtr->commands.constFind(command);
		if (it != DataPtr->commands.cend()) {
			(*it.value())();
		}
	}

	void finish() {
		delete DataPtr;
		DataPtr = nullptr;
	}

}