#include <string.h>

#include "doomkeys.h"

#include "txt_checkbox.h"
#include "txt_io.h"
#include "txt_main.h"
#include "txt_window.h"

static void TXT_CheckBoxSizeCalc(txt_widget_t *widget, int *w, int *h)
{
    txt_checkbox_t *checkbox = (txt_checkbox_t *) widget;

    // Minimum width is the string length + two spaces for padding

    *w = strlen(checkbox->label) + 6;
    *h = 1;
}

static void TXT_CheckBoxDrawer(txt_widget_t *widget, int w, int selected)
{
    txt_checkbox_t *checkbox = (txt_checkbox_t *) widget;
    int i;

    TXT_BGColor(TXT_COLOR_BLUE, 0);
    TXT_FGColor(TXT_COLOR_BRIGHT_CYAN);
    TXT_DrawString(" (");

    TXT_FGColor(TXT_COLOR_BRIGHT_WHITE);

    if (*checkbox->variable)
    {
        TXT_DrawString("\x07");
    }
    else
    {
        TXT_DrawString(" ");
    }

    TXT_FGColor(TXT_COLOR_BRIGHT_CYAN);

    TXT_DrawString(") ");

    if (selected)
    {
        TXT_BGColor(TXT_COLOR_GREY, 0);
    }

    TXT_FGColor(TXT_COLOR_BRIGHT_WHITE);

    TXT_DrawString(checkbox->label);
    
    for (i=strlen(checkbox->label); i < w-6; ++i)
    {
        TXT_DrawString(" ");
    }
}

static void TXT_CheckBoxDestructor(txt_widget_t *widget)
{
    txt_checkbox_t *checkbox = (txt_checkbox_t *) widget;

    free(checkbox->label);
}

static int TXT_CheckBoxKeyPress(txt_widget_t *widget, int key)
{
    txt_checkbox_t *checkbox = (txt_checkbox_t *) widget;

    if (key == KEY_ENTER || key == ' ')
    {
        *checkbox->variable = !*checkbox->variable;
        return 1;
    }
    
    return 0;
}

txt_widget_class_t txt_checkbox_class =
{
    TXT_CheckBoxSizeCalc,
    TXT_CheckBoxDrawer,
    TXT_CheckBoxKeyPress,
    TXT_CheckBoxDestructor,
};

txt_checkbox_t *TXT_NewCheckBox(char *label, int *variable)
{
    txt_checkbox_t *checkbox;

    checkbox = malloc(sizeof(txt_checkbox_t));

    checkbox->widget.widget_class = &txt_checkbox_class;
    checkbox->widget.selectable = 1;
    checkbox->widget.visible = 1;
    checkbox->label = strdup(label);
    checkbox->variable = variable;

    return checkbox;
}