/* Copyright (C) 2014
 * swift project Community / Contributors
 *
 * This file is part of swift Project. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms
 * contained in the LICENSE file.
 */

#include "infoarea.h"
#include "stylesheetutility.h"
#include "guiutility.h"
#include "blackmisc/icons.h"
#include <QMenu>
#include <QListIterator>
#include <QSignalMapper>
#include <QCloseEvent>

using namespace BlackMisc;

namespace BlackGui
{
    CInfoArea::CInfoArea(QWidget *parent) : QMainWindow(parent)
    {
        this->ps_setInfoAreaFloating(this->m_infoAreaFloating);
        this->setWindowIcon(CIcons::swift24());
    }

    void CInfoArea::initInfoArea()
    {
        // after(!) GUI is established
        if (this->m_dockableWidgets.isEmpty())
        {
            this->m_dockableWidgets = this->findChildren<CDockWidgetInfoArea *>();
            Q_ASSERT(!this->m_dockableWidgets.isEmpty());
        }

        this->ps_setDockArea(Qt::TopDockWidgetArea);
        this->setMarginsWhenFloating(5, 5, 5, 5); // left, top, right bottom
        this->setMarginsWhenDocked(1, 1, 1, 1);   // top has no effect
        this->connectAllWidgets();
        this->setFeaturesForDockableWidgets(QDockWidget::DockWidgetFloatable | QDockWidget::DockWidgetMovable | QDockWidget::DockWidgetClosable);
        this->tabifyAllWidgets();

        // context menu
        this->setContextMenuPolicy(Qt::CustomContextMenu);
        connect(this, &CInfoArea::customContextMenuRequested, this, &CInfoArea::ps_showContextMenu);
        connect(&CStyleSheetUtility::instance(), &CStyleSheetUtility::styleSheetsChanged, this, &CInfoArea::ps_onStyleSheetChanged);

        // initial style sheet setting
        this->ps_onStyleSheetChanged();
    }

    void CInfoArea::addToContextMenu(QMenu *menu) const
    {
        if (!menu) return;
        menu->addAction(CIcons::dockTop16(), "Dock all", this, SLOT(dockAllWidgets()));
        menu->addAction(CIcons::floatAll16(), "Float all", this, SLOT(floatAllWidgets()));
        menu->addAction(CIcons::floatOne16(), "Dock / float info area", this, SLOT(toggleFloating()));
        QAction *lockTabBarMenuAction = new QAction(menu);
        lockTabBarMenuAction->setObjectName(this->objectName().append("LockTabBar"));
        lockTabBarMenuAction->setIconText("Lock tab bar");
        lockTabBarMenuAction->setIcon(CIcons::lockClosed16());
        lockTabBarMenuAction->setCheckable(true);
        lockTabBarMenuAction->setChecked(this->m_lockTabBar);
        menu->addAction(lockTabBarMenuAction);
        connect(lockTabBarMenuAction, &QAction::toggled, this, &CInfoArea::ps_toggleTabBarLocked);

        if (!this->m_dockableWidgets.isEmpty())
        {
            bool c = false;
            menu->addSeparator();
            QMenu *subMenuToggleFloat = new QMenu("Toggle Float/Dock", menu);
            QMenu *subMenuDisplay = new QMenu("Display", menu);

            QSignalMapper *signalMapperToggleFloating = new QSignalMapper(menu);
            QSignalMapper *signalMapperDisplay = new QSignalMapper(menu);

            for (int i = 0; i < this->m_dockableWidgets.size(); i++)
            {
                const CDockWidgetInfoArea *dw = this->m_dockableWidgets.at(i);
                const QPixmap pm = indexToPixmap(i);
                const QString t = dw->windowTitleBackup();
                QAction *checkableMenuAction = new QAction(menu);
                checkableMenuAction->setObjectName(QString(t).append("ToggleFloatingAction"));
                checkableMenuAction->setIconText(t);
                checkableMenuAction->setIcon(pm);
                checkableMenuAction->setData(QVariant(i));
                checkableMenuAction->setCheckable(true);
                checkableMenuAction->setChecked(!dw->isFloating());
                subMenuToggleFloat->addAction(checkableMenuAction);
                c = connect(checkableMenuAction, SIGNAL(toggled(bool)), signalMapperToggleFloating, SLOT(map()));
                Q_ASSERT(c);
                signalMapperToggleFloating->setMapping(checkableMenuAction, i);

                QAction *displayMenuAction = new QAction(menu);
                displayMenuAction->setObjectName(QString(t).append("DisplayAction"));
                displayMenuAction->setIconText(t);
                displayMenuAction->setIcon(pm);
                displayMenuAction->setData(QVariant(i));
                displayMenuAction->setCheckable(false);

                subMenuDisplay->addAction(displayMenuAction);
                c = connect(displayMenuAction, SIGNAL(triggered(bool)), signalMapperDisplay, SLOT(map()));
                Q_ASSERT(c);
                signalMapperDisplay->setMapping(displayMenuAction, i); // action to index
            }
            c = connect(signalMapperToggleFloating, SIGNAL(mapped(int)), this, SLOT(toggleFloating(int)));
            Q_ASSERT(c);

            c = connect(signalMapperDisplay, SIGNAL(mapped(int)), this, SLOT(selectArea(int)));
            Q_ASSERT(c);

            menu->addMenu(subMenuDisplay);
            menu->addMenu(subMenuToggleFloat);

            // where and how to display tab
            menu->addSeparator();
            QAction *showMenuText = new QAction(menu);
            showMenuText->setObjectName("ShowDockedWidgetTextAction");
            showMenuText->setIconText("Show tab text");
            showMenuText->setIcon(CIcons::headingOne16());
            showMenuText->setCheckable(true);
            showMenuText->setChecked(this->m_showTabTexts);
            menu->addAction(showMenuText);
            connect(showMenuText, &QAction::toggled, this, &CInfoArea::ps_showTabTexts);

            // auto adjust floating widgets
            QAction *showTabbar = new QAction(menu);
            showTabbar->setObjectName("ShowTabBar");
            showTabbar->setIconText("Show tab bar");
            showTabbar->setIcon(CIcons::dockBottom16());
            showTabbar->setCheckable(true);
            showTabbar->setChecked(this->m_showTabBar);
            menu->addAction(showTabbar);
            connect(showTabbar, &QAction::toggled, this, &CInfoArea::ps_showTabBar);

            // tab bar position
            menu->addAction(CIcons::dockBottom16(), "Toogle tabbar position", this, SLOT(ps_toggleTabBarPosition()));
        }
    }

    int CInfoArea::getSelectedTabBarIndex() const
    {
        Q_ASSERT(this->m_tabBar);
        if (!this->m_tabBar || this->m_tabBar->count() < 1) return -1;
        return this->m_tabBar->currentIndex();
    }

    bool CInfoArea::isSelectedInfoArea(const CDockWidgetInfoArea *infoArea) const
    {
        if (!infoArea) return false;
        if (infoArea->isFloating()) return false;
        int tabBarIndex = getSelectedTabBarIndex();
        if (tabBarIndex < 0 || tabBarIndex >= this->m_dockableWidgets.size()) return false;
        return this->getDockableWidgetByTabBarIndex(tabBarIndex) == infoArea;
    }

    void CInfoArea::dockAllWidgets()
    {
        this->tabifyAllWidgets();
    }

    void CInfoArea::adjustSizeForAllDockWidgets()
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            dw->adjustSize();
        }
    }

    void CInfoArea::floatAllWidgets()
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            if (dw->isFloating()) continue;
            dw->toggleFloating();
        }
    }

    void CInfoArea::toggleFloating()
    {
        this->ps_setInfoAreaFloating(!this->m_infoAreaFloating);
    }

    void CInfoArea::toggleFloating(int areaIndex)
    {
        if (areaIndex < 0 || areaIndex >= this->m_dockableWidgets.size()) return;
        CDockWidgetInfoArea *dw = this->m_dockableWidgets.at(areaIndex);
        Q_ASSERT(dw);
        if (!dw) return;
        dw->toggleFloating();
    }

    void CInfoArea::selectArea(int areaIndex)
    {
        CDockWidgetInfoArea *dw = this->m_dockableWidgets.at(areaIndex);
        Q_ASSERT(dw);
        if (!dw) return;
        Q_ASSERT(this->m_tabBar);
        if (m_tabBar->count() < 1) return;

        if (dw->isFloating())
        {
            dw->show();
        }
        else
        {
            int index = this->widgetToTabBarIndex(dw);
            Q_ASSERT(index >= 0);
            if (index >= 0 && index < m_tabBar->count())
            {
                m_tabBar->setCurrentIndex(index);
            }
        }
    }

    void CInfoArea::ps_setDockArea(Qt::DockWidgetArea area)
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            Qt::DockWidgetAreas newAreas = static_cast<Qt::DockWidgetAreas>(area);
            Qt::DockWidgetAreas oldAreas = dw->allowedAreas();
            if (oldAreas == newAreas) continue;
            dw->setAllowedAreas(newAreas);
            this->addDockWidget(area, dw);
        }
    }

    void CInfoArea::ps_setInfoAreaFloating(bool floating)
    {
        // float whole info area
        this->m_infoAreaFloating = floating;
        if (this->m_infoAreaFloating)
        {
            QPoint p = CGuiUtility::mainWindowPosition();
            this->setWindowFlags(Qt::Dialog);
            this->move(p.rx() + 20, p.ry() + 20);
            this->show(); // not working without show
        }
        else
        {
            // make this compliant as QWidget
            // https://qt-project.org/forums/viewthread/17519
            // http://www.qtcentre.org/threads/12569-QMainWindow-as-a-child-of-QMainWindow

            // this->setParent(this->m_originalParent, this->windowFlags() & ~Qt::Window);
            this->setWindowFlags(this->windowFlags() & ~Qt::Window);
            this->setVisible(true); // after redocking this is required
        }
    }

    void CInfoArea::tabifyAllWidgets()
    {
        // this->setDockArea(Qt::LeftDockWidgetArea);
        this->setTabPosition(Qt::LeftDockWidgetArea, QTabWidget::East);
        bool init = this->m_tabBar ? false : true;

        if (!this->m_dockableWidgets.isEmpty())
        {
            for (int i = 0; i < this->m_dockableWidgets.size(); i++)
            {
                CDockWidgetInfoArea *first = i > 0 ? this->m_dockableWidgets.at(i - 1) : nullptr;
                CDockWidgetInfoArea *after = this->m_dockableWidgets.at(i);
                Q_ASSERT(after);

                // trick, init widget as floating
                // this completely initializes the tab bar and all docked widgets
                if (init)
                {
                    QPoint offset(i * 25, i * 20);
                    after->setVisible(false);
                    after->setFloating(true);
                    after->setOffsetWhenFloating(offset);
                    after->setPreferredSizeWhenFloating(getPreferredSizeWhenFloating(i));
                    after->setFloating(false);
                    after->setVisible(true);
                    after->resetWasAlreadyFLoating();
                }
                else
                {
                    after->setFloating(false);
                }
                if (!first) { continue; }
                this->tabifyDockWidget(first, after);
            }
        }

        // as now tabified, now set tab
        if (!this->m_tabBar)
        {
            this->m_tabBar = this->findChild<QTabBar *>();
            Q_ASSERT(m_tabBar);
            QString qss = CStyleSheetUtility::instance().style(CStyleSheetUtility::fileNameDockWidgetTab());
            this->m_tabBar->setStyleSheet(qss);
            this->m_tabBar->setObjectName("comp_MainInfoAreaDockWidgetTab");
            this->m_tabBar->setMovable(false);
            this->m_tabBar->setElideMode(Qt::ElideNone);
            this->setTabPixmaps();

            // East / West does not work (shown, but area itself empty)
            // South does not have any effect
            this->m_tabBar->setShape(QTabBar::TriangularSouth);

            // signals
            connect(this->m_tabBar, &QTabBar::tabBarDoubleClicked, this, &CInfoArea::ps_tabBarDoubleClicked);
            connect(this->m_tabBar, &QTabBar::currentChanged, this, &CInfoArea::tabBarCurrentChanged);
        }

        if (this->countDockedWidgets() > 0)
        {
            this->m_tabBar->setCurrentIndex(0);
        }
    }

    void CInfoArea::unTabifyAllWidgets()
    {
        if (this->m_dockableWidgets.size() < 2) return;
        CDockWidgetInfoArea *first = this->m_dockableWidgets.first();
        for (int i = 1; i < this->m_dockableWidgets.size(); i++)
        {
            CDockWidgetInfoArea *after = this->m_dockableWidgets.at(i);
            Q_ASSERT(after);
            this->splitDockWidget(first, after, Qt::Horizontal);
        }
    }

    void CInfoArea::connectAllWidgets()
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            connect(dw, &CDockWidgetInfoArea::widgetTopLevelChanged, this, &CInfoArea::ps_onWidgetTopLevelChanged);
        }
    }

    void CInfoArea::setMarginsWhenFloating(int left, int top, int right, int bottom)
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            //! Margins when window is floating
            dw->setMarginsWhenFloating(left, top, right, bottom);
        }
    }

    void CInfoArea::setMarginsWhenDocked(int left, int top, int right, int bottom)
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            //! Margins when window is docked
            dw->setMarginsWhenDocked(left, top, right, bottom);
        }
    }

    int CInfoArea::countDockedWidgets() const
    {
        if (!this->m_tabBar) return 0;
        return this->m_tabBar->count();
    }

    CDockWidgetInfoArea *CInfoArea::getDockableWidgetByTabBarIndex(int tabBarIndex) const
    {
        if (tabBarIndex >= this->m_dockableWidgets.count() || tabBarIndex < 0) return nullptr;
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            if (dw->isFloating()) continue; // not in tab bar
            if (tabBarIndex == 0) return dw;
            tabBarIndex--;
        }
        return nullptr;
    }

    int CInfoArea::widgetToTabBarIndex(const CDockWidgetInfoArea *dockWidget)
    {
        if (!dockWidget) return -1;
        if (dockWidget->isFloating()) return -1;
        int tabBarIndex = 0;
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            if (dw->isFloating()) continue; // not in tab bar
            if (dw == dockWidget) return tabBarIndex;
            tabBarIndex++;
        }
        return -1;
    }

    void CInfoArea::setFeaturesForDockableWidgets(QDockWidget::DockWidgetFeatures features)
    {
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            dw->setFeatures(features);
        }
    }

    void CInfoArea::setTabPixmaps()
    {
        if (!this->m_tabBar) return;
        for (int i = 0; i < this->m_tabBar->count(); i++)
        {
            const QPixmap p = indexToPixmap(i);
            this->m_tabBar->setTabIcon(i, p);
        }
    }

    void CInfoArea::ps_tabBarDoubleClicked(int tabBarIndex)
    {
        if (this->m_lockTabBar) return;
        CDockWidgetInfoArea *dw = this->getDockableWidgetByTabBarIndex(tabBarIndex);
        if (!dw) return;
        dw->toggleFloating();
    }

    void CInfoArea::ps_onWidgetTopLevelChanged(CDockWidget *widget, bool topLevel)
    {
        Q_ASSERT(widget);
        Q_UNUSED(topLevel);
        if (!widget) return;

        // fix pixmaps
        this->setTabPixmaps();
    }

    void CInfoArea::ps_onStyleSheetChanged()
    {
        QString qss = CStyleSheetUtility::instance().style(CStyleSheetUtility::fileNameMainInfoArea());
        this->setStyleSheet(qss);
        if (this->m_tabBar)
        {
            QString qss = CStyleSheetUtility::instance().style(CStyleSheetUtility::fileNameDockWidgetTab());
            this->m_tabBar->setStyleSheet(qss);
        }
    }

    void CInfoArea::ps_showContextMenu(const QPoint &pos)
    {
        QPoint globalPos = this->mapToGlobal(pos);
        QScopedPointer<QMenu> contextMenu(new QMenu(this));
        this->addToContextMenu(contextMenu.data());

        QAction *selectedItem = contextMenu.data()->exec(globalPos);
        Q_UNUSED(selectedItem);
    }

    void CInfoArea::ps_showTabTexts(bool show)
    {
        if (show == this->m_showTabTexts) return;
        this->m_showTabTexts = show;
        for (CDockWidgetInfoArea *dw : this->m_dockableWidgets)
        {
            dw->showTitleWhenDocked(show);
        }
    }

    void CInfoArea::ps_showTabBar(bool show)
    {
        if (show == this->m_showTabBar) return;
        this->m_showTabBar = show;
        if (!this->m_tabBar) return;
        this->m_tabBar->setVisible(show); // not working, but setting right value will not harm anything
        this->m_tabBar->setMaximumHeight(show ? 10000 : 0); // does the trick
        this->adjustSizeForAllDockWidgets();
    }

    void CInfoArea::ps_toggleTabBarLocked(bool locked)
    {
        this->m_lockTabBar = locked;
    }

    void CInfoArea::ps_setTabBarPosition(QTabWidget::TabPosition position)
    {
        Q_ASSERT(position == QTabWidget::North || position == QTabWidget::South);
        this->setTabPosition(Qt::TopDockWidgetArea, position);
    }

    void CInfoArea::ps_toggleTabBarPosition()
    {
        QTabWidget::TabPosition p = (this->tabPosition(Qt::TopDockWidgetArea) == QTabWidget::North) ?
                                    QTabWidget::South : QTabWidget::North;
        this->ps_setTabBarPosition(p);
    }

    void CInfoArea::closeEvent(QCloseEvent *event)
    {
        if (this->isFloating())
        {
            this->toggleFloating();
            event->setAccepted(false); // refuse -> do not close
        }
        else
        {
            QMainWindow::closeEvent(event);
        }
    }
} // namespace