/*
 * Copyright (c) Contributors to the Open 3D Engine Project
 * 
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#pragma once

#if !defined(Q_MOC_RUN)
#include <AzCore/Memory/SystemAllocator.h>
#include <Atom/Document/ShaderManagementConsoleDocumentNotificationBus.h>

#include <Atom/RPI.Public/Shader/Shader.h>
#include <Atom/RPI.Edit/Shader/ShaderVariantListSourceData.h>

AZ_PUSH_DISABLE_WARNING(4251 4800, "-Wunknown-warning-option") // disable warnings spawned by QT
#include <AzQtComponents/Components/DockMainWindow.h>
#include <AzQtComponents/Components/FancyDocking.h>
#include <AzQtComponents/Components/StyledDockWidget.h>
#include <AzQtComponents/Components/Widgets/TabWidget.h>

#include <Window/ShaderManagementConsoleBrowserWidget.h>
#include <Window/ToolBar/ShaderManagementConsoleToolBar.h>

#include <QMenuBar>
#include <QToolBar>
#include <QStandardItemModel>
AZ_POP_DISABLE_WARNING
#endif

namespace AzToolsFramework
{
    class CScriptTermDialog;
}

namespace ShaderManagementConsole
{
    /**
     * ShaderManagementConsoleWindow is the main class. Its responsibility is limited to initializing and connecting
     * its panels, managing selection of assets, and performing high-level actions like saving. It contains...
     */
    class ShaderManagementConsoleWindow
        : public AzQtComponents::DockMainWindow
        , private ShaderManagementConsoleDocumentNotificationBus::Handler
    {
        Q_OBJECT
    public:
        AZ_CLASS_ALLOCATOR(ShaderManagementConsoleWindow, AZ::SystemAllocator, 0);

        ShaderManagementConsoleWindow(QWidget* parent = 0);
        ~ShaderManagementConsoleWindow();

    private:
        // ShaderManagementConsoleDocumentNotificationBus::Handler overrides...
        void OnDocumentOpened(const AZ::Uuid& documentId) override;
        void OnDocumentClosed(const AZ::Uuid& documentId) override;
        void OnDocumentModified(const AZ::Uuid& documentId) override;
        void OnDocumentUndoStateChanged(const AZ::Uuid& documentId) override;
        void OnDocumentSaved(const AZ::Uuid& documentId) override;

        void SetupMenu();

        void SetupTabs();
        void AddTabForDocumentId(const AZ::Uuid& documentId);
        void RemoveTabForDocumentId(const AZ::Uuid& documentId);
        void UpdateTabForDocumentId(const AZ::Uuid& documentId);
        AZ::Uuid GetDocumentIdFromTab(const int tabIndex) const;

        void OpenTabContextMenu();
        void SelectPreviousTab();
        void SelectNextTab();

        void SelectDocumentForTab(const int tabIndex);
        void CloseDocumentForTab(const int tabIndex);
        void CloseAllExceptDocumentForTab(const int tabIndex);

        void closeEvent(QCloseEvent* closeEvent) override;

        void CreateDocumentContent(const AZ::Uuid& documentId, QStandardItemModel* model);

        AzQtComponents::FancyDocking* m_advancedDockManager = nullptr;
        QMenuBar* m_menuBar = nullptr;
        QWidget* m_centralWidget = nullptr;
        AzQtComponents::TabWidget* m_tabWidget = nullptr;
        ShaderManagementConsoleBrowserWidget* m_assetBrowser = nullptr;
        ShaderManagementConsoleToolBar* m_toolBar = nullptr;
        AzToolsFramework::CScriptTermDialog* m_pythonTerminal = nullptr;

        AzQtComponents::StyledDockWidget* m_assetBrowserDockWidget = nullptr;
        AzQtComponents::StyledDockWidget* m_pythonTerminalDockWidget = nullptr;

        QMenu* m_menuFile = {};
        QMenu* m_menuNew = {};
        QAction* m_actionOpen = {};
        QAction* m_actionOpenRecent = {};
        QAction* m_actionClose = {};
        QAction* m_actionCloseAll = {};
        QAction* m_actionCloseOthers = {};
        QAction* m_actionSave = {};
        QAction* m_actionSaveAsCopy = {};
        QAction* m_actionSaveAll = {};
        QAction* m_actionExit = {};

        QMenu* m_menuEdit = {};
        QAction* m_actionUndo = {};
        QAction* m_actionRedo = {};
        QAction* m_actionPreferences = {};

        QMenu* m_menuView = {};
        QAction* m_actionAssetBrowser = {};
        QAction* m_actionPythonTerminal = {};
        QAction* m_actionNextTab = {};
        QAction* m_actionPreviousTab = {};

        QMenu* m_menuHelp = {};
        QAction* m_actionHelp = {};
        QAction* m_actionAbout = {};
    };
} // namespace ShaderManagementConsole