src/kitemviews/private/kfileitemmodelfilter.cpp

   1 /*
   2  * SPDX-FileCopyrightText: 2011 Janardhan Reddy <annapareddyjanardhanreddy@gmail.com>
   3  *
   4  * SPDX-License-Identifier: GPL-2.0-or-later
   5  */
   6
   7 #include "kfileitemmodelfilter.h"
   8
   9 #include <QRegularExpression>
  10
  11 #include <algorithm>
  12
  13 #include <KFileItem>
  14
  15 KFileItemModelFilter::KFileItemModelFilter()
  16     : m_useRegExp(false)
  17     , m_regExp(nullptr)
  18     , m_lowerCasePattern()
  19     , m_pattern()
  20 {
  21 }
  22
  23 KFileItemModelFilter::~KFileItemModelFilter()
  24 {
  25     delete m_regExp;
  26     m_regExp = nullptr;
  27 }
  28
  29 void KFileItemModelFilter::setPattern(const QString &filter)
  30 {
  31     m_pattern = filter;
  32     m_lowerCasePattern = filter.toLower();
  33
  34     if (filter.contains(QLatin1Char('*')) || filter.contains(QLatin1Char('?')) || filter.contains(QLatin1Char('['))) {
  35         if (!m_regExp) {
  36             m_regExp = new QRegularExpression();
  37             m_regExp->setPatternOptions(QRegularExpression::CaseInsensitiveOption);
  38         }
  39         m_regExp->setPattern(QRegularExpression::wildcardToRegularExpression(filter));
  40         m_useRegExp = m_regExp->isValid();
  41     } else {
  42         m_useRegExp = false;
  43     }
  44 }
  45
  46 QString KFileItemModelFilter::pattern() const
  47 {
  48     return m_pattern;
  49 }
  50
  51 void KFileItemModelFilter::setMimeTypes(const QStringList &types)
  52 {
  53     m_mimeTypes = types;
  54 }
  55
  56 QStringList KFileItemModelFilter::mimeTypes() const
  57 {
  58     return m_mimeTypes;
  59 }
  60
  61 void KFileItemModelFilter::setExcludeMimeTypes(const QStringList &types)
  62 {
  63     m_excludeMimeTypes = types;
  64 }
  65
  66 QStringList KFileItemModelFilter::excludeMimeTypes() const
  67 {
  68     return m_excludeMimeTypes;
  69 }
  70
  71 bool KFileItemModelFilter::hasSetFilters() const
  72 {
  73     return (!m_pattern.isEmpty() || !m_mimeTypes.isEmpty() || !m_excludeMimeTypes.isEmpty());
  74 }
  75
  76 bool KFileItemModelFilter::matches(const KFileItem &item) const
  77 {
  78     const bool hasPatternFilter = !m_pattern.isEmpty();
  79     const bool hasMimeTypesFilter = !m_mimeTypes.isEmpty() || !m_excludeMimeTypes.isEmpty();
  80
  81     // If no filter is set, return true.
  82     if (!hasPatternFilter && !hasMimeTypesFilter) {
  83         return true;
  84     }
  85
  86     // If both filters are set, return true when both filters are matched
  87     if (hasPatternFilter && hasMimeTypesFilter) {
  88         return (matchesPattern(item) && matchesType(item));
  89     }
  90
  91     // If only one filter is set, return true when that filter is matched
  92     if (hasPatternFilter) {
  93         return matchesPattern(item);
  94     }
  95
  96     return matchesType(item);
  97 }
  98
  99 bool KFileItemModelFilter::matchesPattern(const KFileItem &item) const
 100 {
 101     if (m_useRegExp) {
 102         return m_regExp->match(item.text()).hasMatch();
 103     } else {
 104         return item.text().toLower().contains(m_lowerCasePattern);
 105     }
 106 }
 107
 108 bool KFileItemModelFilter::matchesType(const KFileItem &item) const
 109 {
 110     bool excluded = std::any_of(m_excludeMimeTypes.constBegin(), m_excludeMimeTypes.constEnd(), [item](const QString &excludeMimetype) {
 111         return item.mimetype() == excludeMimetype;
 112     });
 113     if (excluded) {
 114         return false;
 115     }
 116
 117     bool included = std::any_of(m_mimeTypes.constBegin(), m_mimeTypes.constEnd(), [item](const QString &mimeType) {
 118         return item.mimetype() == mimeType;
 119     });
 120     if (included) {
 121         return true;
 122     }
 123
 124     return m_mimeTypes.isEmpty();
 125 }