mirror of
https://github.com/d47081/qBittorrent.git
synced 2025-01-22 20:44:15 +00:00
- Added RSS support header
This commit is contained in:
Arnaud Demaiziere
parent
4ca852c2b3
commit
456f57e32f
460
src/rss.h
Normal file
460
src/rss.h
Normal file
@ -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…
x
Reference in New Issue
Block a user