#include <ncurses.h>

#include "../state.h"
#include "pages.h"
#include "text.h"

const char *options_screen_title = "OPTIONS";

const char *options_screen_width = " Board width:";
const char *options_screen_height = "Board height:";
const char *options_screen_minecount = "  # of Mines:";

const char *options_screen_back = "Back";

int digit_input(int line, int col) {
    // input of two digits for int
    move(line, col);
    clrtoeol();

    char in1, in2;
    uint8_t n = 0;

    // digit one (tens)
    attron(A_UNDERLINE);
    mvaddch(line, col, ' ');
    while ((in1 = getch())) {
        if (in1 >= '0' && in1 <= '9') {
            n = 10 * (in1 - '0');
            break;
        } else // invalid input, beep
            beep();
    }
    attroff(A_UNDERLINE);
    mvaddch(line, col, in1);

    // digit two (ones)
    attron(A_UNDERLINE);
    mvaddch(line, col + 1, ' ');
    while ((in2 = getch())) {
        if (in2 >= '0' && in2 <= '9') {
            n += in2 - '0';
            break;
        } else // invalid input, beep
            beep();
    }
    attroff(A_UNDERLINE);

    return n;
}

int draw_options_screen(game_state *state, int ch) {
    if (ch == 'j' || ch == 'J') {
        if (state->page_selection < 3)
            state->page_selection++;
        else
            beep();
    } else if (ch == 'k' || ch == 'K') {
        if (state->page_selection > 0)
            state->page_selection--;
        else
            beep();
    } else if (ch == ' ' || ch == 10) {
        if (state->page_selection == 3) {
            state->page = Title;
            state->page_selection = 1;
            return draw_title_screen(state, 0);
        }
    }

    int y = centery();

    attron(A_BOLD);
    mvaddstr(y - 3, centerx(options_screen_title), options_screen_title);
    attroff(A_BOLD);

    int x = (int)(COLS / 2);
    mvaddstr(y - 1, x - 10, options_screen_width);
    mvaddstr(y, x - 10, options_screen_height);
    mvaddstr(y + 1, x - 10, options_screen_minecount);

    if (state->page_selection == 0) attron(A_STANDOUT);
    mvprintw(y - 1, x + 4, "%02d", state->board->width);
    if (state->page_selection == 0) attroff(A_STANDOUT);

    if (state->page_selection == 1) attron(A_STANDOUT);
    mvprintw(y, x + 4, "%02d", state->board->height);
    if (state->page_selection == 1) attroff(A_STANDOUT);

    if (state->page_selection == 2) attron(A_STANDOUT);
    mvprintw(y + 1, x + 4, "%02d", state->board->mine_count);
    if (state->page_selection == 2) attroff(A_STANDOUT);

    if (state->page_selection == 3) attron(A_STANDOUT);
    mvaddstr(y + 3, centerx(options_screen_back), options_screen_back);
    if (state->page_selection == 3) attroff(A_STANDOUT);

    if (ch == ' ' || ch == 10) {
        switch (state->page_selection) {
            case 0: {
                state->board->width = digit_input(y - 1, x + 4);
                attron(A_STANDOUT);
                mvprintw(y - 1, x + 4, "%02d", state->board->width);
                attroff(A_STANDOUT);
                break;
            }
            case 1: {
                state->board->height = digit_input(y, x + 4);
                attron(A_STANDOUT);
                mvprintw(y, x + 4, "%02d", state->board->height);
                attroff(A_STANDOUT);
                break;
            }
            case 2: {
                state->board->mine_count = digit_input(y + 1, x + 4);
                attron(A_STANDOUT);
                mvprintw(y + 1, x + 4, "%02d", state->board->mine_count);
                attroff(A_STANDOUT);
                break;
            }
        }
    }

    return 0;
}