diff options
Diffstat (limited to 'plugins/rssfeed/rssfeed.cpp')
| -rw-r--r-- | plugins/rssfeed/rssfeed.cpp | 359 |
1 files changed, 359 insertions, 0 deletions
diff --git a/plugins/rssfeed/rssfeed.cpp b/plugins/rssfeed/rssfeed.cpp new file mode 100644 index 0000000..0d1244b --- /dev/null +++ b/plugins/rssfeed/rssfeed.cpp @@ -0,0 +1,359 @@ +/*************************************************************************** + * Copyright (C) 2006 by Alan Jones * + * skyphyr@gmail.com * + * * + * 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. * + ***************************************************************************/ +#include "rssfeed.h" + +#include <kglobal.h> +#include <kstandarddirs.h> +#include <krfcdate.h> +#include <kio/netaccess.h> +#include <qfile.h> +#include <qapplication.h> +#include <qdir.h> + +namespace kt +{ + + void RssFeed::startFeed() + { + if (m_active) + { + refreshFeed(); + refreshTimer.start(QTime().msecsTo(m_autoRefresh)); + } + else + { + refreshTimer.stop(); + } + } + + void RssFeed::initialize() + { + feedLoading = false; + loadArticles(); + + connect(&refreshTimer, SIGNAL(timeout()), this, SLOT( refreshFeed() ) ); + connect(this, SIGNAL(articlesChanged(const RssArticle::List&)), this, SLOT( saveArticles() ) ); + + startFeed(); + } + + RssFeed::RssFeed(QObject * parent) : QObject(parent) + { + m_active = false; + m_articleAge = 365; + m_ignoreTTL = false; + m_title = "New"; + + initialize(); + } + + RssFeed::RssFeed(KURL feedUrl, QString title, bool active, int articleAge, bool ignoreTTL, QTime autoRefresh ) + { + m_feedUrl = feedUrl; + m_title = title; + m_active = active; + m_articleAge = articleAge; + m_ignoreTTL = ignoreTTL; + m_autoRefresh = autoRefresh; + + initialize(); + } + + RssFeed::RssFeed(const RssFeed &other) : QObject() + { + *this = other; + } + + RssFeed &RssFeed::operator=(const RssFeed &other) + { + if (&other != this) + { + m_feedUrl = other.feedUrl(); + m_title = other.title(); + m_active = other.active(); + m_articleAge = other.articleAge(); + m_ignoreTTL = other.ignoreTTL(); + m_autoRefresh = other.autoRefresh(); + } + + initialize(); + + return *this; + } + + void RssFeed::setFeedUrl( const KURL& url ) + { + if (m_feedUrl != url) + { + m_feedUrl = url; + loadArticles(); + startFeed(); + emit feedUrlChanged(url); + } + } + + void RssFeed::setFeedUrl( const QString& url ) + { + if (m_feedUrl != url) + { + m_feedUrl = url; + loadArticles(); + startFeed(); + emit feedUrlChanged(url); + } + } + + void RssFeed::setActive( bool active ) + { + if (m_active != active) + { + m_active = active; + + startFeed(); + + emit activeChanged(active); + } + } + + void RssFeed::setArticleAge( int articleAge ) + { + if (m_articleAge != articleAge) + { + if (articleAge < m_articleAge) + { + cleanArticles(); + } + + m_articleAge = articleAge; + emit articleAgeChanged(articleAge); + } + } + + void RssFeed::setTitle( const QString& title ) + { + if (m_title != title) + { + m_title = title; + emit titleChanged(title); + } + } + + void RssFeed::setAutoRefresh( const QTime& autoRefresh ) + { + if (m_autoRefresh != autoRefresh) + { + m_autoRefresh = autoRefresh; + if (m_active) + { + refreshTimer.changeInterval(QTime().msecsTo(m_autoRefresh)); + } + + emit autoRefreshChanged(autoRefresh); + } + } + + void RssFeed::setIgnoreTTL( bool ignoreTTL ) + { + if (m_ignoreTTL != ignoreTTL) + { + m_ignoreTTL = ignoreTTL; + emit ignoreTTLChanged(ignoreTTL); + } + } + + QString RssFeed::getFilename() + { + QDir directory; + directory.mkdir(KGlobal::dirs()->saveLocation("data","ktorrent") + "rssfeeds"); + return KGlobal::dirs()->saveLocation("data","ktorrent") + "rssfeeds/" + m_feedUrl.prettyURL(-1).replace("/", "_").replace(":", "_") + ".ktr"; + + } + + void RssFeed::loadArticles() + { + QString filename = getFilename(); + + //load articles from disk + QFile file(filename); + + if (file.exists()) + { + file.open( IO_ReadOnly ); + QDataStream in(&file); + + in >> m_articles; + emit articlesChanged( m_articles ); + } + } + + void RssFeed::saveArticles() + { + QString filename = getFilename(); + + //load articles from disk + QFile file(filename); + + file.open( IO_WriteOnly ); + QDataStream out(&file); + + out << m_articles; + } + + void RssFeed::cleanArticles() + { + bool removed = false; + + RssArticle::List::iterator it; + for ( it = m_articles.begin(); it != m_articles.end(); ) + { + if ((*it).pubDate().daysTo(QDateTime::currentDateTime()) > m_articleAge) + { + it = m_articles.erase(it); + removed = true; + } + else + { + it++; + } + } + + if (removed) + { + emit articlesChanged(m_articles); + } + + } + + void RssFeed::clearArticles() + { + m_articles.clear(); + } + + void RssFeed::refreshFeed() + { + if (feedLoading) + return; + + feedLoading = true; + cleanArticles(); + Loader * feedLoader = Loader::create(); + connect( feedLoader, SIGNAL( loadingComplete( Loader *, Document, Status ) ), + this, SLOT( feedLoaded( Loader *, Document, Status ) ) ); + feedLoader->loadFrom( m_feedUrl, new FileRetriever ); + } + + void RssFeed::feedLoaded(Loader *feedLoader, Document doc, Status status) + { + feedLoading = false; + + if ( status == Success ) + { + bool added = false; + + if (m_title.isEmpty() || m_title == QString("New")) + { + setTitle(doc.title()); + emit updateTitle(doc.title()); + } + + if (!m_ignoreTTL) + { + if (doc.ttl() < 0) + { + setAutoRefresh(QTime().addSecs(3600)); + } + else + { + setAutoRefresh(QTime().addSecs(doc.ttl() * 60)); + } + } + + RssArticle curArticle; + + for (int i=doc.articles().count()-1; i>=0; i--) + { + curArticle = doc.articles()[i]; + if (curArticle.pubDate().daysTo(QDateTime::currentDateTime()) < m_articleAge && !m_articles.contains(curArticle)) + { + m_articles.prepend(curArticle); + emit scanRssArticle(curArticle); + added = true; + } + } + + if (added) + { + emit articlesChanged(m_articles); + } + } else { + qDebug( "There was and error loading the feed\n"); + } + + disconnect( feedLoader, SIGNAL( loadingComplete( Loader *, Document, Status ) ), + this, SLOT( feedLoaded( Loader *, Document, Status ) ) ); + feedLoader->deleteLater(); + + } + + void RssFeed::setDownloaded(QString link, int downloaded) + { + bool changed = false; + + RssArticle::List::iterator it; + for ( it = m_articles.begin(); it != m_articles.end(); it++ ) + { + if ((*it).link().prettyURL() == link) + { + (*it).setDownloaded( downloaded ); + changed = true; + } + } + + if (changed) + { + emit articlesChanged(m_articles); + } + } + + QDataStream &operator<<( QDataStream &out, const RssFeed &feed ) + { + out << feed.feedUrl() << feed.title() << int(feed.active()) << feed.articleAge() << int(feed.ignoreTTL()) << feed.autoRefresh(); + + return out; + } + + QDataStream &operator>>( QDataStream &in, RssFeed &feed ) + { + KURL feedUrl; + QString title; + int active; + int articleAge; + int ignoreTTL; + QTime autoRefresh; + in >> feedUrl >> title >> active >> articleAge >> ignoreTTL >> autoRefresh; + feed = RssFeed(feedUrl, title, active, articleAge, ignoreTTL, autoRefresh); + + return in; + } + + RssFeed::~RssFeed() + { + } +} |