/*

 *  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);

}