Christophe Dumez
18 years ago
10 changed files with 1121 additions and 1031 deletions
@ -0,0 +1,386 @@
@@ -0,0 +1,386 @@
|
||||
<ui version="4.0" > |
||||
<class>search_engine</class> |
||||
<widget class="QWidget" name="search_engine" > |
||||
<property name="geometry" > |
||||
<rect> |
||||
<x>0</x> |
||||
<y>0</y> |
||||
<width>811</width> |
||||
<height>453</height> |
||||
</rect> |
||||
</property> |
||||
<property name="windowTitle" > |
||||
<string>Search</string> |
||||
</property> |
||||
<layout class="QVBoxLayout" > |
||||
<property name="margin" > |
||||
<number>9</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<layout class="QHBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QGroupBox" name="groupEngines" > |
||||
<property name="minimumSize" > |
||||
<size> |
||||
<width>131</width> |
||||
<height>132</height> |
||||
</size> |
||||
</property> |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>125</width> |
||||
<height>132</height> |
||||
</size> |
||||
</property> |
||||
<property name="title" > |
||||
<string>Search Engines</string> |
||||
</property> |
||||
<layout class="QVBoxLayout" > |
||||
<property name="margin" > |
||||
<number>9</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QCheckBox" name="mininova" > |
||||
<property name="text" > |
||||
<string/> |
||||
</property> |
||||
<property name="checked" > |
||||
<bool>true</bool> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QCheckBox" name="piratebay" > |
||||
<property name="text" > |
||||
<string/> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QCheckBox" name="isohunt" > |
||||
<property name="text" > |
||||
<string/> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QCheckBox" name="meganova" > |
||||
<property name="text" > |
||||
<string/> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<layout class="QVBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<layout class="QHBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QLabel" name="search_lbl" > |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>35</height> |
||||
</size> |
||||
</property> |
||||
<property name="font" > |
||||
<font> |
||||
<family>Sans Serif</family> |
||||
<pointsize>9</pointsize> |
||||
<weight>75</weight> |
||||
<italic>false</italic> |
||||
<bold>true</bold> |
||||
<underline>false</underline> |
||||
<strikeout>false</strikeout> |
||||
</font> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Search Pattern:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QLineEdit" name="search_pattern" > |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>22</height> |
||||
</size> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QPushButton" name="search_button" > |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>29</height> |
||||
</size> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Search</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QPushButton" name="stop_search_button" > |
||||
<property name="enabled" > |
||||
<bool>false</bool> |
||||
</property> |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>29</height> |
||||
</size> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Stop</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<layout class="QHBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QLabel" name="status_lbl" > |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>35</height> |
||||
</size> |
||||
</property> |
||||
<property name="font" > |
||||
<font> |
||||
<family>Sans Serif</family> |
||||
<pointsize>9</pointsize> |
||||
<weight>75</weight> |
||||
<italic>false</italic> |
||||
<bold>true</bold> |
||||
<underline>false</underline> |
||||
<strikeout>false</strikeout> |
||||
</font> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Status:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QLabel" name="search_status" > |
||||
<property name="minimumSize" > |
||||
<size> |
||||
<width>400</width> |
||||
<height>0</height> |
||||
</size> |
||||
</property> |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>35</height> |
||||
</size> |
||||
</property> |
||||
<property name="font" > |
||||
<font> |
||||
<family>Sans Serif</family> |
||||
<pointsize>9</pointsize> |
||||
<weight>50</weight> |
||||
<italic>true</italic> |
||||
<bold>false</bold> |
||||
<underline>false</underline> |
||||
<strikeout>false</strikeout> |
||||
</font> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Stopped</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer> |
||||
<property name="orientation" > |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" > |
||||
<size> |
||||
<width>40</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<layout class="QVBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<layout class="QHBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QLabel" name="results_lbl" > |
||||
<property name="maximumSize" > |
||||
<size> |
||||
<width>16777215</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
<property name="font" > |
||||
<font> |
||||
<family>Sans Serif</family> |
||||
<pointsize>9</pointsize> |
||||
<weight>75</weight> |
||||
<italic>false</italic> |
||||
<bold>true</bold> |
||||
<underline>false</underline> |
||||
<strikeout>false</strikeout> |
||||
</font> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Results:</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer> |
||||
<property name="orientation" > |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" > |
||||
<size> |
||||
<width>721</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
<item> |
||||
<widget class="QTreeView" name="resultsBrowser" > |
||||
<property name="minimumSize" > |
||||
<size> |
||||
<width>0</width> |
||||
<height>200</height> |
||||
</size> |
||||
</property> |
||||
<property name="contextMenuPolicy" > |
||||
<enum>Qt::CustomContextMenu</enum> |
||||
</property> |
||||
<property name="autoScroll" > |
||||
<bool>true</bool> |
||||
</property> |
||||
<property name="selectionMode" > |
||||
<enum>QAbstractItemView::ExtendedSelection</enum> |
||||
</property> |
||||
<property name="indentation" > |
||||
<number>1</number> |
||||
</property> |
||||
<property name="itemsExpandable" > |
||||
<bool>false</bool> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<layout class="QHBoxLayout" > |
||||
<property name="margin" > |
||||
<number>0</number> |
||||
</property> |
||||
<property name="spacing" > |
||||
<number>6</number> |
||||
</property> |
||||
<item> |
||||
<widget class="QPushButton" name="download_button" > |
||||
<property name="enabled" > |
||||
<bool>false</bool> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Download</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<widget class="QPushButton" name="clear_button" > |
||||
<property name="enabled" > |
||||
<bool>false</bool> |
||||
</property> |
||||
<property name="text" > |
||||
<string>Clear</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
<item> |
||||
<spacer> |
||||
<property name="orientation" > |
||||
<enum>Qt::Horizontal</enum> |
||||
</property> |
||||
<property name="sizeHint" > |
||||
<size> |
||||
<width>601</width> |
||||
<height>20</height> |
||||
</size> |
||||
</property> |
||||
</spacer> |
||||
</item> |
||||
<item> |
||||
<widget class="QPushButton" name="update_nova_button" > |
||||
<property name="text" > |
||||
<string>Update search plugin</string> |
||||
</property> |
||||
</widget> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</item> |
||||
</layout> |
||||
</widget> |
||||
<resources/> |
||||
<connections/> |
||||
</ui> |
||||
@ -0,0 +1,607 @@
@@ -0,0 +1,607 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* 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. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#include <QStandardItemModel> |
||||
#include <QHeaderView> |
||||
#include <QCompleter> |
||||
#include <QSettings> |
||||
#include <QMessageBox> |
||||
#include <QSystemTrayIcon> |
||||
#include <QTemporaryFile> |
||||
#include <QSystemTrayIcon> |
||||
#include <curl/curl.h> |
||||
#include <iostream> |
||||
|
||||
#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) : QWidget(){ |
||||
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; i<SearchListModel->columnCount(); ++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<QPair<int, qlonglong> > lines; |
||||
// Insertion sorting
|
||||
for(int i=0; i<SearchListModel->rowCount(); ++i){ |
||||
misc::insertSort(lines, QPair<int,qlonglong>(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; row<lines.size(); ++row){ |
||||
SearchListModel->insertRow(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<QPair<int, QString> > lines; |
||||
// Insetion sorting
|
||||
for(int i=0; i<SearchListModel->rowCount(); ++i){ |
||||
misc::insertSortString(lines, QPair<int, QString>(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; row<nbRows_old; ++row){ |
||||
SearchListModel->insertRow(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; i<SearchListModel->columnCount(); ++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; i<width_list.size(); ++i){ |
||||
resultsBrowser->header()->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")+" <i>(0)</i>:"); |
||||
// 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<QByteArray> 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")+" <i>("+QString(misc::toString(nb_search_results).c_str())+")</i>:"); |
||||
} |
||||
|
||||
// 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: %.1f", 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: %f", getNovaVersion(filePath)); |
||||
float version_on_server = getNovaVersion(filePath); |
||||
if(version_on_server == 0.0){ |
||||
//First server is down, try 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(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")+" <i>("+QString(misc::toString(nb_search_results).c_str())+")</i>:"); |
||||
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"); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,84 @@
@@ -0,0 +1,84 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* 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. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#ifndef SEARCH_H |
||||
#define SEARCH_H |
||||
|
||||
#define TIME_TRAY_BALLOON 5000 |
||||
|
||||
#include <QProcess> |
||||
#include "ui_search.h" |
||||
|
||||
class QStandardItemModel; |
||||
class SearchListDelegate; |
||||
class bittorrent; |
||||
class QSystemTrayIcon; |
||||
|
||||
class SearchEngine : public QWidget, public Ui::search_engine{ |
||||
Q_OBJECT |
||||
|
||||
private: |
||||
// Search related
|
||||
QHash<QString, QString> searchResultsUrls; |
||||
QProcess *searchProcess; |
||||
bool search_stopped; |
||||
bool no_search_results; |
||||
QByteArray search_result_line_truncated; |
||||
unsigned long nb_search_results; |
||||
QCompleter *searchCompleter; |
||||
QStringList searchHistory; |
||||
QStandardItemModel *SearchListModel; |
||||
SearchListDelegate *SearchDelegate; |
||||
bittorrent *BTSession; |
||||
QSystemTrayIcon *myTrayIcon; |
||||
|
||||
public: |
||||
SearchEngine(bittorrent *BTSession, QSystemTrayIcon *myTrayIcon); |
||||
~SearchEngine(); |
||||
float getNovaVersion(const QString& novaPath) const; |
||||
QByteArray getNovaChangelog(const QString& novaPath) const; |
||||
bool loadColWidthSearchList(); |
||||
|
||||
public slots: |
||||
// Search slots
|
||||
void on_search_button_clicked(); |
||||
void on_stop_search_button_clicked(); |
||||
void on_clear_button_clicked(); |
||||
void on_download_button_clicked(); |
||||
void on_update_nova_button_clicked(); |
||||
void appendSearchResult(const QString& line); |
||||
void searchFinished(int exitcode,QProcess::ExitStatus); |
||||
void readSearchOutput(); |
||||
void setRowColor(int row, const QString& color); |
||||
void searchStarted(); |
||||
void downloadSelectedItem(const QModelIndex& index); |
||||
void startSearchHistory(); |
||||
void loadCheckedSearchEngines(); |
||||
void updateNova() const; |
||||
void saveSearchHistory(); |
||||
void saveColWidthSearchList() const; |
||||
void saveCheckedSearchEngines(int) const; |
||||
void sortSearchList(int index); |
||||
void sortSearchListInt(int index, Qt::SortOrder sortOrder); |
||||
void sortSearchListString(int index, Qt::SortOrder sortOrder); |
||||
}; |
||||
|
||||
#endif |
||||
Loading…
Reference in new issue