Add Duplicate feature

[dolphin.git] / src / search / dolphinquery.cpp

diff --git a/src/search/dolphinquery.cpp b/src/search/dolphinquery.cpp
index 09a841859188cd18a219329d95d7cbde32b2ec50..ab107f43fab736e3a39407017dd5a7dd28f2abbf 100644 (file)
--- a/src/search/dolphinquery.cpp
+++ b/src/search/dolphinquery.cpp
@@ -19,28 +19,53 @@
 
 #include "dolphinquery.h"
 
 
 #include "dolphinquery.h"
 

+#include <QRegularExpression>
+

 #include <config-baloo.h>
 #ifdef HAVE_BALOO
 #include <Baloo/Query>
 #endif
 
 namespace {
 #include <config-baloo.h>
 #ifdef HAVE_BALOO
 #include <Baloo/Query>
 #endif
 
 namespace {

-    /** Checks if a given term in the Baloo::Query::searchString() is a special search term.
-     * This is a copy of `DolphinFacetsWidget::isRatingTerm()` method.
+    /** Checks if a given term in the Baloo::Query::searchString() is a special search term
+     * @return: the specific search token of the term, or an empty QString() if none is found

      */
      */

-    bool isSearchTerm(const QString& term)
+    QString searchTermToken(const QString& term)

     {
         static const QLatin1String searchTokens[] {
     {
         static const QLatin1String searchTokens[] {

+            QLatin1String("filename:"),

             QLatin1String("modified>="),
             QLatin1String("modified>="),

-            QLatin1String("rating>=")
+            QLatin1String("rating>="),
+            QLatin1String("tag:"), QLatin1String("tag=")

         };
 
         for (const auto &searchToken : searchTokens) {
             if (term.startsWith(searchToken)) {
         };
 
         for (const auto &searchToken : searchTokens) {
             if (term.startsWith(searchToken)) {

-                return true;
+                return searchToken;

             }
         }
             }
         }

-        return false;
+        return QString();
+    }
+
+    QString stripQuotes(const QString& text)
+    {
+        if (text.length() >= 2 && text.at(0) == QLatin1Char('"')
+                               && text.back() == QLatin1Char('"')) {
+            return text.mid(1, text.size() - 2);
+        }
+        return text;
+    }
+
+    QStringList splitOutsideQuotes(const QString& text)
+    {
+        const QRegularExpression subTermsRegExp("(\\S*?\"[^\"]*?\"|(?<=\\s|^)\\S+(?=\\s|$))");
+        auto subTermsMatchIterator = subTermsRegExp.globalMatch(text);
+
+        QStringList textParts;
+        while (subTermsMatchIterator.hasNext()) {
+            textParts << subTermsMatchIterator.next().captured(0);
+        }
+        return textParts;

     }
 }
 
     }
 }
 


@@ -54,20 +79,44 @@ DolphinQuery DolphinQuery::fromBalooSearchUrl(const QUrl& searchUrl)
 
     model.m_includeFolder = query.includeFolder();
 
 
     model.m_includeFolder = query.includeFolder();
 

-    model.m_searchText = query.searchString();
-

     const QStringList types = query.types();
     model.m_fileType = types.isEmpty() ? QString() : types.first();
 
     const QStringList types = query.types();
     model.m_fileType = types.isEmpty() ? QString() : types.first();
 

-    const QStringList subTerms = query.searchString().split(' ', QString::SkipEmptyParts);
+    QStringList textParts;
+    QString fileName;
+
+    const QStringList subTerms = splitOutsideQuotes(query.searchString());

     foreach (const QString& subTerm, subTerms) {
     foreach (const QString& subTerm, subTerms) {

-        if (subTerm.startsWith(QLatin1String("filename:"))) {
-            const QString value = subTerm.mid(9);
-            model.m_searchText = value;
-        } else if (isSearchTerm(subTerm)) {
-            model.m_searchTerms << subTerm;
+        const QString token = searchTermToken(subTerm);
+        const QString value = stripQuotes(subTerm.mid(token.length()));
+
+        if (token == QLatin1String("filename:")) {
+            if (!value.isEmpty()) {
+                fileName = value;
+                model.m_hasFileName = true;
+            }
+            continue;
+        } else if (!token.isEmpty()) {
+            model.m_searchTerms << token + value;
+            continue;
+        } else if (subTerm == QLatin1String("AND") && subTerm != subTerms.at(0) && subTerm != subTerms.back()) {
+            continue;
+        } else if (!value.isEmpty()) {
+            textParts << value;
+            model.m_hasContentSearch = true;
+        }
+    }
+
+    if (model.m_hasFileName) {
+        if (model.m_hasContentSearch) {
+            textParts << QStringLiteral("filename:\"%1\"").arg(fileName);
+        } else {
+            textParts << fileName;

         }
     }
         }
     }

+
+    model.m_searchText = textParts.join(QLatin1Char(' '));
+

 #endif
     return model;
 }
 #endif
     return model;
 }


@@ -96,3 +145,13 @@ QString DolphinQuery::includeFolder() const
 {
     return m_includeFolder;
 }
 {
     return m_includeFolder;
 }

+
+bool DolphinQuery::hasContentSearch() const
+{
+    return m_hasContentSearch;
+}
+
+bool DolphinQuery::hasFileName() const
+{
+    return m_hasFileName;
+}