#include "viewer.h"
// #include "image_handler.h"
// #include "radiation_handler.h"
// #include "scatterometer_handler.h"
// #include "core/config.h"
// #include "products/dataset.h"
#include "common/utils.h"
// #include "resources.h"
#include "main_ui.h"

#include "products2/handler/dummy_handler.h"
#include "products2/handler/product/image_product_handler.h" // TODOREWORK CLEAN
#include "products2/handler/dataset/dataset_handler.h"
#include "products2/handler/vector/shapefile_handler.h"
#include "products2/handler/projection/projection_handler.h"
#include "products2/handler/trash/trash_handler.h"
#include "dsp/dsp_flowgraph_handler.h"
// TODOREWORK
#include "resources.h"

// TODOREWORK
#include "dsp/waterfall_test.h"
#include "dsp/newrec.h"

namespace satdump
{
    namespace viewer
    {
        ViewerApplication::ViewerApplication()
            : Application("viewer")
        {
            master_handler = std::make_shared<DummyHandler>("MasterHandlerViewer");

            // Add trashcan
            trash_handler = std::make_shared<DummyHandler>("TrashHandlerViewer");
            auto trash_h = std::make_shared<TrashHandler>();
            trash_h->setCanBeDragged(false);
            trash_handler->addSubHandler(trash_h);
            trash_handler->setCanBeDraggedTo(false);

            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/metop_test/dataset.json");
        }

        ViewerApplication::~ViewerApplication()
        {
            if (file_open_thread.joinable())
                file_open_thread.join();
        }

        void ViewerApplication::drawPanel()
        {
            ImGui::SetNextItemWidth(ImGui::GetWindowWidth() / 2);

            if (ImGui::CollapsingHeader("General", ImGuiTreeNodeFlags_DefaultOpen))
            {
                if (ImGui::BeginTable("##masterhandlertable", 1, ImGuiTableFlags_RowBg | ImGuiTableFlags_Borders))
                {
                    ImGui::TableSetupColumn("##masterhandlertable_col", ImGuiTableColumnFlags_None);
                    ImGui::TableNextColumn();

                    master_handler->drawTreeMenu(curr_handler);
                    trash_handler->drawTreeMenu(curr_handler);

                    ImGui::EndTable();
                }

                if (curr_handler)
                    curr_handler->drawMenu();
            }
        }

        void ViewerApplication::drawMenuBar()
        {
            // Handle File Stuff TODOREWORK?
            {
                if (file_open_dialog && file_open_dialog->ready(0))
                {
                    std::string prod_path = (file_open_dialog->result().size() == 0 ? "" : file_open_dialog->result()[0]);
                    delete file_open_dialog;
                    file_open_dialog = nullptr;
                    openProductOrDataset(prod_path);
                }
            }

            if (ImGui::BeginMenuBar())
            {
                if (ImGui::BeginMenu("File"))
                {
                    if (ImGui::MenuItem("Open Product/Dataset (.cbor/.json)") && !file_open_dialog)
                        file_open_dialog = new pfd::open_file("Dataset/Product (.cbor/.json)", ".", {"CBOR Files", "*.cbor", "JSON Files", "*.json", "SHP Files", "*.shp"}, pfd::opt::force_path); // TODOREWORK remember path?

                    ImGui::MenuItem("Open Image");

                    if (ImGui::BeginMenu("Hardcoded"))
                    {
                        if (ImGui::MenuItem("Load KMSS"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/KMSS_24/dataset.json");
                        if (ImGui::MenuItem("Load Sterna"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/aws_pfm_cadu/dataset.json");
                        if (ImGui::MenuItem("Load MSUGS"))
                            openProductOrDataset("/home/alan/Downloads/20241231_132953_ARKTIKA-M 2_dat/MSUGS_VIS1/product.cbor");
                        if (ImGui::MenuItem("Load MSUGS 2"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/20250104_071415_ELEKTRO-L_3_dat/dataset.json");
                        if (ImGui::MenuItem("Load MetOp"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/metop_test/dataset.json");
                        if (ImGui::MenuItem("Load L3"))
                            openProductOrDataset("/data_ssd/ELEKTRO-L3/20250104_071415_ELEKTRO-L 3.dat_OUT/dataset.json");
                        if (ImGui::MenuItem("Load JPSS-1"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/202411271228_NOAA_20/dataset.json");
                        if (ImGui::MenuItem("Load JPSS-2"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/n21_day/dataset.json");
                        if (ImGui::MenuItem("Load APT"))
                            openProductOrDataset("/home/alan/Downloads/audio_137912500Hz_11-39-56_04-03-2024_wav/dataset.json");
                        if (ImGui::MenuItem("Load GOES"))
                            openProductOrDataset("/home/alan/Downloads/SatDump_NEWPRODS/goes_hrit_jvital2013_cadu/IMAGES/GOES-16/Full Disk/2024-04-17_18-00-20/product.cbor");
                        if (ImGui::MenuItem("Load Shapefile"))
                        {
                            auto shp_h = std::make_shared<ShapefileHandler>(resources::getResourcePath("maps/ne_10m_admin_0_countries.shp"));
                            master_handler->addSubHandler(shp_h);
                        }

                        if (ImGui::MenuItem("Load Shapefile FRA_1"))
                        {
                            auto shp_h = std::make_shared<ShapefileHandler>("/home/alan/Downloads/gadm41_FRA_shp/gadm41_FRA_1.shp");
                            master_handler->addSubHandler(shp_h);
                        }

                        if (ImGui::MenuItem("Load Shapefile FRA_2"))
                        {
                            auto shp_h = std::make_shared<ShapefileHandler>("/home/alan/Downloads/gadm41_FRA_shp/gadm41_FRA_2.shp");
                            master_handler->addSubHandler(shp_h);
                        }

                        ImGui::EndMenu();
                    }

                    ImGui::EndMenu();
                }
                if (ImGui::BeginMenu("Handler"))
                {
                    if (ImGui::BeginMenu("Add"))
                    {
                        if (ImGui::MenuItem("Dataset"))
                            logger->error("Dummy Menu!");
                        if (ImGui::MenuItem("Projection"))
                            master_handler->addSubHandler(std::make_shared<ProjectionHandler>());
                        if (ImGui::MenuItem("DSP Flowgraph"))
                            master_handler->addSubHandler(std::make_shared<DSPFlowGraphHandler>());
                        if (ImGui::MenuItem("BitView TEST"))
                        {
                            std::vector<std::shared_ptr<Handler>> e;
                            eventBus->fire_event<RequestHandlersEvent>({e});
                            if (e.size() > 0)
                                master_handler->addSubHandler(e[0]);
                        }
                        if (ImGui::MenuItem("Waterfall TEST"))
                            master_handler->addSubHandler(std::make_shared<WaterfallTestHandler>());
                        if (ImGui::MenuItem("NewRec TEST"))
                            master_handler->addSubHandler(std::make_shared<NewRecHandler>());
                        ImGui::EndMenu();
                    }
                    if (curr_handler && ImGui::BeginMenu("Config"))
                    {
                        if (ImGui::MenuItem("JSON To Clipboard"))
                            ImGui::SetClipboardText(curr_handler->getConfig().dump(4).c_str());
                        if (ImGui::MenuItem("JSON From Clipboard"))
                            curr_handler->setConfig(nlohmann::json::parse(ImGui::GetClipboardText()));
                        ImGui::EndMenu();
                    }
                    ImGui::EndMenu();
                }

                if (curr_handler)
                {
                    if (ImGui::BeginMenu("Current"))
                    {
                        curr_handler->drawMenuBar();
                        ImGui::EndMenu();
                    }
                }

                eventBus->fire_event<RenderLoadMenuElementsEvent>({});

                ImGui::EndMenuBar();
            }
        }

        void ViewerApplication::drawContents()
        {
            ImGui::Text("No handler selected!");
        }

        void ViewerApplication::drawUI()
        {
            ImVec2 viewer_size = ImGui::GetContentRegionAvail();

            if (ImGui::BeginTable("##wiever_table", 2, ImGuiTableFlags_Resizable | ImGuiTableFlags_SizingStretchProp | ImGuiTableFlags_BordersInnerV))
            {
                ImGui::TableSetupColumn("##panel_v", ImGuiTableColumnFlags_None, viewer_size.x * panel_ratio);
                ImGui::TableSetupColumn("##view", ImGuiTableColumnFlags_None, viewer_size.x * (1.0f - panel_ratio));
                ImGui::TableNextColumn();

                float left_width = ImGui::GetColumnWidth(0);
                float right_width = viewer_size.x - left_width;
                if (left_width != last_width && last_width != -1)
                    panel_ratio = left_width / viewer_size.x;
                last_width = left_width;

                ImGui::BeginChild("ViewerChildPanel", {left_width, float(viewer_size.y - 10)}, false, ImGuiWindowFlags_MenuBar);
                drawMenuBar();
                drawPanel();
                ImGui::EndChild();

                ImGui::TableNextColumn();
                ImGui::BeginGroup();

                if (curr_handler)
                    curr_handler->drawContents({float(right_width - 4), float(viewer_size.y)});
                else
                    drawContents();

                ImGui::EndGroup();
                ImGui::EndTable();
            }
        }

        void ViewerApplication::openProductOrDataset(std::string path) // TODOREWORK
        {
            if (file_open_thread.joinable())
                file_open_thread.join();
            auto fun = [this, path]()
            {
                if (!std::filesystem::path(path).has_extension())
                {
                    logger->error("Invalid file, no extension!");
                    return;
                }

                // TODOREWORK Load more than just image products
                if (std::filesystem::path(path).extension().string() == ".cbor")
                {
                    try
                    {
                        std::shared_ptr<ProductHandler> prod_h = std::make_shared<ImageProductHandler>(products::loadProduct(path));
                        master_handler->addSubHandler(prod_h);
                    }
                    catch (std::exception &e)
                    {
                        logger->error("Error loading product! Maybe not a valid product? Details : %s", e.what());
                    }
                }
                else if (std::filesystem::path(path).extension().string() == ".json")
                {
                    try
                    {
                        products::DataSet dataset;
                        dataset.load(path);
                        std::shared_ptr<DatasetHandler> dat_h = std::make_shared<DatasetHandler>(std::filesystem::path(path).parent_path().string(), dataset);
                        master_handler->addSubHandler(dat_h);
                    }
                    catch (std::exception &e)
                    {
                        logger->error("Error loading dataset! Maybe not a valid dataset? Details : %s", e.what());
                    }
                }
                else if (std::filesystem::path(path).extension().string() == ".shp")
                {
                    master_handler->addSubHandler(std::make_shared<ShapefileHandler>(path));
                }
            };
            file_open_thread = std::thread(fun);
        }
    }
};