#include "app_base.h"
#include "imgui.h"

#include "gfx/shader_canvas.h"
#include "gfx/gl_shader.h"
#include "gfx/uniform_parser.h"
#include "gfx/uniform_panel.h"
#include "gfx/dag_catalog.h"
#include "gfx/dag_compile.h"
#include "gfx/dag_uniforms.h"
#include "gfx/dag_panel.h"
#include "core/fps_overlay.h"
#include "seed_shaders.h"

#include <chrono>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>

enum class Source { Code, Dag };

static Source g_active = Source::Code;
static fn::gfx::ShaderCanvas g_canvas;
static std::string g_source = PLASMA;
static std::string g_last_err;
static int g_last_err_line = -1;
static std::chrono::steady_clock::time_point g_last_edit;
static bool g_dirty = true;
static std::vector<fn::gfx::UniformDescriptor> g_descs;
static fn::gfx::UniformStore g_store;
static std::vector<fn::gfx::DagStep> g_pipeline;
static std::string g_dag_glsl;

static void try_compile() {
    if (g_active == Source::Code) {
        auto r = fn::gfx::compile_fragment(g_source);
        if (r.ok) {
            g_descs = fn::gfx::parse_uniforms(g_source);
            fn::gfx::uniforms_sync(g_store, g_descs);
            fn::gfx::canvas_set_program(g_canvas, r.program);
            g_last_err.clear();
            g_last_err_line = -1;
        } else {
            g_last_err = r.err_msg;
            g_last_err_line = r.err_line;
        }
    } else {
        g_dag_glsl = fn::gfx::compile_dag_to_glsl(g_pipeline);
        auto r = fn::gfx::compile_fragment(g_dag_glsl);
        if (r.ok) {
            fn::gfx::canvas_set_program(g_canvas, r.program);
            g_last_err.clear();
            g_last_err_line = -1;
        } else {
            g_last_err = r.err_msg;
            g_last_err_line = r.err_line;
        }
    }
}

static void mark_dirty() {
    g_last_edit = std::chrono::steady_clock::now();
    g_dirty = true;
}

static void set_active(Source s) {
    if (g_active != s) {
        g_active = s;
        try_compile();
    }
}

static void load_preset(const char* src) {
    g_source = src;
    mark_dirty();
}

static void ensure_dag_default() {
    if (g_pipeline.empty()) {
        const fn::gfx::DagNodeDef* def = fn::gfx::dag_find("plasma");
        if (def) {
            fn::gfx::DagStep step;
            step.id     = "n_default";
            step.name   = def->name;
            step.params = def->param_defaults;
            g_pipeline.push_back(step);
        }
    }
}

static void draw_err_footer() {
    if (g_last_err.empty()) return;
    ImGui::Separator();
    if (g_last_err_line > 0) {
        ImGui::TextColored(ImVec4(1, 0.4f, 0.4f, 1), "line %d: %s",
                           g_last_err_line, g_last_err.c_str());
    } else {
        ImGui::TextColored(ImVec4(1, 0.4f, 0.4f, 1), "%s", g_last_err.c_str());
    }
}

static void render() {
    if (!g_canvas.initialized) fn::gfx::canvas_init(g_canvas);

    if (g_dirty) {
        auto now = std::chrono::steady_clock::now();
        auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(now - g_last_edit).count();
        if (elapsed > 250) {
            try_compile();
            g_dirty = false;
        }
    }

    ImGui::DockSpaceOverViewport(0, ImGui::GetMainViewport());

    // --- Code window ---
    if (ImGui::Begin("Code")) {
        if (ImGui::IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows)) {
            set_active(Source::Code);
        }

        if (ImGui::Button("Plasma"))  { load_preset(PLASMA); }
        ImGui::SameLine();
        if (ImGui::Button("Circle"))  { load_preset(CIRCLE); }
        ImGui::SameLine();
        if (ImGui::Button("Checker")) { load_preset(CHECKER); }

        bool has_err_here = !g_last_err.empty() && g_active == Source::Code;
        ImVec2 avail = ImGui::GetContentRegionAvail();
        float footer_height = has_err_here ? ImGui::GetTextLineHeightWithSpacing() + 8.0f : 0.0f;
        ImVec2 editor_size(avail.x, avail.y - footer_height);

        char buf[1 << 16];
        size_t copy_len = g_source.size() < sizeof(buf) - 1 ? g_source.size() : sizeof(buf) - 1;
        memcpy(buf, g_source.c_str(), copy_len);
        buf[copy_len] = '\0';

        if (ImGui::InputTextMultiline("##code", buf, sizeof(buf), editor_size,
                                      ImGuiInputTextFlags_AllowTabInput)) {
            g_source = buf;
            mark_dirty();
        }

        if (has_err_here) draw_err_footer();
    }
    ImGui::End();

    // --- DAG Pipeline window ---
    if (ImGui::Begin("DAG Pipeline")) {
        if (ImGui::IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows)) {
            set_active(Source::Dag);
        }
        bool topo_changed = fn::gfx::dag_panel(g_pipeline);
        if (topo_changed) mark_dirty();

        if (!g_last_err.empty() && g_active == Source::Dag) draw_err_footer();
    }
    ImGui::End();

    // --- Canvas window ---
    char canvas_title[64];
    snprintf(canvas_title, sizeof(canvas_title), "Canvas [%s]###canvas",
             g_active == Source::Code ? "Code" : "DAG");
    if (ImGui::Begin(canvas_title)) {
        if (g_active == Source::Code) {
            fn::gfx::canvas_render(g_canvas, static_cast<float>(ImGui::GetTime()),
                [](unsigned int program) {
                    fn::gfx::uniforms_apply(g_store, g_descs, program);
                });
        } else {
            fn::gfx::canvas_render(g_canvas, static_cast<float>(ImGui::GetTime()),
                [](unsigned int program) {
                    fn::gfx::dag_uniforms_apply(g_pipeline, program);
                });
        }
    }
    ImGui::End();

    // --- Controls window (Code uniforms) ---
    if (ImGui::Begin("Controls")) {
        if (g_descs.empty()) {
            ImGui::TextDisabled("No uniforms declared in Code.");
            ImGui::TextDisabled("Use // @slider, @color, @toggle, @xy annotations.");
        } else {
            fn::gfx::uniforms_panel(g_store, g_descs);
        }
        ImGui::Spacing();
        fps_overlay();
    }
    ImGui::End();

    // --- Generated GLSL window (DAG compiled output, read-only) ---
    if (ImGui::Begin("Generated GLSL")) {
        if (g_dag_glsl.empty()) {
            ImGui::TextDisabled("(DAG not compiled yet — focus the DAG Pipeline window)");
        } else {
            ImVec2 avail = ImGui::GetContentRegionAvail();
            ImGui::InputTextMultiline("##dag_glsl",
                const_cast<char*>(g_dag_glsl.c_str()),
                g_dag_glsl.size() + 1,
                avail,
                ImGuiInputTextFlags_ReadOnly);
        }
    }
    ImGui::End();
}

int main() {
    ensure_dag_default();
    fn::AppConfig cfg;
    cfg.title  = "shaders_lab";
    cfg.width  = 1400;
    cfg.height = 860;
    int rc = fn::run_app(cfg, render);
    fn::gfx::canvas_destroy(g_canvas);
    return rc;
}