#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/event_bus.h"
#include "common/utils.h"
// #include "resources.h"
#include "core/plugin.h"

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

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

#include "common/image/io.h"

#include "tools_todoreworkmove/lut_generator.h"

#include "imgui/imgui_filedrop.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);

            // Enable dropping files onto the viewer. TODOREWORK, check the viewer IS active!?
            eventBus->register_handler<imgui_utils::FileDropEvent>(
                [this](const imgui_utils::FileDropEvent &v)
                {
                    for (auto &f : v.files)
                        openProductOrDataset(f);
                });

            eventBus->register_handler<GetLastSelectedOfTypeEvent>(
                [this](const GetLastSelectedOfTypeEvent &v)
                {
                    if (last_selected_handler.count(v.type))
                        v.h = last_selected_handler[v.type];
                    else
                        v.h = 0;
                });

            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))
            {
                auto prev_curr = curr_handler;

                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 (prev_curr != curr_handler)
                    last_selected_handler.insert_or_assign(curr_handler->getID(), curr_handler);

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

        void ViewerApplication::drawMenuBar()
        {
            // Handle File Stuff TODOREWORK?
            {
                if (file_open_dialog && file_open_dialog->is_ready())
                {
                    std::string prod_path = file_open_dialog->result();
                    delete file_open_dialog;
                    file_open_dialog = nullptr;
                    if (prod_path.size() > 0)
                        openProductOrDataset(prod_path);
                    else
                        logger->trace("No file selected");
                }
            }

            if (ImGui::BeginMenuBar())
            {
                if (ImGui::BeginMenu("File"))
                {
                    if (ImGui::MenuItem("Open File") && !file_open_dialog)                     // TODOREWORK switch to general thing
                        file_open_dialog = new fileutils::FileSelTh({{"All Files", "*"}}, ""); // TODOREWORK remember path?

                    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("Waterfall TEST"))
                            master_handler->addSubHandler(std::make_shared<WaterfallTestHandler>());
                        if (ImGui::MenuItem("NewRec TEST"))
                            master_handler->addSubHandler(std::make_shared<NewRecHandler>());
                        if (ImGui::MenuItem("CycloHelper TEST"))
                            master_handler->addSubHandler(std::make_shared<CycloHelperHandler>());
                        if (ImGui::MenuItem("Lut Generator"))
                            master_handler->addSubHandler(std::make_shared<LutGeneratorHandler>());
                        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>({curr_handler, master_handler});

                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 - 9 * ui_scale), float(viewer_size.y)});
                else
                    drawContents();

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

        void ViewerApplication::openProductOrDataset(std::string path) // TODOREWORK Rename!
        {
            if (file_open_thread.joinable())
                file_open_thread.join();
            auto fun = [this, path]()
            {
                try
                {
                    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")
                    {
                        logger->trace("Viewer loading product " + path);

                        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")
                    {
                        logger->trace("Viewer loading dataset " + path);

                        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")
                    {
                        logger->trace("Viewer loading shapefile " + path);

                        master_handler->addSubHandler(std::make_shared<ShapefileHandler>(path));
                    }
                    else
                    {
                        logger->trace("Viewer loading image " + path);

                        image::Image img;
                        image::load_img(img, path);
                        if (img.size() > 0)
                            master_handler->addSubHandler(std::make_shared<ImageHandler>(img, std::filesystem::path(path).stem().string()));
                        else
                            logger->error("Could not open this file as image!"); // TODOREWORK Probably check before this?
                    }
                }
                catch (std::exception &e)
                {
                    logger->error("Error opening file! => %s", e.what());
                }
            };
            file_open_thread = std::thread(fun);
        }
    } // namespace viewer
}; // namespace satdump