Arnaud Demaiziere
18 years ago
1 changed files with 460 additions and 0 deletions
@ -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