/** LICENSE ********************************************************************
 ** 
 ** 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 "webviewtabbar.h"

WebViewTabBar::WebViewTabBar(QWidget *parent) :
    QTabBar(parent)
{
    setTabsClosable(true);
    connect(this, SIGNAL(tabCloseRequested(int)), this, SLOT(handleTabClose(int)));
    connect(this, SIGNAL(currentChanged(int)), this, SLOT(handleCurrentChanged(int)));
}

WebViewTabBar::~WebViewTabBar()
{
    // cleanup
    qDeleteAll(m_views);
    m_views.clear();
}

int WebViewTabBar::addTab(QWebEngineProfile *profile, const QUrl &url)
{
    QWebEngineView *view = new QWebEngineView(0);
    QWebEnginePage *page = new QWebEnginePage(profile);
    view->setPage(page);
    page->load(url);
    m_views.append(view);

    //connect(view, SIGNAL(titleChanged()), this, SLOT(updateTabText()));
    connect(view, &QWebEngineView::titleChanged, [this, view](const QString &title) {
       int index = m_views.indexOf(view);
       setTabText(index, title);
    });
    connect(view, &QWebEngineView::iconChanged, [this, view](const QIcon &icon) {
        int index = m_views.indexOf(view);
        setTabIcon(index, icon);
    });

    return QTabBar::addTab("New Tab");
}

void WebViewTabBar::setProfile(QWebEngineProfile *profile)
{
    for(QWebEngineView *view : m_views) {
        QWebEnginePage *page = new QWebEnginePage(profile);
        page->load(view->url());
        view->setPage(page);
    }
}

QWebEngineView *WebViewTabBar::currentView()
{
    return m_views.at(currentIndex());
}

void WebViewTabBar::handleCurrentChanged(int index)
{
    emit(currentTabChanged(m_views.at(index)));
}

void WebViewTabBar::handleTabClose(int index)
{
    removeTab(index);
    m_views.at(index)->deleteLater();
    m_views.remove(index);
}

void WebViewTabBar::updateTabText(QWebEngineView *view, const QString &text)
{
    int index = m_views.indexOf(view);
    setTabText(index, text);
}