Browse Source
Add a Tags system to the GUI to expand torrent organization / filtering options. Closes #13.adaptive-webui-19844
Vladimir Golovnev
7 years ago
committed by
GitHub
27 changed files with 1314 additions and 34 deletions
@ -0,0 +1,337 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "tagfiltermodel.h" |
||||||
|
|
||||||
|
#include <QDebug> |
||||||
|
#include <QHash> |
||||||
|
#include <QIcon> |
||||||
|
|
||||||
|
#include "base/bittorrent/session.h" |
||||||
|
#include "base/bittorrent/torrenthandle.h" |
||||||
|
#include "guiiconprovider.h" |
||||||
|
|
||||||
|
namespace |
||||||
|
{ |
||||||
|
QString getSpecialAllTag() |
||||||
|
{ |
||||||
|
static const QString *const ALL_TAG = new QString(" "); |
||||||
|
Q_ASSERT(!BitTorrent::Session::isValidTag(*ALL_TAG)); |
||||||
|
return *ALL_TAG; |
||||||
|
} |
||||||
|
|
||||||
|
QString getSpecialUntaggedTag() |
||||||
|
{ |
||||||
|
static const QString *const UNTAGGED_TAG = new QString(" "); |
||||||
|
Q_ASSERT(!BitTorrent::Session::isValidTag(*UNTAGGED_TAG)); |
||||||
|
return *UNTAGGED_TAG; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
class TagModelItem |
||||||
|
{ |
||||||
|
public: |
||||||
|
TagModelItem(const QString &tag, int torrentsCount = 0) |
||||||
|
: m_tag(tag) |
||||||
|
, m_torrentsCount(torrentsCount) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
QString tag() const |
||||||
|
{ |
||||||
|
return m_tag; |
||||||
|
} |
||||||
|
|
||||||
|
int torrentsCount() const |
||||||
|
{ |
||||||
|
return m_torrentsCount; |
||||||
|
} |
||||||
|
|
||||||
|
void increaseTorrentsCount() |
||||||
|
{ |
||||||
|
++m_torrentsCount; |
||||||
|
} |
||||||
|
|
||||||
|
void decreaseTorrentsCount() |
||||||
|
{ |
||||||
|
Q_ASSERT(m_torrentsCount > 0); |
||||||
|
--m_torrentsCount; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
const QString m_tag; |
||||||
|
int m_torrentsCount; |
||||||
|
}; |
||||||
|
|
||||||
|
TagFilterModel::TagFilterModel(QObject *parent) |
||||||
|
: QAbstractListModel(parent) |
||||||
|
{ |
||||||
|
using Session = BitTorrent::Session; |
||||||
|
auto session = Session::instance(); |
||||||
|
|
||||||
|
connect(session, &Session::tagAdded, this, &TagFilterModel::tagAdded); |
||||||
|
connect(session, &Session::tagRemoved, this, &TagFilterModel::tagRemoved); |
||||||
|
connect(session, &Session::torrentTagAdded, this, &TagFilterModel::torrentTagAdded); |
||||||
|
connect(session, &Session::torrentTagRemoved, this, &TagFilterModel::torrentTagRemoved); |
||||||
|
connect(session, &Session::torrentAdded, this, &TagFilterModel::torrentAdded); |
||||||
|
connect(session, &Session::torrentAboutToBeRemoved, this, &TagFilterModel::torrentAboutToBeRemoved); |
||||||
|
populate(); |
||||||
|
} |
||||||
|
|
||||||
|
TagFilterModel::~TagFilterModel() = default; |
||||||
|
|
||||||
|
bool TagFilterModel::isSpecialItem(const QModelIndex &index) |
||||||
|
{ |
||||||
|
// the first two items are special items: 'All' and 'Untagged'
|
||||||
|
return (!index.parent().isValid() && (index.row() <= 1)); |
||||||
|
} |
||||||
|
|
||||||
|
QVariant TagFilterModel::data(const QModelIndex &index, int role) const |
||||||
|
{ |
||||||
|
if (!index.isValid() || index.column() != 0) |
||||||
|
return QVariant(); |
||||||
|
|
||||||
|
const int row = index.internalId(); |
||||||
|
Q_ASSERT(isValidRow(row)); |
||||||
|
const TagModelItem &item = m_tagItems[row]; |
||||||
|
|
||||||
|
switch (role) { |
||||||
|
case Qt::DecorationRole: |
||||||
|
return GuiIconProvider::instance()->getIcon("inode-directory"); |
||||||
|
case Qt::DisplayRole: |
||||||
|
return QString(QLatin1String("%1 (%2)")) |
||||||
|
.arg(tagDisplayName(item.tag())).arg(item.torrentsCount()); |
||||||
|
case Qt::UserRole: |
||||||
|
return item.torrentsCount(); |
||||||
|
default: |
||||||
|
return QVariant(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
Qt::ItemFlags TagFilterModel::flags(const QModelIndex &index) const |
||||||
|
{ |
||||||
|
if (!index.isValid()) |
||||||
|
return 0; |
||||||
|
return Qt::ItemIsEnabled | Qt::ItemIsSelectable; |
||||||
|
} |
||||||
|
|
||||||
|
QVariant TagFilterModel::headerData(int section, Qt::Orientation orientation, int role) const |
||||||
|
{ |
||||||
|
if ((orientation == Qt::Horizontal) && (role == Qt::DisplayRole)) |
||||||
|
if (section == 0) |
||||||
|
return tr("Tags"); |
||||||
|
return QVariant(); |
||||||
|
} |
||||||
|
|
||||||
|
QModelIndex TagFilterModel::index(int row, int, const QModelIndex &) const |
||||||
|
{ |
||||||
|
if (!isValidRow(row)) |
||||||
|
return QModelIndex(); |
||||||
|
return createIndex(row, 0, row); |
||||||
|
} |
||||||
|
|
||||||
|
int TagFilterModel::rowCount(const QModelIndex &parent) const |
||||||
|
{ |
||||||
|
if (!parent.isValid()) |
||||||
|
return m_tagItems.count(); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
bool TagFilterModel::isValidRow(int row) const |
||||||
|
{ |
||||||
|
return (row >= 0) && (row < m_tagItems.size()); |
||||||
|
} |
||||||
|
|
||||||
|
QModelIndex TagFilterModel::index(const QString &tag) const |
||||||
|
{ |
||||||
|
const int row = findRow(tag); |
||||||
|
if (!isValidRow(row)) |
||||||
|
return QModelIndex(); |
||||||
|
return index(row, 0, QModelIndex()); |
||||||
|
} |
||||||
|
|
||||||
|
QString TagFilterModel::tag(const QModelIndex &index) const |
||||||
|
{ |
||||||
|
if (!index.isValid()) |
||||||
|
return QString(); |
||||||
|
const int row = index.internalId(); |
||||||
|
Q_ASSERT(isValidRow(row)); |
||||||
|
return m_tagItems[row].tag(); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::tagAdded(const QString &tag) |
||||||
|
{ |
||||||
|
const int row = m_tagItems.count(); |
||||||
|
beginInsertRows(QModelIndex(), row, row); |
||||||
|
addToModel(tag, 0); |
||||||
|
endInsertRows(); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::tagRemoved(const QString &tag) |
||||||
|
{ |
||||||
|
QModelIndex i = index(tag); |
||||||
|
beginRemoveRows(i.parent(), i.row(), i.row()); |
||||||
|
removeFromModel(i.row()); |
||||||
|
endRemoveRows(); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::torrentTagAdded(BitTorrent::TorrentHandle *const torrent, const QString &tag) |
||||||
|
{ |
||||||
|
if (torrent->tags().count() == 1) |
||||||
|
untaggedItem()->decreaseTorrentsCount(); |
||||||
|
|
||||||
|
const int row = findRow(tag); |
||||||
|
Q_ASSERT(isValidRow(row)); |
||||||
|
TagModelItem &item = m_tagItems[row]; |
||||||
|
|
||||||
|
item.increaseTorrentsCount(); |
||||||
|
const QModelIndex i = index(row, 0, QModelIndex()); |
||||||
|
emit dataChanged(i, i); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::torrentTagRemoved(BitTorrent::TorrentHandle* const torrent, const QString &tag) |
||||||
|
{ |
||||||
|
Q_ASSERT(torrent->tags().count() >= 0); |
||||||
|
if (torrent->tags().count() == 0) |
||||||
|
untaggedItem()->increaseTorrentsCount(); |
||||||
|
|
||||||
|
const int row = findRow(tag); |
||||||
|
Q_ASSERT(isValidRow(row)); |
||||||
|
TagModelItem &item = m_tagItems[row]; |
||||||
|
|
||||||
|
item.decreaseTorrentsCount(); |
||||||
|
const QModelIndex i = index(row, 0, QModelIndex()); |
||||||
|
emit dataChanged(i, i); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::torrentAdded(BitTorrent::TorrentHandle *const torrent) |
||||||
|
{ |
||||||
|
allTagsItem()->increaseTorrentsCount(); |
||||||
|
|
||||||
|
const QVector<TagModelItem *> items = findItems(torrent->tags()); |
||||||
|
if (items.isEmpty()) |
||||||
|
untaggedItem()->increaseTorrentsCount(); |
||||||
|
|
||||||
|
foreach (TagModelItem *item, items) |
||||||
|
item->increaseTorrentsCount(); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::torrentAboutToBeRemoved(BitTorrent::TorrentHandle *const torrent) |
||||||
|
{ |
||||||
|
allTagsItem()->decreaseTorrentsCount(); |
||||||
|
|
||||||
|
if (torrent->tags().isEmpty()) |
||||||
|
untaggedItem()->decreaseTorrentsCount(); |
||||||
|
|
||||||
|
foreach (TagModelItem *item, findItems(torrent->tags())) |
||||||
|
item->decreaseTorrentsCount(); |
||||||
|
} |
||||||
|
|
||||||
|
QString TagFilterModel::tagDisplayName(const QString &tag) |
||||||
|
{ |
||||||
|
if (tag == getSpecialAllTag()) |
||||||
|
return tr("All"); |
||||||
|
if (tag == getSpecialUntaggedTag()) |
||||||
|
return tr("Untagged"); |
||||||
|
return tag; |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::populate() |
||||||
|
{ |
||||||
|
using Torrent = BitTorrent::TorrentHandle; |
||||||
|
|
||||||
|
auto session = BitTorrent::Session::instance(); |
||||||
|
auto torrents = session->torrents(); |
||||||
|
|
||||||
|
// All torrents
|
||||||
|
addToModel(getSpecialAllTag(), torrents.count()); |
||||||
|
|
||||||
|
const int untaggedCount = std::count_if(torrents.begin(), torrents.end(), |
||||||
|
[](Torrent *torrent) { return torrent->tags().isEmpty(); }); |
||||||
|
addToModel(getSpecialUntaggedTag(), untaggedCount); |
||||||
|
|
||||||
|
foreach (const QString &tag, session->tags()) { |
||||||
|
const int count = std::count_if(torrents.begin(), torrents.end(), |
||||||
|
[tag](Torrent *torrent) { return torrent->hasTag(tag); }); |
||||||
|
addToModel(tag, count); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::addToModel(const QString &tag, int count) |
||||||
|
{ |
||||||
|
m_tagItems.append(TagModelItem(tag, count)); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterModel::removeFromModel(int row) |
||||||
|
{ |
||||||
|
Q_ASSERT(isValidRow(row)); |
||||||
|
m_tagItems.removeAt(row); |
||||||
|
} |
||||||
|
|
||||||
|
int TagFilterModel::findRow(const QString &tag) const |
||||||
|
{ |
||||||
|
for (int i = 0; i < m_tagItems.size(); ++i) { |
||||||
|
if (m_tagItems[i].tag() == tag) |
||||||
|
return i; |
||||||
|
} |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
TagModelItem *TagFilterModel::findItem(const QString &tag) |
||||||
|
{ |
||||||
|
const int row = findRow(tag); |
||||||
|
if (!isValidRow(row)) |
||||||
|
return nullptr; |
||||||
|
return &m_tagItems[row]; |
||||||
|
} |
||||||
|
|
||||||
|
QVector<TagModelItem *> TagFilterModel::findItems(const QSet<QString> &tags) |
||||||
|
{ |
||||||
|
QVector<TagModelItem *> items; |
||||||
|
items.reserve(tags.size()); |
||||||
|
foreach (const QString &tag, tags) { |
||||||
|
TagModelItem *item = findItem(tag); |
||||||
|
if (item) |
||||||
|
items.push_back(item); |
||||||
|
else |
||||||
|
qWarning() << QString("Requested tag '%1' missing from the model.").arg(tag); |
||||||
|
} |
||||||
|
return items; |
||||||
|
} |
||||||
|
|
||||||
|
TagModelItem *TagFilterModel::allTagsItem() |
||||||
|
{ |
||||||
|
Q_ASSERT(m_tagItems.size() > 0); |
||||||
|
return &m_tagItems[0]; |
||||||
|
} |
||||||
|
|
||||||
|
TagModelItem *TagFilterModel::untaggedItem() |
||||||
|
{ |
||||||
|
Q_ASSERT(m_tagItems.size() > 1); |
||||||
|
return &m_tagItems[1]; |
||||||
|
} |
||||||
@ -0,0 +1,88 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef TAGFILTERMODEL_H |
||||||
|
#define TAGFILTERMODEL_H |
||||||
|
|
||||||
|
#include <QAbstractListModel> |
||||||
|
#include <QHash> |
||||||
|
#include <QModelIndex> |
||||||
|
#include <QSet> |
||||||
|
#include <QVector> |
||||||
|
|
||||||
|
namespace BitTorrent |
||||||
|
{ |
||||||
|
class TorrentHandle; |
||||||
|
} |
||||||
|
|
||||||
|
class TagModelItem; |
||||||
|
|
||||||
|
class TagFilterModel: public QAbstractListModel |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
public: |
||||||
|
explicit TagFilterModel(QObject *parent = nullptr); |
||||||
|
~TagFilterModel(); |
||||||
|
|
||||||
|
static bool isSpecialItem(const QModelIndex &index); |
||||||
|
|
||||||
|
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override; |
||||||
|
Qt::ItemFlags flags(const QModelIndex &index) const override; |
||||||
|
QVariant headerData(int section, Qt::Orientation orientation, int role) const override; |
||||||
|
QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const override; |
||||||
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override; |
||||||
|
|
||||||
|
QModelIndex index(const QString &tag) const; |
||||||
|
QString tag(const QModelIndex &index) const; |
||||||
|
|
||||||
|
private slots: |
||||||
|
void tagAdded(const QString &tag); |
||||||
|
void tagRemoved(const QString &tag); |
||||||
|
void torrentTagAdded(BitTorrent::TorrentHandle *const torrent, const QString &tag); |
||||||
|
void torrentTagRemoved(BitTorrent::TorrentHandle *const, const QString &tag); |
||||||
|
void torrentAdded(BitTorrent::TorrentHandle *const torrent); |
||||||
|
void torrentAboutToBeRemoved(BitTorrent::TorrentHandle *const torrent); |
||||||
|
|
||||||
|
private: |
||||||
|
static QString tagDisplayName(const QString &tag); |
||||||
|
|
||||||
|
void populate(); |
||||||
|
void addToModel(const QString &tag, int count); |
||||||
|
void removeFromModel(int row); |
||||||
|
bool isValidRow(int row) const; |
||||||
|
int findRow(const QString &tag) const; |
||||||
|
TagModelItem *findItem(const QString &tag); |
||||||
|
QVector<TagModelItem *> findItems(const QSet<QString> &tags); |
||||||
|
TagModelItem *allTagsItem(); |
||||||
|
TagModelItem *untaggedItem(); |
||||||
|
|
||||||
|
QList<TagModelItem> m_tagItems; // Index corresponds to its row
|
||||||
|
}; |
||||||
|
|
||||||
|
#endif // TAGFILTERMODEL_H
|
||||||
@ -0,0 +1,56 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "tagfilterproxymodel.h" |
||||||
|
|
||||||
|
#include "base/utils/string.h" |
||||||
|
#include "tagfiltermodel.h" |
||||||
|
|
||||||
|
TagFilterProxyModel::TagFilterProxyModel(QObject *parent) |
||||||
|
: QSortFilterProxyModel(parent) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
QModelIndex TagFilterProxyModel::index(const QString &tag) const |
||||||
|
{ |
||||||
|
return mapFromSource(static_cast<TagFilterModel *>(sourceModel())->index(tag)); |
||||||
|
} |
||||||
|
|
||||||
|
QString TagFilterProxyModel::tag(const QModelIndex &index) const |
||||||
|
{ |
||||||
|
return static_cast<TagFilterModel *>(sourceModel())->tag(mapToSource(index)); |
||||||
|
} |
||||||
|
|
||||||
|
bool TagFilterProxyModel::lessThan(const QModelIndex &left, const QModelIndex &right) const |
||||||
|
{ |
||||||
|
// "All" and "Untagged" must be left in place
|
||||||
|
if (TagFilterModel::isSpecialItem(left) || TagFilterModel::isSpecialItem(right)) |
||||||
|
return left.row() < right.row(); |
||||||
|
return Utils::String::naturalCompareCaseInsensitive( |
||||||
|
left.data().toString(), right.data().toString()); |
||||||
|
} |
||||||
@ -0,0 +1,52 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef TAGFILTERPROXYMODEL_H |
||||||
|
#define TAGFILTERPROXYMODEL_H |
||||||
|
|
||||||
|
#include <QSortFilterProxyModel> |
||||||
|
#include <QString> |
||||||
|
|
||||||
|
class TagFilterProxyModel: public QSortFilterProxyModel |
||||||
|
{ |
||||||
|
public: |
||||||
|
explicit TagFilterProxyModel(QObject *parent = nullptr); |
||||||
|
|
||||||
|
// TagFilterModel methods which we need to relay
|
||||||
|
QModelIndex index(const QString &tag) const; |
||||||
|
QString tag(const QModelIndex &index) const; |
||||||
|
|
||||||
|
protected: |
||||||
|
bool lessThan(const QModelIndex &left, const QModelIndex &right) const override; |
||||||
|
|
||||||
|
private: |
||||||
|
// we added another overload of index(), hence this using directive:
|
||||||
|
using QSortFilterProxyModel::index; |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // TAGFILTERPROXYMODEL_H
|
||||||
@ -0,0 +1,224 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "tagfilterwidget.h" |
||||||
|
|
||||||
|
#include <QAction> |
||||||
|
#include <QDebug> |
||||||
|
#include <QHeaderView> |
||||||
|
#include <QLayout> |
||||||
|
#include <QMenu> |
||||||
|
#include <QMessageBox> |
||||||
|
|
||||||
|
#include "base/bittorrent/session.h" |
||||||
|
#include "base/utils/misc.h" |
||||||
|
#include "autoexpandabledialog.h" |
||||||
|
#include "guiiconprovider.h" |
||||||
|
#include "tagfiltermodel.h" |
||||||
|
#include "tagfilterproxymodel.h" |
||||||
|
|
||||||
|
namespace |
||||||
|
{ |
||||||
|
QString getTagFilter(const TagFilterProxyModel *const model, const QModelIndex &index) |
||||||
|
{ |
||||||
|
QString tagFilter; // Defaults to All
|
||||||
|
if (index.isValid()) { |
||||||
|
if (index.row() == 1) |
||||||
|
tagFilter = ""; // Untagged
|
||||||
|
else if (index.row() > 1) |
||||||
|
tagFilter = model->tag(index); |
||||||
|
} |
||||||
|
return tagFilter; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
TagFilterWidget::TagFilterWidget(QWidget *parent) |
||||||
|
: QTreeView(parent) |
||||||
|
{ |
||||||
|
TagFilterProxyModel *proxyModel = new TagFilterProxyModel(this); |
||||||
|
proxyModel->setSortCaseSensitivity(Qt::CaseInsensitive); |
||||||
|
proxyModel->setSourceModel(new TagFilterModel(this)); |
||||||
|
setModel(proxyModel); |
||||||
|
setFrameShape(QFrame::NoFrame); |
||||||
|
setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff); |
||||||
|
setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff); |
||||||
|
setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed); |
||||||
|
setUniformRowHeights(true); |
||||||
|
setHeaderHidden(true); |
||||||
|
setIconSize(Utils::Misc::smallIconSize()); |
||||||
|
#if defined(Q_OS_MAC) |
||||||
|
setAttribute(Qt::WA_MacShowFocusRect, false); |
||||||
|
#endif |
||||||
|
setContextMenuPolicy(Qt::CustomContextMenu); |
||||||
|
sortByColumn(0, Qt::AscendingOrder); |
||||||
|
setCurrentIndex(model()->index(0, 0)); |
||||||
|
|
||||||
|
connect(this, &TagFilterWidget::collapsed, this, &TagFilterWidget::callUpdateGeometry); |
||||||
|
connect(this, &TagFilterWidget::expanded, this, &TagFilterWidget::callUpdateGeometry); |
||||||
|
connect(this, &TagFilterWidget::customContextMenuRequested, this, &TagFilterWidget::showMenu); |
||||||
|
connect(selectionModel(), &QItemSelectionModel::currentRowChanged, this |
||||||
|
, &TagFilterWidget::onCurrentRowChanged); |
||||||
|
connect(model(), &QAbstractItemModel::modelReset, this, &TagFilterWidget::callUpdateGeometry); |
||||||
|
} |
||||||
|
|
||||||
|
QString TagFilterWidget::currentTag() const |
||||||
|
{ |
||||||
|
QModelIndex current; |
||||||
|
auto selectedRows = selectionModel()->selectedRows(); |
||||||
|
if (!selectedRows.isEmpty()) |
||||||
|
current = selectedRows.first(); |
||||||
|
|
||||||
|
return getTagFilter(static_cast<TagFilterProxyModel *>(model()), current); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::onCurrentRowChanged(const QModelIndex ¤t, const QModelIndex &previous) |
||||||
|
{ |
||||||
|
Q_UNUSED(previous); |
||||||
|
|
||||||
|
emit tagChanged(getTagFilter(static_cast<TagFilterProxyModel *>(model()), current)); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::showMenu(QPoint) |
||||||
|
{ |
||||||
|
QMenu menu(this); |
||||||
|
|
||||||
|
QAction *addAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("list-add") |
||||||
|
, tr("Add tag...")); |
||||||
|
connect(addAct, &QAction::triggered, this, &TagFilterWidget::addTag); |
||||||
|
|
||||||
|
auto selectedRows = selectionModel()->selectedRows(); |
||||||
|
if (!selectedRows.empty() && !TagFilterModel::isSpecialItem(selectedRows.first())) { |
||||||
|
QAction *removeAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("list-remove") |
||||||
|
, tr("Remove tag")); |
||||||
|
connect(removeAct, &QAction::triggered, this, &TagFilterWidget::removeTag); |
||||||
|
} |
||||||
|
|
||||||
|
QAction *removeUnusedAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("list-remove") |
||||||
|
, tr("Remove unused tags")); |
||||||
|
connect(removeUnusedAct, &QAction::triggered, this, &TagFilterWidget::removeUnusedTags); |
||||||
|
|
||||||
|
menu.addSeparator(); |
||||||
|
|
||||||
|
QAction *startAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("media-playback-start") |
||||||
|
, tr("Resume torrents")); |
||||||
|
connect(startAct, &QAction::triggered |
||||||
|
, this, &TagFilterWidget::actionResumeTorrentsTriggered); |
||||||
|
|
||||||
|
QAction *pauseAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("media-playback-pause") |
||||||
|
, tr("Pause torrents")); |
||||||
|
connect(pauseAct, &QAction::triggered, this |
||||||
|
, &TagFilterWidget::actionPauseTorrentsTriggered); |
||||||
|
|
||||||
|
QAction *deleteTorrentsAct = menu.addAction( |
||||||
|
GuiIconProvider::instance()->getIcon("edit-delete") |
||||||
|
, tr("Delete torrents")); |
||||||
|
connect(deleteTorrentsAct, &QAction::triggered, this |
||||||
|
, &TagFilterWidget::actionDeleteTorrentsTriggered); |
||||||
|
|
||||||
|
menu.exec(QCursor::pos()); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::callUpdateGeometry() |
||||||
|
{ |
||||||
|
updateGeometry(); |
||||||
|
} |
||||||
|
|
||||||
|
QSize TagFilterWidget::sizeHint() const |
||||||
|
{ |
||||||
|
return viewportSizeHint(); |
||||||
|
} |
||||||
|
|
||||||
|
QSize TagFilterWidget::minimumSizeHint() const |
||||||
|
{ |
||||||
|
QSize size = sizeHint(); |
||||||
|
size.setWidth(6); |
||||||
|
return size; |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::rowsInserted(const QModelIndex &parent, int start, int end) |
||||||
|
{ |
||||||
|
QTreeView::rowsInserted(parent, start, end); |
||||||
|
updateGeometry(); |
||||||
|
} |
||||||
|
|
||||||
|
QString TagFilterWidget::askTagName() |
||||||
|
{ |
||||||
|
bool ok = false; |
||||||
|
QString tag = ""; |
||||||
|
bool invalid = true; |
||||||
|
while (invalid) { |
||||||
|
invalid = false; |
||||||
|
tag = AutoExpandableDialog::getText( |
||||||
|
this, tr("New Tag"), tr("Tag:"), QLineEdit::Normal, tag, &ok).trimmed(); |
||||||
|
if (ok && !tag.isEmpty()) { |
||||||
|
if (!BitTorrent::Session::isValidTag(tag)) { |
||||||
|
QMessageBox::warning( |
||||||
|
this, tr("Invalid tag name") |
||||||
|
, tr("Tag name '%1' is invalid").arg(tag)); |
||||||
|
invalid = true; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return ok ? tag : QString(); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::addTag() |
||||||
|
{ |
||||||
|
const QString tag = askTagName(); |
||||||
|
if (tag.isEmpty()) return; |
||||||
|
|
||||||
|
if (BitTorrent::Session::instance()->tags().contains(tag)) |
||||||
|
QMessageBox::warning(this, tr("Tag exists"), tr("Tag name already exists.")); |
||||||
|
else |
||||||
|
BitTorrent::Session::instance()->addTag(tag); |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::removeTag() |
||||||
|
{ |
||||||
|
auto selectedRows = selectionModel()->selectedRows(); |
||||||
|
if (!selectedRows.empty() && !TagFilterModel::isSpecialItem(selectedRows.first())) { |
||||||
|
BitTorrent::Session::instance()->removeTag( |
||||||
|
static_cast<TagFilterProxyModel *>(model())->tag(selectedRows.first())); |
||||||
|
updateGeometry(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void TagFilterWidget::removeUnusedTags() |
||||||
|
{ |
||||||
|
auto session = BitTorrent::Session::instance(); |
||||||
|
foreach (const QString &tag, session->tags()) |
||||||
|
if (model()->data(static_cast<TagFilterProxyModel *>(model())->index(tag), Qt::UserRole) == 0) |
||||||
|
session->removeTag(tag); |
||||||
|
updateGeometry(); |
||||||
|
} |
||||||
@ -0,0 +1,64 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2017 Tony Gregerson <tony.gregerson@gmail.com> |
||||||
|
* |
||||||
|
* This program 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 2 |
||||||
|
* of the License, or (at your option) any later version. |
||||||
|
* |
||||||
|
* This program 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 this program; if not, write to the Free Software |
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
||||||
|
* |
||||||
|
* In addition, as a special exception, the copyright holders give permission to |
||||||
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||||
|
* modified versions of it that use the same license as the "OpenSSL" library), |
||||||
|
* and distribute the linked executables. You must obey the GNU General Public |
||||||
|
* License in all respects for all of the code used other than "OpenSSL". If you |
||||||
|
* modify file(s), you may extend this exception to your version of the file(s), |
||||||
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||||
|
* exception statement from your version. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef TAGFILTERWIDGET_H |
||||||
|
#define TAGFILTERWIDGET_H |
||||||
|
|
||||||
|
#include <QTreeView> |
||||||
|
|
||||||
|
class TagFilterWidget: public QTreeView |
||||||
|
{ |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
public: |
||||||
|
explicit TagFilterWidget(QWidget *parent = nullptr); |
||||||
|
|
||||||
|
QString currentTag() const; |
||||||
|
|
||||||
|
signals: |
||||||
|
void tagChanged(const QString &tag); |
||||||
|
void actionResumeTorrentsTriggered(); |
||||||
|
void actionPauseTorrentsTriggered(); |
||||||
|
void actionDeleteTorrentsTriggered(); |
||||||
|
|
||||||
|
private slots: |
||||||
|
void onCurrentRowChanged(const QModelIndex ¤t, const QModelIndex &previous); |
||||||
|
void showMenu(QPoint); |
||||||
|
void callUpdateGeometry(); |
||||||
|
void addTag(); |
||||||
|
void removeTag(); |
||||||
|
void removeUnusedTags(); |
||||||
|
|
||||||
|
private: |
||||||
|
QSize sizeHint() const override; |
||||||
|
QSize minimumSizeHint() const override; |
||||||
|
void rowsInserted(const QModelIndex &parent, int start, int end) override; |
||||||
|
QString askTagName(); |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // TAGFILTERWIDGET_H
|
||||||
Loading…
Reference in new issue