/* ============================================================
*
* This file is a part of the rekonq project
*
* Copyright (C) 2008-2009 by Andrea Diamantini <adjam7 at gmail dot com>
* Copyright (C) 2009 by Domrachev Alexandr <alexandr.domrachev@gmail.com>
* Copyright (C) 2009 by Paweł Prażak <pawelprazak at gmail dot com>
* Copyright (C) 2009 by Lionel Chauvin <megabigbug@yahoo.fr>
*
*
* 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/>.
*
* ============================================================ */


// Self Includes
#include "urlbar.h"
#include "urlbar.moc"

// Auto Includes
#include "rekonq.h"

// Local Includes
#include "mainwindow.h"
#include "webtab.h"
#include "webview.h"
#include "completionwidget.h"

// KDE Includes
#include <KDebug>
#include <KCompletionBox>

// Qt Includes
#include <QPainter>
#include <QPaintEvent>
#include <QPalette>
#include <QTimer>
#include <QVBoxLayout>

// Defines
#define QL1S(x)  QLatin1String(x)


IconButton::IconButton(QWidget *parent)
    : QToolButton(parent)
{
    setToolButtonStyle(Qt::ToolButtonIconOnly);
    setStyleSheet("IconButton { background-color:transparent; border: none; padding: 0px}");
    setCursor(Qt::ArrowCursor);
}

void IconButton::mouseReleaseEvent(QMouseEvent* event)
{
    emit clicked(event->globalPos());
}

// -----------------------------------------------------------------------------------------------------------


UrlBar::UrlBar(QWidget *parent)
    : KLineEdit(parent)
    , _tab(0)
    , _privateMode(false)
    , _icon( new IconButton(this) )
{
    // initial style
    setStyleSheet( QString("UrlBar { padding: 0 0 0 %1px;} ").arg(_icon->sizeHint().width()) );

    // cosmetic
    setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Fixed);
    setMinimumWidth(200);
    setMinimumHeight(26);

    // doesn't show the clear button
    setClearButtonShown(false);
    
    // trap Key_Enter & Key_Return events, while emitting the returnPressed signal
    setTrapReturnKey(true);
    
    // insert decoded URLs
    setUrlDropsEnabled(true);

    // accept focus, via tabbing, clicking & wheeling
    setFocusPolicy(Qt::WheelFocus);
    
    // disable completion object (we have our own :) )
    setCompletionObject(0);
    
    _tab = qobject_cast<WebTab *>(parent);
    
    connect(_tab->view(), SIGNAL(urlChanged(const QUrl &)), this, SLOT(setQUrl(const QUrl &)));
    connect(_tab->view(), SIGNAL(loadFinished(bool)), this, SLOT(loadFinished()));
    connect(_tab->view(), SIGNAL(loadStarted()), this, SLOT(clearRightIcons()));
    
    // load typed urls
    connect(this, SIGNAL(returnPressed(const QString &)), this, SLOT(loadTyped(const QString &)));

    activateSuggestions(true);
}


UrlBar::~UrlBar()
{
    activateSuggestions(false);
    delete _icon;
    _box.clear();
}


void UrlBar::setQUrl(const QUrl& url)
{
    if(url.scheme() == QL1S("about") )
    {
        _icon->setIcon( KIcon("arrow-right") );
        clear();
        setFocus();
    }
    else
    {
        clearFocus();
        KLineEdit::setUrl(url);
        setCursorPosition(0);
        _icon->setIcon( Application::icon(url) );
    }
}


void UrlBar::activated(const KUrl& url, Rekonq::OpenType type)
{
    activateSuggestions(false);
    
    clearFocus();
    setUrl(url);
    Application::instance()->loadUrl(url, type);
}


void UrlBar::paintEvent(QPaintEvent *event)
{
    QColor backgroundColor;
    if( _privateMode )
    {
        backgroundColor = QColor(220, 220, 220);  // light gray
    }
    else
    {
        backgroundColor = Application::palette().color(QPalette::Base);
    }
    
    // set background color of UrlBar
    QPalette p = palette();

    int progr = _tab->progress();
    if (progr == 0) 
    {
        if( _tab->url().scheme() == QL1S("https") )
        {
            backgroundColor = QColor(255, 255, 171);  // light yellow
        }
        p.setBrush(QPalette::Base, backgroundColor);
    } 
    else 
    {
        QColor loadingColor = QColor(116, 192, 250);
        
        QLinearGradient gradient(0, 0, width(), 0);
        gradient.setColorAt(0, loadingColor);
        gradient.setColorAt(((double)progr)/100, backgroundColor);
        p.setBrush(QPalette::Base, gradient);
    }
    setPalette(p);
    
    // you need this before our code to draw inside the line edit..
    KLineEdit::paintEvent(event);
    
    if (text().isEmpty()) 
    {       
        QStyleOptionFrame option;
        initStyleOption(&option);
        QRect textRect = style()->subElementRect(QStyle::SE_LineEditContents, &option, this);
        QPainter painter(this);
        painter.setPen(Qt::gray);
        painter.drawText( textRect, 
                          Qt::AlignCenter, 
                          i18n("Start typing here to search your bookmarks, history and the web...")
                        );
    }
}


void UrlBar::keyPressEvent(QKeyEvent *event)
{
    // this handles the Modifiers + Return key combinations
    QString currentText = text().trimmed();
    if ((event->key() == Qt::Key_Enter || event->key() == Qt::Key_Return)
        && !currentText.startsWith(QLatin1String("http://"), Qt::CaseInsensitive))
    {
        QString append;
        if (event->modifiers() == Qt::ControlModifier)
        {
            append = QLatin1String(".com");
        }
        else if (event->modifiers() == (Qt::ControlModifier | Qt::ShiftModifier))
        {
            append = QLatin1String(".org");
        }
        else if (event->modifiers() == Qt::ShiftModifier)
        {
            append = QLatin1String(".net");
        }

        QUrl url(QLatin1String("http://www.") + currentText);
        QString host = url.host();
        if (!host.endsWith(append, Qt::CaseInsensitive))
        {
            host += append;
            url.setHost(host);
            setText(url.toString());
        }
    }
    
    if (event->key() == Qt::Key_Escape)
    {
        clearFocus();
        event->accept();
    }

    KLineEdit::keyPressEvent(event);
}


void UrlBar::focusInEvent(QFocusEvent *event)
{
    activateSuggestions(true);
    
    KLineEdit::focusInEvent(event);
}


void UrlBar::setPrivateMode(bool on)
{
    _privateMode = on;
}


void UrlBar::dropEvent(QDropEvent *event)
{
    KLineEdit::dropEvent(event);
    activated(text());
}


void UrlBar::loadFinished()
{
    if(_tab->progress() != 0)
        return;
    
    if(_tab->url().scheme() == QL1S("about") )
    {
        update();
        return;
    }
    
    // show KGet downloads??
    if(ReKonfig::kgetList())
    {
        IconButton *bt = addRightIcon(UrlBar::KGet);
        connect(bt, SIGNAL(clicked()), _tab->page(), SLOT(downloadAllContentsWithKGet()));
    }
    
    // show RSS
    if(_tab->hasRSSInfo())
    {
        IconButton *bt = addRightIcon(UrlBar::RSS);
        connect(bt, SIGNAL(clicked(QPoint)), _tab, SLOT(showRSSInfo(QPoint)));
    }
    
    // show SSL
    if(_tab->url().scheme() == QL1S("https") )
    {
        IconButton *bt = addRightIcon(UrlBar::SSL);
        connect(bt, SIGNAL(clicked()), _tab->page(), SLOT(showSSLInfo()));
    }
    
    update();
}


void UrlBar::loadTyped(const QString &text)
{
    activated(KUrl(text));
}


void UrlBar::activateSuggestions(bool b)
{
    if(b)
    {
        if(_box.isNull())
        {
            _box = new CompletionWidget(this);
            installEventFilter(_box.data());
            connect(_box.data(), SIGNAL(chosenUrl(const KUrl &, Rekonq::OpenType)), this, SLOT(activated(const KUrl &, Rekonq::OpenType)));

            // activate suggestions on edit text
            connect(this, SIGNAL(textChanged(const QString &)), _box.data(), SLOT(suggestUrls(const QString &)));
        }
    }
    else
    {
        removeEventFilter(_box.data());
        _box.data()->deleteLater();
    }
}


void UrlBar::mouseDoubleClickEvent(QMouseEvent *)
{
    selectAll();
}


IconButton *UrlBar::addRightIcon(UrlBar::icon ic)
{
    IconButton *rightIcon = new IconButton(this);
    
    switch(ic)
    {
    case UrlBar::KGet:
        rightIcon->setIcon( KIcon("download") );
        rightIcon->setToolTip( i18n("List all links with KGet") );
        break;
    case UrlBar::RSS:
        rightIcon->setIcon( KIcon("application-rss+xml") );
        rightIcon->setToolTip( i18n("List all available RSS feeds") );
        break;
    case UrlBar::SSL:
        rightIcon->setIcon( KIcon("object-locked") );
        rightIcon->setToolTip( i18n("Show SSL Infos") );
        break;
    default:
        kDebug() << "ERROR.. default non extant case!!";
        break;
    }
    
    _rightIconsList << rightIcon;
    int iconsCount = _rightIconsList.count();
    rightIcon->move( width() - 23*iconsCount, 6);
    rightIcon->show();
    
    return rightIcon;
}


void UrlBar::clearRightIcons()
{
    qDeleteAll(_rightIconsList);
    _rightIconsList.clear();
}


void UrlBar::resizeEvent(QResizeEvent *event)
{
    int newHeight = ( height() - 19 )/2;
    _icon->move(4, newHeight );
    
    int iconsCount = _rightIconsList.count();
    int w = width();
    
    for(int i = 0; i < iconsCount; ++i)
    {
        IconButton *bt = _rightIconsList.at(i);
        bt->move( w - 25*(i+1), newHeight );
    }

    KLineEdit::resizeEvent(event);

}