1 /*
2  * Bittorrent Client using Qt and libtorrent.
3  * Copyright (C) 2013  sledgehammer999 <hammered999@gmail.com>
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version 2
8  * of the License, or (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
18  *
19  * In addition, as a special exception, the copyright holders give permission to
20  * link this program with the OpenSSL project's "OpenSSL" library (or with
21  * modified versions of it that use the same license as the "OpenSSL" library),
22  * and distribute the linked executables. You must obey the GNU General Public
23  * License in all respects for all of the code used other than "OpenSSL".  If you
24  * modify file(s), you may extend this exception to your version of the file(s),
25  * but you are not obligated to do so. If you do not wish to do so, delete this
26  * exception statement from your version.
27  */
28 
29 #pragma once
30 
31 #include <QSortFilterProxyModel>
32 #include <QStringList>
33 
34 class SearchSortModel final : public QSortFilterProxyModel
35 {
36     using base = QSortFilterProxyModel;
37 
38 public:
39     enum SearchColumn
40     {
41         NAME,
42         SIZE,
43         SEEDS,
44         LEECHES,
45         ENGINE_URL,
46         DL_LINK,
47         DESC_LINK,
48         NB_SEARCH_COLUMNS
49     };
50 
51     enum SearchDataRole
52     {
53         UnderlyingDataRole = Qt::UserRole
54     };
55 
56     explicit SearchSortModel(QObject *parent = nullptr);
57 
58     void enableNameFilter(bool enabled);
59     void setNameFilter(const QString &searchTerm = {});
60 
61     //! \brief Sets parameters for filtering by size
62     //! \param minSize minimal size in bytes
63     //! \param maxSize maximal size in bytes, negative value to disable filtering
64     void setSizeFilter(qint64 minSize, qint64 maxSize);
65 
66     //! \brief Sets parameters for filtering by seeds number
67     //! \param minSeeds minimal number of seeders
68     //! \param maxSeeds maximal number of seeders, negative value to disable filtering
69     void setSeedsFilter(int minSeeds, int maxSeeds);
70 
71     //! \brief Sets parameters for filtering by leeches number
72     //! \param minLeeches minimal number of leechers
73     //! \param maxLeeches maximal number of leechers, negative value to disable filtering
74     void setLeechesFilter(int minLeeches, int maxLeeches);
75 
76     bool isNameFilterEnabled() const;
77 
78     QString searchTerm() const;
79 
80     int minSeeds() const;
81     int maxSeeds() const;
82 
83     qint64 minSize() const;
84     qint64 maxSize() const;
85 
86 protected:
87     bool lessThan(const QModelIndex &left, const QModelIndex &right) const override;
88     bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const override;
89 
90 private:
91     bool m_isNameFilterEnabled;
92     QString m_searchTerm;
93     QStringList m_searchTermWords;
94     int m_minSeeds, m_maxSeeds;
95     int m_minLeeches, m_maxLeeches;
96     qint64 m_minSize, m_maxSize;
97 };
98