#ifndef TOS_UI_ATTRIBUTE_H
#define TOS_UI_ATTRIBUTE_H

#include "../../stdlib/Types.h"
#include "../../utils/StringUtils.h"
#include "../../math/Evaluator.h"
#include "UIAttributeType.h"
#include "UIAttributeDimension.h"

enum UIAttributeDataType : byte {
    UI_ATTRIBUTE_DATA_TYPE_INT,
    UI_ATTRIBUTE_DATA_TYPE_F32,
    UI_ATTRIBUTE_DATA_TYPE_STR,
    UI_ATTRIBUTE_DATA_TYPE_V4_F32,
};

struct UIAttribute {
    // Attributes use ids (=type name) instead of strings
    UIAttributeType attribute_id;
    UIAttributeDataType datatype;

    union {
        // @performance The string makes this struct really large when it is not needed in 95% of the cases
        char value_str[32];
        int32 value_int;
        f32 value_float;
        v4_f32 value_v4_f32;
    };
};

struct UIAttributeGroup {
    int32 attribute_size;
    UIAttribute* attributes;
};

UIAttribute* ui_attribute_from_group(UIAttributeGroup* group, UIAttributeType type)
{
    if (!group->attributes) {
        return NULL;
    }

    int32 left = 0;
    int32 right = type;

    // Binary search since attributes are sorted by attribute_id
    // @performance Consider Eytzinger
    while (left <= right) {
        int32 mid = left + (right - left) / 2;

        if (group->attributes[mid].attribute_id == type) {
            return &group->attributes[mid];
        }  else if (group->attributes[mid].attribute_id < type) {
            left = mid + 1;
        }  else {
            right = mid - 1;
        }
    }

    return NULL;
}

constexpr
int32 ui_attribute_type_to_id(const char* attribute_name)
{
    if (str_compare(attribute_name, "x") == 0) {
        return UI_ATTRIBUTE_TYPE_DIMENSION_X;
    } else if (str_compare(attribute_name, "y") == 0) {
        return UI_ATTRIBUTE_TYPE_DIMENSION_Y;
    } else if (str_compare(attribute_name, "width") == 0) {
        return UI_ATTRIBUTE_TYPE_DIMENSION_WIDTH;
    } else if (str_compare(attribute_name, "height") == 0) {
        return UI_ATTRIBUTE_TYPE_DIMENSION_HEIGHT;
    } else if (str_compare(attribute_name, "font_name") == 0) {
        return UI_ATTRIBUTE_TYPE_FONT_NAME;
    } else if (str_compare(attribute_name, "font_color") == 0) {
        return UI_ATTRIBUTE_TYPE_FONT_COLOR;
    } else if (str_compare(attribute_name, "font_size") == 0) {
        return UI_ATTRIBUTE_TYPE_FONT_SIZE;
    } else if (str_compare(attribute_name, "font_weight") == 0) {
        return UI_ATTRIBUTE_TYPE_FONT_WEIGHT;
    } else if (str_compare(attribute_name, "font_line_height") == 0) {
        return UI_ATTRIBUTE_TYPE_FONT_LINE_HEIGHT;
    } else if (str_compare(attribute_name, "align_h") == 0) {
        return UI_ATTRIBUTE_TYPE_ALIGN_H;
    } else if (str_compare(attribute_name, "align_v") == 0) {
        return UI_ATTRIBUTE_TYPE_ALIGN_V;
    } else if (str_compare(attribute_name, "zindex") == 0) {
        return UI_ATTRIBUTE_TYPE_ZINDEX;
    } else if (str_compare(attribute_name, "style1") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE1;
    } else if (str_compare(attribute_name, "style2") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE2;
    } else if (str_compare(attribute_name, "style3") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE3;
    } else if (str_compare(attribute_name, "style4") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE4;
    } else if (str_compare(attribute_name, "style5") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE5;
    } else if (str_compare(attribute_name, "style6") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE6;
    } else if (str_compare(attribute_name, "style7") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE7;
    } else if (str_compare(attribute_name, "style8") == 0) {
        return UI_ATTRIBUTE_TYPE_STYLE8;
    } else if (str_compare(attribute_name, "foreground_color") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_COLOR;
    } else if (str_compare(attribute_name, "foreground_img") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_IMG;
    } else if (str_compare(attribute_name, "foreground_img_opacity") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_IMG_OPACITY;
    } else if (str_compare(attribute_name, "foreground_img_position_v") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_IMG_POSITION_V;
    } else if (str_compare(attribute_name, "foreground_img_position_h") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_IMG_POSITION_H;
    } else if (str_compare(attribute_name, "foreground_img_style") == 0) {
        return UI_ATTRIBUTE_TYPE_FOREGROUND_IMG_STYLE;
    } else if (str_compare(attribute_name, "background_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_COLOR;
    } else if (str_compare(attribute_name, "background_img") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_IMG;
    } else if (str_compare(attribute_name, "background_img_opacity") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_IMG_OPACITY;
    } else if (str_compare(attribute_name, "background_img_position_v") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_IMG_POSITION_V;
    } else if (str_compare(attribute_name, "background_img_position_h") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_IMG_POSITION_H;
    } else if (str_compare(attribute_name, "background_img_style") == 0) {
        return UI_ATTRIBUTE_TYPE_BACKGROUND_IMG_STYLE;
    } else if (str_compare(attribute_name, "border_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_COLOR;
    } else if (str_compare(attribute_name, "border_width") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_WIDTH;
    } else if (str_compare(attribute_name, "border_top_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_TOP_COLOR;
    } else if (str_compare(attribute_name, "border_top_width") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_TOP_WIDTH;
    } else if (str_compare(attribute_name, "border_right_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_RIGHT_COLOR;
    } else if (str_compare(attribute_name, "border_right_width") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_RIGHT_WIDTH;
    } else if (str_compare(attribute_name, "border_bottom_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_BOTTOM_COLOR;
    } else if (str_compare(attribute_name, "border_bottom_width") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_BOTTOM_WIDTH;
    } else if (str_compare(attribute_name, "border_left_color") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_LEFT_COLOR;
    } else if (str_compare(attribute_name, "border_left_width") == 0) {
        return UI_ATTRIBUTE_TYPE_BORDER_LEFT_WIDTH;
    } else if (str_compare(attribute_name, "padding") == 0) {
        return UI_ATTRIBUTE_TYPE_PADDING;
    } else if (str_compare(attribute_name, "padding_top") == 0) {
        return UI_ATTRIBUTE_TYPE_PADDING_TOP;
    } else if (str_compare(attribute_name, "padding_right") == 0) {
        return UI_ATTRIBUTE_TYPE_PADDING_RIGHT;
    } else if (str_compare(attribute_name, "padding_bottom") == 0) {
        return UI_ATTRIBUTE_TYPE_PADDING_BOTTOM;
    } else if (str_compare(attribute_name, "padding_left") == 0) {
        return UI_ATTRIBUTE_TYPE_PADDING_LEFT;
    } else if (str_compare(attribute_name, "scroll_style") == 0) {
        return UI_ATTRIBUTE_TYPE_SCROLL_STYLE;
    } else if (str_compare(attribute_name, "scroll_x") == 0) {
        return UI_ATTRIBUTE_TYPE_SCROLL_X;
    } else if (str_compare(attribute_name, "scroll_y") == 0) {
        return UI_ATTRIBUTE_TYPE_SCROLL_Y;
    } else if (str_compare(attribute_name, "shadow_inner_color") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_INNER_COLOR;
    } else if (str_compare(attribute_name, "shadow_inner_angle") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_INNER_ANGLE;
    } else if (str_compare(attribute_name, "shadow_inner_distance") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_INNER_DISTANCE;
    } else if (str_compare(attribute_name, "shadow_outer_color") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_OUTER_COLOR;
    } else if (str_compare(attribute_name, "shadow_outer_angle") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_OUTER_ANGLE;
    } else if (str_compare(attribute_name, "shadow_outer_distance") == 0) {
        return UI_ATTRIBUTE_TYPE_SHADOW_OUTER_DISTANCE;
    } else if (str_compare(attribute_name, "transition_animation") == 0) {
        return UI_ATTRIBUTE_TYPE_TRANSITION_ANIMATION;
    } else if (str_compare(attribute_name, "transition_duration") == 0) {
        return UI_ATTRIBUTE_TYPE_TRANSITION_DURATION;
    }

    return -1;
}

inline
void ui_attribute_parse_value(UIAttribute* attr, const char* attribute_name, const char* pos)
{
    attr->attribute_id = (UIAttributeType) ui_attribute_type_to_id(attribute_name);
    char value[64];

    str_copy_until(pos, value, "\r\n");

    if (is_integer(value)) {
        attr->value_int = strtoul(value, NULL, 10);
        attr->datatype = UI_ATTRIBUTE_DATA_TYPE_INT;
    } else if (is_float(value)) {
        attr->value_float = strtof(value, NULL);
        attr->datatype = UI_ATTRIBUTE_DATA_TYPE_F32;
    } else if (is_hex_color(value)) {
        ++pos; // Skip '#'
        hexstr_to_rgba(&attr->value_v4_f32, pos);
        attr->datatype = UI_ATTRIBUTE_DATA_TYPE_V4_F32;
    } else {
        str_copy_until(attr->value_str, value, "\r\n");
        attr->datatype = UI_ATTRIBUTE_DATA_TYPE_STR;
    }
}

inline
void ui_theme_assign_f32(f32* a, const UIAttribute* attr, int32 variable_count, const EvaluatorVariable* variables)
{
    if (attr->datatype == UI_ATTRIBUTE_DATA_TYPE_INT) {
        *a = (f32) attr->value_int;
    } else if (attr->datatype == UI_ATTRIBUTE_DATA_TYPE_F32) {
        *a = (f32) attr->value_float;
    } else if (attr->datatype == UI_ATTRIBUTE_DATA_TYPE_STR) {
        char value[32];
        memcpy(value, attr->value_str, ARRAY_COUNT(attr->value_str));
        *a = (f32) evaluator_evaluate(value, variable_count, variables);
    }
}

inline
void ui_theme_assign_dimension(UIAttributeDimension* dimension, const UIAttribute* attr, int32 variable_count, const EvaluatorVariable* variables)
{
    switch (attr->attribute_id) {
        case UI_ATTRIBUTE_TYPE_DIMENSION_X: {
                ui_theme_assign_f32(&dimension->dimension.x, attr, variable_count, variables);
            } break;
        case UI_ATTRIBUTE_TYPE_DIMENSION_WIDTH: {
                ui_theme_assign_f32(&dimension->dimension.width, attr, variable_count, variables);
            } break;
        case UI_ATTRIBUTE_TYPE_DIMENSION_Y: {
                ui_theme_assign_f32(&dimension->dimension.y, attr, variable_count, variables);
            } break;
        case UI_ATTRIBUTE_TYPE_DIMENSION_HEIGHT: {
                ui_theme_assign_f32(&dimension->dimension.height, attr, variable_count, variables);
            } break;
        default: {
            UNREACHABLE();
        }
    }
}

#endif