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