modorganizer/src/pluginlistsortproxy.cpp

/*
Copyright (C) 2012 Sebastian Herbord. All rights reserved.

This file is part of Mod Organizer.

Mod Organizer 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.

Mod Organizer 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.

You should have received a copy of the GNU General Public License
along with Mod Organizer.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "pluginlistsortproxy.h"
#include "messagedialog.h"
#include <QMenu>
#include <QCheckBox>
#include <QApplication>
#include <QWidgetAction>
#include <QTreeView>


PluginListSortProxy::PluginListSortProxy(QObject *parent)
  : QSortFilterProxyModel(parent),
    m_SortIndex(0), m_SortOrder(Qt::AscendingOrder)
{
  m_EnabledColumns.set(PluginList::COL_NAME);
  m_EnabledColumns.set(PluginList::COL_PRIORITY);
  m_EnabledColumns.set(PluginList::COL_MODINDEX);
  this->setDynamicSortFilter(true);
}





void PluginListSortProxy::setEnabledColumns(unsigned int columns)
{
  emit layoutAboutToBeChanged();
  for (int i = 0; i <= PluginList::COL_LASTCOLUMN; ++i) {
    m_EnabledColumns.set(i, (columns & (1 << i)) != 0);
  }
  emit layoutChanged();
}


Qt::ItemFlags PluginListSortProxy::flags(const QModelIndex &modelIndex) const
{
/*  Qt::ItemFlags flags;
  QModelIndex index = mapToSource(modelIndex);
  if (index.isValid()) {
    flags = sourceModel()->flags(index);
  }
  return flags;*/

  return sourceModel()->flags(mapToSource(modelIndex));
}


void PluginListSortProxy::updateFilter(const QString &filter)
{
  m_CurrentFilter = filter;
  invalidateFilter();
}


bool PluginListSortProxy::filterAcceptsRow(int row, const QModelIndex&) const
{
  return filterMatchesPlugin(sourceModel()->data(sourceModel()->index(row, 0)).toString());
}


bool PluginListSortProxy::lessThan(const QModelIndex &left,
                                   const QModelIndex &right) const
{
  PluginList *plugins = qobject_cast<PluginList*>(sourceModel());
  switch (left.column()) {
    case PluginList::COL_NAME: {
      return QString::compare(plugins->getName(left.row()), plugins->getName(right.row()), Qt::CaseInsensitive) < 0;
    } break;
    case PluginList::COL_FLAGS: {
      QVariantList lhsList = left.data(Qt::UserRole + 1).toList();
      QVariantList rhsList = right.data(Qt::UserRole + 1).toList();
      if (lhsList.size() != rhsList.size()) {
        return lhsList.size() < rhsList.size();
      } else {
        for (int i = 0; i < lhsList.size(); ++i) {
          if (lhsList.at(i) != rhsList.at(i)) {
            return lhsList.at(i) < rhsList.at(i);
          }
        }
        return false;
      }
    } break;
    case PluginList::COL_MODINDEX: {
      QString leftVal = plugins->getIndexPriority(left.row());
      QString rightVal = plugins->getIndexPriority(right.row());
      return leftVal < rightVal;
    } break;
    default: {
      return plugins->getPriority(left.row()) < plugins->getPriority(right.row());
    } break;
  }
}


bool PluginListSortProxy::dropMimeData(const QMimeData *data, Qt::DropAction action,
                                       int row, int column, const QModelIndex &parent)
{
    if ((sortColumn() != PluginList::COL_PRIORITY)
        && (sortColumn() != PluginList::COL_MODINDEX)) {
      QWidget *wid = qApp->activeWindow()->findChild<QTreeView*>("espList");
      MessageDialog::showMessage(tr("Drag&Drop is only supported when sorting by priority or mod index"), wid);
      return false;
    }

    if ((row == -1) && (column == -1)) {
      return this->sourceModel()->dropMimeData(data, action, -1, -1, mapToSource(parent));
    }
    // in the regular model, when dropping between rows, the row-value passed to
    // the sourceModel is inconsistent between ascending and descending ordering.
    // This should fix that
    if (sortOrder() == Qt::DescendingOrder) {
      --row;
    }

    QModelIndex proxyIndex = index(row, column, parent);
    QModelIndex sourceIndex = mapToSource(proxyIndex);
    return this->sourceModel()->dropMimeData(data, action, sourceIndex.row(), sourceIndex.column(),
                                             sourceIndex.parent());
}


bool PluginListSortProxy::filterMatchesPlugin(const QString &plugin) const
{
  if (!m_CurrentFilter.isEmpty()) {

    bool display = false;
    QString filterCopy = QString(m_CurrentFilter);
    filterCopy.replace("||", ";").replace("OR", ";").replace("|", ";");
    QStringList ORList = filterCopy.split(";", QString::SkipEmptyParts);

    bool segmentGood = true;

    //split in ORSegments that internally use AND logic
    for (auto& ORSegment : ORList) {
      QStringList ANDKeywords = ORSegment.split(" ", QString::SkipEmptyParts);
      segmentGood = true;

      //check each word in the segment for match, each word needs to be matched but it doesn't matter where.
      for (auto& currentKeyword : ANDKeywords) {
        if (!plugin.contains(currentKeyword, Qt::CaseInsensitive)) {
          segmentGood = false;
          break;
        }
      }

      if (segmentGood) {
        //the last AND loop didn't break so the ORSegments is true so mod matches filter
        return true;
      }

    }//for ORList loop

    return false;
  }
  else
    return true;
}