File indexing completed on 2024-05-12 15:42:12

 /*
     This file is part of the KDE project
     SPDX-FileCopyrightText: 1999 Simon Hausmann <hausmann@kde.org>
     SPDX-FileCopyrightText: 2000 Yves Arrouye <yves@realnames.com>
     SPDX-FileCopyrightText: 2002, 2003 Dawit Alemayehu <adawit@kde.org>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "kuriikwsfilter.h"
 #include "ikwsopts.h"
 #include "kuriikwsfiltereng.h"
 #include "searchprovider.h"
 
 #include <KLocalizedString>
 #include <KPluginFactory>
 
 #include <QDBusConnection>
 #include <QLoggingCategory>
 
 namespace
 {
 Q_LOGGING_CATEGORY(category, "kf.kio.urifilters.ikws", QtWarningMsg)
 }
 
 /**
  * IMPORTANT: If you change anything here, make sure you run the kurifiltertest
  * regression test (this should be included as part of "make test").
  */
 
 K_PLUGIN_CLASS_WITH_JSON(KAutoWebSearch, "kuriikwsfilter.json")
 
 KAutoWebSearch::KAutoWebSearch(QObject *parent, const QVariantList &)
     : KUriFilterPlugin(QStringLiteral("kuriikwsfilter"), parent)
 {
     QDBusConnection::sessionBus()
         .connect(QString(), QStringLiteral("/"), QStringLiteral("org.kde.KUriFilterPlugin"), QStringLiteral("configure"), this, SLOT(configure()));
 }
 
 KAutoWebSearch::~KAutoWebSearch()
 {
 }
 
 void KAutoWebSearch::configure()
 {
     qCDebug(category) << "Config reload requested...";
     KURISearchFilterEngine::self()->loadConfig();
 }
 
 void KAutoWebSearch::populateProvidersList(QList<KUriFilterSearchProvider *> &searchProviders, const KUriFilterData &data, bool allproviders) const
 {
     QList<SearchProvider *> providers;
     KURISearchFilterEngine *filter = KURISearchFilterEngine::self();
 
     if (allproviders) {
         providers = filter->registry()->findAll();
     } else {
         // Start with the search engines marked as preferred...
         QStringList favEngines = filter->favoriteEngineList();
         if (favEngines.isEmpty()) {
             favEngines = data.alternateSearchProviders();
         }
 
         // Get rid of duplicates...
         favEngines.removeDuplicates();
 
         // Sort the items...
         std::stable_sort(favEngines.begin(), favEngines.end());
 
         // Add the search engine set as the default provider...
         const QString defaultEngine = filter->defaultSearchEngine();
         if (!defaultEngine.isEmpty()) {
             favEngines.removeAll(defaultEngine);
             favEngines.insert(0, defaultEngine);
         }
 
         QStringListIterator it(favEngines);
         while (it.hasNext()) {
             SearchProvider *favProvider = filter->registry()->findByDesktopName(it.next());
             if (favProvider) {
                 providers << favProvider;
             }
         }
     }
 
     for (SearchProvider *p : std::as_const(providers)) {
         searchProviders << p;
     }
 }
 
 bool KAutoWebSearch::filterUri(KUriFilterData &data) const
 {
     qCDebug(category) << data.typedString();
 
     KUriFilterData::SearchFilterOptions option = data.searchFilteringOptions();
 
     // Handle the flag to retrieve only preferred providers, no filtering...
     if (option & KUriFilterData::RetrievePreferredSearchProvidersOnly) {
         QList<KUriFilterSearchProvider *> searchProviders;
         populateProvidersList(searchProviders, data);
         if (searchProviders.isEmpty()) {
             if (!(option & KUriFilterData::RetrieveSearchProvidersOnly)) {
                 setUriType(data, KUriFilterData::Error);
                 setErrorMsg(data, i18n("No preferred search providers were found."));
                 return false;
             }
         } else {
             setSearchProvider(data, QString(), data.typedString(), QLatin1Char(KURISearchFilterEngine::self()->keywordDelimiter()));
             setSearchProviders(data, searchProviders);
             return true;
         }
     }
 
     if (option & KUriFilterData::RetrieveSearchProvidersOnly) {
         QList<KUriFilterSearchProvider *> searchProviders;
         populateProvidersList(searchProviders, data, true);
         if (searchProviders.isEmpty()) {
             setUriType(data, KUriFilterData::Error);
             setErrorMsg(data, i18n("No search providers were found."));
             return false;
         }
 
         setSearchProvider(data, QString(), data.typedString(), QLatin1Char(KURISearchFilterEngine::self()->keywordDelimiter()));
         setSearchProviders(data, searchProviders);
         return true;
     }
 
     if (data.uriType() == KUriFilterData::Unknown && data.uri().password().isEmpty()) {
         KURISearchFilterEngine *filter = KURISearchFilterEngine::self();
         SearchProvider *provider = filter->autoWebSearchQuery(data.typedString(), data.alternateDefaultSearchProvider());
         if (provider) {
             const QUrl result = filter->formatResult(provider->query(), provider->charset(), QString(), data.typedString(), true);
             setFilteredUri(data, result);
             setUriType(data, KUriFilterData::NetProtocol);
             setSearchProvider(data, provider->name(), data.typedString(), QLatin1Char(filter->keywordDelimiter()));
 
             QList<KUriFilterSearchProvider *> searchProviders;
             populateProvidersList(searchProviders, data);
             setSearchProviders(data, searchProviders);
             return true;
         }
     }
     return false;
 }
 
 #include "kuriikwsfilter.moc"
 
 #include "moc_kuriikwsfilter.cpp"