Christophe Dumez
15 years ago
19 changed files with 621 additions and 525 deletions
|
Before Width: | Height: | Size: 813 B |
|
After Width: | Height: | Size: 1.2 KiB |
@ -1,5 +1,5 @@ |
|||||||
isohunt: 1.21 |
isohunt: 1.21 |
||||||
torrentreactor: 1.11 |
torrentreactor: 1.11 |
||||||
btjunkie: 2.11 |
btjunkie: 2.11 |
||||||
mininova: 1.23 |
mininova: 1.31 |
||||||
piratebay: 1.11 |
piratebay: 1.11 |
||||||
|
|||||||
@ -0,0 +1,173 @@ |
|||||||
|
/*
|
||||||
|
* 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. |
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
* Contact : chris@qbittorrent.org |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef SEARCHENGINES_H |
||||||
|
#define SEARCHENGINES_H |
||||||
|
|
||||||
|
#include <QHash> |
||||||
|
#include <QStringList> |
||||||
|
#include <QDomDocument> |
||||||
|
#include <QDomNode> |
||||||
|
#include <QDomElement> |
||||||
|
#include <QProcess> |
||||||
|
#include <QSettings> |
||||||
|
|
||||||
|
#include "misc.h" |
||||||
|
|
||||||
|
class SearchCategories: public QObject, public QHash<QString, QString> { |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
public: |
||||||
|
SearchCategories() { |
||||||
|
(*this)["all"] = tr("All categories"); |
||||||
|
(*this)["movies"] = tr("Movies"); |
||||||
|
(*this)["tv"] = tr("TV shows"); |
||||||
|
(*this)["music"] = tr("Music"); |
||||||
|
(*this)["games"] = tr("Games"); |
||||||
|
(*this)["anime"] = tr("Anime"); |
||||||
|
(*this)["software"] = tr("Software"); |
||||||
|
(*this)["pictures"] = tr("Pictures"); |
||||||
|
(*this)["books"] = tr("Books"); |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
class SupportedEngine { |
||||||
|
private: |
||||||
|
QString name; |
||||||
|
QString full_name; |
||||||
|
QString url; |
||||||
|
QStringList supported_categories; |
||||||
|
bool enabled; |
||||||
|
|
||||||
|
public: |
||||||
|
SupportedEngine(QDomElement engine_elem) { |
||||||
|
name = engine_elem.tagName(); |
||||||
|
full_name = engine_elem.elementsByTagName("name").at(0).toElement().text(); |
||||||
|
url = engine_elem.elementsByTagName("url").at(0).toElement().text(); |
||||||
|
supported_categories = engine_elem.elementsByTagName("categories").at(0).toElement().text().split(" "); |
||||||
|
QSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
||||||
|
QVariantList disabled_engines = settings.value(QString::fromUtf8("SearchEngines/disabledEngines"), QVariantList()).toList(); |
||||||
|
enabled = !disabled_engines.contains(QVariant(name)); |
||||||
|
} |
||||||
|
|
||||||
|
QString getName() const { return name; } |
||||||
|
QString getUrl() const { return url; } |
||||||
|
QString getFullName() const { return full_name; } |
||||||
|
QStringList getSupportedCategories() const { return supported_categories; } |
||||||
|
bool isEnabled() const { return enabled; } |
||||||
|
void setEnabled(bool _enabled) { |
||||||
|
enabled = _enabled; |
||||||
|
// Save to Hard disk
|
||||||
|
QSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
||||||
|
QVariantList disabled_engines = settings.value(QString::fromUtf8("SearchEngines/disabledEngines"), QVariantList()).toList(); |
||||||
|
if(enabled) { |
||||||
|
disabled_engines.removeAll(QVariant(name)); |
||||||
|
} else { |
||||||
|
disabled_engines.append(QVariant(name)); |
||||||
|
} |
||||||
|
settings.setValue("SearchEngines/disabledEngines", disabled_engines); |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
class SupportedEngines: public QObject, public QHash<QString, SupportedEngine*> { |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
signals: |
||||||
|
void newSupportedEngine(QString name); |
||||||
|
|
||||||
|
public: |
||||||
|
SupportedEngines() { |
||||||
|
update(); |
||||||
|
} |
||||||
|
|
||||||
|
~SupportedEngines() { |
||||||
|
qDeleteAll(this->values()); |
||||||
|
} |
||||||
|
|
||||||
|
QStringList enginesEnabled() const { |
||||||
|
QStringList engines; |
||||||
|
foreach(SupportedEngine *engine, values()) { |
||||||
|
if(engine->isEnabled()) |
||||||
|
engines << engine->getName(); |
||||||
|
} |
||||||
|
return engines; |
||||||
|
} |
||||||
|
|
||||||
|
QStringList supportedCategories() const { |
||||||
|
QStringList supported_cat; |
||||||
|
foreach(SupportedEngine *engine, values()) { |
||||||
|
QStringList s = engine->getSupportedCategories(); |
||||||
|
foreach(QString cat, s) { |
||||||
|
cat = cat.trimmed(); |
||||||
|
if(!cat.isEmpty() && !supported_cat.contains(cat)) |
||||||
|
supported_cat << cat; |
||||||
|
} |
||||||
|
} |
||||||
|
return supported_cat; |
||||||
|
} |
||||||
|
|
||||||
|
public slots: |
||||||
|
void update() { |
||||||
|
QProcess nova; |
||||||
|
QStringList params; |
||||||
|
params << misc::qBittorrentPath()+"search_engine"+QDir::separator()+"nova2.py"; |
||||||
|
params << "--capabilities"; |
||||||
|
nova.start("python", params, QIODevice::ReadOnly); |
||||||
|
nova.waitForStarted(); |
||||||
|
nova.waitForFinished(); |
||||||
|
QString capabilities = QString(nova.readAll()); |
||||||
|
QDomDocument xml_doc; |
||||||
|
if(!xml_doc.setContent(capabilities)) { |
||||||
|
std::cerr << "Could not parse Nova search engine capabilities, msg: " << capabilities.toLocal8Bit().data() << std::endl; |
||||||
|
return; |
||||||
|
} |
||||||
|
QDomElement root = xml_doc.documentElement(); |
||||||
|
if(root.tagName() != "capabilities") { |
||||||
|
std::cout << "Invalid XML file for Nova search engine capabilities, msg: " << capabilities.toLocal8Bit().data() << std::endl; |
||||||
|
return; |
||||||
|
} |
||||||
|
for(QDomNode engine_node = root.firstChild(); !engine_node.isNull(); engine_node = engine_node.nextSibling()) { |
||||||
|
QDomElement engine_elem = engine_node.toElement(); |
||||||
|
if(!engine_elem.isNull()) { |
||||||
|
SupportedEngine *s = new SupportedEngine(engine_elem); |
||||||
|
if(this->contains(s->getName())) { |
||||||
|
// Already in the list
|
||||||
|
delete s; |
||||||
|
} else { |
||||||
|
qDebug("Supported search engine: %s", s->getFullName().toLocal8Bit().data()); |
||||||
|
(*this)[s->getName()] = s; |
||||||
|
emit newSupportedEngine(s->getName()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
#endif // SEARCHENGINES_H
|
||||||
Loading…
Reference in new issue