/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/gitea/aqua/smolbote
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#include "tabwidget.h"
#include "browser.h"
#include "webengine/webview.h"
#include <QAction>
#include <QContextMenuEvent>
#include <QMenu>
#include <QTabBar>
#include <web/webprofile.h>
#include <QWebEngineHistory>

inline WebView *createViewFromInfo(TabWidget::TabInformation &tab, QWidget *parent)
{
    auto *view = new WebView(tab.profile, parent);
    QDataStream stream(&tab.historyBuffer, QIODevice::ReadOnly);
    stream >> *view->history();
    view->history()->goToItem(view->history()->itemAt(tab.historyIndex));
    return view;
}

TabWidget::TabWidget(QWidget *parent)
    : QTabWidget(parent)
{
    setStyleSheet("QTabBar::tab { width: 200px; }");

    setTabsClosable(true);
    //setTabBarAutoHide(true);
    setElideMode(Qt::ElideRight);
    setMovable(true);

    connect(this, &TabWidget::tabCloseRequested, this, &TabWidget::deleteTab);

    // when changing tabs, give focus to the widget
    // otherwise when closing tabs, the tabwidget will retain focus
    connect(this, &TabWidget::currentChanged, this, [this](int index) {
        if(widget(index))
            widget(index)->setFocus();
    });

    // context menu
    tabContextMenu = new QMenu(this);
    auto *closeTab = tabContextMenu->addAction(tr("Close Tab"));
    connect(closeTab, &QAction::triggered, this, [this]() {
        deleteTab(this->tabBar()->tabAt(mapFromGlobal(tabContextMenu->pos())));
    });

    auto *closeTabsLeft = tabContextMenu->addAction(tr("Close Tabs left"));
    connect(closeTabsLeft, &QAction::triggered, this, [this]() {
        int idx = this->tabBar()->tabAt(mapFromGlobal(tabContextMenu->pos()));
        for(int i = idx - 1; i >= 0; --i) {
            deleteTab(i);
        }
    });

    auto *closeTabsRight = tabContextMenu->addAction(tr("Close Tabs right"));
    connect(closeTabsRight, &QAction::triggered, this, [this]() {
        int idx = this->tabBar()->tabAt(mapFromGlobal(tabContextMenu->pos()));
        for(int i = count() - 1; i > idx; --i) {
            deleteTab(i);
        }
    });
}

TabWidget::~TabWidget()
{
    for(int i = count() - 1; i >= 0; i--) {
        delete widget(i);
    }
}

QMenu *TabWidget::createTabMenu(QWidget *parent)
{
    auto *menu = new QMenu(parent);

    auto *reopenTabAction = menu->addAction(tr("Reopen last tab"));
    connect(reopenTabAction, &QAction::triggered, this, [this]() {
        if(!m_closedTabs.isEmpty()) {
            TabInformation tab = m_closedTabs.takeLast();
            addTab(createViewFromInfo(tab, this));
        }
    });

    auto *tabHistoryMenu = menu->addMenu(tr("Tab History"));
    connect(tabHistoryMenu, &QMenu::aboutToShow, this, [this, tabHistoryMenu]() {
        tabHistoryMenu->clear();
        for(int i = 0; i < m_closedTabs.count(); ++i) {
            auto *openAction = tabHistoryMenu->addAction(m_closedTabs.at(i).title);

            connect(openAction, &QAction::triggered, this, [this, i]() {
                TabInformation tab = m_closedTabs.takeAt(i);
                addTab(createViewFromInfo(tab, this));
            });
        }

        tabHistoryMenu->addSeparator();

        auto *clearTabHistory = tabHistoryMenu->addAction(tr("Clear"));
        connect(clearTabHistory, &QAction::triggered, this, [this]() {
            m_closedTabs.clear();
        });
    });

    connect(menu, &QMenu::aboutToShow, this, [this, reopenTabAction, tabHistoryMenu]() {
        if(m_closedTabs.isEmpty()) {
            reopenTabAction->setEnabled(false);
            tabHistoryMenu->setEnabled(false);
        } else {
            reopenTabAction->setEnabled(true);
            tabHistoryMenu->setEnabled(true);
        }
    });

    return menu;
}

int TabWidget::addTab(WebView *view)
{
    Q_ASSERT_X(view != nullptr, "TabWidget::addTab", "Tried to add null view");

    int idx = QTabWidget::addTab(view, view->title());
    connect(view, &WebView::titleChanged, [this, view](const QString &title) {
        int idx = this->indexOf(view);
        Q_ASSERT(idx != -1);

        this->setTabText(idx, title);
    });
    connect(view, &WebView::iconChanged, [this, view](const QIcon &icon) {
        int idx = this->indexOf(view);
        Q_ASSERT(idx != -1);

        this->setTabIcon(idx, icon);
    });

    return idx;
}

void TabWidget::deleteTab(int index)
{
    // deleting the widget automatically removes the tab?
    WebView *w = qobject_cast<WebView *>(widget(index));

    TabInformation tab;
    tab.profile = w->profile();
    tab.title = w->title();
    tab.historyIndex = w->history()->currentItemIndex();
    QDataStream stream(&tab.historyBuffer, QIODevice::WriteOnly);
    stream << *w->history();
    m_closedTabs.enqueue(tab);

    while(m_closedTabs.length() > 5) {
        m_closedTabs.dequeue();
    }

    disconnect(w);
    delete w;

    if(count() == 0)
        parentWidget()->close();
}

void TabWidget::contextMenuEvent(QContextMenuEvent *event)
{
    // check if the context menu was called on a tab
    int tabIndex = tabBar()->tabAt(event->pos());
    if(tabIndex < 0) {
        return;
    }

    tabContextMenu->exec(event->globalPos());
}

void TabWidget::mousePressEvent(QMouseEvent *event)
{
    if(event->button() == Qt::MiddleButton) {
        int index = tabBar()->tabAt(event->pos());
        if(index >= 0) {
            deleteTab(index);
        }
        event->accept();
        return;
    }

    QTabWidget::mousePressEvent(event);
}