2 * Copyright (C) 2013 IBM Corporation
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; version 2 of the License.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software
15 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include <linux/input.h> /* This must be included before ncurses.h */
23 #if defined HAVE_NCURSESW_CURSES_H
24 # include <ncursesw/curses.h>
25 #elif defined HAVE_NCURSESW_H
26 # include <ncursesw.h>
27 #elif defined HAVE_NCURSES_CURSES_H
28 # include <ncurses/curses.h>
29 #elif defined HAVE_NCURSES_H
31 #elif defined HAVE_CURSES_H
34 # error "Curses header file not found."
37 #if defined HAVE_NCURSESW_FORM_H
38 # include <ncursesw/form.h>
39 #elif defined HAVE_NCURSES_FORM_H
40 # include <ncurses/form.h>
41 #elif defined HAVE_FORM_H
44 # error "Curses form.h not found."
50 #include <talloc/talloc.h>
51 #include <types/types.h>
53 #include <util/util.h>
57 #include "nc-widgets.h"
61 #define to_checkbox(w) container_of(w, struct nc_widget_checkbox, widget)
62 #define to_textbox(w) container_of(w, struct nc_widget_textbox, widget)
63 #define to_button(w) container_of(w, struct nc_widget_button, widget)
64 #define to_select(w) container_of(w, struct nc_widget_select, widget)
66 static const char *checkbox_checked_str = "[*]";
67 static const char *checkbox_unchecked_str = "[ ]";
69 static const char *select_selected_str = "(*)";
70 static const char *select_unselected_str = "( )";
77 int n_fields, n_alloc_fields;
78 void (*widget_focus)(struct nc_widget *, void *);
79 void *widget_focus_arg;
85 bool (*process_key)(struct nc_widget *, FORM *, int);
86 void (*set_visible)(struct nc_widget *, bool);
87 void (*move)(struct nc_widget *, int, int);
88 void (*field_focus)(struct nc_widget *, FIELD *);
98 struct nc_widget_label {
99 struct nc_widget widget;
103 struct nc_widget_checkbox {
104 struct nc_widget widget;
108 struct nc_widget_textbox {
109 struct nc_widget widget;
112 struct nc_widget_select {
113 struct nc_widget widget;
114 struct select_option {
120 int n_options, selected_option;
121 struct nc_widgetset *set;
122 void (*on_change)(void *, int);
126 struct nc_widget_button {
127 struct nc_widget widget;
128 void (*click)(void *arg);
132 static void widgetset_add_field(struct nc_widgetset *set, FIELD *field);
133 static void widgetset_remove_field(struct nc_widgetset *set, FIELD *field);
135 static bool process_key_nop(struct nc_widget *widget __attribute__((unused)),
136 FORM *form __attribute((unused)),
137 int key __attribute__((unused)))
142 static void field_set_visible(FIELD *field, bool visible)
144 int opts = field_opts(field) & ~O_VISIBLE;
147 set_field_opts(field, opts);
150 static void field_move(FIELD *field, int y, int x)
152 move_field(field, y, x);
155 static int label_destructor(void *ptr)
157 struct nc_widget_label *label = ptr;
158 free_field(label->widget.field);
163 struct nc_widget_label *widget_new_label(struct nc_widgetset *set,
164 int y, int x, char *str)
166 struct nc_widget_label *label;
172 label = talloc_zero(set, struct nc_widget_label);
173 label->widget.height = 1;
174 label->widget.width = len;
177 label->widget.process_key = process_key_nop;
178 label->widget.field = f = new_field(1, len, y, x, 0, 0);
179 label->widget.focussed_attr = A_NORMAL;
180 label->widget.unfocussed_attr = A_NORMAL;
182 field_opts_off(f, O_ACTIVE);
183 set_field_buffer(f, 0, str);
184 set_field_userptr(f, &label->widget);
186 widgetset_add_field(set, label->widget.field);
187 talloc_set_destructor(label, label_destructor);
192 bool widget_checkbox_get_value(struct nc_widget_checkbox *checkbox)
194 return checkbox->checked;
197 static void checkbox_set_buffer(struct nc_widget_checkbox *checkbox)
200 str = checkbox->checked ? checkbox_checked_str : checkbox_unchecked_str;
201 set_field_buffer(checkbox->widget.field, 0, str);
204 static bool checkbox_process_key(struct nc_widget *widget,
205 FORM *form __attribute__((unused)), int key)
207 struct nc_widget_checkbox *checkbox = to_checkbox(widget);
212 checkbox->checked = !checkbox->checked;
213 checkbox_set_buffer(checkbox);
218 static int checkbox_destructor(void *ptr)
220 struct nc_widget_checkbox *checkbox = ptr;
221 free_field(checkbox->widget.field);
225 struct nc_widget_checkbox *widget_new_checkbox(struct nc_widgetset *set,
226 int y, int x, bool checked)
228 struct nc_widget_checkbox *checkbox;
231 checkbox = talloc_zero(set, struct nc_widget_checkbox);
232 checkbox->checked = checked;
233 checkbox->widget.height = 1;
234 checkbox->widget.width = strlen(checkbox_checked_str);
235 checkbox->widget.x = x;
236 checkbox->widget.y = y;
237 checkbox->widget.process_key = checkbox_process_key;
238 checkbox->widget.focussed_attr = A_REVERSE;
239 checkbox->widget.unfocussed_attr = A_NORMAL;
240 checkbox->widget.field = f = new_field(1, strlen(checkbox_checked_str),
243 field_opts_off(f, O_EDIT);
244 set_field_userptr(f, &checkbox->widget);
245 checkbox_set_buffer(checkbox);
247 widgetset_add_field(set, checkbox->widget.field);
248 talloc_set_destructor(checkbox, checkbox_destructor);
253 static char *strip_string(char *str)
259 /* clear trailing space */
260 for (i = len - 1; i >= 0; i--) {
261 if (!isspace(str[i]))
266 /* increment str past leading space */
267 for (i = 0; i < len; i++) {
268 if (str[i] == '\0' || !isspace(str[i]))
275 char *widget_textbox_get_value(struct nc_widget_textbox *textbox)
277 char *str = field_buffer(textbox->widget.field, 0);
278 return str ? strip_string(str) : NULL;
281 static bool textbox_process_key(
282 struct nc_widget *widget __attribute__((unused)),
287 form_driver(form, REQ_BEG_FIELD);
290 form_driver(form, REQ_END_FIELD);
293 form_driver(form, REQ_LEFT_CHAR);
296 form_driver(form, REQ_RIGHT_CHAR);
299 if (form_driver(form, REQ_LEFT_CHAR) != E_OK)
303 form_driver(form, REQ_DEL_CHAR);
306 form_driver(form, key);
313 static int textbox_destructor(void *ptr)
315 struct nc_widget_textbox *textbox = ptr;
316 free_field(textbox->widget.field);
320 struct nc_widget_textbox *widget_new_textbox(struct nc_widgetset *set,
321 int y, int x, int len, char *str)
323 struct nc_widget_textbox *textbox;
326 textbox = talloc_zero(set, struct nc_widget_textbox);
327 textbox->widget.height = 1;
328 textbox->widget.width = len;
329 textbox->widget.x = x;
330 textbox->widget.y = y;
331 textbox->widget.process_key = textbox_process_key;
332 textbox->widget.field = f = new_field(1, len, y, x, 0, 0);
333 textbox->widget.focussed_attr = A_REVERSE;
334 textbox->widget.unfocussed_attr = A_UNDERLINE;
336 field_opts_off(f, O_STATIC | O_WRAP | O_BLANK);
337 set_field_buffer(f, 0, str);
338 set_field_back(f, textbox->widget.unfocussed_attr);
339 set_field_userptr(f, &textbox->widget);
341 widgetset_add_field(set, textbox->widget.field);
342 talloc_set_destructor(textbox, textbox_destructor);
347 static void select_option_change(struct select_option *opt, bool selected)
351 str = selected ? select_selected_str : select_unselected_str;
353 memcpy(opt->str, str, strlen(str));
354 set_field_buffer(opt->field, 0, opt->str);
357 static bool select_process_key(struct nc_widget *w, FORM *form, int key)
359 struct nc_widget_select *select = to_select(w);
360 struct select_option *new_opt, *old_opt;
372 field = current_field(form);
375 for (i = 0; i < select->n_options; i++) {
376 if (select->options[i].field == field) {
377 new_opt = &select->options[i];
386 if (new_idx == select->selected_option)
389 old_opt = &select->options[select->selected_option];
391 select_option_change(old_opt, false);
392 select_option_change(new_opt, true);
394 select->selected_option = new_idx;
396 if (select->on_change)
397 select->on_change(select->on_change_arg, new_opt->val);
402 static void select_set_visible(struct nc_widget *widget, bool visible)
404 struct nc_widget_select *select = to_select(widget);
407 for (i = 0; i < select->n_options; i++)
408 field_set_visible(select->options[i].field, visible);
411 static void select_move(struct nc_widget *widget, int y, int x)
413 struct nc_widget_select *select = to_select(widget);
416 for (i = 0; i < select->n_options; i++)
417 field_move(select->options[i].field, y + i, x);
420 static void select_field_focus(struct nc_widget *widget, FIELD *field)
422 struct nc_widget_select *select = to_select(widget);
425 for (i = 0; i < select->n_options; i++) {
426 if (field != select->options[i].field)
433 static int select_destructor(void *ptr)
435 struct nc_widget_select *select = ptr;
438 for (i = 0; i < select->n_options; i++)
439 free_field(select->options[i].field);
444 struct nc_widget_select *widget_new_select(struct nc_widgetset *set,
445 int y, int x, int len)
447 struct nc_widget_select *select;
449 select = talloc_zero(set, struct nc_widget_select);
450 select->widget.width = len;
451 select->widget.height = 0;
452 select->widget.x = x;
453 select->widget.y = y;
454 select->widget.process_key = select_process_key;
455 select->widget.set_visible = select_set_visible;
456 select->widget.move = select_move;
457 select->widget.field_focus = select_field_focus;
458 select->widget.focussed_attr = A_REVERSE;
459 select->widget.unfocussed_attr = A_NORMAL;
465 talloc_set_destructor(select, select_destructor);
470 void widget_select_add_option(struct nc_widget_select *select, int value,
471 const char *text, bool selected)
477 /* if we never see an option with selected set, we want the first
478 * one to be selected */
479 if (select->n_options == 0)
482 select_option_change(&select->options[select->selected_option],
486 select->selected_option = select->n_options;
487 str = select_selected_str;
489 str = select_unselected_str;
491 i = select->n_options++;
492 select->widget.height = select->n_options;
494 select->options = talloc_realloc(select, select->options,
495 struct select_option, i + 2);
496 select->options[i].val = value;
497 select->options[i].str = talloc_asprintf(select->options,
500 select->options[i].field = f = new_field(1, select->size,
504 field_opts_off(f, O_WRAP | O_EDIT);
505 set_field_userptr(f, &select->widget);
506 set_field_buffer(f, 0, select->options[i].str);
508 widgetset_add_field(select->set, f);
511 int widget_select_get_value(struct nc_widget_select *select)
513 if (!select->n_options)
515 return select->options[select->selected_option].val;
518 int widget_select_height(struct nc_widget_select *select)
520 return select->n_options;
523 void widget_select_on_change(struct nc_widget_select *select,
524 void (*on_change)(void *, int), void *arg)
526 select->on_change = on_change;
527 select->on_change_arg = arg;
530 void widget_select_drop_options(struct nc_widget_select *select)
532 struct nc_widgetset *set = select->set;
535 for (i = 0; i < select->n_options; i++) {
536 FIELD *field = select->options[i].field;
537 widgetset_remove_field(set, field);
538 if (field == set->cur_field)
539 set->cur_field = NULL;
540 free_field(select->options[i].field);
543 talloc_free(select->options);
544 select->options = NULL;
545 select->n_options = 0;
546 select->widget.height = 0;
547 select->widget.focus_y = 0;
551 static bool button_process_key(struct nc_widget *widget,
552 FORM *form __attribute__((unused)), int key)
554 struct nc_widget_button *button = to_button(widget);
563 button->click(button->arg);
570 static int button_destructor(void *ptr)
572 struct nc_widget_button *button = ptr;
573 free_field(button->widget.field);
577 struct nc_widget_button *widget_new_button(struct nc_widgetset *set,
578 int y, int x, int size, const char *str,
579 void (*click)(void *), void *arg)
581 struct nc_widget_button *button;
586 button = talloc_zero(set, struct nc_widget_button);
587 button->widget.height = 1;
588 button->widget.width = size;
589 button->widget.x = x;
590 button->widget.y = y;
591 button->widget.field = f = new_field(1, size + 2, y, x, 0, 0);
592 button->widget.process_key = button_process_key;
593 button->widget.focussed_attr = A_REVERSE;
594 button->widget.unfocussed_attr = A_NORMAL;
595 button->click = click;
598 field_opts_off(f, O_EDIT);
599 set_field_userptr(f, &button->widget);
601 /* center str in a size-char buffer, but don't overrun */
603 len = min(len, size);
604 idx = (size - len) / 2;
606 text = talloc_array(button, char, size + 3);
607 memset(text, ' ', size + 2);
608 memcpy(text + idx + 1, str, len);
610 text[size + 1] = ']';
611 text[size + 2] = '\0';
613 set_field_buffer(f, 0, text);
615 widgetset_add_field(set, button->widget.field);
616 talloc_set_destructor(button, button_destructor);
621 static void widget_focus_change(struct nc_widget *widget, FIELD *field,
624 int attr = focussed ? widget->focussed_attr : widget->unfocussed_attr;
625 set_field_back(field, attr);
628 bool widgetset_process_key(struct nc_widgetset *set, int key)
630 struct nc_widget *widget;
634 field = current_field(set->form);
637 /* handle field change events */
641 req = REQ_PREV_FIELD;
645 req = REQ_NEXT_FIELD;
648 req = REQ_FIRST_FIELD;
651 req = REQ_LAST_FIELD;
655 widget = field_userptr(field);
657 widget_focus_change(widget, field, false);
658 form_driver(set->form, req);
659 form_driver(set->form, REQ_END_FIELD);
660 field = current_field(set->form);
661 widget = field_userptr(field);
662 widget_focus_change(widget, field, true);
663 if (widget->field_focus)
664 widget->field_focus(widget, field);
665 if (set->widget_focus)
666 set->widget_focus(widget, set->widget_focus_arg);
670 if (!widget->process_key)
673 return widget->process_key(widget, set->form, key);
676 static int widgetset_destructor(void *ptr)
678 struct nc_widgetset *set = ptr;
679 free_form(set->form);
683 struct nc_widgetset *widgetset_create(void *ctx, WINDOW *main, WINDOW *sub)
685 struct nc_widgetset *set;
687 set = talloc_zero(ctx, struct nc_widgetset);
688 set->n_alloc_fields = 8;
691 set->fields = talloc_array(set, FIELD *, set->n_alloc_fields);
692 talloc_set_destructor(set, widgetset_destructor);
697 void widgetset_set_windows(struct nc_widgetset *set,
698 WINDOW *main, WINDOW *sub)
704 void widgetset_set_widget_focus(struct nc_widgetset *set,
705 widget_focus_cb cb, void *arg)
707 set->widget_focus = cb;
708 set->widget_focus_arg = arg;
711 void widgetset_post(struct nc_widgetset *set)
713 struct nc_widget *widget;
716 set->form = new_form(set->fields);
717 set_form_win(set->form, set->mainwin);
718 set_form_sub(set->form, set->subwin);
719 post_form(set->form);
720 form_driver(set->form, REQ_END_FIELD);
722 if (set->cur_field) {
723 set_current_field(set->form, set->cur_field);
724 field = set->cur_field;
727 field = current_field(set->form);
728 widget = field_userptr(field);
729 widget_focus_change(widget, field, true);
730 if (set->widget_focus)
731 set->widget_focus(widget, set->widget_focus_arg);
734 void widgetset_unpost(struct nc_widgetset *set)
736 set->cur_field = current_field(set->form);
737 unpost_form(set->form);
738 free_form(set->form);
742 static void widgetset_add_field(struct nc_widgetset *set, FIELD *field)
744 if (set->n_fields == set->n_alloc_fields - 1) {
745 set->n_alloc_fields *= 2;
746 set->fields = talloc_realloc(set, set->fields,
747 FIELD *, set->n_alloc_fields);
751 set->fields[set->n_fields - 1] = field;
752 set->fields[set->n_fields] = NULL;
755 static void widgetset_remove_field(struct nc_widgetset *set, FIELD *field)
759 for (i = 0; i < set->n_fields; i++) {
760 if (set->fields[i] == field)
764 if (i == set->n_fields)
767 memmove(&set->fields[i], &set->fields[i+i],
768 (set->n_fields - i) * sizeof(set->fields[i]));
772 #define DECLARE_BASEFN(type) \
773 struct nc_widget *widget_ ## type ## _base \
774 (struct nc_widget_ ## type *w) \
775 { return &w->widget; }
777 DECLARE_BASEFN(textbox);
778 DECLARE_BASEFN(checkbox);
779 DECLARE_BASEFN(select);
780 DECLARE_BASEFN(label);
781 DECLARE_BASEFN(button);
783 void widget_set_visible(struct nc_widget *widget, bool visible)
785 if (widget->set_visible)
786 widget->set_visible(widget, visible);
788 field_set_visible(widget->field, visible);
791 void widget_move(struct nc_widget *widget, int y, int x)
794 widget->move(widget, y, x);
796 field_move(widget->field, y, x);
802 int widget_height(struct nc_widget *widget)
804 return widget->height;
807 int widget_width(struct nc_widget *widget)
809 return widget->width;
812 int widget_x(struct nc_widget *widget)
817 int widget_y(struct nc_widget *widget)
822 int widget_focus_y(struct nc_widget *widget)
824 return widget->focus_y;