#include "toonzqt/addfxcontextmenu.h"

// TnzQt includes
#include "toonzqt/fxselection.h"

// TnzLib includes
#include "toonz/toonzfolders.h"
#include "toonz/txsheethandle.h"
#include "toonz/tcolumnhandle.h"
#include "toonz/tframehandle.h"
#include "toonz/tfxhandle.h"
#include "toonz/fxcommand.h"
#include "toonz/txsheet.h"
#include "toonz/fxdag.h"
#include "toonz/tapplication.h"
#include "tw/stringtable.h"

// TnzBase includes
#include "texternfx.h"
#include "tmacrofx.h"
#include "tfxattributes.h"

// TnzCore includes
#include "tstream.h"
#include "tsystem.h"

// Qt includes
#include <QMenu>
#include <QAction>
#include <QStack>

#include "pluginhost.h"
#include <map>
#include <string>

std::map<std::string, PluginInformation *> plugin_dict_;

namespace
{

TFx *createFxByName(std::string fxId)
{
	if (fxId.find("_ext_") == 0)
		return TExternFx::create(fxId.substr(5));
	if (fxId.find("_plg_") == 0) {
		std::string id = fxId.substr(5);
		std::map<std::string, PluginInformation *>::iterator it = plugin_dict_.find(id);
		if (it != plugin_dict_.end()) {
			RasterFxPluginHost *plugin = new RasterFxPluginHost(it->second);
			plugin->notify();
			return plugin;
		}
		return NULL;
	} else {
		return TFx::create(fxId);
	}
}

//-----------------------------------------------------------------------------

TFx *createPresetFxByName(TFilePath path)
{
	std::string id = path.getParentDir().getName();
	TFx *fx = createFxByName(id);
	if (fx) {
		TIStream is(path);
		fx->loadPreset(is);
		fx->setName(path.getWideName());
	}
	return fx;
}

//-----------------------------------------------------------------------------

TFx *createMacroFxByPath(TFilePath path, TXsheet *xsheet)
{
	try {
		TIStream is(path);
		TPersist *p = 0;
		is >> p;
		TMacroFx *fx = dynamic_cast<TMacroFx *>(p);
		if (!fx)
			return 0;
		fx->setName(path.getWideName());
		//Assign a unic ID to each fx in the macro!
		if (!xsheet)
			return fx;
		FxDag *fxDag = xsheet->getFxDag();
		if (!fxDag)
			return fx;
		std::vector<TFxP> fxs;
		fxs = fx->getFxs();
		QMap<std::wstring, std::wstring> oldNewId;
		int i;
		for (i = 0; i < (int)fxs.size(); i++) {
			std::wstring oldId = fxs[i]->getFxId();
			fxDag->assignUniqueId(fxs[i].getPointer());
			std::wstring newId = fxs[i]->getFxId();
			oldNewId[oldId] = newId;

			//cambiando l'id degli effetti interni di una macro si rompono i legami tra il nome della porta
			//e la porta a cui e' legato: devo cambiare i nomei delle porte e rimapparli all'interno della macro
			int j;
			for (j = 0; j < fx->getInputPortCount(); j++) {
				QString inputName = QString::fromStdString(fx->getInputPortName(j));
				if (inputName.endsWith(QString::fromStdWString(oldId))) {
					QString newInputName = inputName;
					newInputName.replace(QString::fromStdWString(oldId), QString::fromStdWString(newId));
					fx->renamePort(inputName.toStdString(), newInputName.toStdString());
				}
			}
		}
		/* QStack<QPair<std::string, TFxPort*> > newPortNames;

      //Devo cambiare il nome alle porte: contengono l'id dei vecchi effetti
      for(i=fx->getInputPortCount()-1; i>=0; i--)
      {
        string oldPortName = fx->getInputPortName(i);
        string inFxOldId = oldPortName;
        inFxOldId.erase(0,inFxOldId.find_last_of("_")+1);
        assert(oldNewId.contains(toWideString(inFxOldId)));
        string inFxNewId = toString(oldNewId[toWideString(inFxOldId)]);
        string newPortName = oldPortName;
        newPortName.erase(newPortName.find_last_of("_")+1,newPortName.size()-1);
        newPortName.append(inFxNewId);
        TFxPort* fxPort = fx->getInputPort(i);
        newPortNames.append(QPair<std::string, TFxPort*>(newPortName,fxPort));
        fx->removeInputPort(oldPortName);
      }
      while(!newPortNames.isEmpty())
      {
        QPair<std::string, TFxPort*> newPort = newPortNames.pop();
        fx->addInputPort(newPort.first,*newPort.second);
      }*/

		return fx;
	} catch (...) {
		return 0;
	}
}

//-----------------------------------------------------------------------------

TFx *createFx(QAction *action, TXsheetHandle *xshHandle)
{
	TXsheet *xsh = xshHandle->getXsheet();
	QString text = action->data().toString();

	if (text.isEmpty())
		return 0;

	TFx *fx = 0;

	TFilePath path = TFilePath(text.toStdWString());

	if (TFileStatus(path).doesExist() && TFileStatus(path.getParentDir()).isDirectory()) {
		std::string folder = path.getParentDir().getName();
		if (folder == "macroFx") //have to load a Macro
			fx = createMacroFxByPath(path, xsh);
		else {
			folder = path.getParentDir().getParentDir().getName();
			if (folder == "presets") //have to load a preset
				fx = createPresetFxByName(path);
		}
	} else
		fx = createFxByName(text.toStdString());

	return fx;
}

} //namespace

//***************************************************
//
//AddFxContextMenu
//
//***************************************************

AddFxContextMenu::AddFxContextMenu()
	: QObject(), m_app(0), m_currentCursorScenePos(0, 0), m_againCommand(0)
{
	m_fxListPath = TFilePath(ToonzFolder::getProfileFolder() + "layouts" + "fxs" + "fxs.lst");
	m_presetPath = TFilePath(ToonzFolder::getFxPresetFolder() + "presets");

	m_insertMenu = new QMenu(tr("Insert FX"), 0);
	m_insertActionGroup = new QActionGroup(m_insertMenu);
	m_addMenu = new QMenu(tr("Add FX"), 0);
	m_addActionGroup = new QActionGroup(m_addMenu);
	m_replaceMenu = new QMenu(tr("Replace FX"), 0);
	m_replaceActionGroup = new QActionGroup(m_replaceMenu);

	connect(m_insertActionGroup, SIGNAL(triggered(QAction *)), this, SLOT(onInsertFx(QAction *)));
	connect(m_addActionGroup, SIGNAL(triggered(QAction *)), this, SLOT(onAddFx(QAction *)));
	connect(m_replaceActionGroup, SIGNAL(triggered(QAction *)), this, SLOT(onReplaceFx(QAction *)));

	fillMenus();
}

//---------------------------------------------------
static void clear_all_plugins()
{
	for (std::map<std::string, PluginInformation *>::iterator it = plugin_dict_.begin();
		 it != plugin_dict_.end(); ++it) {
		it->second->release();
	}
	plugin_dict_.clear();
}

AddFxContextMenu::~AddFxContextMenu()
{
	clear_all_plugins();
}

//---------------------------------------------------

void AddFxContextMenu::setApplication(TApplication *app)
{
	m_app = app;

	if (TFxHandle *fxHandle = app->getCurrentFx()) {
		connect(fxHandle, SIGNAL(fxPresetSaved()), this, SLOT(onFxPresetHandled()));
		connect(fxHandle, SIGNAL(fxPresetRemoved()), this, SLOT(onFxPresetHandled()));
	}
}

//---------------------------------------------------

void AddFxContextMenu::fillMenus()
{
	loadFxs();
	loadMacro();
}

//---------------------------------------------------

void scan_all_plugins(const std::string &basedir, QObject *listener)
{
	//clear_all_plugins();
	new PluginLoadController(basedir, listener);
}

void AddFxContextMenu::result(PluginInformation *pi)
{
	/* slot receives PluginInformation on the main thread たぶん */
	printf("AddFxContextMenu::result() pi:%p\n", pi);
	/* addfxcontextmenu.cpp の dict に登録する */
	if (pi)
		plugin_dict_.insert(std::pair<std::string, PluginInformation *>(pi->desc_->id_, pi));
	//RasterFxPluginHost* plug = new RasterFxPluginHost(pi);
	//pi->handler_->create(plug);
}

void AddFxContextMenu::fixup()
{
	loadFxPluginGroup();
}

void AddFxContextMenu::loadFxs()
{
	TIStream is(m_fxListPath);

	try {
		std::string tagName;
		if (is.matchTag(tagName) && tagName == "fxs") {
			loadFxGroup(&is);
			is.closeChild();
		}
	} catch (...) {
	}

	scan_all_plugins("", this);
}

//---------------------------------------------------

void AddFxContextMenu::loadFxPluginGroup()
{
	QString groupName = QString::fromStdString("Plugins");

	std::auto_ptr<QMenu> insertFxGroup(new QMenu(groupName, m_insertMenu));
	std::auto_ptr<QMenu> addFxGroup(new QMenu(groupName, m_addMenu));
	std::auto_ptr<QMenu> replaceFxGroup(new QMenu(groupName, m_replaceMenu));

	loadFxPlugins(insertFxGroup.get(), addFxGroup.get(), replaceFxGroup.get());

	if (!insertFxGroup->isEmpty())
		m_insertMenu->addMenu(insertFxGroup.release());
	if (!addFxGroup->isEmpty())
		m_addMenu->addMenu(addFxGroup.release());
	if (!replaceFxGroup->isEmpty())
		m_replaceMenu->addMenu(replaceFxGroup.release());
}

void AddFxContextMenu::loadFxGroup(TIStream *is)
{
	while (!is->eos()) {
		std::string tagName;
		if (is->matchTag(tagName)) {
			QString groupName = QString::fromStdString(tagName);

			std::auto_ptr<QMenu> insertFxGroup(new QMenu(groupName, m_insertMenu));
			std::auto_ptr<QMenu> addFxGroup(new QMenu(groupName, m_addMenu));
			std::auto_ptr<QMenu> replaceFxGroup(new QMenu(groupName, m_replaceMenu));

			loadFx(is, insertFxGroup.get(), addFxGroup.get(), replaceFxGroup.get());

			if (!insertFxGroup->isEmpty())
				m_insertMenu->addMenu(insertFxGroup.release());
			if (!addFxGroup->isEmpty())
				m_addMenu->addMenu(addFxGroup.release());
			if (!replaceFxGroup->isEmpty())
				m_replaceMenu->addMenu(replaceFxGroup.release());

			is->closeChild();
		}
	}
}

//---------------------------------------------------
void AddFxContextMenu::loadFxPlugins(QMenu *insertFxGroup, QMenu *addFxGroup, QMenu *replaceFxGroup)
{
	// list vendors
	std::vector<std::string> vendors;
	for (auto &&plugin : plugin_dict_) {
		PluginDescription *desc = plugin.second->desc_;
		vendors.push_back(desc->vendor_);
	}
	std::sort(std::begin(vendors), std::end(vendors));

	// add vendor folders
	std::map<std::string, QMenu *> insVendors;
	std::map<std::string, QMenu *> addVendors;
	std::map<std::string, QMenu *> repVendors;
	for (std::string vendor : vendors) {
		std::map<std::string, QMenu *>::iterator v = insVendors.find(vendor);
		if (v == insVendors.end()) {
			QString vendorQStr = QString::fromStdString(vendor);
			insVendors.insert(std::make_pair(vendor, insertFxGroup->addMenu(vendorQStr)));
			addVendors.insert(std::make_pair(vendor, addFxGroup->addMenu(vendorQStr)));
			repVendors.insert(std::make_pair(vendor, replaceFxGroup->addMenu(vendorQStr)));
		}
	}

	// add actions
	for (auto &&plugin : plugin_dict_) {
		PluginDescription *desc = plugin.second->desc_;
		QString label = QString::fromStdString(desc->name_);

		QAction *insertAction = new QAction(label, insertFxGroup);
		QAction *addAction = new QAction(label, addFxGroup);
		QAction *replaceAction = new QAction(label, replaceFxGroup);

		insertAction->setData(QVariant("_plg_" + QString::fromStdString(desc->id_)));
		addAction->setData(QVariant("_plg_" + QString::fromStdString(desc->id_)));
		replaceAction->setData(QVariant("_plg_" + QString::fromStdString(desc->id_)));

		(*insVendors.find(desc->vendor_)).second->addAction(insertAction);
		(*addVendors.find(desc->vendor_)).second->addAction(addAction);
		(*repVendors.find(desc->vendor_)).second->addAction(replaceAction);

		m_insertActionGroup->addAction(insertAction);
		m_addActionGroup->addAction(addAction);
		m_replaceActionGroup->addAction(replaceAction);
	}

	// sort actions
	auto const comp = [](QAction *lhs, QAction *rhs) {
		return lhs->text() < rhs->text();
	};

	for (auto &&ins : insVendors) {
		QList<QAction *> actions = ins.second->actions();
		ins.second->clear();
		qSort(actions.begin(), actions.end(), comp);
		ins.second->addActions(actions);
	}

	for (auto &&ins : addVendors) {
		QList<QAction *> actions = ins.second->actions();
		ins.second->clear();
		qSort(actions.begin(), actions.end(), comp);
		ins.second->addActions(actions);
	}

	for (auto &&ins : repVendors) {
		QList<QAction *> actions = ins.second->actions();
		ins.second->clear();
		qSort(actions.begin(), actions.end(), comp);
		ins.second->addActions(actions);
	}
}

void AddFxContextMenu::loadFx(TIStream *is, QMenu *insertFxGroup, QMenu *addFxGroup, QMenu *replaceFxGroup)
{
	while (!is->eos()) {
		std::string fxName;
		*is >> fxName;

		if (!fxName.empty()) {
			if (!loadPreset(fxName, insertFxGroup, addFxGroup, replaceFxGroup)) {
				QString translatedName = QString::fromStdWString(TStringTable::translate(fxName));

				QAction *insertAction = new QAction(translatedName, insertFxGroup);
				QAction *addAction = new QAction(translatedName, addFxGroup);
				QAction *replaceAction = new QAction(translatedName, replaceFxGroup);

				insertAction->setData(QVariant(QString::fromStdString(fxName)));
				addAction->setData(QVariant(QString::fromStdString(fxName)));
				replaceAction->setData(QVariant(QString::fromStdString(fxName)));

				insertFxGroup->addAction(insertAction);
				addFxGroup->addAction(addAction);
				replaceFxGroup->addAction(replaceAction);

				m_insertActionGroup->addAction(insertAction);
				m_addActionGroup->addAction(addAction);
				m_replaceActionGroup->addAction(replaceAction);
			}
		}
	}
}

//---------------------------------------------------

bool AddFxContextMenu::loadPreset(const std::string &name,
								  QMenu *insertFxGroup, QMenu *addFxGroup, QMenu *replaceFxGroup)
{
	TFilePath presetsFilepath(m_presetPath + name);
	if (TFileStatus(presetsFilepath).isDirectory()) {
		TFilePathSet presets = TSystem::readDirectory(presetsFilepath, false);
		if (!presets.empty()) {
			QMenu *inserMenu = new QMenu(QString::fromStdWString(TStringTable::translate(name)), insertFxGroup);
			insertFxGroup->addMenu(inserMenu);
			QMenu *addMenu = new QMenu(QString::fromStdWString(TStringTable::translate(name)), addFxGroup);
			addFxGroup->addMenu(addMenu);
			QMenu *replaceMenu = new QMenu(QString::fromStdWString(TStringTable::translate(name)), replaceFxGroup);
			replaceFxGroup->addMenu(replaceMenu);

			//This is a workaround to set the bold style to the first element of this menu
			//Setting a font directly to a QAction is not enought; style sheet definitions
			//preval over QAction font settings.
			inserMenu->setObjectName("fxMenu");
			addMenu->setObjectName("fxMenu");
			replaceMenu->setObjectName("fxMenu");

			QAction *insertAction = new QAction(QString::fromStdWString(TStringTable::translate(name)), inserMenu);
			QAction *addAction = new QAction(QString::fromStdWString(TStringTable::translate(name)), addMenu);
			QAction *replaceAction = new QAction(QString::fromStdWString(TStringTable::translate(name)), replaceMenu);

			insertAction->setCheckable(true);
			addAction->setCheckable(true);
			replaceAction->setCheckable(true);

			insertAction->setData(QVariant(QString::fromStdString(name)));
			addAction->setData(QVariant(QString::fromStdString(name)));
			replaceAction->setData(QVariant(QString::fromStdString(name)));

			inserMenu->addAction(insertAction);
			addMenu->addAction(addAction);
			replaceMenu->addAction(replaceAction);

			m_insertActionGroup->addAction(insertAction);
			m_addActionGroup->addAction(addAction);
			m_replaceActionGroup->addAction(replaceAction);

			for (TFilePathSet::iterator it2 = presets.begin(); it2 != presets.end(); ++it2) {
				TFilePath presetName = *it2;
				QString qPresetName = QString::fromStdWString(presetName.getWideName());

				insertAction = new QAction(qPresetName, inserMenu);
				addAction = new QAction(qPresetName, addMenu);
				replaceAction = new QAction(qPresetName, replaceMenu);

				insertAction->setData(QVariant(QString::fromStdWString(presetName.getWideString())));
				addAction->setData(QVariant(QString::fromStdWString(presetName.getWideString())));
				replaceAction->setData(QVariant(QString::fromStdWString(presetName.getWideString())));

				inserMenu->addAction(insertAction);
				addMenu->addAction(addAction);
				replaceMenu->addAction(replaceAction);

				m_insertActionGroup->addAction(insertAction);
				m_addActionGroup->addAction(addAction);
				m_replaceActionGroup->addAction(replaceAction);
			}
			return true;
		} else
			return false;
	} else
		return false;
}

//---------------------------------------------------

void AddFxContextMenu::loadMacro()
{
	TFilePath macroDir = m_presetPath + TFilePath("macroFx");
	try {
		if (TFileStatus(macroDir).isDirectory()) {
			TFilePathSet macros = TSystem::readDirectory(macroDir);
			if (macros.empty())
				return;

			QMenu *insertMacroMenu = new QMenu("Macro", m_insertMenu);
			QMenu *addMacroMenu = new QMenu("Macro", m_addMenu);
			QMenu *replaceMacroMenu = new QMenu("Macro", m_replaceMenu);

			m_insertMenu->addMenu(insertMacroMenu);
			m_addMenu->addMenu(addMacroMenu);
			m_replaceMenu->addMenu(replaceMacroMenu);

			for (TFilePathSet::iterator it = macros.begin(); it != macros.end(); ++it) {
				TFilePath macroPath = *it;
				QString name = QString::fromStdWString(macroPath.getWideName());

				QAction *insertAction = new QAction(name, insertMacroMenu);
				QAction *addAction = new QAction(name, addMacroMenu);
				QAction *replaceAction = new QAction(name, replaceMacroMenu);

				insertAction->setData(QVariant(QString::fromStdWString(macroPath.getWideString())));
				addAction->setData(QVariant(QString::fromStdWString(macroPath.getWideString())));
				replaceAction->setData(QVariant(QString::fromStdWString(macroPath.getWideString())));

				insertMacroMenu->addAction(insertAction);
				addMacroMenu->addAction(addAction);
				replaceMacroMenu->addAction(replaceAction);

				m_insertActionGroup->addAction(insertAction);
				m_addActionGroup->addAction(addAction);
				m_replaceActionGroup->addAction(replaceAction);
			}
		}
	} catch (...) {
	}
}

//---------------------------------------------------

void AddFxContextMenu::onInsertFx(QAction *action)
{
	if (action->isCheckable() && action->isChecked())
		action->setChecked(false);
	TFx *fx = createFx(action, m_app->getCurrentXsheet());
	if (fx) {
		QList<TFxP> fxs = m_selection->getFxs();
		QList<TFxCommand::Link> links = m_selection->getLinks();
		TFxCommand::insertFx(fx, fxs, links, m_app,
							 m_app->getCurrentColumn()->getColumnIndex(), m_app->getCurrentFrame()->getFrameIndex());
		m_app->getCurrentXsheet()->notifyXsheetChanged();
		//memorize the latest operation
		m_app->getCurrentFx()->setPreviousActionString(QString("I ") + action->data().toString());
	}
}

//---------------------------------------------------

void AddFxContextMenu::onAddFx(QAction *action)
{
	if (action->isCheckable() && action->isChecked())
		action->setChecked(false);

	TFx *fx = createFx(action, m_app->getCurrentXsheet());
	if (fx) {
		QList<TFxP> fxs = m_selection->getFxs();
		// try to add node at cursor position
		if (m_currentCursorScenePos.x() != 0 || m_currentCursorScenePos.y() != 0) {
			fx->getAttributes()->setDagNodePos(TPointD(m_currentCursorScenePos.x(), m_currentCursorScenePos.y()));
			m_currentCursorScenePos.setX(0);
			m_currentCursorScenePos.setY(0);
		}

		TFxCommand::addFx(fx, fxs, m_app,
						  m_app->getCurrentColumn()->getColumnIndex(), m_app->getCurrentFrame()->getFrameIndex());

		m_app->getCurrentXsheet()->notifyXsheetChanged();
		//memorize the latest operation
		m_app->getCurrentFx()->setPreviousActionString(QString("A ") + action->data().toString());
	}
}

//---------------------------------------------------

void AddFxContextMenu::onReplaceFx(QAction *action)
{
	if (action->isCheckable() && action->isChecked())
		action->setChecked(false);
	TFx *fx = createFx(action, m_app->getCurrentXsheet());
	if (fx) {
		QList<TFxP> fxs = m_selection->getFxs();
		TFxCommand::replaceFx(fx, fxs, m_app->getCurrentXsheet(), m_app->getCurrentFx());
		m_app->getCurrentXsheet()->notifyXsheetChanged();
		//memorize the latest operation
		m_app->getCurrentFx()->setPreviousActionString(QString("R ") + action->data().toString());
	}
}

//---------------------------------------------------

void AddFxContextMenu::onFxPresetHandled()
{
	m_insertMenu->clear();
	m_addMenu->clear();
	m_replaceMenu->clear();
	fillMenus();
}

//---------------------------------------------------
/*! repeat the last fx creation command done in the schematic.
    arrgument "command" is sum of the ids of available commands(Insert, Add, Replace)
*/
QAction *AddFxContextMenu::getAgainCommand(int command)
{
	QString commandName = m_app->getCurrentFx()->getPreviousActionString();
	//return if the last action is not registered
	if (commandName.isEmpty())
		return 0;

	//classify action by commandName
	Commands com;
	QString commandStr;
	if (commandName.startsWith("I ")) {
		com = Insert;
		commandStr = tr("Insert ");
	} else if (commandName.startsWith("A ")) {
		com = Add;
		commandStr = tr("Add ");
	} else if (commandName.startsWith("R ")) {
		com = Replace;
		commandStr = tr("Replace ");
	} else
		return 0;

	//return if the action is not available
	if (!(command & com))
		return 0;

	QString fxStr = commandName.right(commandName.size() - 2);
	QString translatedCommandName = commandStr + QString::fromStdWString(TStringTable::translate(fxStr.toStdString()));
	//return the action if the command is the exactly same
	if (m_againCommand && translatedCommandName == m_againCommand->text())
		return m_againCommand;

	//create an action
	if (!m_againCommand) {
		m_againCommand = new QAction(translatedCommandName, 0);
		m_againCommand->setData(QVariant(fxStr));
		connect(m_againCommand, SIGNAL(triggered()), this, SLOT(onAgainCommand()));
	}
	//compare the m_againCommand's name and commandName
	else if (translatedCommandName != m_againCommand->text()) {
		//change the action name
		m_againCommand->setText(translatedCommandName);
		m_againCommand->setData(QVariant(fxStr));
	}
	return m_againCommand;
}

//---------------------------------------------------
/*! change the command behavior according to the command name
*/
void AddFxContextMenu::onAgainCommand()
{
	// TODO: 日本語インタフェースの場合、対応が必要 2016/1/8 shun_iwasawa
	if (m_againCommand->text().startsWith("Insert")) {
		onInsertFx(m_againCommand);
	} else if (m_againCommand->text().startsWith("Add")) {
		onAddFx(m_againCommand);
	} else if (m_againCommand->text().startsWith("Replace")) {
		onReplaceFx(m_againCommand);
	}
}