#include "render.h"

#include <stdio.h>
#include <string.h>

// strlen counting utf-8 characters as one
size_t utf8_strlen(char *s) {
    size_t len = 0;
    while (*s) {
        // if the byte is not a continuation byte (10xxxxxx), it's a new character
        if ((*s & 0xC0) != 0x80)
            len++;
        s++;
    }
    return len;
}

void print_row_separator() {
    for (size_t i = 0; i < TABLE_LINE_WIDTH; i++)
        printf("%c", TABLE_ROW_SEPARATOR_CHAR);
    printf("\n");
}

// print string with given color
void print_color(char *str, int color) {
    printf("\033[%dm%s\033[0m", color, str);
}

void print_center(char *text, int color, size_t width, char padding_char) {
    size_t post_char_count = (width - strlen(text)) / 2;
    size_t pre_char_count = width - post_char_count - strlen(text);

    for (size_t i = 0; i < pre_char_count; i++)
        printf("%c", padding_char);

    print_color(text, color);

    for (size_t i = 0; i < post_char_count; i++)
        printf("%c", padding_char);
}

void print_enemy_list(EnemyResults *results, size_t n_enemies) {
    printf(TABLE_LINE_START);
    size_t line_length = TABLE_LINE_START_WIDTH;

    for (size_t i = 0; i < n_enemies; i++) {
        EnemyResults enemy = results[i];
        char enemy_buffer[TABLE_LINE_WIDTH + 1];
        sprintf(enemy_buffer, "%s (%d)%s", enemy.name, enemy.level, i < n_enemies - 1 ? ", " : "");

        // break line if length would be too big
        if (line_length + utf8_strlen(enemy_buffer) > TABLE_LINE_WIDTH - TABLE_LINE_END_WIDTH) {
            printf("%*s\n", (int)(TABLE_LINE_WIDTH - line_length), TABLE_LINE_END);
            printf(TABLE_LINE_START);
            line_length = TABLE_LINE_START_WIDTH;
        }

        printf("%s", enemy_buffer);
        line_length += utf8_strlen(enemy_buffer);
    }

    printf("%*s\n", (int)(TABLE_LINE_WIDTH - line_length), TABLE_LINE_END);
}

void print_column_names() {
    char *names[TABLE_COLUMN_COUNT] = TABLE_COLUMN_NAMES;
    int colors[TABLE_COLUMN_COUNT] = TABLE_COLUMN_COLORS;
    size_t widths[TABLE_COLUMN_COUNT] = TABLE_COLUMN_WIDTHS;

    printf(TABLE_LINE_START);
    for (size_t i = 0; i < TABLE_COLUMN_COUNT; i++) {
        print_center(names[i], colors[i], widths[i], ' ');
        printf(i < TABLE_COLUMN_COUNT - 1 ? TABLE_LINE_SEPARATOR : TABLE_LINE_END);
    }
    printf("\n");
}

int item_type_to_color(ItemType type) {
    switch (type) {
        case NORMAL:
            return COLOR_WHITE;
        case DRIF:
            return COLOR_YELLOW;
        case ORB:
            return COLOR_GOLD;
        case SYNG:
            return COLOR_RED;
        case SET:
            return COLOR_GREEN;
        case RARE:
            return COLOR_AQUA;
        case EPIC:
            return COLOR_PURPLE;
    }
    return COLOR_WHITE;
}

void print_player_rewards(FriendlyResults results) {
    char buffers[TABLE_COLUMN_COUNT][TABLE_LINE_WIDTH];
    size_t widths[TABLE_COLUMN_COUNT] = TABLE_COLUMN_WIDTHS;
    int colors[TABLE_COLUMN_COUNT] = TABLE_COLUMN_COLORS;

    sprintf(buffers[0], "%s (%d)", results.name, results.level);
    sprintf(buffers[1], "%d", results.exp);
    sprintf(buffers[2], "%d", results.gold);

    if (results.psycho > 0)
        sprintf(buffers[3], "%d", results.psycho);
    else
        sprintf(buffers[3], "-");

    if (results.saturation > 0)
        sprintf(buffers[4], "%d (%s)", results.saturation, results.saturation_type);
    else
        sprintf(buffers[4], "-");

    if (results.splinters > 0)
        sprintf(buffers[5], "%d", results.splinters);
    else
        sprintf(buffers[5], "-");

    printf(TABLE_LINE_START);
    for (size_t i = 0; i < TABLE_COLUMN_COUNT; i++) {
        char to_color[widths[i] + 1];
        sprintf(to_color, "%*s", (int)widths[i], buffers[i]);
        print_color(to_color, colors[i]);
        printf(i < TABLE_COLUMN_COUNT - 1 ? TABLE_LINE_SEPARATOR : TABLE_LINE_END);
    }
    printf("\n");

    if (results.num_items > 0) {
        printf(TABLE_LINE_START);
        print_center("ITEMS", COLOR_WHITE, TABLE_ITEMS_WIDTH, ' ');
        printf(TABLE_LINE_END "\n" TABLE_LINE_START);

        size_t line_len = 0;
        for (size_t i = 0; i < results.num_items; i++) {
            char *sep = i < results.num_items - 1 ? ", " : "";
            size_t item_len = utf8_strlen(results.items[i].data) + strlen(sep);
            if (line_len + item_len > TABLE_ITEMS_WIDTH) {
                printf("%*s" TABLE_LINE_END "\n" TABLE_LINE_START, (int)(TABLE_ITEMS_WIDTH - line_len), "");
                line_len = 0;
            }
            print_color(results.items[i].data, item_type_to_color(results.items[i].type));
            printf("%s", sep);
            line_len += item_len;
        }
        printf("%*s" TABLE_LINE_END "\n", (int)(TABLE_ITEMS_WIDTH - line_len), "");
    }
}

void render_fight_results(FightResults results) {
    print_center("ENEMIES", COLOR_RED, TABLE_LINE_WIDTH, TABLE_ROW_SEPARATOR_CHAR);
    printf("\n");

    print_enemy_list(results.enemy_results, results.num_enemy);

    print_center("PLAYER REWARDS", COLOR_GREEN, TABLE_LINE_WIDTH, TABLE_ROW_SEPARATOR_CHAR);
    printf("\n");

    print_column_names();
    print_row_separator();

    for (size_t i = 0; i < results.num_friendly; i++) {
        print_player_rewards(results.friendly_results[i]);
        print_row_separator();
    }
}