1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (C) 2008 Nir Tzachar <nir.tzachar@gmail.com>
4  *
5  * Derived from menuconfig.
6  */
7 #include <xalloc.h>
8 #include "nconf.h"
9 #include "lkc.h"
10 
11 int attr_normal;
12 int attr_main_heading;
13 int attr_main_menu_box;
14 int attr_main_menu_fore;
15 int attr_main_menu_back;
16 int attr_main_menu_grey;
17 int attr_main_menu_heading;
18 int attr_scrollwin_text;
19 int attr_scrollwin_heading;
20 int attr_scrollwin_box;
21 int attr_dialog_text;
22 int attr_dialog_menu_fore;
23 int attr_dialog_menu_back;
24 int attr_dialog_box;
25 int attr_input_box;
26 int attr_input_heading;
27 int attr_input_text;
28 int attr_input_field;
29 int attr_function_text;
30 int attr_function_highlight;
31 
32 #define COLOR_ATTR(_at, _fg, _bg, _hl) \
33 	{ .attr = &(_at), .has_color = true, .color_fg = _fg, .color_bg = _bg, .highlight = _hl }
34 #define NO_COLOR_ATTR(_at, _hl) \
35 	{ .attr = &(_at), .has_color = false, .highlight = _hl }
36 #define COLOR_DEFAULT		-1
37 
38 struct nconf_attr_param {
39 	int *attr;
40 	bool has_color;
41 	int color_fg;
42 	int color_bg;
43 	int highlight;
44 };
45 
46 static const struct nconf_attr_param color_theme_params[] = {
47 	COLOR_ATTR(attr_normal,			COLOR_DEFAULT,	COLOR_DEFAULT,	A_NORMAL),
48 	COLOR_ATTR(attr_main_heading,		COLOR_MAGENTA,	COLOR_DEFAULT,	A_BOLD | A_UNDERLINE),
49 	COLOR_ATTR(attr_main_menu_box,		COLOR_YELLOW,	COLOR_DEFAULT,	A_NORMAL),
50 	COLOR_ATTR(attr_main_menu_fore,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_REVERSE),
51 	COLOR_ATTR(attr_main_menu_back,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_NORMAL),
52 	COLOR_ATTR(attr_main_menu_grey,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_NORMAL),
53 	COLOR_ATTR(attr_main_menu_heading,	COLOR_GREEN,	COLOR_DEFAULT,	A_BOLD),
54 	COLOR_ATTR(attr_scrollwin_text,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_NORMAL),
55 	COLOR_ATTR(attr_scrollwin_heading,	COLOR_GREEN,	COLOR_DEFAULT,	A_BOLD),
56 	COLOR_ATTR(attr_scrollwin_box,		COLOR_YELLOW,	COLOR_DEFAULT,	A_BOLD),
57 	COLOR_ATTR(attr_dialog_text,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_BOLD),
58 	COLOR_ATTR(attr_dialog_menu_fore,	COLOR_RED,	COLOR_DEFAULT,	A_STANDOUT),
59 	COLOR_ATTR(attr_dialog_menu_back,	COLOR_YELLOW,	COLOR_DEFAULT,	A_NORMAL),
60 	COLOR_ATTR(attr_dialog_box,		COLOR_YELLOW,	COLOR_DEFAULT,	A_BOLD),
61 	COLOR_ATTR(attr_input_box,		COLOR_YELLOW,	COLOR_DEFAULT,	A_NORMAL),
62 	COLOR_ATTR(attr_input_heading,		COLOR_GREEN,	COLOR_DEFAULT,	A_BOLD),
63 	COLOR_ATTR(attr_input_text,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_NORMAL),
64 	COLOR_ATTR(attr_input_field,		COLOR_DEFAULT,	COLOR_DEFAULT,	A_UNDERLINE),
65 	COLOR_ATTR(attr_function_text,		COLOR_YELLOW,	COLOR_DEFAULT,	A_REVERSE),
66 	COLOR_ATTR(attr_function_highlight,	COLOR_DEFAULT,	COLOR_DEFAULT,	A_BOLD),
67 	{ /* sentinel */ }
68 };
69 
70 static const struct nconf_attr_param no_color_theme_params[] = {
71 	NO_COLOR_ATTR(attr_normal,		A_NORMAL),
72 	NO_COLOR_ATTR(attr_main_heading,	A_BOLD | A_UNDERLINE),
73 	NO_COLOR_ATTR(attr_main_menu_box,	A_NORMAL),
74 	NO_COLOR_ATTR(attr_main_menu_fore,	A_STANDOUT),
75 	NO_COLOR_ATTR(attr_main_menu_back,	A_NORMAL),
76 	NO_COLOR_ATTR(attr_main_menu_grey,	A_NORMAL),
77 	NO_COLOR_ATTR(attr_main_menu_heading,	A_BOLD),
78 	NO_COLOR_ATTR(attr_scrollwin_text,	A_NORMAL),
79 	NO_COLOR_ATTR(attr_scrollwin_heading,	A_BOLD),
80 	NO_COLOR_ATTR(attr_scrollwin_box,	A_BOLD),
81 	NO_COLOR_ATTR(attr_dialog_text,		A_NORMAL),
82 	NO_COLOR_ATTR(attr_dialog_menu_fore,	A_STANDOUT),
83 	NO_COLOR_ATTR(attr_dialog_menu_back,	A_NORMAL),
84 	NO_COLOR_ATTR(attr_dialog_box,		A_BOLD),
85 	NO_COLOR_ATTR(attr_input_box,		A_BOLD),
86 	NO_COLOR_ATTR(attr_input_heading,	A_BOLD),
87 	NO_COLOR_ATTR(attr_input_text,		A_NORMAL),
88 	NO_COLOR_ATTR(attr_input_field,		A_UNDERLINE),
89 	NO_COLOR_ATTR(attr_function_text,	A_REVERSE),
90 	NO_COLOR_ATTR(attr_function_highlight,	A_BOLD),
91 	{ /* sentinel */ }
92 };
93 
set_colors(void)94 void set_colors(void)
95 {
96 	const struct nconf_attr_param *p;
97 	int pair = 0;
98 
99 	if (has_colors()) {
100 		start_color();
101 		use_default_colors();
102 		p = color_theme_params;
103 	} else {
104 		p = no_color_theme_params;
105 	}
106 
107 	for (; p->attr; p++) {
108 		int attr = p->highlight;
109 
110 		if (p->has_color) {
111 			pair++;
112 			init_pair(pair, p->color_fg, p->color_bg);
113 			attr |= COLOR_PAIR(pair);
114 		}
115 
116 		*p->attr = attr;
117 	}
118 }
119 
120 /* this changes the windows attributes !!! */
print_in_middle(WINDOW * win,int y,int width,const char * str,int attrs)121 void print_in_middle(WINDOW *win, int y, int width, const char *str, int attrs)
122 {
123 	wattrset(win, attrs);
124 	mvwprintw(win, y, (width - strlen(str)) / 2, "%s", str);
125 }
126 
get_line_no(const char * text)127 int get_line_no(const char *text)
128 {
129 	int i;
130 	int total = 1;
131 
132 	if (!text)
133 		return 0;
134 
135 	for (i = 0; text[i] != '\0'; i++)
136 		if (text[i] == '\n')
137 			total++;
138 	return total;
139 }
140 
get_line(const char * text,int line_no)141 const char *get_line(const char *text, int line_no)
142 {
143 	int i;
144 	int lines = 0;
145 
146 	if (!text)
147 		return NULL;
148 
149 	for (i = 0; text[i] != '\0' && lines < line_no; i++)
150 		if (text[i] == '\n')
151 			lines++;
152 	return text+i;
153 }
154 
get_line_length(const char * line)155 int get_line_length(const char *line)
156 {
157 	int res = 0;
158 	while (*line != '\0' && *line != '\n') {
159 		line++;
160 		res++;
161 	}
162 	return res;
163 }
164 
165 /* print all lines to the window. */
fill_window(WINDOW * win,const char * text)166 void fill_window(WINDOW *win, const char *text)
167 {
168 	int x, y;
169 	int total_lines = get_line_no(text);
170 	int i;
171 
172 	getmaxyx(win, y, x);
173 	/* do not go over end of line */
174 	total_lines = min(total_lines, y);
175 	for (i = 0; i < total_lines; i++) {
176 		char tmp[x+10];
177 		const char *line = get_line(text, i);
178 		int len = get_line_length(line);
179 		strncpy(tmp, line, min(len, x));
180 		tmp[len] = '\0';
181 		mvwprintw(win, i, 0, "%s", tmp);
182 	}
183 }
184 
185 /* get the message, and buttons.
186  * each button must be a char*
187  * return the selected button
188  *
189  * this dialog is used for 2 different things:
190  * 1) show a text box, no buttons.
191  * 2) show a dialog, with horizontal buttons
192  */
btn_dialog(WINDOW * main_window,const char * msg,int btn_num,...)193 int btn_dialog(WINDOW *main_window, const char *msg, int btn_num, ...)
194 {
195 	va_list ap;
196 	char *btn;
197 	int btns_width = 0;
198 	int msg_lines = 0;
199 	int msg_width = 0;
200 	int total_width;
201 	int win_rows = 0;
202 	WINDOW *win;
203 	WINDOW *msg_win;
204 	WINDOW *menu_win;
205 	MENU *menu;
206 	ITEM *btns[btn_num+1];
207 	int i, x, y;
208 	int res = -1;
209 
210 
211 	va_start(ap, btn_num);
212 	for (i = 0; i < btn_num; i++) {
213 		btn = va_arg(ap, char *);
214 		btns[i] = new_item(btn, "");
215 		btns_width += strlen(btn)+1;
216 	}
217 	va_end(ap);
218 	btns[btn_num] = NULL;
219 
220 	/* find the widest line of msg: */
221 	msg_lines = get_line_no(msg);
222 	for (i = 0; i < msg_lines; i++) {
223 		const char *line = get_line(msg, i);
224 		int len = get_line_length(line);
225 		if (msg_width < len)
226 			msg_width = len;
227 	}
228 
229 	total_width = max(msg_width, btns_width);
230 	/* place dialog in middle of screen */
231 	y = (getmaxy(stdscr)-(msg_lines+4))/2;
232 	x = (getmaxx(stdscr)-(total_width+4))/2;
233 
234 
235 	/* create the windows */
236 	if (btn_num > 0)
237 		win_rows = msg_lines+4;
238 	else
239 		win_rows = msg_lines+2;
240 
241 	win = newwin(win_rows, total_width+4, y, x);
242 	keypad(win, TRUE);
243 	menu_win = derwin(win, 1, btns_width, win_rows-2,
244 			1+(total_width+2-btns_width)/2);
245 	menu = new_menu(btns);
246 	msg_win = derwin(win, win_rows-2, msg_width, 1,
247 			1+(total_width+2-msg_width)/2);
248 
249 	set_menu_fore(menu, attr_dialog_menu_fore);
250 	set_menu_back(menu, attr_dialog_menu_back);
251 
252 	wattrset(win, attr_dialog_box);
253 	box(win, 0, 0);
254 
255 	/* print message */
256 	wattrset(msg_win, attr_dialog_text);
257 	fill_window(msg_win, msg);
258 
259 	set_menu_win(menu, win);
260 	set_menu_sub(menu, menu_win);
261 	set_menu_format(menu, 1, btn_num);
262 	menu_opts_off(menu, O_SHOWDESC);
263 	menu_opts_off(menu, O_SHOWMATCH);
264 	menu_opts_on(menu, O_ONEVALUE);
265 	menu_opts_on(menu, O_NONCYCLIC);
266 	set_menu_mark(menu, "");
267 	post_menu(menu);
268 
269 
270 	touchwin(win);
271 	refresh_all_windows(main_window);
272 	while ((res = wgetch(win))) {
273 		switch (res) {
274 		case KEY_LEFT:
275 			menu_driver(menu, REQ_LEFT_ITEM);
276 			break;
277 		case KEY_RIGHT:
278 			menu_driver(menu, REQ_RIGHT_ITEM);
279 			break;
280 		case 10: /* ENTER */
281 		case 27: /* ESCAPE */
282 		case ' ':
283 		case KEY_F(F_BACK):
284 		case KEY_F(F_EXIT):
285 			break;
286 		}
287 		touchwin(win);
288 		refresh_all_windows(main_window);
289 
290 		if (res == 10 || res == ' ') {
291 			res = item_index(current_item(menu));
292 			break;
293 		} else if (res == 27 || res == KEY_F(F_BACK) ||
294 				res == KEY_F(F_EXIT)) {
295 			res = KEY_EXIT;
296 			break;
297 		}
298 	}
299 
300 	unpost_menu(menu);
301 	free_menu(menu);
302 	for (i = 0; i < btn_num; i++)
303 		free_item(btns[i]);
304 
305 	delwin(win);
306 	return res;
307 }
308 
dialog_inputbox(WINDOW * main_window,const char * title,const char * prompt,const char * init,char ** resultp,int * result_len)309 int dialog_inputbox(WINDOW *main_window,
310 		const char *title, const char *prompt,
311 		const char *init, char **resultp, int *result_len)
312 {
313 	int prompt_lines = 0;
314 	int prompt_width = 0;
315 	WINDOW *win;
316 	WINDOW *prompt_win;
317 	WINDOW *form_win;
318 	PANEL *panel;
319 	int i, x, y, lines, columns, win_lines, win_cols;
320 	int res = -1;
321 	int cursor_position = strlen(init);
322 	int cursor_form_win;
323 	char *result = *resultp;
324 
325 	getmaxyx(stdscr, lines, columns);
326 
327 	if (strlen(init)+1 > *result_len) {
328 		*result_len = strlen(init)+1;
329 		*resultp = result = xrealloc(result, *result_len);
330 	}
331 
332 	/* find the widest line of msg: */
333 	prompt_lines = get_line_no(prompt);
334 	for (i = 0; i < prompt_lines; i++) {
335 		const char *line = get_line(prompt, i);
336 		int len = get_line_length(line);
337 		prompt_width = max(prompt_width, len);
338 	}
339 
340 	if (title)
341 		prompt_width = max(prompt_width, strlen(title));
342 
343 	win_lines = min(prompt_lines+6, lines-2);
344 	win_cols = min(prompt_width+7, columns-2);
345 	prompt_lines = max(win_lines-6, 0);
346 	prompt_width = max(win_cols-7, 0);
347 
348 	/* place dialog in middle of screen */
349 	y = (lines-win_lines)/2;
350 	x = (columns-win_cols)/2;
351 
352 	strncpy(result, init, *result_len);
353 
354 	/* create the windows */
355 	win = newwin(win_lines, win_cols, y, x);
356 	prompt_win = derwin(win, prompt_lines+1, prompt_width, 2, 2);
357 	form_win = derwin(win, 1, prompt_width, prompt_lines+3, 2);
358 	keypad(form_win, TRUE);
359 
360 	wattrset(form_win, attr_input_field);
361 
362 	wattrset(win, attr_input_box);
363 	box(win, 0, 0);
364 	wattrset(win, attr_input_heading);
365 	if (title)
366 		mvwprintw(win, 0, 3, "%s", title);
367 
368 	/* print message */
369 	wattrset(prompt_win, attr_input_text);
370 	fill_window(prompt_win, prompt);
371 
372 	mvwprintw(form_win, 0, 0, "%*s", prompt_width, " ");
373 	cursor_form_win = min(cursor_position, prompt_width-1);
374 	mvwprintw(form_win, 0, 0, "%s",
375 		  result + cursor_position-cursor_form_win);
376 
377 	/* create panels */
378 	panel = new_panel(win);
379 
380 	/* show the cursor */
381 	curs_set(1);
382 
383 	touchwin(win);
384 	refresh_all_windows(main_window);
385 	while ((res = wgetch(form_win))) {
386 		int len = strlen(result);
387 		switch (res) {
388 		case 10: /* ENTER */
389 		case 27: /* ESCAPE */
390 		case KEY_F(F_HELP):
391 		case KEY_F(F_EXIT):
392 		case KEY_F(F_BACK):
393 			break;
394 		case 8:   /* ^H */
395 		case 127: /* ^? */
396 		case KEY_BACKSPACE:
397 			if (cursor_position > 0) {
398 				memmove(&result[cursor_position-1],
399 						&result[cursor_position],
400 						len-cursor_position+1);
401 				cursor_position--;
402 				cursor_form_win--;
403 				len--;
404 			}
405 			break;
406 		case KEY_DC:
407 			if (cursor_position >= 0 && cursor_position < len) {
408 				memmove(&result[cursor_position],
409 						&result[cursor_position+1],
410 						len-cursor_position+1);
411 				len--;
412 			}
413 			break;
414 		case KEY_UP:
415 		case KEY_RIGHT:
416 			if (cursor_position < len) {
417 				cursor_position++;
418 				cursor_form_win++;
419 			}
420 			break;
421 		case KEY_DOWN:
422 		case KEY_LEFT:
423 			if (cursor_position > 0) {
424 				cursor_position--;
425 				cursor_form_win--;
426 			}
427 			break;
428 		case KEY_HOME:
429 			cursor_position = 0;
430 			cursor_form_win = 0;
431 			break;
432 		case KEY_END:
433 			cursor_position = len;
434 			cursor_form_win = min(cursor_position, prompt_width-1);
435 			break;
436 		default:
437 			if ((isgraph(res) || isspace(res))) {
438 				/* one for new char, one for '\0' */
439 				if (len+2 > *result_len) {
440 					*result_len = len+2;
441 					*resultp = result = realloc(result,
442 								*result_len);
443 				}
444 				/* insert the char at the proper position */
445 				memmove(&result[cursor_position+1],
446 						&result[cursor_position],
447 						len-cursor_position+1);
448 				result[cursor_position] = res;
449 				cursor_position++;
450 				cursor_form_win++;
451 				len++;
452 			} else {
453 				mvprintw(0, 0, "unknown key: %d\n", res);
454 			}
455 			break;
456 		}
457 		if (cursor_form_win < 0)
458 			cursor_form_win = 0;
459 		else if (cursor_form_win > prompt_width-1)
460 			cursor_form_win = prompt_width-1;
461 
462 		wmove(form_win, 0, 0);
463 		wclrtoeol(form_win);
464 		mvwprintw(form_win, 0, 0, "%*s", prompt_width, " ");
465 		mvwprintw(form_win, 0, 0, "%s",
466 			result + cursor_position-cursor_form_win);
467 		wmove(form_win, 0, cursor_form_win);
468 		touchwin(win);
469 		refresh_all_windows(main_window);
470 
471 		if (res == 10) {
472 			res = 0;
473 			break;
474 		} else if (res == 27 || res == KEY_F(F_BACK) ||
475 				res == KEY_F(F_EXIT)) {
476 			res = KEY_EXIT;
477 			break;
478 		} else if (res == KEY_F(F_HELP)) {
479 			res = 1;
480 			break;
481 		}
482 	}
483 
484 	/* hide the cursor */
485 	curs_set(0);
486 	del_panel(panel);
487 	delwin(prompt_win);
488 	delwin(form_win);
489 	delwin(win);
490 	return res;
491 }
492 
493 /* refresh all windows in the correct order */
refresh_all_windows(WINDOW * main_window)494 void refresh_all_windows(WINDOW *main_window)
495 {
496 	update_panels();
497 	touchwin(main_window);
498 	refresh();
499 }
500 
show_scroll_win(WINDOW * main_window,const char * title,const char * text)501 void show_scroll_win(WINDOW *main_window,
502 		const char *title,
503 		const char *text)
504 {
505 	(void)show_scroll_win_ext(main_window, title, (char *)text, NULL, NULL, NULL, NULL);
506 }
507 
508 /* layman's scrollable window... */
show_scroll_win_ext(WINDOW * main_window,const char * title,char * text,int * vscroll,int * hscroll,extra_key_cb_fn extra_key_cb,void * data)509 int show_scroll_win_ext(WINDOW *main_window, const char *title, char *text,
510 			int *vscroll, int *hscroll,
511 			extra_key_cb_fn extra_key_cb, void *data)
512 {
513 	int res;
514 	int total_lines = get_line_no(text);
515 	int x, y, lines, columns;
516 	int start_x = 0, start_y = 0;
517 	int text_lines = 0, text_cols = 0;
518 	int total_cols = 0;
519 	int win_cols = 0;
520 	int win_lines = 0;
521 	int i = 0;
522 	WINDOW *win;
523 	WINDOW *pad;
524 	PANEL *panel;
525 	bool done = false;
526 
527 	if (hscroll)
528 		start_x = *hscroll;
529 	if (vscroll)
530 		start_y = *vscroll;
531 
532 	getmaxyx(stdscr, lines, columns);
533 
534 	/* find the widest line of msg: */
535 	total_lines = get_line_no(text);
536 	for (i = 0; i < total_lines; i++) {
537 		const char *line = get_line(text, i);
538 		int len = get_line_length(line);
539 		total_cols = max(total_cols, len+2);
540 	}
541 
542 	/* create the pad */
543 	pad = newpad(total_lines+10, total_cols+10);
544 	wattrset(pad, attr_scrollwin_text);
545 	fill_window(pad, text);
546 
547 	win_lines = min(total_lines+4, lines-2);
548 	win_cols = min(total_cols+2, columns-2);
549 	text_lines = max(win_lines-4, 0);
550 	text_cols = max(win_cols-2, 0);
551 
552 	/* place window in middle of screen */
553 	y = (lines-win_lines)/2;
554 	x = (columns-win_cols)/2;
555 
556 	win = newwin(win_lines, win_cols, y, x);
557 	keypad(win, TRUE);
558 	/* show the help in the help window, and show the help panel */
559 	wattrset(win, attr_scrollwin_box);
560 	box(win, 0, 0);
561 	wattrset(win, attr_scrollwin_heading);
562 	mvwprintw(win, 0, 3, " %s ", title);
563 	panel = new_panel(win);
564 
565 	/* handle scrolling */
566 	while (!done) {
567 		copywin(pad, win, start_y, start_x, 2, 2, text_lines,
568 				text_cols, 0);
569 		print_in_middle(win,
570 				text_lines+2,
571 				text_cols,
572 				"<OK>",
573 				attr_dialog_menu_fore);
574 		wrefresh(win);
575 
576 		res = wgetch(win);
577 		switch (res) {
578 		case KEY_NPAGE:
579 		case ' ':
580 		case 'd':
581 			start_y += text_lines-2;
582 			break;
583 		case KEY_PPAGE:
584 		case 'u':
585 			start_y -= text_lines+2;
586 			break;
587 		case KEY_HOME:
588 			start_y = 0;
589 			break;
590 		case KEY_END:
591 			start_y = total_lines-text_lines;
592 			break;
593 		case KEY_DOWN:
594 		case 'j':
595 			start_y++;
596 			break;
597 		case KEY_UP:
598 		case 'k':
599 			start_y--;
600 			break;
601 		case KEY_LEFT:
602 		case 'h':
603 			start_x--;
604 			break;
605 		case KEY_RIGHT:
606 		case 'l':
607 			start_x++;
608 			break;
609 		default:
610 			if (extra_key_cb) {
611 				size_t start = (get_line(text, start_y) - text);
612 				size_t end = (get_line(text, start_y + text_lines) - text);
613 
614 				if (extra_key_cb(res, start, end, data)) {
615 					done = true;
616 					break;
617 				}
618 			}
619 		}
620 		if (res == 0 || res == 10 || res == 27 || res == 'q' ||
621 			res == KEY_F(F_HELP) || res == KEY_F(F_BACK) ||
622 			res == KEY_F(F_EXIT))
623 			break;
624 		if (start_y < 0)
625 			start_y = 0;
626 		if (start_y >= total_lines-text_lines)
627 			start_y = total_lines-text_lines;
628 		if (start_x < 0)
629 			start_x = 0;
630 		if (start_x >= total_cols-text_cols)
631 			start_x = total_cols-text_cols;
632 	}
633 
634 	if (hscroll)
635 		*hscroll = start_x;
636 	if (vscroll)
637 		*vscroll = start_y;
638 	del_panel(panel);
639 	delwin(win);
640 	refresh_all_windows(main_window);
641 	return res;
642 }
643