X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fuisupport%2Fbufferviewfilter.cpp;h=780d0bf6ec6cb6e3014f9fb56850bef2c019d8ff;hp=7179f262537405a054a3af3b0a6f95258b0d5f52;hb=52209badc8e769e50aa3019b63689dda0e79e9d0;hpb=a3e9fc8ef08339fb19eacd0c9c35733792f770ca diff --git a/src/uisupport/bufferviewfilter.cpp b/src/uisupport/bufferviewfilter.cpp index 7179f262..780d0bf6 100644 --- a/src/uisupport/bufferviewfilter.cpp +++ b/src/uisupport/bufferviewfilter.cpp @@ -1,5 +1,5 @@ /*************************************************************************** - * Copyright (C) 2005-09 by the Quassel Project * + * Copyright (C) 2005-2019 by the Quassel Project * * devel@quassel-irc.org * * * * This program is free software; you can redistribute it and/or modify * @@ -15,492 +15,554 @@ * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the * * Free Software Foundation, Inc., * - * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * ***************************************************************************/ #include "bufferviewfilter.h" #include -#include #include +#include #include "bufferinfo.h" #include "buffermodel.h" #include "buffersettings.h" #include "client.h" -#include "iconloader.h" +#include "clientbufferviewconfig.h" +#include "graphicalui.h" #include "networkmodel.h" - -class CheckRemovalEvent : public QEvent { -public: - CheckRemovalEvent(const QModelIndex &source_index) : QEvent(QEvent::User), index(source_index) {}; - QPersistentModelIndex index; -}; +#include "uistyle.h" +#include "util.h" /***************************************** -* The Filter for the Tree View -*****************************************/ -BufferViewFilter::BufferViewFilter(QAbstractItemModel *model, BufferViewConfig *config) - : QSortFilterProxyModel(model), - _config(0), - _sortOrder(Qt::AscendingOrder), - _userOfflineIcon(SmallIcon("im-user-offline")), - _userAwayIcon(SmallIcon("im-user-away")), - _userOnlineIcon(SmallIcon("im-user")), - _editMode(false), - _enableEditMode(tr("Show / Hide buffers"), this) + * The Filter for the Tree View + *****************************************/ +BufferViewFilter::BufferViewFilter(QAbstractItemModel* model, BufferViewConfig* config) + : QSortFilterProxyModel(model) + , _config(nullptr) + , _sortOrder(Qt::AscendingOrder) + , _showServerQueries(false) + , _editMode(false) + , _enableEditMode(tr("Show / Hide Chats"), this) { - setConfig(config); - setSourceModel(model); + setConfig(config); + setSourceModel(model); - setDynamicSortFilter(true); + setDynamicSortFilter(true); + // Sort case-insensitively (primarily for network names; channels/nicks handled elsewhere) + setSortCaseSensitivity(Qt::CaseInsensitive); - connect(this, SIGNAL(_dataChanged(const QModelIndex &, const QModelIndex &)), - this, SLOT(_q_sourceDataChanged(QModelIndex,QModelIndex))); + _enableEditMode.setCheckable(true); + _enableEditMode.setChecked(_editMode); + connect(&_enableEditMode, &QAction::toggled, this, &BufferViewFilter::enableEditMode); - _enableEditMode.setCheckable(true); - _enableEditMode.setChecked(_editMode); - connect(&_enableEditMode, SIGNAL(toggled(bool)), this, SLOT(enableEditMode(bool))); - - BufferSettings bufferSettings; - _showUserStateIcons = bufferSettings.showUserStateIcons(); - bufferSettings.notify("ShowUserStateIcons", this, SLOT(showUserStateIconsChanged())); + BufferSettings defaultSettings; + defaultSettings.notify("ServerNoticesTarget", this, &BufferViewFilter::showServerQueriesChanged); + showServerQueriesChanged(); } -void BufferViewFilter::showUserStateIconsChanged() { - BufferSettings bufferSettings; - _showUserStateIcons = bufferSettings.showUserStateIcons(); +void BufferViewFilter::setConfig(BufferViewConfig* config) +{ + if (_config == config) + return; + + if (_config) { + disconnect(_config, nullptr, this, nullptr); + } + + _config = config; + + if (!config) { + invalidate(); + setObjectName(""); + return; + } + + if (config->isInitialized()) { + configInitialized(); + } + else { + // we use a queued connection here since manipulating the connection list of a sending object + // doesn't seem to be such a good idea while executing a connected slots. + connect(config, &SyncableObject::initDone, this, &BufferViewFilter::configInitialized, Qt::QueuedConnection); + invalidate(); + } } -void BufferViewFilter::setConfig(BufferViewConfig *config) { - if(_config == config) - return; +void BufferViewFilter::configInitialized() +{ + if (!config()) + return; - if(_config) { - disconnect(_config, 0, this, 0); - } + connect(config(), &BufferViewConfig::configChanged, this, &QSortFilterProxyModel::invalidate); - _config = config; + disconnect(config(), &SyncableObject::initDone, this, &BufferViewFilter::configInitialized); + + setObjectName(config()->bufferViewName()); - if(!config) { - invalidate(); - setObjectName(""); - return; - } - - if(config->isInitialized()) { - configInitialized(); - } else { - connect(config, SIGNAL(initDone()), this, SLOT(configInitialized())); invalidate(); - } + emit configChanged(); } -void BufferViewFilter::configInitialized() { - if(!config()) - return; - - connect(config(), SIGNAL(bufferViewNameSet(const QString &)), this, SLOT(invalidate())); - connect(config(), SIGNAL(networkIdSet(const NetworkId &)), this, SLOT(invalidate())); - connect(config(), SIGNAL(addNewBuffersAutomaticallySet(bool)), this, SLOT(invalidate())); - connect(config(), SIGNAL(sortAlphabeticallySet(bool)), this, SLOT(invalidate())); - connect(config(), SIGNAL(hideInactiveBuffersSet(bool)), this, SLOT(invalidate())); - connect(config(), SIGNAL(allowedBufferTypesSet(int)), this, SLOT(invalidate())); - connect(config(), SIGNAL(minimumActivitySet(int)), this, SLOT(invalidate())); - connect(config(), SIGNAL(bufferListSet()), this, SLOT(invalidate())); - connect(config(), SIGNAL(bufferAdded(const BufferId &, int)), this, SLOT(invalidate())); - connect(config(), SIGNAL(bufferMoved(const BufferId &, int)), this, SLOT(invalidate())); - connect(config(), SIGNAL(bufferRemoved(const BufferId &)), this, SLOT(invalidate())); - connect(config(), SIGNAL(bufferPermanentlyRemoved(const BufferId &)), this, SLOT(invalidate())); - - disconnect(config(), SIGNAL(initDone()), this, SLOT(configInitialized())); - - setObjectName(config()->bufferViewName()); - - invalidate(); - emit configChanged(); +void BufferViewFilter::showServerQueriesChanged() +{ + BufferSettings bufferSettings; + + bool showQueries = (bufferSettings.serverNoticesTarget() & BufferSettings::DefaultBuffer); + if (_showServerQueries != showQueries) { + _showServerQueries = showQueries; + invalidate(); + } } -QList BufferViewFilter::actions(const QModelIndex &index) { - Q_UNUSED(index) - QList actionList; - actionList << &_enableEditMode; - return actionList; +QList BufferViewFilter::actions(const QModelIndex& index) +{ + Q_UNUSED(index) + QList actionList; + actionList << &_enableEditMode; + return actionList; } -void BufferViewFilter::enableEditMode(bool enable) { - if(_editMode == enable) { - return; - } - _editMode = enable; - - if(!config()) - return; - - if(enable == false) { - int numBuffers = config()->bufferList().count(); - QSet::const_iterator iter; - for(iter = _toAdd.constBegin(); iter != _toAdd.constEnd(); iter++) { - if(config()->bufferList().contains(*iter)) - continue; - config()->requestAddBuffer(*iter, numBuffers); - } - for(iter = _toTempRemove.constBegin(); iter != _toTempRemove.constEnd(); iter++) { - if(config()->temporarilyRemovedBuffers().contains(*iter)) - continue; - config()->requestRemoveBuffer(*iter); +void BufferViewFilter::setFilterString(const QString string) +{ + beginResetModel(); + _filterString = string; + endResetModel(); + enableEditMode(!string.isEmpty()); +} + +void BufferViewFilter::enableEditMode(bool enable) +{ + if (_editMode == enable) { + return; } - for(iter = _toRemove.constBegin(); iter != _toRemove.constEnd(); iter++) { - if(config()->removedBuffers().contains(*iter)) - continue; - config()->requestRemoveBufferPermanently(*iter); + _editMode = enable; + + if (!config()) + return; + + if (enable == false) { + addBuffers(_toAdd.values()); + QSet::const_iterator iter; + for (iter = _toTempRemove.constBegin(); iter != _toTempRemove.constEnd(); ++iter) { + if (config()->temporarilyRemovedBuffers().contains(*iter)) + continue; + config()->requestRemoveBuffer(*iter); + } + for (iter = _toRemove.constBegin(); iter != _toRemove.constEnd(); ++iter) { + if (config()->removedBuffers().contains(*iter)) + continue; + config()->requestRemoveBufferPermanently(*iter); + } } - } - _toAdd.clear(); - _toTempRemove.clear(); - _toRemove.clear(); + _toAdd.clear(); + _toTempRemove.clear(); + _toRemove.clear(); - invalidate(); + invalidate(); } - -Qt::ItemFlags BufferViewFilter::flags(const QModelIndex &index) const { - QModelIndex source_index = mapToSource(index); - Qt::ItemFlags flags = sourceModel()->flags(source_index); - if(_config) { - if(source_index == QModelIndex() || sourceModel()->data(source_index, NetworkModel::ItemTypeRole) == NetworkModel::NetworkItemType) { - flags |= Qt::ItemIsDropEnabled; - } else if(_editMode) { - flags |= Qt::ItemIsUserCheckable | Qt::ItemIsTristate; +Qt::ItemFlags BufferViewFilter::flags(const QModelIndex& index) const +{ + QModelIndex source_index = mapToSource(index); + Qt::ItemFlags flags = sourceModel()->flags(source_index); + if (config()) { + BufferInfo::Type bufferType = (BufferInfo::Type)sourceModel()->data(source_index, NetworkModel::BufferTypeRole).toInt(); + + // We need Status Buffers to be a drop target, to allow for rearranging buffers. + // The Status Buffer "owns" the space between Channel/Query buffers in the tree. + // This DOES mean that it looks like you can merge a buffer into the Status buffer, but that is restricted in BufferView::dropEvent(). + if (bufferType == BufferInfo::StatusBuffer) { + // But only if the layout isn't locked! + auto* clientConf = qobject_cast(config()); + if (clientConf && !clientConf->isLocked()) { + flags |= Qt::ItemIsDropEnabled; + } + } + + // If we're in Edit Mode, everything except Status Buffers should be hideable. + if (_editMode && bufferType != BufferInfo::StatusBuffer) { + flags |= Qt::ItemIsUserCheckable | Qt::ItemIsTristate; + } } - } - return flags; + return flags; } -bool BufferViewFilter::dropMimeData(const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent) { - if(!config() || !NetworkModel::mimeContainsBufferList(data)) - return QSortFilterProxyModel::dropMimeData(data, action, row, column, parent); - - NetworkId droppedNetworkId; - QModelIndex source_parent = mapToSource(parent); - if(sourceModel()->data(source_parent, NetworkModel::ItemTypeRole) == NetworkModel::NetworkItemType) - droppedNetworkId = sourceModel()->data(source_parent, NetworkModel::NetworkIdRole).value(); - - QList< QPair > bufferList = NetworkModel::mimeDataToBufferList(data); - BufferId bufferId; - NetworkId networkId; - int pos; - for(int i = 0; i < bufferList.count(); i++) { - networkId = bufferList[i].first; - bufferId = bufferList[i].second; - if(droppedNetworkId == networkId) { - if(row < 0) - row = 0; - - if(row < rowCount(parent)) { - QModelIndex source_child = mapToSource(index(row, 0, parent)); - BufferId beforeBufferId = sourceModel()->data(source_child, NetworkModel::BufferIdRole).value(); - pos = config()->bufferList().indexOf(beforeBufferId); - if(_sortOrder == Qt::DescendingOrder) - pos++; - } else { - if(_sortOrder == Qt::AscendingOrder) - pos = config()->bufferList().count(); - else - pos = 0; - } - - if(config()->bufferList().contains(bufferId)) { - if(config()->bufferList().indexOf(bufferId) < pos) - pos--; - config()->requestMoveBuffer(bufferId, pos); - } else { - config()->requestAddBuffer(bufferId, pos); - } - - } else { - addBuffer(bufferId); +bool BufferViewFilter::dropMimeData(const QMimeData* data, Qt::DropAction action, int row, int column, const QModelIndex& parent) +{ + if (!config() || !NetworkModel::mimeContainsBufferList(data)) + return QSortFilterProxyModel::dropMimeData(data, action, row, column, parent); + + NetworkId droppedNetworkId; + QModelIndex source_parent = mapToSource(parent); + if (sourceModel()->data(source_parent, NetworkModel::ItemTypeRole) == NetworkModel::NetworkItemType) + droppedNetworkId = sourceModel()->data(source_parent, NetworkModel::NetworkIdRole).value(); + + QList> bufferList = NetworkModel::mimeDataToBufferList(data); + BufferId bufferId; + NetworkId networkId; + int pos; + for (int i = 0; i < bufferList.count(); i++) { + networkId = bufferList[i].first; + bufferId = bufferList[i].second; + if (droppedNetworkId == networkId) { + if (row < 0) + row = 0; + + if (row < rowCount(parent)) { + QModelIndex source_child = mapToSource(index(row, 0, parent)); + BufferId beforeBufferId = sourceModel()->data(source_child, NetworkModel::BufferIdRole).value(); + pos = config()->bufferList().indexOf(beforeBufferId); + if (_sortOrder == Qt::DescendingOrder) + pos++; + } + else { + if (_sortOrder == Qt::AscendingOrder) + pos = config()->bufferList().count(); + else + pos = 0; + } + + if (config()->bufferList().contains(bufferId) && !config()->sortAlphabetically()) { + if (config()->bufferList().indexOf(bufferId) < pos) + pos--; + auto* clientConf = qobject_cast(config()); + if (!clientConf || !clientConf->isLocked()) + config()->requestMoveBuffer(bufferId, pos); + } + else { + config()->requestAddBuffer(bufferId, pos); + } + } + else { + addBuffer(bufferId); + } } - } - return true; + return true; } -void BufferViewFilter::sort(int column, Qt::SortOrder order) { - _sortOrder = order; - QSortFilterProxyModel::sort(column, order); +void BufferViewFilter::sort(int column, Qt::SortOrder order) +{ + _sortOrder = order; + QSortFilterProxyModel::sort(column, order); } -void BufferViewFilter::addBuffer(const BufferId &bufferId) const { - if(!config() || config()->bufferList().contains(bufferId)) - return; - - int pos = config()->bufferList().count(); - bool lt; - for(int i = 0; i < config()->bufferList().count(); i++) { - if(config() && config()->sortAlphabetically()) - lt = bufferIdLessThan(bufferId, config()->bufferList()[i]); - else - lt = bufferId < config()->bufferList()[i]; +void BufferViewFilter::addBuffer(const BufferId& bufferId) const +{ + if (!config() || config()->bufferList().contains(bufferId)) + return; + + int pos = config()->bufferList().count(); + bool lt; + for (int i = 0; i < config()->bufferList().count(); i++) { + if (config() && config()->sortAlphabetically()) + lt = bufferIdLessThan(bufferId, config()->bufferList()[i]); + else + lt = bufferId < config()->bufferList()[i]; + + if (lt) { + pos = i; + break; + } + } + config()->requestAddBuffer(bufferId, pos); +} - if(lt) { - pos = i; - break; +void BufferViewFilter::addBuffers(const QList& bufferIds) const +{ + if (!config()) + return; + + QList bufferList = config()->bufferList(); + foreach (BufferId bufferId, bufferIds) { + if (bufferList.contains(bufferId)) + continue; + + int pos = bufferList.count(); + bool lt; + for (int i = 0; i < bufferList.count(); i++) { + if (config() && config()->sortAlphabetically()) + lt = bufferIdLessThan(bufferId, bufferList[i]); + else + lt = bufferId < config()->bufferList()[i]; + + if (lt) { + pos = i; + bufferList.insert(pos, bufferId); + break; + } + } + config()->requestAddBuffer(bufferId, pos); } - } - config()->requestAddBuffer(bufferId, pos); } -bool BufferViewFilter::filterAcceptBuffer(const QModelIndex &source_bufferIndex) const { - // no config -> "all buffers" -> accept everything - if(!config()) - return true; +bool BufferViewFilter::filterAcceptBuffer(const QModelIndex& source_bufferIndex) const +{ + // no config -> "all buffers" -> accept everything + if (!config()) + return true; + + BufferId bufferId = sourceModel()->data(source_bufferIndex, NetworkModel::BufferIdRole).value(); + Q_ASSERT(bufferId.isValid()); + + int activityLevel = sourceModel()->data(source_bufferIndex, NetworkModel::BufferActivityRole).toInt(); + + if (!config()->bufferList().contains(bufferId) && !_editMode) { + // add the buffer if... + if (config()->isInitialized() && !config()->removedBuffers().contains(bufferId) // it hasn't been manually removed and either + && ((config()->addNewBuffersAutomatically() + && !config()->temporarilyRemovedBuffers().contains(bufferId)) // is totally unknown to us (a new buffer)... + || (config()->temporarilyRemovedBuffers().contains(bufferId) + && activityLevel > BufferInfo::OtherActivity))) { // or was just temporarily hidden and has a new message waiting for us. + addBuffer(bufferId); + } + // note: adding the buffer to the valid list does not temper with the following filters ("show only channels" and stuff) + return false; + } - BufferId bufferId = sourceModel()->data(source_bufferIndex, NetworkModel::BufferIdRole).value(); - Q_ASSERT(bufferId.isValid()); + if (config()->networkId().isValid() + && config()->networkId() != sourceModel()->data(source_bufferIndex, NetworkModel::NetworkIdRole).value()) + return false; - int activityLevel = sourceModel()->data(source_bufferIndex, NetworkModel::BufferActivityRole).toInt(); + int allowedBufferTypes = config()->allowedBufferTypes(); + if (!config()->networkId().isValid()) + allowedBufferTypes &= ~BufferInfo::StatusBuffer; + int bufferType = sourceModel()->data(source_bufferIndex, NetworkModel::BufferTypeRole).toInt(); + if (!(allowedBufferTypes & bufferType)) + return false; - if(!config()->bufferList().contains(bufferId) && !_editMode) { - // add the buffer if... - if(config()->isInitialized() - && !config()->removedBuffers().contains(bufferId) // it hasn't been manually removed and either - && ((config()->addNewBuffersAutomatically() && !config()->temporarilyRemovedBuffers().contains(bufferId)) // is totally unknown to us (a new buffer)... - || (config()->temporarilyRemovedBuffers().contains(bufferId) && activityLevel > BufferInfo::OtherActivity))) { // or was just temporarily hidden and has a new message waiting for us. - addBuffer(bufferId); + if (bufferType & BufferInfo::QueryBuffer && !_showServerQueries + && sourceModel()->data(source_bufferIndex, Qt::DisplayRole).toString().contains('.')) { + return false; } - // note: adding the buffer to the valid list does not temper with the following filters ("show only channels" and stuff) - return false; - } - - if(config()->networkId().isValid() && config()->networkId() != sourceModel()->data(source_bufferIndex, NetworkModel::NetworkIdRole).value()) - return false; - - int allowedBufferTypes = config()->allowedBufferTypes(); - if(!config()->networkId().isValid()) - allowedBufferTypes &= ~BufferInfo::StatusBuffer; - if(!(allowedBufferTypes & sourceModel()->data(source_bufferIndex, NetworkModel::BufferTypeRole).toInt())) - return false; - - // the following dynamic filters may not trigger if the buffer is currently selected. - QModelIndex currentIndex = Client::bufferModel()->standardSelectionModel()->currentIndex(); - if(bufferId == Client::bufferModel()->data(currentIndex, NetworkModel::BufferIdRole).value()) - return true; - if(config()->hideInactiveBuffers() && !sourceModel()->data(source_bufferIndex, NetworkModel::ItemActiveRole).toBool() && activityLevel <= BufferInfo::OtherActivity) - return false; + if (!_filterString.isEmpty()) { + const BufferInfo info = qvariant_cast(Client::bufferModel()->data(source_bufferIndex, NetworkModel::BufferInfoRole)); + QString name = info.bufferName(); + if (name.contains(_filterString, Qt::CaseInsensitive)) { + return true; + } + else { + return false; + } + } - if(config()->minimumActivity() > activityLevel) - return false; + // the following dynamic filters may not trigger if the buffer is currently selected. + QModelIndex currentIndex = Client::bufferModel()->standardSelectionModel()->currentIndex(); + if (bufferId == Client::bufferModel()->data(currentIndex, NetworkModel::BufferIdRole).value()) + return true; - return true; -} + if (config()->hideInactiveBuffers() && !sourceModel()->data(source_bufferIndex, NetworkModel::ItemActiveRole).toBool() + && activityLevel <= BufferInfo::OtherActivity) + return false; -bool BufferViewFilter::filterAcceptNetwork(const QModelIndex &source_index) const { - if(!config()) - return true; + if (config()->minimumActivity() > activityLevel) + return false; - if(!config()->networkId().isValid()) { return true; - } else { - return config()->networkId() == sourceModel()->data(source_index, NetworkModel::NetworkIdRole).value(); - } } -bool BufferViewFilter::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const { - QModelIndex child = sourceModel()->index(source_row, 0, source_parent); - - if(!child.isValid()) { - qWarning() << "filterAcceptsRow has been called with an invalid Child"; - return false; - } - - NetworkModel::ItemType childType = (NetworkModel::ItemType)sourceModel()->data(child, NetworkModel::ItemTypeRole).toInt(); - switch(childType) { - case NetworkModel::NetworkItemType: - return filterAcceptNetwork(child); - case NetworkModel::BufferItemType: - return filterAcceptBuffer(child); - default: - return false; - } -} +bool BufferViewFilter::filterAcceptNetwork(const QModelIndex& source_index) const +{ + if (!config()) + return true; -bool BufferViewFilter::lessThan(const QModelIndex &source_left, const QModelIndex &source_right) const { - int leftItemType = sourceModel()->data(source_left, NetworkModel::ItemTypeRole).toInt(); - int rightItemType = sourceModel()->data(source_right, NetworkModel::ItemTypeRole).toInt(); - int itemType = leftItemType & rightItemType; - switch(itemType) { - case NetworkModel::NetworkItemType: - return networkLessThan(source_left, source_right); - case NetworkModel::BufferItemType: - return bufferLessThan(source_left, source_right); - default: - return QSortFilterProxyModel::lessThan(source_left, source_right); - } -} + if (config()->hideInactiveNetworks() && !(sourceModel()->data(source_index, NetworkModel::ItemActiveRole).toBool())) { + return false; + } -bool BufferViewFilter::bufferLessThan(const QModelIndex &source_left, const QModelIndex &source_right) const { - BufferId leftBufferId = sourceModel()->data(source_left, NetworkModel::BufferIdRole).value(); - BufferId rightBufferId = sourceModel()->data(source_right, NetworkModel::BufferIdRole).value(); - if(config()) { - int leftPos = config()->bufferList().indexOf(leftBufferId); - int rightPos = config()->bufferList().indexOf(rightBufferId); - if(leftPos == -1 && rightPos == -1) - return QSortFilterProxyModel::lessThan(source_left, source_right); - if(leftPos == -1 || rightPos == -1) - return !(leftPos < rightPos); - return leftPos < rightPos; - } else - return bufferIdLessThan(leftBufferId, rightBufferId); + if (!config()->networkId().isValid()) { + return true; + } + else { + return config()->networkId() == sourceModel()->data(source_index, NetworkModel::NetworkIdRole).value(); + } } -bool BufferViewFilter::networkLessThan(const QModelIndex &source_left, const QModelIndex &source_right) const { - NetworkId leftNetworkId = sourceModel()->data(source_left, NetworkModel::NetworkIdRole).value(); - NetworkId rightNetworkId = sourceModel()->data(source_right, NetworkModel::NetworkIdRole).value(); +bool BufferViewFilter::filterAcceptsRow(int source_row, const QModelIndex& source_parent) const +{ + QModelIndex child = sourceModel()->index(source_row, 0, source_parent); - if(config() && config()->sortAlphabetically()) - return QSortFilterProxyModel::lessThan(source_left, source_right); - else - return leftNetworkId < rightNetworkId; -} + if (!child.isValid()) { + qWarning() << "filterAcceptsRow has been called with an invalid Child"; + return false; + } -QVariant BufferViewFilter::data(const QModelIndex &index, int role) const { - switch(role) { - case Qt::DecorationRole: - return icon(index); - case Qt::CheckStateRole: - return checkedState(index); - default: - return QSortFilterProxyModel::data(index, role); - } + NetworkModel::ItemType childType = (NetworkModel::ItemType)sourceModel()->data(child, NetworkModel::ItemTypeRole).toInt(); + switch (childType) { + case NetworkModel::NetworkItemType: + return filterAcceptNetwork(child); + case NetworkModel::BufferItemType: + return filterAcceptBuffer(child); + default: + return false; + } } -QVariant BufferViewFilter::icon(const QModelIndex &index) const { - if(!_showUserStateIcons || (config() && config()->disableDecoration())) - return QVariant(); - - if(index.column() != 0) - return QVariant(); - - QModelIndex source_index = mapToSource(index); - if(sourceModel()->data(source_index, NetworkModel::ItemTypeRole).toInt() != NetworkModel::BufferItemType) - return QVariant(); - - if(sourceModel()->data(source_index, NetworkModel::BufferTypeRole).toInt() != BufferInfo::QueryBuffer) - return QVariant(); - - if(!sourceModel()->data(source_index, NetworkModel::ItemActiveRole).toBool()) - return _userOfflineIcon; - - if(sourceModel()->data(source_index, NetworkModel::UserAwayRole).toBool()) - return _userAwayIcon; - else - return _userOnlineIcon; +bool BufferViewFilter::lessThan(const QModelIndex& source_left, const QModelIndex& source_right) const +{ + int leftItemType = sourceModel()->data(source_left, NetworkModel::ItemTypeRole).toInt(); + int rightItemType = sourceModel()->data(source_right, NetworkModel::ItemTypeRole).toInt(); + int itemType = leftItemType & rightItemType; + switch (itemType) { + case NetworkModel::NetworkItemType: + return networkLessThan(source_left, source_right); + case NetworkModel::BufferItemType: + return bufferLessThan(source_left, source_right); + default: + return QSortFilterProxyModel::lessThan(source_left, source_right); + } +} - return QVariant(); +bool BufferViewFilter::bufferLessThan(const QModelIndex& source_left, const QModelIndex& source_right) const +{ + BufferId leftBufferId = sourceModel()->data(source_left, NetworkModel::BufferIdRole).value(); + BufferId rightBufferId = sourceModel()->data(source_right, NetworkModel::BufferIdRole).value(); + // If filtering, prioritize relevant items first + if (!_filterString.isEmpty()) { + // Get names of the buffers + QString leftBufferName = sourceModel()->data(source_left, NetworkModel::BufferInfoRole).value().bufferName(); + QString rightBufferName = sourceModel()->data(source_right, NetworkModel::BufferInfoRole).value().bufferName(); + // Check if there's any differences across types, most important first + if ((QString::compare(leftBufferName, _filterString, Qt::CaseInsensitive) == 0) + != (QString::compare(rightBufferName, _filterString, Qt::CaseInsensitive) == 0)) { + // One of these buffers is an exact match with the filter string, while the other isn't + // Prioritize whichever one is the exact match + // (If left buffer is exact, return true to set it as less than right) + return (QString::compare(leftBufferName, _filterString, Qt::CaseInsensitive) == 0); + } + else if (leftBufferName.startsWith(_filterString, Qt::CaseInsensitive) + != rightBufferName.startsWith(_filterString, Qt::CaseInsensitive)) { + // One of these buffers starts with the filter string, while the other doesn't + // Prioritize whichever one starts with the filter string + // (If left buffer starts with, return true to set it as less than right) + return leftBufferName.startsWith(_filterString, Qt::CaseInsensitive); + } + // Otherwise, do the normal sorting (sorting happens within each priority bracket) + } + if (config()) { + int leftPos = config()->bufferList().indexOf(leftBufferId); + int rightPos = config()->bufferList().indexOf(rightBufferId); + if (leftPos == -1 && rightPos == -1) + return QSortFilterProxyModel::lessThan(source_left, source_right); + if (leftPos == -1 || rightPos == -1) + return !(leftPos < rightPos); + return leftPos < rightPos; + } + else + return bufferIdLessThan(leftBufferId, rightBufferId); } -QVariant BufferViewFilter::checkedState(const QModelIndex &index) const { - if(!_editMode || !config()) - return QVariant(); +bool BufferViewFilter::networkLessThan(const QModelIndex& source_left, const QModelIndex& source_right) const +{ + // NetworkId leftNetworkId = sourceModel()->data(source_left, NetworkModel::NetworkIdRole).value(); + // NetworkId rightNetworkId = sourceModel()->data(source_right, NetworkModel::NetworkIdRole).value(); - QModelIndex source_index = mapToSource(index); - if(source_index == QModelIndex() || sourceModel()->data(source_index, NetworkModel::ItemTypeRole) == NetworkModel::NetworkItemType) - return QVariant(); + return QSortFilterProxyModel::lessThan(source_left, source_right); +} - BufferId bufferId = sourceModel()->data(source_index, NetworkModel::BufferIdRole).value(); - if(_toAdd.contains(bufferId)) - return Qt::Checked; +QVariant BufferViewFilter::data(const QModelIndex& index, int role) const +{ + switch (role) { + case Qt::FontRole: + case Qt::ForegroundRole: + case Qt::BackgroundRole: + case Qt::DecorationRole: + if ((config() && config()->disableDecoration())) + return QVariant(); + return GraphicalUi::uiStyle()->bufferViewItemData(mapToSource(index), role); + case Qt::CheckStateRole: + return checkedState(index); + default: + return QSortFilterProxyModel::data(index, role); + } +} - if(_toTempRemove.contains(bufferId)) - return Qt::PartiallyChecked; +QVariant BufferViewFilter::checkedState(const QModelIndex& index) const +{ + if (!_editMode || !config()) + return QVariant(); - if(_toRemove.contains(bufferId)) - return Qt::Unchecked; + QModelIndex source_index = mapToSource(index); + if (source_index == QModelIndex() || sourceModel()->data(source_index, NetworkModel::ItemTypeRole) == NetworkModel::NetworkItemType) + return QVariant(); - if(config()->bufferList().contains(bufferId)) - return Qt::Checked; + BufferId bufferId = sourceModel()->data(source_index, NetworkModel::BufferIdRole).value(); + if (_toAdd.contains(bufferId)) + return Qt::Checked; - if(config()->temporarilyRemovedBuffers().contains(bufferId)) - return Qt::PartiallyChecked; + if (_toTempRemove.contains(bufferId)) + return Qt::PartiallyChecked; - return Qt::Unchecked; -} + if (_toRemove.contains(bufferId)) + return Qt::Unchecked; -bool BufferViewFilter::setData(const QModelIndex &index, const QVariant &value, int role) { - switch(role) { - case Qt::CheckStateRole: - return setCheckedState(index, Qt::CheckState(value.toInt())); - default: - return QSortFilterProxyModel::setData(index, value, role); - } -} + if (config()->bufferList().contains(bufferId)) + return Qt::Checked; -bool BufferViewFilter::setCheckedState(const QModelIndex &index, Qt::CheckState state) { - QModelIndex source_index = mapToSource(index); - BufferId bufferId = sourceModel()->data(source_index, NetworkModel::BufferIdRole).value(); - if(!bufferId.isValid()) - return false; - - switch(state) { - case Qt::Unchecked: - _toAdd.remove(bufferId); - _toTempRemove.remove(bufferId); - _toRemove << bufferId; - break; - case Qt::PartiallyChecked: - _toAdd.remove(bufferId); - _toTempRemove << bufferId; - _toRemove.remove(bufferId); - break; - case Qt::Checked: - _toAdd << bufferId; - _toTempRemove.remove(bufferId); - _toRemove.remove(bufferId); - break; - default: - return false; - } - emit dataChanged(index, index); - return true; -} + if (config()->temporarilyRemovedBuffers().contains(bufferId)) + return Qt::PartiallyChecked; -void BufferViewFilter::checkPreviousCurrentForRemoval(const QModelIndex ¤t, const QModelIndex &previous) { - Q_UNUSED(current); - if(previous.isValid()) - QCoreApplication::postEvent(this, new CheckRemovalEvent(previous)); + return Qt::Unchecked; } -void BufferViewFilter::customEvent(QEvent *event) { - if(event->type() != QEvent::User) - return; - - CheckRemovalEvent *removalEvent = static_cast(event); - checkItemForRemoval(removalEvent->index); - - event->accept(); +bool BufferViewFilter::setData(const QModelIndex& index, const QVariant& value, int role) +{ + switch (role) { + case Qt::CheckStateRole: + return setCheckedState(index, Qt::CheckState(value.toInt())); + default: + return QSortFilterProxyModel::setData(index, value, role); + } } -void BufferViewFilter::checkItemsForRemoval(const QModelIndex &topLeft, const QModelIndex &bottomRight) { - QModelIndex source_topLeft = mapToSource(topLeft); - QModelIndex source_bottomRight = mapToSource(bottomRight); - emit _dataChanged(source_topLeft, source_bottomRight); +bool BufferViewFilter::setCheckedState(const QModelIndex& index, Qt::CheckState state) +{ + QModelIndex source_index = mapToSource(index); + BufferId bufferId = sourceModel()->data(source_index, NetworkModel::BufferIdRole).value(); + if (!bufferId.isValid()) + return false; + + switch (state) { + case Qt::Unchecked: + _toAdd.remove(bufferId); + _toTempRemove.remove(bufferId); + _toRemove << bufferId; + break; + case Qt::PartiallyChecked: + _toAdd.remove(bufferId); + _toTempRemove << bufferId; + _toRemove.remove(bufferId); + break; + case Qt::Checked: + _toAdd << bufferId; + _toTempRemove.remove(bufferId); + _toRemove.remove(bufferId); + break; + default: + return false; + } + emit dataChanged(index, index); + return true; } -bool BufferViewFilter::bufferIdLessThan(const BufferId &left, const BufferId &right) { - Q_CHECK_PTR(Client::networkModel()); - if(!Client::networkModel()) - return true; +bool BufferViewFilter::bufferIdLessThan(const BufferId& left, const BufferId& right) +{ + Q_CHECK_PTR(Client::networkModel()); + if (!Client::networkModel()) + return true; - QModelIndex leftIndex = Client::networkModel()->bufferIndex(left); - QModelIndex rightIndex = Client::networkModel()->bufferIndex(right); + QModelIndex leftIndex = Client::networkModel()->bufferIndex(left); + QModelIndex rightIndex = Client::networkModel()->bufferIndex(right); - int leftType = Client::networkModel()->data(leftIndex, NetworkModel::BufferTypeRole).toInt(); - int rightType = Client::networkModel()->data(rightIndex, NetworkModel::BufferTypeRole).toInt(); + int leftType = Client::networkModel()->data(leftIndex, NetworkModel::BufferTypeRole).toInt(); + int rightType = Client::networkModel()->data(rightIndex, NetworkModel::BufferTypeRole).toInt(); - if(leftType != rightType) - return leftType < rightType; - else - return QString::compare(Client::networkModel()->data(leftIndex, Qt::DisplayRole).toString(), Client::networkModel()->data(rightIndex, Qt::DisplayRole).toString(), Qt::CaseInsensitive) < 0; + if (leftType != rightType) + return leftType < rightType; + else + return QString::compare(Client::networkModel()->data(leftIndex, Qt::DisplayRole).toString(), + Client::networkModel()->data(rightIndex, Qt::DisplayRole).toString(), + Qt::CaseInsensitive) + < 0; } -