#include "feedpoller.h"
#include <QtXml>
#include "feedmodel.h"
#include "feednode.h"
 
using namespace Larss;
 
FeedPoller::FeedPoller(QObject *parent, RssParser *parser, FeedModel *model) :
    QThread(parent)
{
    this->parser = parser;
    this->model = model;
 
    workQueue = new QList<QModelIndex> ();
    nowLoading = 0;
 
    rssContent = new QHash<quint32, QString>();
    poll_active = true;
 
    // Create the QNetworkAccessManager and connect the loaded signal
    // with our handler.
    manager = new QNetworkAccessManager ();
    manager->connect(manager, SIGNAL(finished(QNetworkReply*)),
                     this, SLOT(networkManagerReplyFinished(QNetworkReply*)));
 
    // We don't have a flooded bubble queue since there are no bubble queue
    // yet.
    floodedBubbleQueue = false;
}
 
void
FeedPoller::run()
{
 
    // Create the timer that will call the function every second.
    QTimer *timer = new QTimer ();
    timer->setInterval(1000);
    timer->connect(timer, SIGNAL(timeout()),
                   this, SLOT(poll()));
    timer->start();
 
    QTimer* updaterTimer = new QTimer();
    updaterTimer->setInterval(10 * 60000);
    updaterTimer->connect(updaterTimer, SIGNAL(timeout()),
                   this, SLOT(queueAll()));
    updaterTimer->start();
    queueAll();
 
    /* Connect timer to their terminations */
    QObject::connect(this, SIGNAL(finished()),
                   timer, SLOT(stop()));
    QObject::connect(this, SIGNAL(finished()),
                     updaterTimer, SLOT(stop()));
 
    QThread::exec();
}
 
bool
FeedPoller::poll()
{
    // Poll indefinitely until we are requested to exit.
    if (nowLoading == 0)
    {
        if (workQueue->isEmpty())
            return false;
        else
        {
            QModelIndex next_item = workQueue->takeFirst();
            FeedNode *node = model->itemFromIndex(next_item);
            nowLoading = node->id();
            startLoadingFeed(node->name());
            manager->get(QNetworkRequest(QUrl(model->getUrl(next_item))));
            return true;
        }
    }
 
    return false;
}
 
void
FeedPoller::stopPolling ()
{
    poll_active = false;
    QThread::exit();
}
 
void
FeedPoller::queueWork(const QModelIndex &index)
{
    if (!index.isValid())
        return;
    FeedNode *node = model->itemFromIndex(index);
    if (node->type() != FeedNode::Feed)
        return;
    workQueue->append(index);
}
 
void
FeedPoller::queueAll()
{
    QStandardItem *rootItem = model->invisibleRootItem();
    for(qint32 i = 0; i < rootItem->rowCount(); i++)
    {
        // Get the category
        FeedNode *categoryNode = model->itemFromIndex(rootItem->child(i, 0)->index());
 
        // Get all the feed in that category
        for(qint32 j = 0; j < categoryNode->rowCount(); j++)
        {
            QModelIndex feedIndex = categoryNode->child(j, 0)->index ();
            queueWork(feedIndex);
        }
    }
}
 
void
FeedPoller::networkManagerReplyFinished(QNetworkReply *reply)
{
    // Assume that the string is UTF-8 encoded. This is likely to be
    // true, but I should check it in some way.
    rssContent->insert (nowLoading, QString::fromUtf8(reply->readAll()));
 
    // Now update the database with the new data obtained.
    QDomDocument doc;
    if (doc.setContent(rssContent->value(nowLoading)))
    {
        // Get transaction to support, otherwise the db will die
        parser->db.transaction();
 
        // Try to catch other news_feed with the same link, so preload all of them.
        QSqlQuery query(parser->db);
        query.prepare ("SELECT link from news WHERE feed=:feed");
        query.bindValue("feed", nowLoading);
        if (!query.exec ())
            return;
 
        QStringList links;
        if (query.first())
        {
            links.append(query.value(0).toString());
            while (query.next())
                links.append(query.value(0).toString());
        }
 
        QDomElement doc_el = doc.documentElement();
        QDomNodeList items = doc_el.elementsByTagName("item");
 
        for (quint32 i = 0; i < items.length(); i++)
        {
            // Get the i-th news
            QDomNode item = items.item(i);
            QDomElement element = item.toElement();
 
            // Get the data in it
            QString link = element.elementsByTagName("link").item(0).firstChild().nodeValue();
            QString title = element.elementsByTagName("title").item(0).firstChild().nodeValue();
            QString description = element.elementsByTagName("description").item(0).firstChild().nodeValue();
            QString content = description;
            if (element.elementsByTagName("content:encoded").length() > 0)
                content = element.elementsByTagName("content:encoded").item(0).firstChild().nodeValue();
 
            // Check if this is RSS2 or not
            uint pubDate;
            QString guid;
            QString pubDateTimeContent;
 
            // Check if we have Guid, in RSS 1.0 was not present.
            if (element.elementsByTagName("guid").length() != 0)
                guid = element.elementsByTagName("guid").item(0).firstChild().nodeValue();
            else
                guid = link;
 
            // Check if we have pubDate
            if (element.elementsByTagName("pubDate").length() != 0)
            {
                pubDateTimeContent = element.elementsByTagName("pubDate").item(0).firstChild().nodeValue();
                pubDate = pubDateToDateTime(pubDateTimeContent).toTime_t();
            }
            else
            {
                pubDate = QDateTime::currentDateTimeUtc().toTime_t();
            }
 
            if (!links.contains(link))
            {
                // That means that no results were found, so let's insert this one.
                QSqlRecord record = parser->record();
                record.setValue("time", pubDate);
                record.setValue("read", 0);
                record.setValue("title", title);
                record.setValue("link", link);
                record.setValue("description", description);
                record.setValue("content", content);
                record.setValue("feed", nowLoading);
 
                if (!parser->insertRecord(-1, record))
                    qDebug () << "Error inserting record";
 
                // Notify of the newly inserted record.
                if (floodedBubbleQueue)
                {
                    // Check if enought time has passed.
                    QDateTime now = QDateTime::currentDateTimeUtc();
                    if (now >= lastShownBubble.addSecs(10))
                    {
                        lastShownBubble = now;
                        newElementsNotification(record.value("title").toString(),
                                                record.value("description").toString());
                        floodedBubbleQueue = false;
                    }
                }
                else
                {
                    QDateTime now = QDateTime::currentDateTimeUtc();
                    if (now <= lastShownBubble.addSecs(10))
                    {
                        floodedBubbleQueue = true;
                        lastShownBubble = now;
                        newElementsNotification(tr("New items to read"),
                                                tr("New elements have been loaded in Larss."));
                    }
                    else
                    {
                        lastShownBubble = QDateTime::currentDateTimeUtc();
                        newElementsNotification(record.value("title").toString(),
                                               record.value("description").toString());
                    }
                }
 
                // Yield to try making the UI responsive.
                yieldCurrentThread();
            }
        }
 
        if (!parser->submitAll())
            qDebug() << "Error submitting new data";
 
        // Commit the changes.
        parser->db.commit();
    }
    else
        qDebug () << "Error parsing the document";
 
    nowLoading = 0;
    finishedLoadingFeed("");
 
    return;
 
    // Check if there is work in the queue
    if (!workQueue->isEmpty())
    {
        QModelIndex next_item = workQueue->takeFirst();
        FeedNode *node = (FeedNode*) next_item.internalPointer();
        nowLoading = node->id();
        startLoadingFeed(node->name());
        manager->get(QNetworkRequest(QUrl(model->getUrl(next_item))));
    }
}
 
QDateTime
FeedPoller::pubDateToDateTime (QString pubDate)
{
    QString pubDateToParse;
    // If the Day is not present in pubDate set a fake
    // Day on top of it to make parsing working.
    if (!pubDate.contains(","))
        pubDateToParse = QString("Sun, %1").arg(pubDate);
    else
        pubDateToParse = pubDate;
    // Parsing the data, take Sun, 12 Oct 2011 15:21:12 GMT
    // pieces[0] is Sun  --- pieces[1] is 12  --- pieces[2] is Oct
    // pieces[3] is 2011 --- pieces[4] is 15:21:12 --- pieces[6] is GMT
    QStringList pieces = pubDateToParse.split(" ");
    QDateTime date;
 
    int month, year, day;
 
    // Set the day
    day = pieces.at(1).toInt();
 
    // Set the month
    if (pieces.at(2) == "Jan")
        month = 1;
    else if (pieces.at(2) == "Feb")
        month = 2;
    else if (pieces.at(2) == "Mar")
        month = 3;
    else if (pieces.at(2) == "Apr")
        month = 4;
    else if (pieces.at(2) == "May")
        month = 5;
    else if (pieces.at(2) == "Jun")
        month = 6;
    else if (pieces.at(2) == "Jul")
        month = 7;
    else if (pieces.at(2) == "Aug")
        month = 8;
    else if (pieces.at(2) == "Sep")
        month = 9;
    else if (pieces.at(2) == "Oct")
        month = 10;
    else if (pieces.at(2) == "Nov")
        month = 11;
    else if (pieces.at(2) == "Dec")
        month = 12;
 
    // Set the year
    year = pieces.at(3).toInt();
 
    date.setDate(QDate (year, month, day));
 
    // Set the hour
    date.setTime(QTime::fromString(pieces[4], "hh:mm:ss"));
 
    return date;
}