#include "EditorLayer.h"
#include "panels/MenuBar.h"
#include "panels/HierarchyPanel.h"
#include "panels/SceneViewPanel.h"
#include "panels/GameViewPanel.h"
#include "panels/InspectorPanel.h"
#include "panels/ConsolePanel.h"
#include "panels/ProjectPanel.h"
#include "Managers/SelectionManager.h"
#include "Managers/SceneManager.h"
#include <imgui.h>
#include <imgui_internal.h>

namespace XCEngine {
namespace Editor {

EditorLayer::EditorLayer() : Layer("Editor") {}

void EditorLayer::onAttach() {
    m_menuBar = std::make_unique<MenuBar>();
    m_hierarchyPanel = std::make_unique<HierarchyPanel>();
    m_sceneViewPanel = std::make_unique<SceneViewPanel>();
    m_gameViewPanel = std::make_unique<GameViewPanel>();
    m_inspectorPanel = std::make_unique<InspectorPanel>();
    m_consolePanel = std::make_unique<ConsolePanel>();
    m_projectPanel = std::make_unique<ProjectPanel>();
    
    m_projectPanel->Initialize(m_projectPath);
    
    auto& selection = SelectionManager::Get();
    selection.OnSelectionChanged.Subscribe([](uint64_t id) {
    });
}

void EditorLayer::onDetach() {
}

void EditorLayer::onUpdate(float dt) {
}

void EditorLayer::onEvent(void* event) {
    ImGuiIO& io = ImGui::GetIO();
    
    if (ImGui::IsKeyPressed(ImGuiKey_F5)) {
        TogglePlay();
    }
    
    if (ImGui::IsKeyPressed(ImGuiKey_F6)) {
        if (GetEditorMode() != EditorMode::Edit) {
            TogglePause();
        }
    }
}

void EditorLayer::onImGuiRender() {
    setupDockspace();
    renderAllPanels();
}

void EditorLayer::SetProjectPath(const std::string& path) {
    m_projectPath = path;
}

void EditorLayer::setupDockspace() {
    static ImGuiDockNodeFlags dockspaceFlags = ImGuiDockNodeFlags_NoWindowMenuButton;
    
    ImGuiWindowFlags windowFlags = ImGuiWindowFlags_MenuBar | ImGuiWindowFlags_NoDocking;
    
    ImGuiViewport* viewport = ImGui::GetMainViewport();
    ImGui::SetNextWindowPos(viewport->Pos);
    ImGui::SetNextWindowSize(viewport->Size);
    ImGui::SetNextWindowViewport(viewport->ID);
    ImGui::PushStyleVar(ImGuiStyleVar_WindowRounding, 0.0f);
    ImGui::PushStyleVar(ImGuiStyleVar_WindowBorderSize, 0.0f);
    windowFlags |= ImGuiWindowFlags_NoTitleBar | ImGuiWindowFlags_NoCollapse | ImGuiWindowFlags_NoResize | ImGuiWindowFlags_NoMove;
    windowFlags |= ImGuiWindowFlags_NoBringToFrontOnFocus | ImGuiWindowFlags_NoNavFocus;
    
    ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, ImVec2(0.0f, 0.0f));
    ImGui::Begin("MainDockspace", nullptr, windowFlags);
    ImGui::PopStyleVar();
    ImGui::PopStyleVar(2);
    
    ImGuiID dockspaceId = ImGui::GetID("MyDockspace");
    ImGui::DockSpace(dockspaceId, ImVec2(0.0f, 0.0f), dockspaceFlags);
    
    static bool firstTime = true;
    if (firstTime) {
        firstTime = false;
        ImGui::DockBuilderRemoveNode(dockspaceId);
        ImGui::DockBuilderAddNode(dockspaceId, dockspaceFlags | ImGuiDockNodeFlags_DockSpace);
        ImGui::DockBuilderSetNodeSize(dockspaceId, viewport->Size);
        
        ImGuiID dockMain = dockspaceId;
        ImGuiID dockBottom = ImGui::DockBuilderSplitNode(dockMain, ImGuiDir_Down, 0.25f, nullptr, &dockMain);
        ImGuiID dockLeft = ImGui::DockBuilderSplitNode(dockMain, ImGuiDir_Left, 0.15f, nullptr, &dockMain);
        ImGuiID dockRight = ImGui::DockBuilderSplitNode(dockMain, ImGuiDir_Right, 0.25f, nullptr, &dockMain);
        
        ImGui::DockBuilderDockWindow("Hierarchy", dockLeft);
        ImGui::DockBuilderDockWindow("Scene", dockMain);
        ImGui::DockBuilderDockWindow("Game", dockMain);
        ImGui::DockBuilderDockWindow("Inspector", dockRight);
        ImGui::DockBuilderDockWindow("Console", dockBottom);
        ImGui::DockBuilderDockWindow("Project", dockBottom);
        
        ImGui::DockBuilderFinish(dockspaceId);
    }
    
    ImGui::End();
}

void EditorLayer::renderAllPanels() {
    m_menuBar->Render();
    
    m_hierarchyPanel->Render();
    m_sceneViewPanel->Render();
    m_gameViewPanel->Render();
    m_inspectorPanel->Render();
    m_consolePanel->Render();
    m_projectPanel->Render();
}

}
}