/* * Bittorrent Client using Qt4 and libtorrent. * Copyright (C) 2006-2007 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 3 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, see */ #include #include #include #include #include #include #include #include #include #include #include "SearchListDelegate.h" #include "searchEngine.h" #include "bittorrent.h" #define SEARCH_NAME 0 #define SEARCH_SIZE 1 #define SEARCH_SEEDERS 2 #define SEARCH_LEECHERS 3 #define SEARCH_ENGINE 4 #define SEARCHHISTORY_MAXSIZE 50 SearchEngine::SearchEngine(bittorrent *BTSession, QSystemTrayIcon *myTrayIcon, bool systrayIntegration) : QWidget(), systrayIntegration(systrayIntegration){ setupUi(this); this->BTSession = BTSession; this->myTrayIcon = myTrayIcon; // Set Search results list model SearchListModel = new QStandardItemModel(0,5); SearchListModel->setHeaderData(SEARCH_NAME, Qt::Horizontal, tr("Name", "i.e: file name")); SearchListModel->setHeaderData(SEARCH_SIZE, Qt::Horizontal, tr("Size", "i.e: file size")); SearchListModel->setHeaderData(SEARCH_SEEDERS, Qt::Horizontal, tr("Seeders", "i.e: Number of full sources")); SearchListModel->setHeaderData(SEARCH_LEECHERS, Qt::Horizontal, tr("Leechers", "i.e: Number of partial sources")); SearchListModel->setHeaderData(SEARCH_ENGINE, Qt::Horizontal, tr("Search engine")); resultsBrowser->setModel(SearchListModel); SearchDelegate = new SearchListDelegate(); resultsBrowser->setItemDelegate(SearchDelegate); // Make search list header clickable for sorting resultsBrowser->header()->setClickable(true); resultsBrowser->header()->setSortIndicatorShown(true); // Load last columns width for search results list if(!loadColWidthSearchList()){ resultsBrowser->header()->resizeSection(0, 275); } // new qCompleter to the search pattern startSearchHistory(); searchCompleter = new QCompleter(searchHistory, this); searchCompleter->setCaseSensitivity(Qt::CaseInsensitive); search_pattern->setCompleter(searchCompleter); // Boolean initialization search_stopped = false; // Connect signals to slots (search part) connect(resultsBrowser, SIGNAL(doubleClicked(const QModelIndex&)), this, SLOT(downloadSelectedItem(const QModelIndex&))); connect(resultsBrowser->header(), SIGNAL(sectionPressed(int)), this, SLOT(sortSearchList(int))); // Creating Search Process searchProcess = new QProcess(this); connect(searchProcess, SIGNAL(started()), this, SLOT(searchStarted())); connect(searchProcess, SIGNAL(readyReadStandardOutput()), this, SLOT(readSearchOutput())); connect(searchProcess, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(searchFinished(int,QProcess::ExitStatus))); // Set search engines names mininova->setText("Mininova"); piratebay->setText("ThePirateBay"); // reactor->setText("TorrentReactor"); isohunt->setText("Isohunt"); // btjunkie->setText("BTJunkie"); meganova->setText("Meganova"); // Check last checked search engines loadCheckedSearchEngines(); connect(mininova, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); connect(piratebay, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); // connect(reactor, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); connect(isohunt, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); // connect(btjunkie, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); connect(meganova, SIGNAL(stateChanged(int)), this, SLOT(saveCheckedSearchEngines(int))); // Update nova.py search plugin if necessary updateNova(); } SearchEngine::~SearchEngine(){ qDebug("Search destruction"); // save the searchHistory for later uses saveSearchHistory(); saveColWidthSearchList(); searchProcess->kill(); searchProcess->waitForFinished(); delete searchProcess; delete searchCompleter; delete SearchListModel; delete SearchDelegate; } // Set the color of a row in data model void SearchEngine::setRowColor(int row, const QString& color){ for(int i=0; icolumnCount(); ++i){ SearchListModel->setData(SearchListModel->index(row, i), QVariant(QColor(color)), Qt::TextColorRole); } } void SearchEngine::sortSearchList(int index){ static Qt::SortOrder sortOrder = Qt::AscendingOrder; if(resultsBrowser->header()->sortIndicatorSection() == index){ if(sortOrder == Qt::AscendingOrder){ sortOrder = Qt::DescendingOrder; }else{ sortOrder = Qt::AscendingOrder; } } resultsBrowser->header()->setSortIndicator(index, sortOrder); switch(index){ //case SIZE: case SEEDERS: case LEECHERS: case SIZE: sortSearchListInt(index, sortOrder); break; default: sortSearchListString(index, sortOrder); } } void SearchEngine::sortSearchListInt(int index, Qt::SortOrder sortOrder){ QList > lines; // Insertion sorting for(int i=0; irowCount(); ++i){ misc::insertSort(lines, QPair(i, SearchListModel->data(SearchListModel->index(i, index)).toLongLong()), sortOrder); } // Insert items in new model, in correct order int nbRows_old = lines.size(); for(int row=0; rowinsertRow(SearchListModel->rowCount()); int sourceRow = lines[row].first; for(int col=0; col<5; ++col){ SearchListModel->setData(SearchListModel->index(nbRows_old+row, col), SearchListModel->data(SearchListModel->index(sourceRow, col))); SearchListModel->setData(SearchListModel->index(nbRows_old+row, col), SearchListModel->data(SearchListModel->index(sourceRow, col), Qt::TextColorRole), Qt::TextColorRole); } } // Remove old rows SearchListModel->removeRows(0, nbRows_old); } void SearchEngine::sortSearchListString(int index, Qt::SortOrder sortOrder){ QList > lines; // Insetion sorting for(int i=0; irowCount(); ++i){ misc::insertSortString(lines, QPair(i, SearchListModel->data(SearchListModel->index(i, index)).toString()), sortOrder); } // Insert items in new model, in correct order int nbRows_old = lines.size(); for(int row=0; rowinsertRow(SearchListModel->rowCount()); int sourceRow = lines[row].first; for(int col=0; col<5; ++col){ SearchListModel->setData(SearchListModel->index(nbRows_old+row, col), SearchListModel->data(SearchListModel->index(sourceRow, col))); SearchListModel->setData(SearchListModel->index(nbRows_old+row, col), SearchListModel->data(SearchListModel->index(sourceRow, col), Qt::TextColorRole), Qt::TextColorRole); } } // Remove old rows SearchListModel->removeRows(0, nbRows_old); } // Save last checked search engines to a file void SearchEngine::saveCheckedSearchEngines(int) const{ QSettings settings("qBittorrent", "qBittorrent"); settings.beginGroup("SearchEngines"); settings.setValue("mininova", mininova->isChecked()); settings.setValue("piratebay", piratebay->isChecked()); settings.setValue("isohunt", isohunt->isChecked()); settings.setValue("meganova", meganova->isChecked()); settings.endGroup(); qDebug("Saved checked search engines"); } // Save columns width in a file to remember them // (download list) void SearchEngine::saveColWidthSearchList() const{ qDebug("Saving columns width in search list"); QSettings settings("qBittorrent", "qBittorrent"); QStringList width_list; for(int i=0; icolumnCount(); ++i){ width_list << QString(misc::toString(resultsBrowser->columnWidth(i)).c_str()); } settings.setValue("SearchListColsWidth", width_list.join(" ")); qDebug("Search list columns width saved"); } // Load columns width in a file that were saved previously // (search list) bool SearchEngine::loadColWidthSearchList(){ qDebug("Loading columns width for search list"); QSettings settings("qBittorrent", "qBittorrent"); QString line = settings.value("SearchListColsWidth", QString()).toString(); if(line.isEmpty()) return false; QStringList width_list = line.split(' '); if(width_list.size() != SearchListModel->columnCount()) return false; for(int i=0; iheader()->resizeSection(i, width_list.at(i).toInt()); } qDebug("Search list columns width loaded"); return true; } // load last checked search engines from a file void SearchEngine::loadCheckedSearchEngines(){ qDebug("Loading checked search engines"); QSettings settings("qBittorrent", "qBittorrent"); settings.beginGroup("SearchEngines"); mininova->setChecked(settings.value("mininova", true).toBool()); piratebay->setChecked(settings.value("piratebay", false).toBool()); isohunt->setChecked(settings.value("isohunt", false).toBool()); meganova->setChecked(settings.value("meganova", false).toBool()); settings.endGroup(); qDebug("Loaded checked search engines"); } // get the last searchs from a QSettings to a QStringList void SearchEngine::startSearchHistory(){ QSettings settings("qBittorrent", "qBittorrent"); settings.beginGroup("Search"); searchHistory = settings.value("searchHistory",-1).toStringList(); settings.endGroup(); } // Save the history list into the QSettings for the next session void SearchEngine::saveSearchHistory() { QSettings settings("qBittorrent", "qBittorrent"); settings.beginGroup("Search"); settings.setValue("searchHistory",searchHistory); settings.endGroup(); } // Function called when we click on search button void SearchEngine::on_search_button_clicked(){ QString pattern = search_pattern->text().trimmed(); // No search pattern entered if(pattern.isEmpty()){ QMessageBox::critical(0, tr("Empty search pattern"), tr("Please type a search pattern first")); return; } // if the pattern is not in the pattern if(searchHistory.indexOf(pattern) == -1){ //update the searchHistory list searchHistory.append(pattern); // verify the max size of the history if(searchHistory.size() > SEARCHHISTORY_MAXSIZE) searchHistory = searchHistory.mid(searchHistory.size()/2,searchHistory.size()/2); searchCompleter = new QCompleter(searchHistory, this); searchCompleter->setCaseSensitivity(Qt::CaseInsensitive); search_pattern->setCompleter(searchCompleter); } // Getting checked search engines if(!mininova->isChecked() && ! piratebay->isChecked()/* && !reactor->isChecked()*/ && !isohunt->isChecked()/* && !btjunkie->isChecked()*/ && !meganova->isChecked()){ QMessageBox::critical(0, tr("No search engine selected"), tr("You must select at least one search engine.")); return; } QStringList params; QStringList engineNames; search_stopped = false; // Get checked search engines if(mininova->isChecked()){ engineNames << "mininova"; } if(piratebay->isChecked()){ engineNames << "piratebay"; } // if(reactor->isChecked()){ // engineNames << "reactor"; // } if(isohunt->isChecked()){ engineNames << "isohunt"; } // if(btjunkie->isChecked()){ // engineNames << "btjunkie"; // } if(meganova->isChecked()){ engineNames << "meganova"; } params << engineNames.join(","); params << pattern.split(" "); // Update SearchEngine widgets no_search_results = true; nb_search_results = 0; search_result_line_truncated.clear(); results_lbl->setText(tr("Results")+" (0):"); // Launch search searchProcess->start(misc::qBittorrentPath()+"nova.py", params, QIODevice::ReadOnly); } void SearchEngine::searchStarted(){ // Update SearchEngine widgets search_button->setEnabled(false); search_button->repaint(); search_status->setText(tr("Searching...")); search_status->repaint(); stop_search_button->setEnabled(true); stop_search_button->repaint(); // clear results window SearchListModel->removeRows(0, SearchListModel->rowCount()); // Clear previous results urls too searchResultsUrls.clear(); } // Download the given item from search results list void SearchEngine::downloadSelectedItem(const QModelIndex& index){ int row = index.row(); // Get Item url QString url = searchResultsUrls.value(SearchListModel->data(SearchListModel->index(row, NAME)).toString()); // Download from url BTSession->downloadFromUrl(url); // Set item color to RED setRowColor(row, "red"); } // search Qprocess return output as soon as it gets new // stuff to read. We split it into lines and add each // line to search results calling appendSearchResult(). void SearchEngine::readSearchOutput(){ QByteArray output = searchProcess->readAllStandardOutput(); QList lines_list = output.split('\n'); QByteArray line; if(!search_result_line_truncated.isEmpty()){ QByteArray end_of_line = lines_list.takeFirst(); lines_list.prepend(search_result_line_truncated+end_of_line); } search_result_line_truncated = lines_list.takeLast().trimmed(); foreach(line, lines_list){ appendSearchResult(QString(line)); } results_lbl->setText(tr("Results")+" ("+QString(misc::toString(nb_search_results).c_str())+"):"); } // Returns version of nova.py search engine float SearchEngine::getNovaVersion(const QString& novaPath) const{ QFile dest_nova(novaPath); if(!dest_nova.exists()){ return 0.0; } if(!dest_nova.open(QIODevice::ReadOnly | QIODevice::Text)){ return 0.0; } float version = 0.0; while (!dest_nova.atEnd()){ QByteArray line = dest_nova.readLine(); if(line.startsWith("# Version: ")){ line = line.split(' ').last(); line.chop(1); // removes '\n' version = line.toFloat(); qDebug("Search plugin version: %.2f", version); break; } } return version; } // Returns changelog of nova.py search engine QByteArray SearchEngine::getNovaChangelog(const QString& novaPath) const{ QFile dest_nova(novaPath); if(!dest_nova.exists()){ return QByteArray("None"); } if(!dest_nova.open(QIODevice::ReadOnly | QIODevice::Text)){ return QByteArray("None"); } QByteArray changelog; bool in_changelog = false; while (!dest_nova.atEnd()){ QByteArray line = dest_nova.readLine(); line = line.trimmed(); if(line.startsWith("# Changelog:")){ in_changelog = true; }else{ if(in_changelog && line.isEmpty()){ // end of changelog return changelog; } if(in_changelog){ line.remove(0,1); changelog.append(line); } } } return changelog; } // Update nova.py search plugin if necessary void SearchEngine::updateNova() const{ qDebug("Updating nova"); float provided_nova_version = getNovaVersion(":/search_engine/nova.py"); QFile::Permissions perm=QFile::ReadOwner | QFile::WriteOwner | QFile::ExeOwner | QFile::ReadUser | QFile::WriteUser | QFile::ExeUser | QFile::ReadGroup | QFile::ReadGroup; QFile(misc::qBittorrentPath()+"nova.py").setPermissions(perm); if(provided_nova_version > getNovaVersion(misc::qBittorrentPath()+"nova.py")){ qDebug("updating local search plugin with shipped one"); // nova.py needs update QFile::remove(misc::qBittorrentPath()+"nova.py"); qDebug("Old nova removed"); QFile::copy(":/search_engine/nova.py", misc::qBittorrentPath()+"nova.py"); qDebug("New nova copied"); QFile::Permissions perm=QFile::ReadOwner | QFile::WriteOwner | QFile::ExeOwner | QFile::ReadUser | QFile::WriteUser | QFile::ExeUser | QFile::ReadGroup | QFile::ReadGroup; QFile(misc::qBittorrentPath()+"nova.py").setPermissions(perm); qDebug("local search plugin updated"); } } // Download nova.py from qbittorrent.org // Check if our nova.py is outdated and // ask user for action. void SearchEngine::on_update_nova_button_clicked(){ CURL *curl; QString filePath; qDebug("Checking for search plugin updates on qbittorrent.org"); // XXX: Trick to get a unique filename QTemporaryFile *tmpfile = new QTemporaryFile; if (tmpfile->open()) { filePath = tmpfile->fileName(); } delete tmpfile; FILE *file = fopen((const char*)filePath.toUtf8(), "w"); if(!file){ std::cerr << "Error: could not open temporary file...\n"; } // Initilization required by libcurl curl = curl_easy_init(); if(!curl){ std::cerr << "Error: Failed to init curl...\n"; fclose(file); return; } // Set url to download curl_easy_setopt(curl, CURLOPT_URL, "http://www.dchris.eu/nova/nova.zip"); // Define our callback to get called when there's data to be written curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, misc::my_fwrite); // Set destination file curl_easy_setopt(curl, CURLOPT_WRITEDATA, file); // Some SSL mambo jambo curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, FALSE); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0); // Perform Download curl_easy_perform(curl); /* ignores error */ // Cleanup curl_easy_cleanup(curl); // Close tmp file fclose(file); qDebug("Version on qbittorrent.org: %.2f", getNovaVersion(filePath)); float version_on_server = getNovaVersion(filePath); if(version_on_server == 0.0){ //First server is down, try the mirror QFile::remove(filePath); FILE *file = fopen((const char*)filePath.toUtf8(), "w"); if(!file){ std::cerr << "Error: could not open temporary file...\n"; } curl = curl_easy_init(); curl_easy_setopt(curl, CURLOPT_URL, "http://hydr0g3n.free.fr/nova/nova.py"); // Define our callback to get called when there's data to be written curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, misc::my_fwrite); // Set destination file curl_easy_setopt(curl, CURLOPT_WRITEDATA, file); // Some SSL mambo jambo curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, FALSE); curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0); // Perform Download curl_easy_perform(curl); /* ignores error */ // Cleanup curl_easy_cleanup(curl); // Close tmp file fclose(file); version_on_server = getNovaVersion(filePath); } if(version_on_server > getNovaVersion(misc::qBittorrentPath()+"nova.py")){ if(QMessageBox::question(this, tr("Search plugin update -- qBittorrent"), tr("Search plugin can be updated, do you want to update it?\n\nChangelog:\n")+getNovaChangelog(filePath), tr("&Yes"), tr("&No"), QString(), 0, 1)){ return; }else{ qDebug("Updating search plugin from qbittorrent.org"); QFile::remove(misc::qBittorrentPath()+"nova.py"); QFile::copy(filePath, misc::qBittorrentPath()+"nova.py"); QFile::Permissions perm=QFile::ReadOwner | QFile::WriteOwner | QFile::ExeOwner | QFile::ReadUser | QFile::WriteUser | QFile::ExeUser | QFile::ReadGroup | QFile::ReadGroup; QFile(misc::qBittorrentPath()+"nova.py").setPermissions(perm); } }else{ if(version_on_server == 0.0){ QMessageBox::information(this, tr("Search plugin update")+" -- "+tr("qBittorrent"), tr("Sorry, update server is temporarily unavailable.")); }else{ QMessageBox::information(this, tr("Search plugin update -- qBittorrent"), tr("Your search plugin is already up to date.")); } } // Delete tmp file QFile::remove(filePath); } // Slot called when search is Finished // Search can be finished for 3 reasons : // Error | Stopped by user | Finished normally void SearchEngine::searchFinished(int exitcode,QProcess::ExitStatus){ QSettings settings("qBittorrent", "qBittorrent"); int useOSD = settings.value("Options/OSDEnabled", 1).toInt(); if(systrayIntegration && (useOSD == 1 || (useOSD == 2 && (isMinimized() || isHidden())))) { myTrayIcon->showMessage(tr("Search Engine"), tr("Search has finished"), QSystemTrayIcon::Information, TIME_TRAY_BALLOON); } if(exitcode){ search_status->setText(tr("An error occured during search...")); }else{ if(search_stopped){ search_status->setText(tr("Search aborted")); }else{ if(no_search_results){ search_status->setText(tr("Search returned no results")); }else{ search_status->setText(tr("Search has finished")); } } } results_lbl->setText(tr("Results", "i.e: Search results")+" ("+QString(misc::toString(nb_search_results).c_str())+"):"); search_button->setEnabled(true); stop_search_button->setEnabled(false); } // SLOT to append one line to search results list // Line is in the following form : // file url | file name | file size | nb seeds | nb leechers | Search engine url void SearchEngine::appendSearchResult(const QString& line){ QStringList parts = line.split("|"); if(parts.size() != 6){ return; } QString url = parts.takeFirst(); QString filename = parts.first(); // XXX: Two results can't have the same name (right?) if(searchResultsUrls.contains(filename)){ return; } // Add item to search result list int row = SearchListModel->rowCount(); SearchListModel->insertRow(row); for(int i=0; i<5; ++i){ SearchListModel->setData(SearchListModel->index(row, i), QVariant(parts.at(i))); } // Add url to searchResultsUrls associative array searchResultsUrls.insert(filename, url); no_search_results = false; ++nb_search_results; // Enable clear & download buttons clear_button->setEnabled(true); download_button->setEnabled(true); } // Stop search while it is working in background void SearchEngine::on_stop_search_button_clicked(){ // Kill process searchProcess->terminate(); search_stopped = true; } // Clear search results list void SearchEngine::on_clear_button_clicked(){ searchResultsUrls.clear(); SearchListModel->removeRows(0, SearchListModel->rowCount()); // Disable clear & download buttons clear_button->setEnabled(false); download_button->setEnabled(false); } // Download selected items in search results list void SearchEngine::on_download_button_clicked(){ QModelIndexList selectedIndexes = resultsBrowser->selectionModel()->selectedIndexes(); QModelIndex index; foreach(index, selectedIndexes){ if(index.column() == NAME){ // Get Item url QString url = searchResultsUrls.value(index.data().toString()); BTSession->downloadFromUrl(url); setRowColor(index.row(), "red"); } } }