/*
 * Copyright Luka Jankovic 2025
 *
 * This file is part of Kubo.
 *
 * Kubo is free software: you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later
 * version.
 *
 * Kubo is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * Kubo. If not, see <https://www.gnu.org/licenses/>.
 */

#include "kubo_context.h"

void kubo_context_init(struct kubo_context *context) {
    if (!context) {
        return;
    }

    kubo_wall_arr_init(&context->walls);
    context->wall_select_index = 0;
}

void kubo_context_cleanup(struct kubo_context *context) {
    kubo_wall_arr_free(&context->walls);
}

void kubo_context_set_state(struct kubo_context *context,
                            enum kubo_context_state state) {
    context->state = state;

    switch (context->state) {
    case KUBO_CONTEXT_WALL_NEW:
        break;

    case KUBO_CONTEXT_WALL_SELECT:
        wall_select_refresh(context);
        break;

    default:
        break;
    }
}

const char *kubo_context_get_label(enum kubo_context_state state) {
    return kubo_context_labels[state];
}

Color kubo_context_get_color(enum kubo_context_state state) {
    return kubo_context_colors[state];
}

struct kubo_wall *kubo_context_get_pending_wall(struct kubo_context *context) {

    if (context->state != KUBO_CONTEXT_WALL_NEW) {
        return NULL;
    }

    if (context->walls.count > 0) {
        struct kubo_wall *last_wall =
            kubo_wall_arr_get(&context->walls, context->walls.count - 1);
        switch (last_wall->state) {
        case KUBO_WALL_INIT:
        case KUBO_WALL_DRAWING:
            return last_wall;
            break;
        default:
            break;
        }
    }

    struct kubo_wall *wall = kubo_wall_init();
    kubo_wall_arr_add(&context->walls, wall);

    return wall;
}

void kubo_context_select_next_wall(struct kubo_context *context) {
    if (context->state != KUBO_CONTEXT_WALL_SELECT || !context->walls.count) {
        return;
    }

    context->wall_select_index =
        (context->wall_select_index + 1) % context->walls.count;

    wall_select_refresh(context);
}

void kubo_context_select_prev_wall(struct kubo_context *context) {
    if (context->state != KUBO_CONTEXT_WALL_SELECT || !context->walls.count) {
        return;
    }

    size_t count = context->walls.count;

    context->wall_select_index =
        (context->wall_select_index + count - 1) % count;

    wall_select_refresh(context);
}

static inline void wall_select_refresh(struct kubo_context *context) {
    for (int i = 0; i < context->walls.count; i++) {
        struct kubo_wall *wall = kubo_wall_arr_get(&context->walls, i);
        wall->state = i == context->wall_select_index ? KUBO_WALL_SELECTED
                                                      : KUBO_WALL_DONE;
    }
}