/*
 * 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/smolbote.hg
 *
 * 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>

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()) {
            QWebEnginePage *page = m_closedTabs.takeLast();
            auto *view = new WebView(qobject_cast<WebProfile *>(page->profile()), this);
            view->setPage(page);
            addTab(view);
        }
    });

    auto *tabHistoryMenu = menu->addMenu(tr("Tab History"));
    connect(tabHistoryMenu, &QMenu::aboutToShow, this, [this, tabHistoryMenu]() {
        tabHistoryMenu->clear();
        for(QWebEnginePage *page : m_closedTabs) {
            auto *openAction = tabHistoryMenu->addAction(page->title());
            connect(openAction, &QAction::triggered, this, [page, this]() {
                auto *view = new WebView(qobject_cast<WebProfile *>(page->profile()), this);
                view->setPage(page);
                this->addTab(view);
                m_closedTabs.removeOne(page);
            });
        }
        tabHistoryMenu->addSeparator();
        auto *clearTabHistory = tabHistoryMenu->addAction(tr("Clear"));
        connect(clearTabHistory, &QAction::triggered, this, [this]() {
            while(!m_closedTabs.isEmpty()) {
                delete m_closedTabs.dequeue();
            }
        });
    });

    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?
    auto *w = widget(index);

    QWebEnginePage *p = qobject_cast<WebView *>(w)->page();
    p->setParent(nullptr);
    m_closedTabs.enqueue(p);

    while(m_closedTabs.length() > 5) {
        delete 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);
}