/* ============================================================
* 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 "webhistorywrapper.h"

#include <QUrl>
#include <QWebHistory>

WebHistoryWrapper::WebHistoryWrapper()
{
}

QList<QWebHistoryItem> WebHistoryWrapper::forwardItems(int maxItems, QWebHistory* history)
{
    QList<QWebHistoryItem> list;
    QUrl lastUrl = history->currentItem().url();

    int count = 0;
    foreach(const QWebHistoryItem & item, history->forwardItems(maxItems + 5)) {
        if (item.url() == lastUrl || count == maxItems) {
            continue;
        }

        lastUrl = item.url();
        list.append(item);
        count++;
    }

    return list;
}

QList<QWebHistoryItem> WebHistoryWrapper::backItems(int maxItems, QWebHistory* history)
{
    QList<QWebHistoryItem> list;
    QUrl lastUrl = history->currentItem().url();

    int count = 0;
    QList<QWebHistoryItem> bItems = history->backItems(maxItems + 5);
    for (int i = bItems.count() - 1; i >= 0; i--) {
        QWebHistoryItem item = bItems.at(i);
        if (item.url() == lastUrl || count == maxItems) {
            continue;
        }

        lastUrl = item.url();
        list.append(item);
        count++;
    }

    return list;
}

bool WebHistoryWrapper::canGoForward(QWebHistory* history)
{
    return !forwardItems(1, history).isEmpty();
}

bool WebHistoryWrapper::canGoBack(QWebHistory* history)
{
    return !backItems(1, history).isEmpty();
}

void WebHistoryWrapper::goBack(QWebHistory* history)
{
    QList<QWebHistoryItem> items = backItems(1, history);

    if (items.isEmpty()) {
        return;
    }

    history->goToItem(items.at(0));
}

void WebHistoryWrapper::goForward(QWebHistory* history)
{
    QList<QWebHistoryItem> items = forwardItems(1, history);

    if (items.isEmpty()) {
        return;
    }

    history->goToItem(items.at(0));
}

int WebHistoryWrapper::indexOfItem(const QList<QWebHistoryItem> &list, const QWebHistoryItem &item)
{
    for (int i = 0; i < list.count(); i++) {
        QWebHistoryItem it = list.at(i);

        if (it.lastVisited() == item.lastVisited() &&
                it.originalUrl() == item.originalUrl() &&
                it.title() == item.title() &&
                it.url() == item.url()) {
            return i;
        }
    }

    return -1;
}