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 <KFileItem>
  12
  13 KFileItemModelFilter::KFileItemModelFilter()
  14     : m_useRegExp(false)
  15     , m_regExp(nullptr)
  16     , m_lowerCasePattern()
  17     , m_pattern()
  18 {
  19 }
  20
  21 KFileItemModelFilter::~KFileItemModelFilter()
  22 {
  23     delete m_regExp;
  24     m_regExp = nullptr;
  25 }
  26
  27 void KFileItemModelFilter::setPattern(const QString &filter)
  28 {
  29     m_pattern = filter;
  30     m_lowerCasePattern = filter.toLower();
  31
  32     if (filter.contains('*') || filter.contains('?') || filter.contains('[')) {
  33         if (!m_regExp) {
  34             m_regExp = new QRegularExpression();
  35             m_regExp->setPatternOptions(QRegularExpression::CaseInsensitiveOption);
  36         }
  37         m_regExp->setPattern(QRegularExpression::wildcardToRegularExpression(filter));
  38         m_useRegExp = m_regExp->isValid();
  39     } else {
  40         m_useRegExp = false;
  41     }
  42 }
  43
  44 QString KFileItemModelFilter::pattern() const
  45 {
  46     return m_pattern;
  47 }
  48
  49 void KFileItemModelFilter::setMimeTypes(const QStringList &types)
  50 {
  51     m_mimeTypes = types;
  52 }
  53
  54 QStringList KFileItemModelFilter::mimeTypes() const
  55 {
  56     return m_mimeTypes;
  57 }
  58
  59 bool KFileItemModelFilter::hasSetFilters() const
  60 {
  61     return (!m_pattern.isEmpty() || !m_mimeTypes.isEmpty());
  62 }
  63
  64 bool KFileItemModelFilter::matches(const KFileItem &item) const
  65 {
  66     const bool hasPatternFilter = !m_pattern.isEmpty();
  67     const bool hasMimeTypesFilter = !m_mimeTypes.isEmpty();
  68
  69     // If no filter is set, return true.
  70     if (!hasPatternFilter && !hasMimeTypesFilter) {
  71         return true;
  72     }
  73
  74     // If both filters are set, return true when both filters are matched
  75     if (hasPatternFilter && hasMimeTypesFilter) {
  76         return (matchesPattern(item) && matchesType(item));
  77     }
  78
  79     // If only one filter is set, return true when that filter is matched
  80     if (hasPatternFilter) {
  81         return matchesPattern(item);
  82     }
  83
  84     return matchesType(item);
  85 }
  86
  87 bool KFileItemModelFilter::matchesPattern(const KFileItem &item) const
  88 {
  89     if (m_useRegExp) {
  90         return m_regExp->match(item.text()).hasMatch();
  91     } else {
  92         return item.text().toLower().contains(m_lowerCasePattern);
  93     }
  94 }
  95
  96 bool KFileItemModelFilter::matchesType(const KFileItem &item) const
  97 {
  98     for (const QString &mimeType : qAsConst(m_mimeTypes)) {
  99         if (item.mimetype() == mimeType) {
 100             return true;
 101         }
 102     }
 103
 104     return m_mimeTypes.isEmpty();
 105 }