/* ============================================================
*
* This file is a part of the rekonq project
*
* Copyright (C) 2011 by Andrea Diamantini <adjam7 at gmail dot 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) version 3 or any later version
* accepted by the membership of KDE e.V. (or its successor approved
* by the membership of KDE e.V.), which shall act as a proxy
* defined in Section 14 of version 3 of the license.
*
* 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/>.
*
* ============================================================ */


// Auto Includes
#include "sslwidget.h"
#include "sslwidget.moc"

// Local includes
#include "application.h"
#include "historymanager.h"
#include "sslinfodialog.h"

// Qt Includes
#include <QtGui/QDialogButtonBox>
#include <QtGui/QGridLayout>
#include <QtGui/QLabel>
#include <QtGui/QPushButton>


SSLWidget::SSLWidget(const QUrl &url, const WebSslInfo &info, QWidget *parent)
    : QMenu(parent)
    , m_url(url)
    , m_info(info)
{
    setAttribute(Qt::WA_DeleteOnClose);
    setMinimumWidth(400);

    QSslCertificate cert = m_info.certificateChain().first();
    QStringList errorList = SslInfoDialog::errorsFromString(m_info.certificateErrors()).first();
    
    QGridLayout *layout = new QGridLayout(this);

    QLabel *label;
    QLabel *imageLabel;

    int rows = 0;
    // ------------------------------------------------------------------------------------------------------------------
    imageLabel = new QLabel(this);
    layout->addWidget(imageLabel, rows , 0, Qt::AlignCenter);

    label = new QLabel(this);
    label->setWordWrap(true);
    label->setText(i18n("<h4>Identity</h4>"));
    layout->addWidget(label, rows++, 1);

    if(cert.isNull())
    {
        label = new QLabel(this);
        label->setWordWrap(true);
        label->setText(i18n("Warning: this site is carrying a NULL certificate!"));

        imageLabel->setPixmap(KIcon("security-low").pixmap(32));
    }
    else
    {
        if(cert.isValid() && errorList.isEmpty())
        {
            label = new QLabel(this);
            label->setWordWrap(true);
            label->setTextFormat(Qt::PlainText);
            label->setText(i18n("This certificate for this site is valid and has been verified by:\n%1.",
                                cert.issuerInfo(QSslCertificate::CommonName)));


            imageLabel->setPixmap(KIcon("security-high").pixmap(32));
        }
        else
        {
            label = new QLabel(this);
            label->setWordWrap(true);

            QList<QStringList> errorList = SslInfoDialog::errorsFromString(m_info.certificateErrors());
            if(errorList.at(0).isEmpty())
            {
                label->setText(i18n("The certificate for this site is valid, but some on the certificate chain are not!"));
                imageLabel->setPixmap(KIcon("security-medium").pixmap(32));
            }
            else
            {
                QStringList sl = errorList.at(0);
                QString c = QL1S("<ul>");
                Q_FOREACH(const QString & s, sl)
                {
                    c += QL1S("<li>") + s + QL1S("</li>");
                }
                c += QL1S("</ul>");

                label->setText(i18n("The certificate for this site is NOT valid, for the following reasons:\n%1", c));
                label->setTextFormat(Qt::RichText);
                imageLabel->setPixmap(KIcon("security-low").pixmap(32));
            }
        }
    }

    layout->addWidget(label, rows++, 1);

    label = new QLabel(this);
    label->setWordWrap(true);
    label->setText("<a href=\"moresslinfos\">Certificate Information</a>");
    connect(label, SIGNAL(linkActivated(const QString &)), this, SLOT(showMoreSslInfos(const QString &)));
    layout->addWidget(label, rows++, 1);

    // ------------------------------------------------------------------------------------------------------------------
    label = new QLabel(this);
    label->setWordWrap(true);
    label->setText(QL1S("<h4>Encryption</h4>"));   // ----------------------------------------------- //
    layout->addWidget(label, rows, 1);

    imageLabel = new QLabel(this);
    layout->addWidget(imageLabel, rows++ , 0, Qt::AlignCenter);

    if(cert.isNull())
    {
        label = new QLabel(this);
        label->setWordWrap(true);
        label->setText(i18n("Your connection to %1 is NOT encrypted!!\n", m_url.host()));
        layout->addWidget(label, rows++ , 1);

        imageLabel->setPixmap(KIcon("security-low").pixmap(32));
    }
    else
    {
        label = new QLabel(this);
        label->setWordWrap(true);
        label->setTextFormat(Qt::PlainText);
        label->setText(i18n("Your connection to %1 is encrypted with %2-bit encryption.\n", m_url.host(), m_info.supportedChiperBits()));
        layout->addWidget(label, rows++, 1);

        int vers = cert.version().toInt();
        QString sslVersion;
        switch(vers)
        {
        case 0:
            sslVersion = QL1S("SSL 3.0");
            imageLabel->setPixmap(KIcon("security-high").pixmap(32));
            break;
        case 1:
            sslVersion = QL1S("SSL 2.0");
            imageLabel->setPixmap(KIcon("security-medium").pixmap(32));
            break;
        case 2:
        case 3:
            sslVersion = QL1S("TLS 1.0");
            imageLabel->setPixmap(KIcon("security-high").pixmap(32));
            break;
        default:
            sslVersion = QL1S("Unknown");
            imageLabel->setPixmap(KIcon("security-medium").pixmap(32));
        }

        label = new QLabel(this);
        label->setWordWrap(true);
        label->setText(i18n("The connection uses %1.\n", sslVersion));
        layout->addWidget(label, rows++, 1);

        const QStringList cipherInfo = m_info.ciphers().split('\n', QString::SkipEmptyParts);
        label = new QLabel(this);
        label->setWordWrap(true);
        label->setTextFormat(Qt::PlainText);
        label->setText(
            i18n("The connection is encrypted using %1 at %2 bits with %3 for message authentication and %4 as the key exchange mechanism.\n\n",
                 cipherInfo[0],
                 m_info.usedChiperBits(),
                 cipherInfo[3],
                 cipherInfo[1]));
        layout->addWidget(label, rows++, 1);

    }

    // ------------------------------------------------------------------------------------------------------------------
    imageLabel = new QLabel(this);
    layout->addWidget(imageLabel, rows , 0, Qt::AlignCenter);

    label = new QLabel(this);
    label->setWordWrap(true);
    label->setText(i18n("<h4>Site Information</h4>"));
    layout->addWidget(label, rows++, 1);

    label = new QLabel(this);
    label->setWordWrap(true);

    HistoryItem firstVisit = rApp->historyManager()->find(url.toString()).first();

    if(firstVisit.visitCount == 1)
    {
        label->setText(i18n("It's your first time visiting this site!"));
        imageLabel->setPixmap(KIcon("security-medium").pixmap(32));
    }
    else
    {
        label->setText(i18n("You just visited this site!\nYour first visit was on %1.\n", firstVisit.firstDateTimeVisit.toString()));
        imageLabel->setPixmap(KIcon("security-high").pixmap(32));
    }
    layout->addWidget(label, rows++, 1);

    // -----------------------------------------------------------------------------------
    setLayout(layout);
}


void SSLWidget::showAt(const QPoint &pos)
{
    adjustSize();

    QPoint p(pos.x() - width(), pos.y() + 10);
    move(p);
    show();
}


void SSLWidget::accept()
{
    close();
}


void SSLWidget::showMoreSslInfos(const QString &)
{
    QPointer<SslInfoDialog> dlg = new SslInfoDialog(m_url.host(), m_info, this);
    dlg->exec();
    delete dlg;

    return;
}