Arnaud Demaiziere
18 years ago
1 changed files with 460 additions and 0 deletions
@ -0,0 +1,460 @@
@@ -0,0 +1,460 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt4 and libtorrent. |
||||
* Copyright (C) 2006 Christophe Dumez, Arnaud Demaiziere |
||||
* |
||||
* 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, arnaud@qbittorrent.org |
||||
*/ |
||||
|
||||
#define GLOBAL_MAX_ITEM 150 |
||||
#define STREAM_MAX_ITEM 20 |
||||
|
||||
|
||||
#ifndef RSS_H |
||||
#define RSS_H |
||||
|
||||
#include <QFile> |
||||
#include <QList> |
||||
#include <curl/curl.h> |
||||
#include <QTemporaryFile> |
||||
#include <QSettings> |
||||
#include <QDomDocument> |
||||
|
||||
#include "misc.h" |
||||
#include "downloadThread.h" |
||||
|
||||
class RssItem; |
||||
class RssStream; |
||||
class RssManager; |
||||
|
||||
// Item of a rss stream, single information
|
||||
class RssItem{ |
||||
private: |
||||
|
||||
QString title; |
||||
QString link; |
||||
QString description; |
||||
QString image; |
||||
bool read; |
||||
RssStream* parent; |
||||
QString downloadLink; |
||||
|
||||
public: |
||||
// public constructor
|
||||
RssItem(const QDomElement& properties, RssStream* _parent){ |
||||
read = false; |
||||
parent = _parent; |
||||
downloadLink = "none"; |
||||
QDomElement property=properties.firstChild().toElement(); |
||||
while(!property.isNull()) |
||||
{ |
||||
// setters
|
||||
if (property.tagName() == "title") |
||||
title = property.text(); |
||||
else if (property.tagName() == "link") |
||||
link = property.text(); |
||||
else if (property.tagName() == "description") |
||||
description = property.text(); |
||||
else if (property.tagName() == "image") |
||||
image = property.text(); |
||||
// build items
|
||||
property = property.nextSibling().toElement(); |
||||
} |
||||
//displayItem();
|
||||
} |
||||
|
||||
~RssItem(){ |
||||
} |
||||
|
||||
QString getTitle() const{ |
||||
return title; |
||||
} |
||||
|
||||
QString getLink() const{ |
||||
return link; |
||||
} |
||||
|
||||
QString getDescription() const{ |
||||
return description; |
||||
} |
||||
|
||||
QString getImage() const{ |
||||
return image; |
||||
} |
||||
|
||||
QString getDownloadLink() const{ |
||||
return downloadLink; |
||||
} |
||||
|
||||
bool isRead() const{ |
||||
return this->read; |
||||
} |
||||
|
||||
void setRead(){ |
||||
this->read = true; |
||||
} |
||||
|
||||
RssStream* getParent() const{ |
||||
return parent; |
||||
} |
||||
|
||||
void displayItem(){ |
||||
qDebug(" - "+getTitle().toUtf8()+" - "+getLink().toUtf8()); |
||||
} |
||||
}; |
||||
|
||||
// Rss stream, loaded form an xml file
|
||||
class RssStream : public QObject{ |
||||
Q_OBJECT |
||||
private: |
||||
|
||||
QString title; |
||||
QString alias; |
||||
QString link; |
||||
QString description; |
||||
QString image; |
||||
QString url; |
||||
QString filePath; |
||||
QList<RssItem*> listItem; |
||||
downloadThread* downloader; |
||||
|
||||
public slots : |
||||
|
||||
// read and store the downloaded rss' informations
|
||||
void processDownloadedFile(const QString&, const QString& file_path, int return_code, const QString&){ |
||||
QFile::remove(filePath); |
||||
filePath = file_path; |
||||
if(return_code){ |
||||
// Download failed
|
||||
qDebug("(download failure) "+file_path.toUtf8()); |
||||
QFile::remove(file_path); |
||||
return; |
||||
} |
||||
this->openRss(); |
||||
} |
||||
|
||||
public: |
||||
|
||||
RssStream(const QString& _url){ |
||||
url = _url; |
||||
alias = url; |
||||
downloader = new downloadThread(this); |
||||
connect(downloader, SIGNAL(downloadFinished(const QString&, const QString&, int, const QString&)), this, SLOT(processDownloadedFile(const QString&, const QString&, int, const QString&))); |
||||
downloader->downloadUrl(url); |
||||
} |
||||
|
||||
~RssStream(){ |
||||
for(int i=0; i<listItem.size(); i++){ |
||||
delete getItem(i); |
||||
} |
||||
delete downloader; |
||||
if(QFile::exists(filePath)) |
||||
QFile::remove(filePath); |
||||
} |
||||
|
||||
int refresh(){ |
||||
downloader = new downloadThread(this); |
||||
connect(downloader, SIGNAL(downloadFinished(const QString&, const QString&, int, const QString&)), this, SLOT(processDownloadedFile(const QString&, const QString&, int, const QString&))); |
||||
downloader->downloadUrl(url); |
||||
return 1; |
||||
} |
||||
|
||||
QString getTitle() const{ |
||||
return this->title; |
||||
} |
||||
|
||||
QString getAlias() const{ |
||||
return this->alias; |
||||
} |
||||
|
||||
void setAlias(const QString& _alias){ |
||||
this->alias = _alias; |
||||
} |
||||
|
||||
QString getLink() const{ |
||||
return this->link; |
||||
} |
||||
|
||||
QString getUrl() const{ |
||||
return this->url; |
||||
} |
||||
|
||||
QString getDescription() const{ |
||||
return this->description; |
||||
} |
||||
|
||||
QString getImage() const{ |
||||
return this->image; |
||||
} |
||||
|
||||
QString getFilePath() const{ |
||||
return this->filePath; |
||||
} |
||||
|
||||
RssItem* getItem(unsigned short index) const{ |
||||
return this->listItem.at(index); |
||||
} |
||||
|
||||
unsigned short getListSize() const{ |
||||
return this->listItem.size(); |
||||
} |
||||
|
||||
QList<RssItem*> getListItem() const{ |
||||
return this->listItem; |
||||
} |
||||
|
||||
void displayStream(){ |
||||
qDebug(" # "+getTitle().toUtf8()+" - "+getUrl().toUtf8()+" - "+getAlias().toUtf8()); |
||||
for(int i=0; i<listItem.size(); i++){ |
||||
getItem(i)->displayItem(); |
||||
} |
||||
} |
||||
|
||||
private: |
||||
|
||||
short read(const QDomDocument& doc) { |
||||
// is it a rss file ?
|
||||
QDomElement root = doc.documentElement(); |
||||
if(root.tagName() == "html"){ |
||||
qDebug("the file is empty, maybe the url is wrong or the server is too busy"); |
||||
return -1; |
||||
} |
||||
else if(root.tagName() != "rss"){ |
||||
qDebug("the file is not a rss stream, <rss> omitted"+root.tagName().toUtf8()); |
||||
return -1; |
||||
} |
||||
QDomNode rss = root.firstChild(); |
||||
QDomElement channel = root.firstChild().toElement(); |
||||
while(!channel.isNull()) |
||||
{ |
||||
// we are reading the rss'main info
|
||||
if (channel.tagName() == "channel") |
||||
{ |
||||
QDomElement property=channel.firstChild().toElement(); |
||||
while(!property.isNull()) |
||||
{ |
||||
// setters
|
||||
if (property.tagName() == "title") |
||||
this->title = property.text(); |
||||
else if (property.tagName() == "link") |
||||
this->link = property.text(); |
||||
else if (property.tagName() == "description") |
||||
this->description = property.text(); |
||||
else if (property.tagName() == "image") |
||||
this->image = property.text(); |
||||
// build items
|
||||
else if(property.tagName() == "item") |
||||
{ |
||||
RssItem* item = new RssItem(property, this); |
||||
//add it to a list
|
||||
this->listItem.append(item); |
||||
} |
||||
property = property.nextSibling().toElement(); |
||||
} |
||||
} |
||||
channel = channel.nextSibling().toElement(); |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
// existing and opening test after download
|
||||
short openRss(){ |
||||
QDomDocument doc("Rss Seed"); |
||||
QFile fileRss(filePath); |
||||
if(!fileRss.open(QIODevice::ReadOnly | QIODevice::Text)) |
||||
{ |
||||
qDebug("error : open failed, no file or locked"); |
||||
fileRss.remove(); |
||||
return -1; |
||||
} |
||||
if(!doc.setContent(&fileRss)) |
||||
{ |
||||
qDebug("can't read temp file, might be empty"); |
||||
fileRss.close(); |
||||
fileRss.remove(); |
||||
return -1; |
||||
} |
||||
// start reading the xml
|
||||
short return_lecture = read(doc); |
||||
fileRss.close(); |
||||
fileRss.remove(); |
||||
return return_lecture; |
||||
} |
||||
}; |
||||
|
||||
// global class, manage the whole rss stream
|
||||
class RssManager{ |
||||
|
||||
private : |
||||
QList<RssStream*> streamList; |
||||
QList<RssStream*> ignoredStreamList; |
||||
QStringList streamListUrl; |
||||
QStringList streamListAlias; |
||||
|
||||
public : |
||||
RssManager(){ |
||||
loadStreamList(); |
||||
} |
||||
|
||||
~RssManager(){ |
||||
saveStreamList(); |
||||
for(unsigned short i=0; i<streamList.size(); i++){ |
||||
delete getStream(i); |
||||
} |
||||
for(unsigned short i=0; i<ignoredStreamList.size(); i++){ |
||||
delete getIgnored(i); |
||||
} |
||||
} |
||||
|
||||
// load the list of the rss stream
|
||||
void loadStreamList(){ |
||||
QSettings settings("qBittorrent", "qBittorrent"); |
||||
settings.beginGroup("Rss"); |
||||
this->streamListUrl = settings.value("streamList").toStringList(); |
||||
this->streamListAlias = settings.value("streamAlias").toStringList(); |
||||
settings.endGroup(); |
||||
for(unsigned short i=0; i<streamListUrl.size(); i++){ |
||||
RssStream *stream = new RssStream(this->streamListUrl.at(i)); |
||||
stream->setAlias(this->streamListAlias.at(i)); |
||||
this->streamList.append(stream); |
||||
} |
||||
} |
||||
|
||||
// save the list of the rss stream for the next session
|
||||
void saveStreamList(){ |
||||
QSettings settings("qBittorrent", "qBittorrent"); |
||||
settings.beginGroup("Rss"); |
||||
settings.setValue("streamList", this->streamListUrl); |
||||
settings.setValue("streamAlias", this->streamListAlias); |
||||
settings.endGroup(); |
||||
} |
||||
|
||||
// add a stream to the manager
|
||||
void addStream(RssStream* stream){ |
||||
if(hasStream(stream)<0){ |
||||
this->streamList.append(stream); |
||||
this->streamListUrl.append(stream->getUrl()); |
||||
this->streamListAlias.append(stream->getUrl()); |
||||
} |
||||
} |
||||
|
||||
// add a stream to the manager
|
||||
void addStream(QString url){ |
||||
// completion of the address
|
||||
if(!url.endsWith(".xml")){ |
||||
if(url.endsWith(QDir::separator())) |
||||
url.append("rss.xml"); |
||||
else |
||||
{ |
||||
url.append(QDir::separator()); |
||||
url.append("rss.xml"); |
||||
} |
||||
} |
||||
|
||||
if(hasStream(url)<0){ |
||||
RssStream* stream = new RssStream(url); |
||||
this->streamList.append(stream); |
||||
this->streamListUrl.append(url); |
||||
this->streamListAlias.append(url); |
||||
} |
||||
} |
||||
|
||||
// remove a stream from the manager
|
||||
void removeStream(RssStream* stream){ |
||||
short index = hasStream(stream); |
||||
if(index>=0){ |
||||
for(unsigned short i=0; i<ignoredStreamList.size(); i++){ |
||||
if(getIgnored(i)->getUrl()==stream->getUrl()){ |
||||
delete ignoredStreamList.at(i); |
||||
this->ignoredStreamList.removeAt(i); |
||||
} |
||||
} |
||||
for(unsigned short i=0; i<streamList.size(); i++){ |
||||
if(getStream(i)->getUrl()==stream->getUrl()){ |
||||
delete streamList.at(i); |
||||
this->streamList.removeAt(i); |
||||
} |
||||
} |
||||
this->streamListUrl.removeAt(index); |
||||
this->streamListAlias.removeAt(index); |
||||
} |
||||
} |
||||
|
||||
// remove all the streams in the manager
|
||||
void removeAll(){ |
||||
QList<RssStream*> newIgnoredList, newStreamList; |
||||
QStringList newUrlList, newAliasList; |
||||
for(unsigned short i=0; i<streamList.size(); i++){ |
||||
delete getStream(i); |
||||
} |
||||
for(unsigned short i=0; i<ignoredStreamList.size(); i++){ |
||||
delete getIgnored(i); |
||||
} |
||||
this->streamList = newStreamList; |
||||
this->ignoredStreamList = newIgnoredList; |
||||
this->streamListUrl = newUrlList; |
||||
this->streamListAlias = newAliasList; |
||||
} |
||||
|
||||
// return the position index of a stream, if the manager owns it
|
||||
short hasStream(RssStream* stream) const{ |
||||
QString url = stream->getUrl(); |
||||
return hasStream(url); |
||||
} |
||||
|
||||
short hasStream(const QString& url) const{ |
||||
return streamListUrl.indexOf(url); |
||||
} |
||||
|
||||
RssStream* getStream(const int& index) const{ |
||||
return streamList.at(index); |
||||
} |
||||
|
||||
RssStream* getIgnored(const int& index) const{ |
||||
return ignoredStreamList.at(index); |
||||
} |
||||
|
||||
// reload all the xml files from the web
|
||||
void refreshAll(){ |
||||
// first refresh the ignored ones
|
||||
for(unsigned short i=0; i<ignoredStreamList.size(); i++){ |
||||
if(getIgnored(i)->refresh()==0){ |
||||
this->streamList.append(getIgnored(i)); |
||||
this->ignoredStreamList.removeAt(i); |
||||
} |
||||
} |
||||
// then refresh the active ones
|
||||
for(unsigned short i=0; i<streamList.size(); i++){ |
||||
if(getStream(i)->refresh()!=0){ |
||||
this->ignoredStreamList.append(getStream(i)); |
||||
this->streamList.removeAt(i); |
||||
} |
||||
} |
||||
} |
||||
|
||||
void displayManager(){ |
||||
for(unsigned short i=0; i<streamList.size(); i++){ |
||||
getStream(i)->displayStream(); |
||||
} |
||||
qDebug("#### rss ignored streams ####"); |
||||
for(unsigned short i=0; i<ignoredStreamList.size(); i++){ |
||||
qDebug(" _ (ignored) "+getIgnored(i)->getTitle().toUtf8()+" - "+getIgnored(i)->getUrl().toUtf8()+" // "+getIgnored(i)->getAlias().toUtf8()); |
||||
} |
||||
} |
||||
|
||||
}; |
||||
|
||||
#endif |
Loading…
Reference in new issue