#include "ui.h"

#include "state.h"
#include "midi-node.h"
#include "raylib.h"
#include "raygui/raygui.h"
#include "raygui/dark/style_dark.h"

#include "datatypes.h"

#include <memory>
#include <ranges>

namespace VISEQ::APP::UI{
    void AppUI::init(){
        GuiLoadStyleDark();
    }

    void AppUI::drawSideBar(){

        GuiPanel((Rectangle){-1,-30,380,(float)state->settings.height + 31}, "");
        const int s_controls = 25;
        //GuiSetStyle(BUTTON, STATE_FOCUSED, 0xFF0000FF);
        if(GuiButton((Rectangle){20, (float)s_controls, 80, 30}, "PLAY")) sm->play ? sm->pause() : sm->start();
        if(GuiButton((Rectangle){120, (float)s_controls, 60, 30}, "STOP")) sm->stop();
        if(GuiButton((Rectangle){200, (float)s_controls, 60, 30}, "RESET")) sm->reset();

        const int s_bpm = s_controls + 45;
        GuiLabel((Rectangle){20, (float)s_bpm, 120, 20}, "TIMING");
        GuiLabel((Rectangle){200, (float)s_bpm, 40, 20}, std::format("{}",state->settings.bpm).c_str());
        int _bpm = state->settings.bpm;
//			GuiSlider((Rectangle){20, (float)s_bpm + 25, 240, 20}, "", "", &_bpm, 1, 300);
        if(GuiSpinner((Rectangle){20, (float)s_bpm + 25, 240, 20}, "", &_bpm, 1, 300, bpm_spinner_edit_mode)) bpm_spinner_edit_mode = !bpm_spinner_edit_mode;
        state->settings.bpm = _bpm;

        int s_midi = s_bpm + 75;
        GuiLabel((Rectangle){20, (float)s_midi, 240, 20}, "MIDI OUT");
        if(GuiButton((Rectangle){200, (float)s_midi, 60, 20}, "RELOAD")) sm->midi.enumerate();

        int i = 0;
        for(auto& mo : sm->midi.midi_out_ports){
            GuiCheckBox((Rectangle){20, (float)s_midi + 25 + i*20.0f, 20, 20}, mo.port.port_name.substr(0,32).c_str(), &mo.enabled);
            i++;
        }

        if(GuiButton((Rectangle){20, (float)s_midi + 45 + i*20.0f, 240, 20}, "ENABLE FOR ALL")) {
            for(auto &midis : sm->midi.midi_out_ports){
                if(midis.enabled){
                    gm->setMidiPortForAll(midis.number, true);
                }
            }
        }

        int s_signal = s_midi + 100 + i*20;
        GuiLabel((Rectangle){20, (float)s_signal, 240, 20}, "SIGNALS");
        if(GuiButton((Rectangle){20, (float)s_signal + 35, 240, 20}, "ADD RANDOM SIGNAL")){
            std::shared_ptr<Node> ni = gm->getRandomNode(); 
            sm->addSignalAtNode(ni);
        }
        
        int signal_configs = 0;
        for(auto &s : sm->signals){
            if(GuiButton((Rectangle){20,(float)s_signal+70 + signal_configs*20, 20, 20},"X")) sm->removeSignal(s);
            DrawCircle(55, s_signal+80 + signal_configs*20, 6.0F, s->color);

            int s_speed = round(s->speed);
            s->multiplier = state->settings.bpm / 120.0f;
            float signal_bpm = round(state->settings.bpm * s->multiplier * (s->speed / state->settings.speed));
            if(GuiSpinner((Rectangle){70, (float)s_signal+70 + signal_configs*20, 100, 20}, "", &s_speed, 1, 16 * state->settings.grid * 4, s->ui_edit_speed)) s->ui_edit_speed = !s->ui_edit_speed;
            if(GuiButton((Rectangle){200,(float)s_signal+70 + signal_configs*20, 30, 20}, "1/2")) s_speed = state->settings.speed / 2; 
            //if(GuiButton((Rectangle){200,(float)s_signal+70 + signal_configs*20, 30, 20},"1/4")) s_speed = state->settings.speed / 4; 
            GuiLabel((Rectangle){195, (float)s_signal+70 + signal_configs*20, 30, 20}, std::format("{}",signal_bpm).c_str());
            GuiToggle((Rectangle){240, (float)s_signal+70 + signal_configs*20, 20, 20},"P", &s->run); 
            s->speed = s_speed;
            signal_configs++;
        }
        
        int s_nodes = s_signal + 70 + signal_configs*20;
        GuiLabel((Rectangle){20, (float)s_nodes + 30, 240, 20}, "NODES");
        if(GuiButton((Rectangle){20, (float)s_nodes + 60, 240, 20}, "ADD RANDOM NODE")){
            gm->createMidiNode(Vec2Df{200.0f, 200.0f});
        }
        DrawText(std::format("{}", gm->getNodesSize()).c_str(), 20, state->settings.height - 70, 15, VS_COLOR_WHITE);
        DrawFPS(20, state->settings.height - 30);

        std::string shorts[6] = {"NT","CC","PC","PB","AT","PP"};
        for(int i = 0; i < 6; ++i){
            Color col = state->settings.mode_colors[i];
            Vec2Df pos = {state->settings.width - (6.0f*75.0f) + i*75.0f, state->settings.height - 75.0f};
            DrawRing(pos, 30.0f, 34.0f, 0.0f, 360.0f, 24, state->node_mode == i ? VS_COLOR_WHITE : (Color){0,0,0,0}); 
            DrawCircleV(pos, 30.0f, col);
            DrawCenteredText(shorts[i], pos, 20, WHITE);
        }
    }

    void AppUI::DrawCenteredText(std::string text, Vec2Df position, int size, Color color){	
        float offset = MeasureText(text.c_str(), size);
        DrawText(text.c_str(), position.x-offset/2.0f, position.y-(float)size/2.0f+1, size, color );
    }
    
    void AppUI::drawContextMenu(Camera2D camera){
        for(auto n : gm->getSelection() | std::views::filter([](std::shared_ptr<Node> n){ return n->config; })){
            Vector2 np = GetWorldToScreen2D((Vector2){n->position.x, n->position.y}, camera);

            state->context_menu = !GuiWindowBox((Rectangle){np.x + 10, np.y + 10, 240, 440}, "config");
            n->config = state->context_menu;

            int option = n->distributionMode;
            GuiToggleGroup((Rectangle){ np.x + 20, np.y + 40, 73, 20 }, "GEO;RR;RAND", &option);
            n->distributionMode = (Distribution)option;

            GuiLabel((Rectangle){np.x + 20, np.y + 65, 220, 20}, "SIGNALS");
            if(GuiButton((Rectangle){np.x + 20, np.y + 90, 220, 20}, "ADD SIGNAL")){
                sm->addSignalAtNode(n);
            }

            int y_midi = 120;
            GuiLabel((Rectangle){np.x + 20, np.y + y_midi, 220, 20}, "MIDI OUT");

            int i = 0;
            if(n->getNodeType() == NodeType::MIDINODE){
                std::shared_ptr<MidiNode> mn = std::dynamic_pointer_cast<MidiNode>(n);
                for(auto& mo : mn->midi_outs){
                    GuiCheckBox((Rectangle){np.x + 20, np.y + y_midi + 20 + i*20.0f, 20, 20}, mo.second.name.c_str(), &mo.second.enabled);
                    i++;
                }

                int y_channel = y_midi + 30 + i*20;
                GuiCheckBox((Rectangle){np.x + 20, np.y + y_channel, 20, 20}, "NOTE", &mn->send_note);
                GuiCheckBox((Rectangle){np.x + 90, np.y + y_channel, 20, 20}, "CC", &mn->send_cc);
                GuiCheckBox((Rectangle){np.x + 160, np.y + y_channel, 20, 20}, "PITCH", &mn->send_pitch_bend);

                int y_cc_note = y_channel + 120;
                switch(state->node_mode){
                    case 0:{
                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "NOTE CHANNEL");
                               GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &mn->midiNote.channel);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "NOTE");
                               float v = mn->midiNote.note;
                               GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note , 20, 20}, MIDI::Notes[(int)round(v)].c_str());
                               GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",mn->midiNote.note).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 127);
                               mn->midiNote.note = round(v);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 60, 100, 20}, "VELOCITY");
                               v = mn->midiNote.velocity;
                               GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 60, 20, 20}, std::format("{}",mn->midiNote.velocity).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 90, 220, 20}, "", "", &v, 0, 127);
                               mn->midiNote.velocity = round(v);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 120, 100, 20}, "GATE");
                               v = mn->midiNote.gate;
                               GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 120, 20, 20}, std::format("{}",(float)mn->midiNote.gate/1000.0f).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 150, 220, 20}, "", "", &v, 0, 30000);
                               mn->midiNote.gate= round(v);
                               break;}
                    case 1:
                           {
                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "CC CHANNEL");
                               GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &mn->midiCC.channel);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "CC");
                               float v = mn->midiCC.control;
                               GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",mn->midiCC.control).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 127);
                               mn->midiCC.control = round(v);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note + 60, 100, 20}, "CC VALUE");
                               v = mn->midiCC.value;
                               GuiLabel((Rectangle){np.x + 210, np.y + y_cc_note + 60, 20, 20}, std::format("{}",mn->midiCC.value).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 90, 220, 20}, "", "", &v, 0, 127);
                               mn->midiCC.value = round(v);

                               break;}
                    case 2:
                           {
                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note - 75, 220, 20}, "PITCH BEND CHANNEL");
                               GuiToggleGroup((Rectangle){np.x + 20, np.y + y_cc_note - 50, 25, 20}, "1;2;3;4;5;6;7;8\n9;10;11;12;13;14;15;16", &mn->midiPitchBend.channel);

                               GuiLabel((Rectangle){np.x + 20, np.y + y_cc_note , 100, 20}, "CC");
                               float v = mn->midiPitchBend.value;
                               GuiLabel((Rectangle){np.x + 160, np.y + y_cc_note , 40, 20}, std::format("{}",mn->midiPitchBend.value).c_str());
                               GuiSlider((Rectangle){np.x + 20, np.y + y_cc_note + 30, 220, 20}, "", "", &v, 0, 16383);
                               mn->midiPitchBend.value = round(v);
                               break;}
                }
            }
        }
    }

    void AppUI::draw(Vec2Df mouse_pos, Camera2D camera){
        drawSideBar();
        if(state->context_menu) drawContextMenu(camera);
    }
}