/* ============================================================
* QupZilla - WebKit based browser
* Copyright (C) 2010-2012  David Rosca <nowrep@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 3 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, see <http://www.gnu.org/licenses/>.
* ============================================================ */
#include "qupzillaschemehandler.h"
#include "globalfunctions.h"
#include "qupzilla.h"
#include "mainapplication.h"
#include "webpage.h"
#include "speeddial.h"
#include "pluginproxy.h"

QString authorString(const QString &name, const QString &mail)
{
    return QString("%1 &lt;<a href=\"mailto:%2\">%2</a>&gt;").arg(name, mail);
}

QupZillaSchemeHandler::QupZillaSchemeHandler(QObject* parent)
    : QObject(parent)
{
}

QNetworkReply* QupZillaSchemeHandler::createRequest(QNetworkAccessManager::Operation op, const QNetworkRequest &request, QIODevice* outgoingData)
{
    Q_UNUSED(outgoingData)

    if (op != QNetworkAccessManager::GetOperation) {
        return 0;
    }

    QupZillaSchemeReply* reply = new QupZillaSchemeReply(request);
    return reply;
}

QupZillaSchemeReply::QupZillaSchemeReply(const QNetworkRequest &req, QObject* parent)
    : QNetworkReply(parent)
{
    setOperation(QNetworkAccessManager::GetOperation);
    setRequest(req);
    setUrl(req.url());

    m_pageName = req.url().path();
    if (m_pageName == "about" || m_pageName == "reportbug" || m_pageName == "start" || m_pageName == "speeddial") {
        m_buffer.open(QIODevice::ReadWrite);
        setError(QNetworkReply::NoError, tr("No Error"));

        QTimer::singleShot(0, this, SLOT(loadPage()));
        open(QIODevice::ReadOnly);
    }
    else {
        setError(QNetworkReply::HostNotFoundError, tr("Not Found"));
        QTimer::singleShot(0, this, SLOT(delayedFinish()));
    }
}

void QupZillaSchemeReply::loadPage()
{
    QTextStream stream(&m_buffer);

    if (m_pageName == "about") {
        stream << aboutPage();
    }
    else if (m_pageName == "reportbug") {
        stream << reportbugPage();
    }
    else if (m_pageName == "start") {
        stream << startPage();
    }
    else if (m_pageName == "speeddial") {
        stream << speeddialPage();
    }

    stream.flush();
    m_buffer.reset();

    setHeader(QNetworkRequest::ContentTypeHeader, QByteArray("text/html"));
    setHeader(QNetworkRequest::ContentLengthHeader, m_buffer.bytesAvailable());
    setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 200);
    setAttribute(QNetworkRequest::HttpReasonPhraseAttribute, QByteArray("Ok"));
    emit metaDataChanged();
    emit downloadProgress(m_buffer.size(), m_buffer.size());

    emit readyRead();
    emit finished();
}

void QupZillaSchemeReply::delayedFinish()
{
    emit error(QNetworkReply::HostNotFoundError);
    emit finished();
}

qint64 QupZillaSchemeReply::bytesAvailable() const
{
    return m_buffer.bytesAvailable() + QNetworkReply::bytesAvailable();
}

qint64 QupZillaSchemeReply::readData(char* data, qint64 maxSize)
{
    return m_buffer.read(data, maxSize);
}

QString QupZillaSchemeReply::reportbugPage()
{
    static QString bPage;

    if (!bPage.isEmpty()) {
        return bPage;
    }

    bPage.append(qz_readAllFileContents(":html/reportbug.html"));
    bPage.replace("%FAVICON%", "qrc:icons/qupzilla.png");
    bPage.replace("%BOX-BORDER%", "qrc:html/box-border.png");

    bPage.replace("%TITLE%", tr("Report Issue"));
    bPage.replace("%REPORT-ISSUE%", tr("Report Issue"));
    bPage.replace("%PLUGINS-TEXT%", tr("If you are experiencing problems with QupZilla, please try to disable"
                                       " all plugins first. <br/>If this does not fix it, then please fill out this form: "));
    bPage.replace("%EMAIL%", tr("Your E-mail"));
    bPage.replace("%TYPE%", tr("Issue type"));
    bPage.replace("%DESCRIPTION%", tr("Issue description"));
    bPage.replace("%SEND%", tr("Send"));
    bPage.replace("%E-MAIL-OPTIONAL%", tr("E-mail is optional<br/><b>Note: </b>Please use English language only."));
    bPage.replace("%FIELDS-ARE-REQUIRED%", tr("Please fill out all required fields!"));

    return bPage;
}

QString QupZillaSchemeReply::startPage()
{
    static QString sPage;

    if (!sPage.isEmpty()) {
        return sPage;
    }

    sPage.append(qz_readAllFileContents(":html/start.html"));
    sPage.replace("%FAVICON%", "qrc:icons/qupzilla.png");
    sPage.replace("%BOX-BORDER%", "qrc:html/box-border.png");
    sPage.replace("%ABOUT-IMG%", "qrc:icons/other/about.png");

    sPage.replace("%TITLE%", tr("Start Page"));
    sPage.replace("%BUTTON-LABEL%", tr("Google Search"));
    sPage.replace("%SEARCH-BY-GOOGLE%", tr("Search results provided by Google"));
    sPage.replace("%WWW%", QupZilla::WIKIADDRESS);
    sPage.replace("%ABOUT-QUPZILLA%", tr("About QupZilla"));

    return sPage;
}

QString QupZillaSchemeReply::aboutPage()
{
    static QString aPage;

    if (!aPage.isEmpty()) {
        return aPage;
    }

    aPage.append(qz_readAllFileContents(":html/about.html"));
    aPage.replace("%FAVICON%", "qrc:icons/qupzilla.png");
    aPage.replace("%BOX-BORDER%", "qrc:html/box-border.png");
    aPage.replace("%ABOUT-IMG%", "qrc:icons/other/about.png");
    aPage.replace("%COPYRIGHT-INCLUDE%", Qt::escape(qz_readAllFileContents(":html/copyright")));

    aPage.replace("%TITLE%", tr("About QupZilla"));
    aPage.replace("%ABOUT-QUPZILLA%", tr("About QupZilla"));
    aPage.replace("%INFORMATIONS-ABOUT-VERSION%", tr("Informations about version"));
    aPage.replace("%BROWSER-IDENTIFICATION%", tr("Browser Identification"));
    aPage.replace("%PATHS%", tr("Paths"));
    aPage.replace("%COPYRIGHT%", tr("Copyright"));

    aPage.replace("%VERSION-INFO%",
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Version"), QupZilla::VERSION) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("WebKit version"), QupZilla::WEBKITVERSION) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Build time"), QupZilla::BUILDTIME) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Platform"), qz_buildSystem()));
    aPage.replace("%USER-AGENT%", WebPage::UserAgent);
    aPage.replace("%PATHS-TEXT%",
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Profile"), mApp->getActiveProfilPath()) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Settings"), mApp->getActiveProfilPath() + "settings.ini") +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Saved session"), mApp->getActiveProfilPath() + "session.dat") +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Pinned tabs"), mApp->getActiveProfilPath() + "pinnedtabs.dat") +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Data"), mApp->DATADIR) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Themes"), mApp->THEMESDIR) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Plugins"), mApp->PLUGINSDIR) +
                  QString("<dt>%1</dt><dd>%2<dd>").arg(tr("Translations"), mApp->TRANSLATIONSDIR));
    aPage.replace("%MAIN-DEVELOPER%", tr("Main developer"));
    aPage.replace("%MAIN-DEVELOPER-TEXT%", authorString(QupZilla::AUTHOR, "nowrep@gmail.com"));
    aPage.replace("%CONTRIBUTORS%", tr("Contributors"));
    aPage.replace("%CONTRIBUTORS-TEXT%",
                  authorString("Mladen Pejaković", "pejakm@gmail.com") + "<br/>" +
                  authorString("Bryan M Dunsmore", "dunsmoreb@gmail.com") + "<br/>" +
                  authorString("Mariusz Fik", "fisiu@opensuse.org") + "<br/>" +
                  authorString("Jan Rajnoha", "honza.rajny@hotmail.com")  + "<br/>" +
                  authorString("Daniele Cocca", "jmc@chakra-project.org")
                 );
    aPage.replace("%TRANSLATORS%", tr("Translators"));
    aPage.replace("%TRANSLATORS-TEXT%",
                  authorString("Heimen Stoffels", "vistausss@gmail.com") + " (Dutch)<br/>" +
                  authorString("Peter Vacula", "pvacula1989@gmail.com") + " (Slovak)<br/>" +
                  authorString("Ján Ďanovský", "dagsoftware@yahoo.com") + " (Slovak)<br/>" +
                  authorString("Jonathan Hooverman", "jonathan.hooverman@gmail.com") + " (German)<br/>" +
                  authorString("Unink-Lio", "unink4451@163.com") + " (Chinese)<br/>" +
                  authorString("Federico Fabiani", "federico.fabiani85@gmail.com") + " (Italy)<br/>" +
                  authorString("Francesco Marinucci", "framarinucci@gmail.com") + " (Italy)<br/>" +
                  authorString("Jorge Sevilla", "jsevi@ozu.es") + " (Spanish)<br/>" +
                  authorString("Michał Szymanowski", "tylkobuba@gmail.com") + " (Polish)<br/>" +
                  authorString("Mariusz Fik", "fisiu@opensuse.org") + " (Polish)<br/>" +
                  authorString("Jérôme Giry", "baikalink@hotmail.fr") + " (French)<br/>" +
                  authorString("Nicolas Ourceau", "lamessen@hotmail.fr") + " (French)<br/>" +
                  authorString("Vasilis Tsivikis", "vasitsiv.dev@gmail.com") + " (Greek)<br/>" +
                  authorString("Alexander Maslov", "it@delta-z.ru") + " (Russian)<br/>" +
                  authorString("Oleg Brezhnev", "oleg-423@yandex.ru") + " (Russian)<br/>" +
                  authorString("Sérgio Marques", "smarquespt@gmail.com") + " (Portuguese)<br/>" +
                  authorString("Mladen Pejaković", "pejakm@gmail.com") + " (Serbian)"
                 );

    return aPage;
}

QString QupZillaSchemeReply::speeddialPage()
{
    static QString dPage;

    if (dPage.isEmpty()) {
        dPage.append(qz_readAllFileContents(":html/speeddial.html"));
        dPage.replace("%FAVICON%", "qrc:icons/qupzilla.png");
        dPage.replace("%IMG_PLUS%", "qrc:html/plus.png");
        dPage.replace("%BOX-BORDER%", "qrc:html/box-border-small.png");
        dPage.replace("%IMG_CLOSE%", "qrc:html/close.png");
        dPage.replace("%IMG_EDIT%", "qrc:html/edit.png");
        dPage.replace("%IMG_RELOAD%", "qrc:html/reload.png");

        dPage.replace("%SITE-TITLE%", tr("Speed Dial"));
        dPage.replace("%ADD-TITLE%", tr("Add New Page"));
        dPage.replace("%TITLE-EDIT%", tr("Edit"));
        dPage.replace("%TITLE-REMOVE%", tr("Remove"));
        dPage.replace("%TITLE-RELOAD%", tr("Reload"));
        dPage.replace("%TITLE-FETCHTITLE%", tr("Load title from page"));
        dPage.replace("%JQUERY%", "qrc:html/jquery.js");
        dPage.replace("%JQUERY-UI%", "qrc:html/jquery-ui.js");
        dPage.replace("%LOADING-IMG%", "qrc:html/loading.gif");
        dPage.replace("%URL%", tr("Url"));
        dPage.replace("%TITLE%", tr("Title"));
        dPage.replace("%APPLY%", tr("Apply"));
        dPage.replace("%NEW-PAGE%", tr("New Page"));
        dPage.replace("%IMG_SETTINGS%", "qrc:html/setting.png");
        dPage.replace("%SETTINGS-TITLE%", tr("Speed Dial settings"));
        dPage.replace("%TXT_PLACEMENT%", tr("Placement: "));
        dPage.replace("%TXT_AUTO%", tr("Auto"));
        dPage.replace("%TXT_COVER%", tr("Cover"));
        dPage.replace("%TXT_FIT%", tr("Fit"));
        dPage.replace("%TXT_FWIDTH%", tr("Fit Width"));
        dPage.replace("%TXT_FHEIGHT%", tr("Fit Height"));
        dPage.replace("%TXT_NOTE%", tr("Use background image"));
        dPage.replace("%TXT_SELECTIMAGE%", tr("Select image"));
        dPage.replace("%TXT_NRROWS%", tr("Max pages in row:"));
    }

    QString page = dPage;
    page.replace("%INITIAL-SCRIPT%", mApp->plugins()->speedDial()->initialScript());
    page.replace("%IMG_BACKGROUND%", mApp->plugins()->speedDial()->backgroundImage());
    page.replace("%B_SIZE%", mApp->plugins()->speedDial()->backgroundImageSize());
    page.replace("%ROW-PAGES%", QString::number(mApp->plugins()->speedDial()->pagesInRow()));
    return page;
}