src/gui/search/searchsortmodel.h

   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 #include "base/utils/compare.h"
  35
  36 class SearchSortModel final : public QSortFilterProxyModel
  37 {
  38     using base = QSortFilterProxyModel;
  39
  40 public:
  41     enum SearchColumn
  42     {
  43         NAME,
  44         SIZE,
  45         SEEDS,
  46         LEECHES,
  47         ENGINE_URL,
  48         DL_LINK,
  49         DESC_LINK,
  50         NB_SEARCH_COLUMNS
  51     };
  52
  53     enum SearchDataRole
  54     {
  55         UnderlyingDataRole = Qt::UserRole
  56     };
  57
  58     explicit SearchSortModel(QObject *parent = nullptr);
  59
  60     void enableNameFilter(bool enabled);
  61     void setNameFilter(const QString &searchTerm = {});
  62
  63     //! \brief Sets parameters for filtering by size
  64     //! \param minSize minimal size in bytes
  65     //! \param maxSize maximal size in bytes, negative value to disable filtering
  66     void setSizeFilter(qint64 minSize, qint64 maxSize);
  67
  68     //! \brief Sets parameters for filtering by seeds number
  69     //! \param minSeeds minimal number of seeders
  70     //! \param maxSeeds maximal number of seeders, negative value to disable filtering
  71     void setSeedsFilter(int minSeeds, int maxSeeds);
  72
  73     //! \brief Sets parameters for filtering by leeches number
  74     //! \param minLeeches minimal number of leechers
  75     //! \param maxLeeches maximal number of leechers, negative value to disable filtering
  76     void setLeechesFilter(int minLeeches, int maxLeeches);
  77
  78     bool isNameFilterEnabled() const;
  79
  80     QString searchTerm() const;
  81
  82     int minSeeds() const;
  83     int maxSeeds() const;
  84
  85     qint64 minSize() const;
  86     qint64 maxSize() const;
  87
  88 protected:
  89     bool lessThan(const QModelIndex &left, const QModelIndex &right) const override;
  90     bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const override;
  91
  92 private:
  93     bool m_isNameFilterEnabled = false;
  94     QString m_searchTerm;
  95     QStringList m_searchTermWords;
  96     int m_minSeeds = 0, m_maxSeeds = -1;
  97     int m_minLeeches = 0, m_maxLeeches = -1;
  98     qint64 m_minSize = 0, m_maxSize = -1;
  99
 100     Utils::Compare::NaturalLessThan<Qt::CaseInsensitive> m_naturalLessThan;
 101 };