/* * Bittorrent Client using Qt and libtorrent. * Copyright (C) 2015, 2018 Vladimir Golovnev * Copyright (C) 2006 Christophe Dumez * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * * In addition, as a special exception, the copyright holders give permission to * link this program with the OpenSSL project's "OpenSSL" library (or with * modified versions of it that use the same license as the "OpenSSL" library), * and distribute the linked executables. You must obey the GNU General Public * License in all respects for all of the code used other than "OpenSSL". If you * modify file(s), you may extend this exception to your version of the file(s), * but you are not obligated to do so. If you do not wish to do so, delete this * exception statement from your version. */ #include "searchwidget.h" #include #ifdef Q_OS_WIN #include #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "base/bittorrent/session.h" #include "base/preferences.h" #include "base/search/searchpluginmanager.h" #include "base/search/searchhandler.h" #include "base/utils/fs.h" #include "base/utils/misc.h" #include "addnewtorrentdialog.h" #include "guiiconprovider.h" #include "mainwindow.h" #include "pluginselectdlg.h" #include "searchlistdelegate.h" #include "searchsortmodel.h" #include "searchtab.h" #include "ui_searchwidget.h" #define SEARCHHISTORY_MAXSIZE 50 #define URL_COLUMN 5 namespace { QString statusIconName(SearchTab::Status st) { switch (st) { case SearchTab::Status::Ongoing: return QLatin1String("task-ongoing"); case SearchTab::Status::Finished: return QLatin1String("task-complete"); case SearchTab::Status::Aborted: return QLatin1String("task-reject"); case SearchTab::Status::Error: return QLatin1String("task-attention"); case SearchTab::Status::NoResults: return QLatin1String("task-attention"); default: return QString(); } } } SearchWidget::SearchWidget(MainWindow *mainWindow) : QWidget(mainWindow) , m_ui(new Ui::SearchWidget()) , m_tabStatusChangedMapper(new QSignalMapper(this)) , m_mainWindow(mainWindow) , m_isNewQueryString(false) { m_ui->setupUi(this); QString searchPatternHint; QTextStream stream(&searchPatternHint, QIODevice::WriteOnly); stream << "

" << tr("A phrase to search for.") << "
" << tr("Spaces in a search term may be protected by double quotes.") << "

" << tr("Example:", "Search phrase example") << "
" << tr("foo bar: search for foo and bar", "Search phrase example, illustrates quotes usage, a pair of " "space delimited words, individal words are highlighted") << "
" << tr(""foo bar": search for foo bar", "Search phrase example, illustrates quotes usage, double quoted" "pair of space delimited words, the whole pair is highlighted") << "

" << flush; m_ui->m_searchPattern->setToolTip(searchPatternHint); #ifndef Q_OS_MAC // Icons m_ui->searchButton->setIcon(GuiIconProvider::instance()->getIcon("edit-find")); m_ui->downloadButton->setIcon(GuiIconProvider::instance()->getIcon("download")); m_ui->goToDescBtn->setIcon(GuiIconProvider::instance()->getIcon("application-x-mswinurl")); m_ui->pluginsButton->setIcon(GuiIconProvider::instance()->getIcon("preferences-system-network")); m_ui->copyURLBtn->setIcon(GuiIconProvider::instance()->getIcon("edit-copy")); #else // On macOS the icons overlap the text otherwise QSize iconSize = m_ui->tabWidget->iconSize(); iconSize.setWidth(iconSize.width() + 16); m_ui->tabWidget->setIconSize(iconSize); #endif connect(m_ui->tabWidget, &QTabWidget::tabCloseRequested, this, &SearchWidget::closeTab); connect(m_ui->tabWidget, &QTabWidget::currentChanged, this, &SearchWidget::tabChanged); connect(m_tabStatusChangedMapper, static_cast(&QSignalMapper::mapped) , this, &SearchWidget::tabStatusChanged); // NOTE: Although SearchManager is Application-wide component now, we still create it the legacy way. auto *searchManager = new SearchPluginManager; const auto onPluginChanged = [this]() { fillPluginComboBox(); fillCatCombobox(); selectActivePage(); }; connect(searchManager, &SearchPluginManager::pluginInstalled, this, onPluginChanged); connect(searchManager, &SearchPluginManager::pluginUninstalled, this, onPluginChanged); connect(searchManager, &SearchPluginManager::pluginUpdated, this, onPluginChanged); connect(searchManager, &SearchPluginManager::pluginEnabled, this, onPluginChanged); // Fill in category combobox onPluginChanged(); connect(m_ui->m_searchPattern, &LineEdit::returnPressed, m_ui->searchButton, &QPushButton::click); connect(m_ui->m_searchPattern, &LineEdit::textEdited, this, &SearchWidget::searchTextEdited); connect(m_ui->selectPlugin, static_cast(&QComboBox::currentIndexChanged) , this, &SearchWidget::selectMultipleBox); connect(m_ui->selectPlugin, static_cast(&QComboBox::currentIndexChanged) , this, &SearchWidget::fillCatCombobox); } void SearchWidget::fillCatCombobox() { m_ui->comboCategory->clear(); m_ui->comboCategory->addItem(SearchPluginManager::categoryFullName("all"), QVariant("all")); using QStrPair = QPair; QList tmpList; foreach (const QString &cat, SearchPluginManager::instance()->getPluginCategories(selectedPlugin())) tmpList << qMakePair(SearchPluginManager::categoryFullName(cat), cat); std::sort(tmpList.begin(), tmpList.end(), [](const QStrPair &l, const QStrPair &r) { return (QString::localeAwareCompare(l.first, r.first) < 0); }); foreach (const QStrPair &p, tmpList) { qDebug("Supported category: %s", qUtf8Printable(p.second)); m_ui->comboCategory->addItem(p.first, QVariant(p.second)); } if (m_ui->comboCategory->count() > 1) m_ui->comboCategory->insertSeparator(1); } void SearchWidget::fillPluginComboBox() { m_ui->selectPlugin->clear(); m_ui->selectPlugin->addItem(tr("Only enabled"), QVariant("enabled")); m_ui->selectPlugin->addItem(tr("All plugins"), QVariant("all")); m_ui->selectPlugin->addItem(tr("Select..."), QVariant("multi")); using QStrPair = QPair; QList tmpList; foreach (const QString &name, SearchPluginManager::instance()->enabledPlugins()) tmpList << qMakePair(SearchPluginManager::instance()->pluginFullName(name), name); std::sort(tmpList.begin(), tmpList.end(), [](const QStrPair &l, const QStrPair &r) { return (l.first < r.first); } ); foreach (const QStrPair &p, tmpList) m_ui->selectPlugin->addItem(p.first, QVariant(p.second)); if (m_ui->selectPlugin->count() > 3) m_ui->selectPlugin->insertSeparator(3); } QString SearchWidget::selectedCategory() const { return m_ui->comboCategory->itemData(m_ui->comboCategory->currentIndex()).toString(); } QString SearchWidget::selectedPlugin() const { return m_ui->selectPlugin->itemData(m_ui->selectPlugin->currentIndex()).toString(); } void SearchWidget::selectActivePage() { if (SearchPluginManager::instance()->allPlugins().isEmpty()) { m_ui->stackedPages->setCurrentWidget(m_ui->emptyPage); m_ui->m_searchPattern->setEnabled(false); m_ui->comboCategory->setEnabled(false); m_ui->selectPlugin->setEnabled(false); m_ui->searchButton->setEnabled(false); } else { m_ui->stackedPages->setCurrentWidget(m_ui->searchPage); m_ui->m_searchPattern->setEnabled(true); m_ui->comboCategory->setEnabled(true); m_ui->selectPlugin->setEnabled(true); m_ui->searchButton->setEnabled(true); } } SearchWidget::~SearchWidget() { qDebug("Search destruction"); delete SearchPluginManager::instance(); delete m_ui; } void SearchWidget::updateButtons() { if (m_currentSearchTab && (m_currentSearchTab->visibleResultsCount() > 0)) { m_ui->downloadButton->setEnabled(true); m_ui->goToDescBtn->setEnabled(true); m_ui->copyURLBtn->setEnabled(true); } else { m_ui->downloadButton->setEnabled(false); m_ui->goToDescBtn->setEnabled(false); m_ui->copyURLBtn->setEnabled(false); } } void SearchWidget::tabChanged(int index) { // when we switch from a tab that is not empty to another that is empty // the download button doesn't have to be available m_currentSearchTab = ((index < 0) ? nullptr : m_allTabs.at(m_ui->tabWidget->currentIndex())); updateButtons(); } void SearchWidget::selectMultipleBox(int index) { Q_UNUSED(index); if (selectedPlugin() == "multi") on_pluginsButton_clicked(); } void SearchWidget::on_pluginsButton_clicked() { new PluginSelectDlg(SearchPluginManager::instance(), this); } void SearchWidget::searchTextEdited(QString) { // Enable search button m_ui->searchButton->setText(tr("Search")); m_isNewQueryString = true; } void SearchWidget::giveFocusToSearchInput() { m_ui->m_searchPattern->setFocus(); } // Function called when we click on search button void SearchWidget::on_searchButton_clicked() { if (Utils::Misc::pythonVersion() < 0) { m_mainWindow->showNotificationBaloon(tr("Search Engine"), tr("Please install Python to use the Search Engine.")); return; } if (m_activeSearchTab) { m_activeSearchTab->cancelSearch(); if (!m_isNewQueryString) { m_ui->searchButton->setText(tr("Search")); return; } } m_isNewQueryString = false; const QString pattern = m_ui->m_searchPattern->text().trimmed(); // No search pattern entered if (pattern.isEmpty()) { QMessageBox::critical(this, tr("Empty search pattern"), tr("Please type a search pattern first")); return; } QStringList plugins; if (selectedPlugin() == "all") plugins = SearchPluginManager::instance()->allPlugins(); else if (selectedPlugin() == "enabled") plugins = SearchPluginManager::instance()->enabledPlugins(); else if (selectedPlugin() == "multi") plugins = SearchPluginManager::instance()->enabledPlugins(); else plugins << selectedPlugin(); qDebug("Search with category: %s", qUtf8Printable(selectedCategory())); // Launch search auto *searchHandler = SearchPluginManager::instance()->startSearch(pattern, selectedCategory(), plugins); // Tab Addition auto *newTab = new SearchTab(searchHandler, this); m_allTabs.append(newTab); QString tabName = pattern; tabName.replace(QRegularExpression("&{1}"), "&&"); m_ui->tabWidget->addTab(newTab, tabName); m_ui->tabWidget->setCurrentWidget(newTab); connect(newTab, &SearchTab::resultsCountUpdated, this, &SearchWidget::resultsCountUpdated); connect(newTab, &SearchTab::statusChanged , m_tabStatusChangedMapper, static_cast(&QSignalMapper::map)); m_tabStatusChangedMapper->setMapping(newTab, newTab); m_ui->searchButton->setText(tr("Stop")); m_activeSearchTab = newTab; tabStatusChanged(newTab); } void SearchWidget::resultsCountUpdated() { updateButtons(); } void SearchWidget::tabStatusChanged(QWidget *tab) { const int tabIndex = m_ui->tabWidget->indexOf(tab); m_ui->tabWidget->setTabToolTip(tabIndex, tab->statusTip()); m_ui->tabWidget->setTabIcon(tabIndex, GuiIconProvider::instance()->getIcon( statusIconName(static_cast(tab)->status()))); if ((tab == m_activeSearchTab) && (m_activeSearchTab->status() != SearchTab::Status::Ongoing)) { Q_ASSERT(m_activeSearchTab->status() != SearchTab::Status::Ongoing); if (m_mainWindow->isNotificationsEnabled() && (m_mainWindow->currentTabWidget() != this)) { if (m_activeSearchTab->status() == SearchTab::Status::Error) m_mainWindow->showNotificationBaloon(tr("Search Engine"), tr("Search has failed")); else m_mainWindow->showNotificationBaloon(tr("Search Engine"), tr("Search has finished")); } m_activeSearchTab = nullptr; m_ui->searchButton->setText(tr("Search")); } } void SearchWidget::closeTab(int index) { SearchTab *tab = m_allTabs.takeAt(index); if (tab == m_activeSearchTab) m_ui->searchButton->setText(tr("Search")); delete tab; } // Download selected items in search results list void SearchWidget::on_downloadButton_clicked() { m_allTabs.at(m_ui->tabWidget->currentIndex())->downloadTorrents(); } void SearchWidget::on_goToDescBtn_clicked() { m_allTabs.at(m_ui->tabWidget->currentIndex())->openTorrentPages(); } void SearchWidget::on_copyURLBtn_clicked() { m_allTabs.at(m_ui->tabWidget->currentIndex())->copyTorrentURLs(); }