#include <stdbool.h>

#include "game/camera.h"
#include "game/level/boxes.h"
#include "game/level/level_editor/proto_rect.h"
#include "game/level/level_editor/color_picker.h"
#include "system/stacktrace.h"
#include "system/nth_alloc.h"

#include "level_editor.h"

struct LevelEditor
{
    Vec camera_position;
    float camera_scale;
    ProtoRect proto_rect;
    ColorPicker color_picker;
    Boxes *boxes;
};

LevelEditor *create_level_editor(Boxes *boxes)
{
    LevelEditor *level_editor = nth_calloc(1, sizeof(LevelEditor));
    if (level_editor == NULL) {
        return NULL;
    }

    level_editor->camera_position = vec(0.0f, 0.0f);
    level_editor->camera_scale = 1.0f;
    level_editor->boxes = boxes;
    level_editor->proto_rect.color = rgba(1.0f, 0.0f, 0.0f, 1.0f);
    level_editor->color_picker.position = vec(0.0f, 0.0f);
    level_editor->color_picker.proto_rect = &level_editor->proto_rect;

    return level_editor;
}

void destroy_level_editor(LevelEditor *level_editor)
{
    trace_assert(level_editor);
    free(level_editor);
}

int level_editor_render(const LevelEditor *level_editor,
                        Camera *camera)
{
    trace_assert(level_editor);
    trace_assert(camera);

    if (proto_rect_render(&level_editor->proto_rect, camera) < 0) {
        return -1;
    }

    if (color_picker_render(&level_editor->color_picker, camera) < 0) {
        return -1;
    }

    return 0;
}

int level_editor_update(LevelEditor *level_editor,
                        float delta_time)
{
    trace_assert(level_editor);

    if (proto_rect_update(&level_editor->proto_rect, delta_time) < 0) {
        return -1;
    }

    return 0;
}

int level_editor_event(LevelEditor *level_editor,
                       const SDL_Event *event,
                       const Camera *camera)
{
    trace_assert(level_editor);
    trace_assert(event);

    (void) camera;

    switch (event->type) {
    case SDL_MOUSEWHEEL: {
        // TODO(#679): zooming in edit mode is not smooth enough
        if (event->wheel.y > 0) {
            level_editor->camera_scale += 0.1f;
        } else if (event->wheel.y < 0) {
            level_editor->camera_scale = fmaxf(0.1f, level_editor->camera_scale - 0.1f);
        }
    } break;

    case SDL_MOUSEBUTTONUP:
    case SDL_MOUSEBUTTONDOWN: {
        bool selected = false;
        if (color_picker_mouse_button(
                &level_editor->color_picker,
                &event->button,
                &selected) < 0) {
            return -1;
        }

        if (!selected && proto_rect_mouse_button(
                &level_editor->proto_rect,
                &event->button,
                level_editor->boxes,
                camera) < 0) {
            return -1;
        }
    } break;

    case SDL_MOUSEMOTION: {
        const float sens = 1.0f / level_editor->camera_scale * 0.25f;
        vec_add(&level_editor->camera_position,
                vec((float) event->motion.xrel * sens, (float) event->motion.yrel * sens));

        if (proto_rect_mouse_motion(&level_editor->proto_rect, &event->motion, camera) < 0) {
            return -1;
        }
    } break;
    }

    return 0;
}

int level_editor_focus_camera(LevelEditor *level_editor,
                              Camera *camera)
{
    camera_center_at(camera, level_editor->camera_position);
    camera_scale(camera, level_editor->camera_scale);
    return 0;
}