/**
  * @RDE_HEADER@
  * from FileManangement - Folder model
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Iuliana Ojog, SLSE, 2017-07-11
  */

#include "lts_dialog_model_folder.h"
#include <nglwidgets/ngltreeitem.h>

#include <FileManagement/flm_translation.h>

//#include <QDebug>

namespace LtsWidgets
{
//-----------------------------------------------------------------------------------------------------------
// DialogModelFolder
//-----------------------------------------------------------------------------------------------------------
DialogModelFolder::DialogModelFolder(QObject *parent
                                     , NGLTreeItem *rootItem
                                     , const std::shared_ptr<FileManagement::Interface::Dtc> &dtc
                                     , const std::shared_ptr<ExerciseManagement::Interface::Dtc> &exmDtc
                                     , const QStringList &header)
  : NGLTreeModel(parent, rootItem, header)
  , m_flmDtc(dtc)
  , m_exmDtc(exmDtc)
{
}

//-----------------------------------------------------------------------------------------------------------
QVariant DialogModelFolder::headerData(int section, Qt::Orientation orientation, int role) const
{
  if (role == Qt::DisplayRole && orientation == Qt::Horizontal)
  {
    switch (section)
    {
    case ID:
      return tr("Id");
    case NAME:
      return tr("Folder");
    default:
      break;
    }
  }
  return NGLTreeModel::headerData(section, orientation, role);
}

//-----------------------------------------------------------------------------------------------------------
QVariant DialogModelFolder::data(const QModelIndex &index, int role) const
{
  if (!index.isValid())
    return {};

  NGLTreeItem *item = privateData(index);
  if (item->data().canConvert<FileManagement::Folder>())
  {
    FileManagement::Folder folderData = item->data().value<FileManagement::Folder>();

    switch(role)
    {
    case Qt::DisplayRole:
    {
      switch (index.column())
      {
      case NAME:
        return getFolderName(folderData);
      case ID:
        return folderData.id();
      default:
        break;
      }
    }
    case Qt::UserRole:
      return folderData.id();
    case Qt::DecorationRole:
    {
      if (folderData.isHidden())
        return item->icon();
      break;
    }
    default:
      break;
    }
  }
  return {};
}

//-----------------------------------------------------------------------------------------------------------
FileManagement::Folder DialogModelFolder::getFolder(const QModelIndex &idx) const
{
  return privateData(idx)->data().value<FileManagement::Folder>();
}

//-----------------------------------------------------------------------------------------------------------
QString DialogModelFolder::getFolderName(const FileManagement::Folder &folderData) const
{
  QString folderName = QString::fromStdString(folderData.name());
  if (folderName.startsWith(QStringLiteral("$")))
  {
    folderName.remove(0,1);
    if (folderName.startsWith(QStringLiteral("EX_")))
    {
      int exId = folderName.remove(0,3).toInt();
      return QString::fromStdString(m_exmDtc->exercise(exId).name());
    }
    if (folderName.startsWith(QStringLiteral("PH_")))
    {
      int phId = folderName.remove(0,3).toInt();
      return QString::fromStdString(m_exmDtc->exercisePhase(phId).name());
    }
    return FileManagement::trFolderName(folderName.toStdString());
  }
  return FileManagement::trFolderName(folderName.toStdString());
}

//-----------------------------------------------------------------------------------------------------------
bool DialogModelFolder::isRowRemovable(NGLTreeItem *parentItem) const
{
  if (!parentItem)
    return false;
  FileManagement::Folder folderData = parentItem->data().value<FileManagement::Folder>();
  return folderData.isEditable();
}

//-----------------------------------------------------------------------------------------------------------
bool DialogModelFolder::initModel(int exerciseId, const QVector<FileManagement::MainFolder> &mainFolders)
{
  NGLTreeModel::resetModel();

  QMap<qint32,FileManagement::Folder> tmap;
  QHash<int, QModelIndex> parentItemIndexCache;
  QVector<int> idsToExclude;

  auto cbFolders = [exerciseId, &tmap, &idsToExclude, &mainFolders] (const FileManagement::Folder &data)
  {
    if (data.isDeleted())
      return;
    if (exerciseId > 0)
    {
      QString folderName = QString::fromStdString(data.name());
      if (folderName.startsWith(QStringLiteral("$EX_")))
      {
        int exId = folderName.remove(0, 4).toInt();
        if (exId != exerciseId)
          idsToExclude.append(data.id());
      }
    }
    FileManagement::MainFolder mainFolder = static_cast<FileManagement::MainFolder::type>(data.path()[0]);
    //qCritical() << "Folder = " << QString::fromStdString(data.name()) << ", mainFolder = " << mainFolder.underlying();
    if (!mainFolders.contains(mainFolder))
    {
     // qCritical() << "Excluded id = " << data.id();
      idsToExclude.append(data.id());
    }

    tmap.insert(data.id(), data);
  };
  m_flmDtc->folders(cbFolders);

  parentItemIndexCache.insert(0,  QModelIndex());   //root

  for (QMap<qint32, FileManagement::Folder>::const_iterator iter = tmap.constBegin();
       iter != tmap.constEnd(); ++iter)
  {
    int folderId = iter.key();
    int parentId = 0;
    FileManagement::Folder fld = iter.value();

    if (fld.path().size() > 1)
    {
      parentId = fld.path().at(fld.path().size() - 2);
      int idPos2 = fld.path().at(1);
      if (idsToExclude.contains(idPos2))
        continue;
    }
    else // main Folders
    {
      if (idsToExclude.contains(fld.id()))
        continue;
    }

    QVariant data;
    data.setValue(fld);

    if (!parentItemIndexCache.contains(parentId))
    {
      return false;
    }
    else
    {
      QModelIndex pIndex = parentItemIndexCache.value(parentId);
      int rowPos = -1;
      onInsertRow(rowPos, pIndex, fld);
      parentItemIndexCache.insert(folderId, index(rowPos, 0, pIndex));
    }
  }

  return true;
}

//----------------------------------------------------------------------------
bool DialogModelFolder::onInsertRow(int &row, const QModelIndex &parent, const FileManagement::Folder &folderData)
{
  row = rowCount(parent);
  QVariant data;
  data.setValue(folderData);
  QIcon icon;
  if (folderData.isHidden())
    icon = QIcon(QLatin1String(":/style/lock.svg"));
  NGLTreeItem *item = new NGLTreeItem(data, icon);
  item->setModified(true);

  return NGLTreeModel::insertRow(row, parent, item);
}

//----------------------------------------------------------------------------
bool DialogModelFolder::onUpdateRow(const QModelIndex &index, const FileManagement::Folder &folderData)
{
  QVariant data;
  data.setValue(folderData);

  QIcon icon;
  if (folderData.isHidden())
    icon = QIcon(QLatin1String(":/style/lock.svg"));
  return setItemData(index, data, icon);
}

//----------------------------------------------------------------------------
bool DialogModelFolder::onDeleteRow(const QModelIndex &idx, bool isFirstItem)
{
  while(rowCount(idx) > 0)
  {
    onDeleteRow(NGLTreeModel::index(0, 0, idx),false);
  }
  if (isFirstItem) //edit mode
  {
    NGLTreeItem *item = privateData(idx);
    FileManagement::Folder data = item->data().value<FileManagement::Folder>();
    data.isDeleted(true);
  }
  removeRow(idx.row(), idx.parent());

  return true;
}

//----------------------------------------------------------------------------
QModelIndex DialogModelFolder::getIndex(int folderId) const
{
  QModelIndexList list = match(index(0, 0), Qt::UserRole, QVariant::fromValue(folderId), 1, Qt::MatchRecursive);

  if (!list.isEmpty())
    return list.first();

  return {};
}

//----------------------------------------------------------------------------
void DialogModelProxyFolder::setSourceModel(QAbstractItemModel *model)
{
  m_sourceModel = qobject_cast<DialogModelFolder*>(model);
  QSortFilterProxyModel::setSourceModel(model);
}

//----------------------------------------------------------------------------
bool DialogModelProxyFolder::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const
{
  if (!m_sourceModel || source_left.column() != source_right.column())
    return QSortFilterProxyModel::lessThan(source_left, source_right);

  const NGLTreeItem *itemLeft = m_sourceModel->privateData(source_left);
  const NGLTreeItem *itemRight = m_sourceModel->privateData(source_right);
  if (!itemLeft || !itemRight)
    return QSortFilterProxyModel::lessThan(source_left, source_right);

  const FileManagement::Folder left = itemLeft->data().value<FileManagement::Folder>();
  const FileManagement::Folder right = itemRight->data().value<FileManagement::Folder>();

  if (left.id() == FileManagement::MainFolder::GlobalDocuments   || right.id() == FileManagement::MainFolder::GlobalDocuments ||
      left.id() == FileManagement::MainFolder::UserDocuments     || right.id() == FileManagement::MainFolder::UserDocuments   ||
      left.id() == FileManagement::MainFolder::ExerciseDocuments || right.id() == FileManagement::MainFolder::ExerciseDocuments)
  {
    return left.id() < right.id();
  }

  if (QString::fromStdString(left.name()) != QString::fromStdString(right.name()))
  {
    return (QString::fromStdString(left.name()) < QString::fromStdString(right.name()));
  }

  return left.id() < right.id();
}

//----------------------------------------------------------------------------
void DialogModelProxyFolder::sort(int column, Qt::SortOrder order)
{
  Qt::SortOrder myOrder = order;
  if (column == 0)
    myOrder = Qt::AscendingOrder;
  QSortFilterProxyModel::sort(column, myOrder);
}

//----------------------------------------------------------------------------
bool DialogModelProxyFolder::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
{
  if (!m_sourceModel || !source_parent.isValid())
    return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);

  if (m_operatorId == 0) // no filter for admin
    return true;

  NGLTreeItem *parent = m_sourceModel->privateData(source_parent);
  if (!parent)
    return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);

  NGLTreeItem *child = parent->child(source_row);
  if (!child)
    return QSortFilterProxyModel::filterAcceptsRow(source_row, source_parent);

  const FileManagement::Folder data = child->data().value<FileManagement::Folder>();
  if (data.operatorId() != m_operatorId)
  {
    // it's user folder -> check if it's in user documents and hide if it's not for the current operator
    for (int p : data.path())
    {
      if (p == FileManagement::MainFolder::UserDocuments)
        return false;
    }
  }
  return true;
}

//----------------------------------------------------------------------------
void DialogModelProxyFolder::setOperatorId(int operatorId)
{
  if (m_operatorId == operatorId)
    return;
  m_operatorId = operatorId;
  invalidateFilter();
}

} // namespace LtsWidgets