#include <libultraship/libultraship.h>
#include <libultra/gbi.h>
#include "../CoreMath.h"
#include <libultra/types.h>
#include "../World.h"

#include "Editor.h"
#include "Collision.h"
#include "Light.h"

#include "port/Engine.h"
#include <ship/controller/controldevice/controller/mapping/keyboard/KeyboardScancodes.h>
#include <ship/window/Window.h>

#include "engine/TrackBrowser.h"
#include "engine/actors/Ship.h"
#include "port/Game.h"

extern "C" {
#include "common_structs.h"
#include "main.h"
#include "defines.h"
#include "actors.h"
#include "camera.h"
}

namespace TrackEditor {
    Editor* Editor::Instance;

    Editor::Editor() {
        Instance = this;
    }

    Editor::~Editor() {
        ClearObjects();
        ClearMatrixPool();
    }

    void Editor::Load() {
        printf("Editor: Loading Editor...\n");
        eObjectPicker.Load();
        for (auto& object : eGameObjects) {
            GenerateCollisionMesh(object.get(), (Gfx*)object->Model, 1.0f);
            object->Load();
        }

        printf("Editor: Loading Complete!\n");
    }

    void Editor::Enable() {
        bEditorEnabled = true;
        bIsEditorPaused = true;
        CVarSetInteger("gFreecam", true);
        CM_SetFreeCamera(true);
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Tools")->Show();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Scene Explorer")->Show();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Content Browser")->Show();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Track Properties")->Show();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Properties")->Show();
    }

    void Editor::Disable() {
        bEditorEnabled = false;
        bIsEditorPaused = false;
        CVarSetInteger("gFreecam", false);
        CM_SetFreeCamera(false);
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Tools")->Hide();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Scene Explorer")->Hide();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Content Browser")->Hide();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Track Properties")->Hide();
        Ship::Context::GetInstance()->GetWindow()->GetGui()->GetGuiWindow("Properties")->Hide();
    }

    bool Editor::IsEnabled() {
        return bEditorEnabled;
    }

    void Editor::Play() {
        bIsEditorPaused = false;
    }

    void Editor::Pause() {
        bIsEditorPaused = true;
    }

    bool Editor::IsPaused() {
        return bIsEditorPaused;
    }

    void Editor::TogglePlayState() {
        bIsEditorPaused = !bIsEditorPaused;
    }

    void Editor::GenerateCollision() {
        // for (auto& actor : GetWorld()->Actors) {
        //     GenerateCollisionMesh(actor, (Gfx*)actor->Model, 1.0f);
        // }
    }

    void Editor::Tick() {
        if (Editor_IsEnabled() == true) {
            bEditorEnabled = true;
        } else {
            bEditorEnabled = false;
            bIsEditorPaused = false; // Prevents game being paused with the editor closed.
            return;
        }

        auto wnd = GameEngine::Instance->context->GetWindow();

        static bool wasMouseDown = false;
        static bool isDragging = false;
        static Ship::Coords mouseStartPos;

        Ship::Coords mousePos = wnd->GetMousePos();
        bool isMouseDown = wnd->GetMouseState(Ship::LUS_MOUSE_BTN_LEFT);

        //auto it = std::remove_if(eGameObjects.begin(), eGameObjects.end(),
        //    [](auto& object) {
        //        if (*object->DespawnFlag == object->DespawnValue) {
        //            delete object; // Free the pointed-to memory
        //            return true;   // Remove the pointer from the vector
        //        }
        //        return false;
        //    });

        //eGameObjects.erase(it, eGameObjects.end());

        if (isMouseDown && !wasMouseDown) {  
            // Mouse just pressed (Pressed state)
            mouseStartPos = mousePos;
            isDragging = false;
        }

        if (isMouseDown) {
            // Mouse is being held (Held state)
            int dx = mousePos.x - mouseStartPos.x;
            int dy = mousePos.y - mouseStartPos.y;
            int dragThreshold = 5;  // Adjust as needed

            if ((dx * dx + dy * dy) > (dragThreshold * dragThreshold)) {  
                // Squared distance check to avoid unnecessary sqrt()
                isDragging = true;
                eObjectPicker.DragHandle();  // Call drag logic
            }
        }

        if (!isMouseDown && wasMouseDown) {  
            // Mouse just released (Released state)
            eObjectPicker.eGizmo.SelectedHandle = Gizmo::GizmoHandle::None;
            eObjectPicker.eGizmo.ManipulationStart = true;
            if (!isDragging) {
                std::vector<GameObject*> objects;
                for (auto& object : eGameObjects) {
                    objects.push_back(object.get());
                }
                eObjectPicker.SelectObject(objects);
            }
        }

        wasMouseDown = isMouseDown;  // Update previous state

        eObjectPicker.Tick();

        for (auto& object : eGameObjects) {
            object->Tick();
        }
    }

    void Editor::Draw() {
        if (!bEditorEnabled) {
            return;
        } 
        eObjectPicker.Draw();
        for (auto& object : eGameObjects) {
            object->Draw();
        }
    }

    GameObject* Editor::AddObject(FVector pos, IRotator rot, FVector scale, const char* model, float collScale, GameObject::CollisionType collision, float boundingBoxSize) {
        //printf("After AddObj: Pos(%f, %f, %f), Name: %s, Model: %s\n", 
        // pos->x, pos->y, pos->z, name, model);

        if (nullptr != model && model[0] != '\0') {
            eGameObjects.push_back(std::make_unique<GameObject>(pos, rot, scale, model, std::vector<Triangle>(), collision, boundingBoxSize));
            GenerateCollisionMesh(eGameObjects.back().get(), (Gfx*)LOAD_ASSET_RAW(model), collScale);
        } else { // to bounding box or sphere collision
            eGameObjects.push_back(std::make_unique<GameObject>(pos, rot, scale, model, std::vector<Triangle>(), GameObject::CollisionType::BOUNDING_BOX,
                                                10.0f));
        }
        return eGameObjects.back().get();
    }

    void Editor::AddLight(const char* name, FVector* pos, s8* rot) {
        eGameObjects.push_back(std::make_unique<LightObject>(name, pos, rot));
    }

    void Editor::ClearObjects() {
        ResetGizmo();

        eGameObjects.clear();
    }

    // Reset the gizmo
    void Editor::ResetGizmo() {
        eObjectPicker.eGizmo._selected = static_cast<GameObject*>(nullptr);
        eObjectPicker._selected = static_cast<GameObject*>(nullptr);
        eObjectPicker.eGizmo.Pos = FVector(0, 0, 0);
        eObjectPicker.eGizmo.Enabled = false;
    }

    void Editor::DeleteObject() {

        std::visit([this](auto* obj) {
            if (nullptr != obj) {
                gEditor.ResetGizmo(); // Unselect the object to prevent crashes
                obj->Destroy();
            }
        }, eObjectPicker.eGizmo._selected);
    }

    void Editor::ClearMatrixPool() {
        EditorMatrix.clear();
    }

    void Editor::SelectObjectFromSceneExplorer(std::variant<AActor*, OObject*, GameObject*> object) {
        eObjectPicker._selected = object;
        eObjectPicker.eGizmo.Enabled = true;
        eObjectPicker.eGizmo.SetGizmoNoCursor(object);
    }

    void Editor::SetLevelDimensions(s16 minX, s16 maxX, s16 minZ, s16 maxZ, s16 minY, s16 maxY) {
        eObjectPicker.eGizmo.dimensions.MinX = minX + -1000;
        eObjectPicker.eGizmo.dimensions.MaxX = maxX + 1000;
        eObjectPicker.eGizmo.dimensions.MinY = minY + -100;
        eObjectPicker.eGizmo.dimensions.MaxY = maxY + 500;
        eObjectPicker.eGizmo.dimensions.MinZ = minZ + -1000;
        eObjectPicker.eGizmo.dimensions.MaxZ = maxZ + 1000;
    }
}

/** C BRIDGE FUNCTIONS **/

extern "C" void Editor_Launch(const char* resourceName) {
#if not defined(__SWITCH__) and not defined(__WIIU__)
    TrackBrowser::Instance->SetTrack(std::string(resourceName));
    gEditor.Enable();
#endif
}

extern "C" void Editor_SetLevelDimensions(s16 minX, s16 maxX, s16 minZ, s16 maxZ, s16 minY, s16 maxY) {
    gEditor.SetLevelDimensions(minX, maxX, minZ, maxZ, minY, maxY);
}

extern "C" bool Editor_IsEnabled() {
    return gEditor.IsEnabled();
}

extern "C" bool Editor_IsPaused() {
    return gEditor.IsPaused();
}