#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());
        }
      }
    }

    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();
}

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

static 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::unique_ptr<QMenu> insertFxGroup(new QMenu(groupName, m_insertMenu));
  std::unique_ptr<QMenu> addFxGroup(new QMenu(groupName, m_addMenu));
  std::unique_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::unique_ptr<QMenu> insertFxGroup(new QMenu(groupName, m_insertMenu));
      std::unique_ptr<QMenu> addFxGroup(new QMenu(groupName, m_addMenu));
      std::unique_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);
  }
}