/*
 *  util.c
 *
 *  ORIGINAL AUTHOR: Savio Lam (lam836@cs.cuhk.hk)
 *  MODIFIED FOR LINUX KERNEL CONFIG BY: William Roadcap (roadcap@cfw.com)
 *
 *  This program 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 2
 *  of the License, or (at your option) any later version.
 *
 *  This program 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 this program; if not, write to the Free Software
 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include <stdarg.h>

#include "dialog.h"

struct dialog_info dlg;

static void set_mono_theme(void)
{
    dlg.screen.atr = A_NORMAL;
    dlg.shadow.atr = A_NORMAL;
    dlg.dialog.atr = A_NORMAL;
    dlg.title.atr = A_BOLD;
    dlg.border.atr = A_NORMAL;
    dlg.button_active.atr = A_REVERSE;
    dlg.button_inactive.atr = A_DIM;
    dlg.button_key_active.atr = A_REVERSE;
    dlg.button_key_inactive.atr = A_BOLD;
    dlg.button_label_active.atr = A_REVERSE;
    dlg.button_label_inactive.atr = A_NORMAL;
    dlg.inputbox.atr = A_NORMAL;
    dlg.inputbox_border.atr = A_NORMAL;
    dlg.searchbox.atr = A_NORMAL;
    dlg.searchbox_title.atr = A_BOLD;
    dlg.searchbox_border.atr = A_NORMAL;
    dlg.position_indicator.atr = A_BOLD;
    dlg.menubox.atr = A_NORMAL;
    dlg.menubox_border.atr = A_NORMAL;
    dlg.item.atr = A_NORMAL;
    dlg.item_selected.atr = A_REVERSE;
    dlg.tag.atr = A_BOLD;
    dlg.tag_selected.atr = A_REVERSE;
    dlg.tag_key.atr = A_BOLD;
    dlg.tag_key_selected.atr = A_REVERSE;
    dlg.check.atr = A_BOLD;
    dlg.check_selected.atr = A_REVERSE;
    dlg.uarrow.atr = A_BOLD;
    dlg.darrow.atr = A_BOLD;
}

#define DLG_COLOR(dialog, f, b, h) \
do {                               \
	dlg.dialog.fg = (f);       \
	dlg.dialog.bg = (b);       \
	dlg.dialog.hl = (h);       \
} while (0)

static void set_classic_theme(void)
{
    DLG_COLOR(screen, COLOR_CYAN, COLOR_BLUE, true);
    DLG_COLOR(shadow, COLOR_BLACK, COLOR_BLACK, true);
    DLG_COLOR(dialog, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(title, COLOR_YELLOW, COLOR_WHITE, true);
    DLG_COLOR(border, COLOR_WHITE, COLOR_WHITE, true);
    DLG_COLOR(button_active, COLOR_WHITE, COLOR_BLUE, true);
    DLG_COLOR(button_inactive, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(button_key_active, COLOR_WHITE, COLOR_BLUE, true);
    DLG_COLOR(button_key_inactive, COLOR_RED, COLOR_WHITE, false);
    DLG_COLOR(button_label_active, COLOR_YELLOW, COLOR_BLUE, true);
    DLG_COLOR(button_label_inactive, COLOR_BLACK, COLOR_WHITE, true);
    DLG_COLOR(inputbox, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(inputbox_border, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(searchbox, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(searchbox_title, COLOR_YELLOW, COLOR_WHITE, true);
    DLG_COLOR(searchbox_border, COLOR_WHITE, COLOR_WHITE, true);
    DLG_COLOR(position_indicator, COLOR_YELLOW, COLOR_WHITE, true);
    DLG_COLOR(menubox, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(menubox_border, COLOR_WHITE, COLOR_WHITE, true);
    DLG_COLOR(item, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(item_selected, COLOR_WHITE, COLOR_BLUE, true);
    DLG_COLOR(tag, COLOR_YELLOW, COLOR_WHITE, true);
    DLG_COLOR(tag_selected, COLOR_YELLOW, COLOR_BLUE, true);
    DLG_COLOR(tag_key, COLOR_YELLOW, COLOR_WHITE, true);
    DLG_COLOR(tag_key_selected, COLOR_YELLOW, COLOR_BLUE, true);
    DLG_COLOR(check, COLOR_BLACK, COLOR_WHITE, false);
    DLG_COLOR(check_selected, COLOR_WHITE, COLOR_BLUE, true);
    DLG_COLOR(uarrow, COLOR_GREEN, COLOR_WHITE, true);
    DLG_COLOR(darrow, COLOR_GREEN, COLOR_WHITE, true);
}

static void set_blackbg_theme(void)
{
    DLG_COLOR(screen, COLOR_RED, COLOR_BLACK, true);
    DLG_COLOR(shadow, COLOR_BLACK, COLOR_BLACK, false);
    DLG_COLOR(dialog, COLOR_WHITE, COLOR_BLACK, false);
    DLG_COLOR(title, COLOR_RED, COLOR_BLACK, false);
    DLG_COLOR(border, COLOR_BLACK, COLOR_BLACK, true);

    DLG_COLOR(button_active, COLOR_YELLOW, COLOR_RED, false);
    DLG_COLOR(button_inactive, COLOR_YELLOW, COLOR_BLACK, false);
    DLG_COLOR(button_key_active, COLOR_YELLOW, COLOR_RED, true);
    DLG_COLOR(button_key_inactive, COLOR_RED, COLOR_BLACK, false);
    DLG_COLOR(button_label_active, COLOR_WHITE, COLOR_RED, false);
    DLG_COLOR(button_label_inactive, COLOR_BLACK, COLOR_BLACK, true);

    DLG_COLOR(inputbox, COLOR_YELLOW, COLOR_BLACK, false);
    DLG_COLOR(inputbox_border, COLOR_YELLOW, COLOR_BLACK, false);

    DLG_COLOR(searchbox, COLOR_YELLOW, COLOR_BLACK, false);
    DLG_COLOR(searchbox_title, COLOR_YELLOW, COLOR_BLACK, true);
    DLG_COLOR(searchbox_border, COLOR_BLACK, COLOR_BLACK, true);

    DLG_COLOR(position_indicator, COLOR_RED, COLOR_BLACK, false);

    DLG_COLOR(menubox, COLOR_YELLOW, COLOR_BLACK, false);
    DLG_COLOR(menubox_border, COLOR_BLACK, COLOR_BLACK, true);

    DLG_COLOR(item, COLOR_WHITE, COLOR_BLACK, false);
    DLG_COLOR(item_selected, COLOR_WHITE, COLOR_RED, false);

    DLG_COLOR(tag, COLOR_RED, COLOR_BLACK, false);
    DLG_COLOR(tag_selected, COLOR_YELLOW, COLOR_RED, true);
    DLG_COLOR(tag_key, COLOR_RED, COLOR_BLACK, false);
    DLG_COLOR(tag_key_selected, COLOR_YELLOW, COLOR_RED, true);

    DLG_COLOR(check, COLOR_YELLOW, COLOR_BLACK, false);
    DLG_COLOR(check_selected, COLOR_YELLOW, COLOR_RED, true);

    DLG_COLOR(uarrow, COLOR_RED, COLOR_BLACK, false);
    DLG_COLOR(darrow, COLOR_RED, COLOR_BLACK, false);
}

static void set_bluetitle_theme(void)
{
    set_classic_theme();
    DLG_COLOR(title, COLOR_BLUE, COLOR_WHITE, true);
    DLG_COLOR(button_key_active, COLOR_YELLOW, COLOR_BLUE, true);
    DLG_COLOR(button_label_active, COLOR_WHITE, COLOR_BLUE, true);
    DLG_COLOR(searchbox_title, COLOR_BLUE, COLOR_WHITE, true);
    DLG_COLOR(position_indicator, COLOR_BLUE, COLOR_WHITE, true);
    DLG_COLOR(tag, COLOR_BLUE, COLOR_WHITE, true);
    DLG_COLOR(tag_key, COLOR_BLUE, COLOR_WHITE, true);

}

/*
 * Select color theme
 */
static int set_theme(const char *theme)
{
    int use_color = 1;

    if (!theme)
        set_bluetitle_theme();
    else if (strcmp(theme, "classic") == 0)
        set_classic_theme();
    else if (strcmp(theme, "bluetitle") == 0)
        set_bluetitle_theme();
    else if (strcmp(theme, "blackbg") == 0)
        set_blackbg_theme();
    else if (strcmp(theme, "mono") == 0)
        use_color = 0;

    return use_color;
}

static void init_one_color(struct dialog_color *color)
{
    static int pair = 0;

    pair++;
    init_pair(pair, color->fg, color->bg);
    if (color->hl)
        color->atr = A_BOLD | COLOR_PAIR(pair);
    else
        color->atr = COLOR_PAIR(pair);
}

static void init_dialog_colors(void)
{
    init_one_color(&dlg.screen);
    init_one_color(&dlg.shadow);
    init_one_color(&dlg.dialog);
    init_one_color(&dlg.title);
    init_one_color(&dlg.border);
    init_one_color(&dlg.button_active);
    init_one_color(&dlg.button_inactive);
    init_one_color(&dlg.button_key_active);
    init_one_color(&dlg.button_key_inactive);
    init_one_color(&dlg.button_label_active);
    init_one_color(&dlg.button_label_inactive);
    init_one_color(&dlg.inputbox);
    init_one_color(&dlg.inputbox_border);
    init_one_color(&dlg.searchbox);
    init_one_color(&dlg.searchbox_title);
    init_one_color(&dlg.searchbox_border);
    init_one_color(&dlg.position_indicator);
    init_one_color(&dlg.menubox);
    init_one_color(&dlg.menubox_border);
    init_one_color(&dlg.item);
    init_one_color(&dlg.item_selected);
    init_one_color(&dlg.tag);
    init_one_color(&dlg.tag_selected);
    init_one_color(&dlg.tag_key);
    init_one_color(&dlg.tag_key_selected);
    init_one_color(&dlg.check);
    init_one_color(&dlg.check_selected);
    init_one_color(&dlg.uarrow);
    init_one_color(&dlg.darrow);
}

/*
 * Setup for color display
 */
static void color_setup(const char *theme)
{
    int use_color;

    use_color = set_theme(theme);
    if (use_color && has_colors()) {
        start_color();
        init_dialog_colors();
    } else
        set_mono_theme();
}

/*
 * Set window to attribute 'attr'
 */
void attr_clear(WINDOW * win, int height, int width, chtype attr)
{
    int i, j;

    wattrset(win, attr);
    for (i = 0; i < height; i++) {
        wmove(win, i, 0);
        for (j = 0; j < width; j++)
            waddch(win, ' ');
    }
    touchwin(win);
}

void dialog_clear(void)
{
    attr_clear(stdscr, LINES, COLS, dlg.screen.atr);
    /*
     * Display background title if it exists ... - SLH 
     */
    if (dlg.backtitle != NULL) {
        int i;

        wattrset(stdscr, dlg.screen.atr);
        mvwaddstr(stdscr, 0, 1, (char *)dlg.backtitle);
        wmove(stdscr, 1, 1);
        for (i = 1; i < COLS - 1; i++)
            waddch(stdscr, ACS_HLINE);
    }
    wnoutrefresh(stdscr);
}

/*
 * Do some initialization for dialog
 */
int init_dialog(const char *backtitle)
{
    int height, width;

    initscr();                  /* Init curses */
    getmaxyx(stdscr, height, width);
    if (height < 19 || width < 80) {
        endwin();
        return -ERRDISPLAYTOOSMALL;
    }

    dlg.backtitle = backtitle;
    color_setup(getenv("MENUCONFIG_COLOR"));

    keypad(stdscr, TRUE);
    cbreak();
    noecho();
    dialog_clear();

    return 0;
}

void set_dialog_backtitle(const char *backtitle)
{
    dlg.backtitle = backtitle;
}

/*
 * End using dialog functions.
 */
void end_dialog(int x, int y)
{
    /*
     * move cursor back to original position 
     */
    move(y, x);
    refresh();
    endwin();
}

/* Print the title of the dialog. Center the title and truncate
 * tile if wider than dialog (- 2 chars).
 **/
void print_title(WINDOW * dialog, const char *title, int width)
{
    if (title) {
        int tlen = MIN(width - 2, strlen(title));

        wattrset(dialog, dlg.title.atr);
        mvwaddch(dialog, 0, (width - tlen) / 2 - 1, ' ');
        mvwaddnstr(dialog, 0, (width - tlen) / 2, title, tlen);
        waddch(dialog, ' ');
    }
}

/*
 * Print a string of text in a window, automatically wrap around to the
 * next line if the string is too long to fit on one line. Newline
 * characters '\n' are replaced by spaces.  We start on a new line
 * if there is no room for at least 4 nonblanks following a double-space.
 */
void print_autowrap(WINDOW * win, const char *prompt, int width, int y, int x)
{
    int newl, cur_x, cur_y;

    int i, prompt_len, room, wlen;

    char tempstr[MAX_LEN + 1], *word, *sp, *sp2;

    strcpy(tempstr, prompt);

    prompt_len = strlen(tempstr);

    /*
     * Remove newlines
     */
    for (i = 0; i < prompt_len; i++) {
        if (tempstr[i] == '\n')
            tempstr[i] = ' ';
    }

    if (prompt_len <= width - x * 2) {  /* If prompt is short */
        wmove(win, y, (width - prompt_len) / 2);
        waddstr(win, tempstr);
    } else {
        cur_x = x;
        cur_y = y;
        newl = 1;
        word = tempstr;
        while (word && *word) {
            sp = strchr(word, ' ');
            if (sp)
                *sp++ = 0;

            /*
             * Wrap to next line if either the word does not fit,
             * or it is the first word of a new sentence, and it is
             * short, and the next word does not fit. 
             */
            room = width - cur_x;
            wlen = strlen(word);
            if (wlen > room ||
                (newl && wlen < 4 && sp
                 && wlen + 1 + strlen(sp) > room && (!(sp2 = strchr(sp, ' '))
                                                     || wlen + 1 + (sp2 - sp) >
                                                     room))) {
                cur_y++;
                cur_x = x;
            }
            wmove(win, cur_y, cur_x);
            waddstr(win, word);
            getyx(win, cur_y, cur_x);
            cur_x++;
            if (sp && *sp == ' ') {
                cur_x++;        /* double space */
                while (*++sp == ' ') ;
                newl = 1;
            } else
                newl = 0;
            word = sp;
        }
    }
}

/*
 * Print a button
 */
void print_button(WINDOW * win, const char *label, int y, int x, int selected)
{
    int i, temp;

    wmove(win, y, x);
    wattrset(win, selected ? dlg.button_active.atr : dlg.button_inactive.atr);
    waddstr(win, "<");
    temp = strspn(label, " ");
    label += temp;
    wattrset(win, selected ? dlg.button_label_active.atr
             : dlg.button_label_inactive.atr);
    for (i = 0; i < temp; i++)
        waddch(win, ' ');
    wattrset(win, selected ? dlg.button_key_active.atr
             : dlg.button_key_inactive.atr);
    waddch(win, label[0]);
    wattrset(win, selected ? dlg.button_label_active.atr
             : dlg.button_label_inactive.atr);
    waddstr(win, (char *)label + 1);
    wattrset(win, selected ? dlg.button_active.atr : dlg.button_inactive.atr);
    waddstr(win, ">");
    wmove(win, y, x + temp + 1);
}

/*
 * Draw a rectangular box with line drawing characters
 */
void
draw_box(WINDOW * win, int y, int x, int height, int width,
         chtype box, chtype border)
{
    int i, j;

    wattrset(win, 0);
    for (i = 0; i < height; i++) {
        wmove(win, y + i, x);
        for (j = 0; j < width; j++)
            if (!i && !j)
                waddch(win, border | ACS_ULCORNER);
            else if (i == height - 1 && !j)
                waddch(win, border | ACS_LLCORNER);
            else if (!i && j == width - 1)
                waddch(win, box | ACS_URCORNER);
            else if (i == height - 1 && j == width - 1)
                waddch(win, box | ACS_LRCORNER);
            else if (!i)
                waddch(win, border | ACS_HLINE);
            else if (i == height - 1)
                waddch(win, box | ACS_HLINE);
            else if (!j)
                waddch(win, border | ACS_VLINE);
            else if (j == width - 1)
                waddch(win, box | ACS_VLINE);
            else
                waddch(win, box | ' ');
    }
}

/*
 * Draw shadows along the right and bottom edge to give a more 3D look
 * to the boxes
 */
void draw_shadow(WINDOW * win, int y, int x, int height, int width)
{
    int i;

    if (has_colors()) {         /* Whether terminal supports color? */
        wattrset(win, dlg.shadow.atr);
        wmove(win, y + height, x + 2);
        for (i = 0; i < width; i++)
            waddch(win, winch(win) & A_CHARTEXT);
        for (i = y + 1; i < y + height + 1; i++) {
            wmove(win, i, x + width);
            waddch(win, winch(win) & A_CHARTEXT);
            waddch(win, winch(win) & A_CHARTEXT);
        }
        wnoutrefresh(win);
    }
}

/*
 *  Return the position of the first alphabetic character in a string.
 */
int first_alpha(const char *string, const char *exempt)
{
    int i, in_paren = 0, c;

    for (i = 0; i < strlen(string); i++) {
        c = tolower(string[i]);

        if (strchr("<[(", c))
            ++in_paren;
        if (strchr(">])", c) && in_paren > 0)
            --in_paren;

        if ((!in_paren) && isalpha(c) && strchr(exempt, c) == 0)
            return i;
    }

    return 0;
}

/*
 * ncurses uses ESC to detect escaped char sequences. This resutl in
 * a small timeout before ESC is actually delivered to the application.
 * lxdialog suggest <ESC> <ESC> which is correctly translated to two
 * times esc. But then we need to ignore the second esc to avoid stepping
 * out one menu too much. Filter away all escaped key sequences since
 * keypad(FALSE) turn off ncurses support for escape sequences - and thats
 * needed to make notimeout() do as expected.
 */
int on_key_esc(WINDOW * win)
{
    int key;

    int key2;

    int key3;

    nodelay(win, TRUE);
    keypad(win, FALSE);
    key = wgetch(win);
    key2 = wgetch(win);
    do {
        key3 = wgetch(win);
    } while (key3 != ERR);
    nodelay(win, FALSE);
    keypad(win, TRUE);
    if (key == KEY_ESC && key2 == ERR)
        return KEY_ESC;
    else if (key != ERR && key != KEY_ESC && key2 == ERR)
        ungetch(key);

    return -1;
}

/* redraw screen in new size */
int on_key_resize(void)
{
    dialog_clear();
    return KEY_RESIZE;
}

struct dialog_list *item_cur;

struct dialog_list item_nil;

struct dialog_list *item_head;

void item_reset(void)
{
    struct dialog_list *p, *next;

    for (p = item_head; p; p = next) {
        next = p->next;
        free(p);
    }
    item_head = NULL;
    item_cur = &item_nil;
}

void item_make(const char *fmt, ...)
{
    va_list ap;

    struct dialog_list *p = malloc(sizeof (*p));

    if (item_head)
        item_cur->next = p;
    else
        item_head = p;
    item_cur = p;
    memset(p, 0, sizeof (*p));

    va_start(ap, fmt);
    vsnprintf(item_cur->node.str, sizeof (item_cur->node.str), fmt, ap);
    va_end(ap);
}

void item_add_str(const char *fmt, ...)
{
    va_list ap;

    size_t avail;

    avail = sizeof (item_cur->node.str) - strlen(item_cur->node.str);

    va_start(ap, fmt);
    vsnprintf(item_cur->node.str + strlen(item_cur->node.str), avail, fmt, ap);
    item_cur->node.str[sizeof (item_cur->node.str) - 1] = '\0';
    va_end(ap);
}

void item_set_tag(char tag)
{
    item_cur->node.tag = tag;
}

void item_set_data(void *ptr)
{
    item_cur->node.data = ptr;
}

void item_set_selected(int val)
{
    item_cur->node.selected = val;
}

int item_activate_selected(void)
{
    item_foreach()
        if (item_is_selected())
        return 1;
    return 0;
}

void *item_data(void)
{
    return item_cur->node.data;
}

char item_tag(void)
{
    return item_cur->node.tag;
}

int item_count(void)
{
    int n = 0;

    struct dialog_list *p;

    for (p = item_head; p; p = p->next)
        n++;
    return n;
}

void item_set(int n)
{
    int i = 0;

    item_foreach()
        if (i++ == n)
        return;
}

int item_n(void)
{
    int n = 0;

    struct dialog_list *p;

    for (p = item_head; p; p = p->next) {
        if (p == item_cur)
            return n;
        n++;
    }
    return 0;
}

const char *item_str(void)
{
    return item_cur->node.str;
}

int item_is_selected(void)
{
    return (item_cur->node.selected != 0);
}

int item_is_tag(char tag)
{
    return (item_cur->node.tag == tag);
}
