/*******************************************************************************
 **
 ** smolbote: yet another qute browser
 ** Copyright (C) 2017  Xian Nox
 **
 ** 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 "navigationbutton.h"

#include <QStyle>
#include <QWebEngineHistory>
#include <QMenu>

NavigationButton::NavigationButton(Type type, QWidget *parent) :
    QToolButton(parent)
{
    m_type = type;
    menu = new QMenu(this);

    switch (type) {
    case BackButton:
        setIcon(style()->standardIcon(QStyle::SP_ArrowBack));
        setMenu(menu);
        connect(menu, &QMenu::aboutToShow, this, &NavigationButton::prepareMenu);
        break;
    case ForwardButton:
        setIcon(style()->standardIcon(QStyle::SP_ArrowForward));
        setMenu(menu);
        connect(menu, &QMenu::aboutToShow, this, &NavigationButton::prepareMenu);
        break;
    case ReloadButton:
        setIcon(style()->standardIcon(QStyle::SP_BrowserReload));
        break;
    case StopButton:
        setIcon(style()->standardIcon(QStyle::SP_BrowserStop));
        break;
    }

    connect(this, &NavigationButton::clicked, this, &NavigationButton::doAction);

}


void NavigationButton::setView(WebView *view)
{
    disconnect(loadStartedConnection);
    disconnect(loadFinishedConnection);

    m_view = view;
    if(m_type == BackButton || m_type == ForwardButton) {
        updateOnLoadFinished();
    }

    loadStartedConnection = connect(view, &WebView::loadStarted, this, &NavigationButton::updateOnLoadStarted);
    loadFinishedConnection = connect(view, &WebView::loadFinished, this, &NavigationButton::updateOnLoadFinished);
}

void NavigationButton::updateOnLoadStarted()
{
    switch (m_type) {
    case BackButton:
        break;
    case ForwardButton:
        break;
    case ReloadButton:
        m_type = StopButton;
        setIcon(style()->standardIcon(QStyle::SP_BrowserStop));
        break;
    case StopButton:
        break;
    }
}

void NavigationButton::updateOnLoadFinished()
{
    switch (m_type) {
    case BackButton:
        if(m_view->history()->canGoBack()) {
            setEnabled(true);
        } else {
            setEnabled(false);
        }
        break;
    case ForwardButton:
        if(m_view->history()->canGoForward()) {
            setEnabled(true);
        } else {
            setEnabled(false);
        }
        break;
    case ReloadButton:
        break;
    case StopButton:
        m_type = ReloadButton;
        setIcon(style()->standardIcon(QStyle::SP_BrowserReload));
        break;
    }
}

void NavigationButton::doAction()
{
    switch (m_type) {
    case BackButton:
        m_view->history()->back();
        break;
    case ForwardButton:
        m_view->history()->forward();
        break;
    case ReloadButton:
        m_view->reload();
        break;
    case StopButton:
        m_view->stop();
        break;
    }
}

void NavigationButton::prepareMenu()
{
    menu->clear();

    QList<QWebEngineHistoryItem> items;
    switch (m_type) {
    case BackButton:
        items = m_view->history()->backItems(10);
        break;
    case ForwardButton:
        items = m_view->history()->forwardItems(10);
        break;
    default:
        break;
    }

    for(QWebEngineHistoryItem i : items) {
        QAction *a = menu->addAction(i.title());
        connect(a, &QAction::triggered, [i, this]() {
            m_view->history()->goToItem(i);
        });
    }
}