/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include "Tests/UI/ModalPopupHandler.h"
#include "Tests/UI/UIFixture.h"

#include <QApplication>
#include <QTimer>
#include <QMenu>
#include <QObject>
#include <QtTest>

namespace EMotionFX
{
    static const int WaitTickTime = 10;

    ModalPopupHandler::ModalPopupHandler(QObject* pParent)
        : QObject(pParent)
    {
    }

    void ModalPopupHandler::ShowContextMenuAndTriggerAction(QWidget* widget, const QString& actionName, int timeout, ActionCompletionCallback completionCallback)
    {
        MenuActiveCallback menuCallback = [=](QMenu* menu)
        {
            ASSERT_TRUE(menu) << "Failed to find context menu.";

            m_seenTargetWidget = true;

            QAction* action = menu->findChild<QAction*>(actionName);
            ASSERT_TRUE(action) << "Unable to find context menu action " << actionName.toUtf8().data();
            action->trigger();

            menu->close();

            if (m_actionCompletionCallback)
            {
                m_actionCompletionCallback(actionName);
            }
        };

        m_totalTime = 0;
        m_actionCompletionCallback = completionCallback;
        m_MenuActiveCallback = menuCallback;
        m_timeout = timeout;

        // Kick a timer off to check whether the menu is open.
        QTimer::singleShot(WaitTickTime, this, SLOT(CheckForContextMenu()));

        // Open the modal menu.
        UIFixture::BringUpContextMenu(widget, QPoint(10, 10), widget->mapToGlobal(QPoint(10, 10)));
    }

    void ModalPopupHandler::CheckForContextMenu()
    {
        m_totalTime += WaitTickTime;

        if (m_totalTime >= m_timeout)
        {
            if (m_actionCompletionCallback)
            {
                m_actionCompletionCallback("");
                return;
            }
        }

        // Check for the active widget being a popup widget.
        QWidget* qpop = QApplication::activePopupWidget();
        if (!qpop)
        {
            QTimer::singleShot(WaitTickTime, this, SLOT(CheckForContextMenu()));
            return;
        }

        // If the active widget is not a menu, keep waiting.
        QMenu* menu = qobject_cast<QMenu*>(qpop);
        if (!menu)
        {
            QTimer::singleShot(WaitTickTime, this, SLOT(CheckForContextMenu()));
            return;
        }

        // The menu is now active, inform the calling object.
        m_MenuActiveCallback(menu);
    }

    void ModalPopupHandler::CheckForPopupWidget()
    {
        m_totalTime += WaitTickTime;

        if (m_totalTime >= m_timeout)
        {
            if (m_actionCompletionCallback)
            {
                m_actionCompletionCallback("");
                m_complete = true;
                return;
            }
        }

        // Check for the active widget being a popup widget.
        QWidget* modalWidget = QApplication::activeModalWidget();
        if (!modalWidget)
        {
            QTimer::singleShot(WaitTickTime, this, SLOT(CheckForPopupWidget()));
            return;
        }

        m_seenTargetWidget = true;

        // Inform the calling object.
        m_widgetActiveCallback(modalWidget);
    }

    bool ModalPopupHandler::GetSeenTargetWidget()
    {
        return m_seenTargetWidget;
    }

    void ModalPopupHandler::ResetSeenTargetWidget()
    {
        m_seenTargetWidget = false;
    }

    bool ModalPopupHandler::GetIsComplete()
    {
        return m_complete;
    }

    void ModalPopupHandler::WaitForCompletion(const int timeout)
    {
        static_cast<void>(QTest::qWaitFor([&]() {
            return m_complete;
        }, timeout));
    }
} // namespace EMotionFX

#include <Tests/UI/ModalPopupHandler.moc>