Add ncurses UI support
[petitboot] / ui / ncurses / nc-ked.c
1 /*
2  *  Copyright (C) 2009 Sony Computer Entertainment Inc.
3  *  Copyright 2009 Sony Corp.
4  *
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; version 2 of the License.
8  *
9  *  This program is distributed in the hope that it will be useful,
10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *  GNU General Public License for more details.
13  *
14  *  You should have received a copy of the GNU General Public License
15  *  along with this program; if not, write to the Free Software
16  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17  */
18
19 #define _GNU_SOURCE
20
21 #include <assert.h>
22 #include <string.h>
23
24 #include "log/log.h"
25 #include "talloc/talloc.h"
26 #include "nc-ked.h"
27
28 static struct ked *ked_from_scr(struct nc_scr *scr)
29 {
30         struct ked *ked;
31
32         assert(scr->sig == pb_ked_sig);
33         ked = (struct ked *)((char *)scr - (size_t)&((struct ked *)0)->scr);
34         assert(ked->scr.sig == pb_ked_sig);
35         return ked;
36 }
37
38 static struct ked *ked_from_arg(void *arg)
39 {
40         struct ked *ked = arg;
41
42         assert(ked->scr.sig == pb_ked_sig);
43         return ked;
44 }
45
46 /**
47  * ked_move_cursor - Move the cursor, setting correct attributes.
48  * @req: An ncurses request or char to send to form_driver().
49  */
50
51 static void ked_move_cursor(struct ked *ked, int req)
52 {
53         wchgat(ked->scr.sub_ncw, 1, ked_attr_field_selected, 0, 0);
54         form_driver(ked->ncf, req);
55         wchgat(ked->scr.sub_ncw, 1, ked->attr_cursor, 0, 0);
56         wrefresh(ked->scr.main_ncw);
57 }
58
59 /**
60  * ked_insert_mode_set - Set the insert mode.
61  */
62
63 static void ked_insert_mode_set(struct ked *ked, int req)
64 {
65         switch (req) {
66         case REQ_INS_MODE:
67                 ked->attr_cursor = ked_attr_cursor_ins;
68                 break;
69         case REQ_OVL_MODE:
70                 ked->attr_cursor = ked_attr_cursor_ovl;
71                 break;
72         default:
73                 assert(0 && "bad req");
74                 break;
75         }
76         ked_move_cursor(ked, req);
77 }
78
79 /**
80  * ked_insert_mode_tog - Toggle the insert mode.
81  */
82
83 static void ked_insert_mode_tog(struct ked *ked)
84 {
85         if (ked->attr_cursor == ked_attr_cursor_ins)
86                 ked_insert_mode_set(ked, REQ_OVL_MODE);
87         else
88                 ked_insert_mode_set(ked, REQ_INS_MODE);
89 }
90
91 /**
92  * ked_move_field - Move selected field, setting correct attributes.
93  * @req: An ncurses request to send to form_driver().
94  */
95
96 static void ked_move_field(struct ked *ked, int req)
97 {
98         set_field_back(current_field(ked->ncf), ked_attr_field_normal);
99         form_driver(ked->ncf, req);
100         set_field_back(current_field(ked->ncf), ked_attr_field_selected);
101         ked_move_cursor(ked, REQ_END_FIELD);
102 }
103
104 static int ked_post(struct nc_scr *scr)
105 {
106         struct ked *ked = ked_from_scr(scr);
107
108         post_form(ked->ncf);
109
110         nc_scr_frame_draw(scr);
111         ked_move_field(ked, REQ_FIRST_FIELD);
112         ked_move_field(ked, REQ_END_FIELD);
113         ked_insert_mode_set(ked, REQ_INS_MODE);
114
115         redrawwin(ked->scr.main_ncw);
116         wrefresh(ked->scr.main_ncw);
117
118         return 0;
119 }
120
121 static int ked_unpost(struct nc_scr *scr)
122 {
123         return unpost_form(ked_from_scr(scr)->ncf);
124 }
125
126 static void ked_resize(struct nc_scr *scr)
127 {
128         /* FIXME: forms can't be resized, need to recreate here */
129         ked_unpost(scr);
130         ked_post(scr);
131 }
132
133 /**
134  * ked_chomp - Eat leading and trailing WS.
135  */
136
137 static char *ked_chomp(char *s)
138 {
139         char *start;
140         char *end;
141         char *const s_end = s + strlen(s);
142
143         for (; s < s_end; s++)
144                 if (*s != ' ' && *s != '\t')
145                         break;
146         start = s;
147
148         for (++s; s < s_end; s++)
149                 if (*s != ' ' && *s != '\t')
150                         end = s;
151         *(end + 1) = 0;
152         return start;
153 }
154
155 static struct pb_kexec_data *ked_prepare_data(struct ked *ked)
156 {
157         struct pb_kexec_data *kd;
158         char *s;
159
160         kd = talloc(ked, struct pb_kexec_data);
161
162         if (!kd)
163                 return NULL;
164
165         s = ked_chomp(field_buffer(ked->fields[0], 0));
166         kd->image = *s ? talloc_strdup(kd, s) : NULL;
167
168         s = ked_chomp(field_buffer(ked->fields[1], 0));
169         kd->initrd = *s ? talloc_strdup(kd, s) : NULL;
170
171         s = ked_chomp(field_buffer(ked->fields[2], 0));
172         kd->args = *s ? talloc_strdup(kd, s) : NULL;
173
174         return kd;
175 }
176
177 /**
178  * ked_process_key - Process a user keystroke.
179  *
180  * Called from the cui via the scr:process_key method.
181  */
182
183 static void ked_process_key(struct nc_scr *scr)
184 {
185         struct ked *ked = ked_from_scr(scr);
186
187         while (1) {
188                 int c = getch();
189
190                 switch (c) {
191                 default:
192                         ked_move_cursor(ked, c);
193                 break;
194                 case ERR:
195                         return;
196
197                 /* hot keys */
198                 case 2: { /* CTRL-B */
199                         struct pb_kexec_data *kd;
200
201                         form_driver(ked->ncf, REQ_VALIDATION);
202                         kd = ked_prepare_data(ked);
203                         ked->on_exit(ked, ked_boot, kd);
204                         nc_flush_keys();
205                         return;
206                 }
207                 case 27: /* ESC */
208                         ked->on_exit(ked, ked_cancel, NULL);
209                         nc_flush_keys();
210                         return;
211                 case '\n':
212                 case '\r': {
213                         struct pb_kexec_data *kd;
214
215                         form_driver(ked->ncf, REQ_VALIDATION);
216                         kd = ked_prepare_data(ked);
217                         ked->on_exit(ked, ked_update, kd);
218                         nc_flush_keys();
219                         return;
220                 }
221
222                 /* insert mode */
223                 case KEY_IC:
224                         ked_insert_mode_tog(ked);
225                         break;
226
227                 /* form nav */
228                 case KEY_PPAGE:
229                         ked_move_field(ked, REQ_FIRST_FIELD);
230                         break;
231                 case KEY_NPAGE:
232                         ked_move_field(ked, REQ_LAST_FIELD);
233                         break;
234                 case KEY_DOWN:
235                         ked_move_field(ked, REQ_NEXT_FIELD);
236                         break;
237                 case KEY_UP:
238                         ked_move_field(ked, REQ_PREV_FIELD);
239                         break;
240
241                 /* field nav */
242                 case KEY_HOME:
243                         ked_move_cursor(ked, REQ_BEG_FIELD);
244                         break;
245                 case KEY_END:
246                         ked_move_cursor(ked, REQ_END_FIELD);
247                         break;
248                 case KEY_LEFT:
249                         ked_move_cursor(ked, REQ_LEFT_CHAR);
250                         break;
251                 case KEY_RIGHT:
252                         ked_move_cursor(ked, REQ_RIGHT_CHAR);
253                         break;
254                 case KEY_BACKSPACE:
255                         ked_move_cursor(ked, REQ_LEFT_CHAR);
256                         ked_move_cursor(ked, REQ_DEL_CHAR);
257                         break;
258                 case KEY_DC:
259                         ked_move_cursor(ked, REQ_DEL_CHAR);
260                         break;
261                 }
262         }
263 }
264
265 /**
266  * ked_destructor - The talloc destructor for a ked.
267  */
268
269 static int ked_destructor(void *arg)
270 {
271         struct ked *ked = ked_from_arg(arg);
272         FIELD **f;
273
274         for (f = ked->fields; *f; f++)
275                 free_field(*f);
276
277         free_form(ked->ncf);
278         ked->scr.sig = pb_removed_sig;
279
280         return 0;
281 }
282
283 static FIELD *ked_setup_field(unsigned int y, unsigned int x, char *str)
284 {
285         FIELD *f;
286
287         f = new_field(1, COLS - 1 - x, y, x, 0, 0);
288         field_opts_off(f, O_STATIC | O_WRAP);
289         set_max_field(f, 256);
290         set_field_buffer(f, 0, str);
291         set_field_status(f, 0);
292         return f;
293 }
294
295 static FIELD *ked_setup_label(unsigned int y, unsigned int x, char *str)
296 {
297         FIELD *f;
298
299         f = new_field(1, strlen(str), y, x, 0, 0);
300         field_opts_off(f, O_ACTIVE);
301         set_field_buffer(f, 0, str);
302         return f;
303 }
304
305 struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd,
306         void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *))
307 {
308         struct ked *ked;
309
310         pb_log("%s: image:  '%s'\n", __func__, kd->image);
311         pb_log("%s: initrd: '%s'\n", __func__, kd->initrd);
312         pb_log("%s: args:   '%s'\n", __func__, kd->args);
313
314         assert(on_exit);
315
316         ked = talloc_zero(ui_ctx, struct ked);
317
318         if (!ked)
319                 return NULL;
320
321         talloc_set_destructor(ked, ked_destructor);
322
323         nc_scr_init(&ked->scr, pb_ked_sig, 0, ui_ctx, ked_process_key,
324                 ked_post, ked_unpost, ked_resize);
325
326         ked->scr.frame.title = talloc_strdup(ked, "Petitboot Option Editor");
327         ked->scr.frame.help = talloc_strdup(ked,
328                 "ESC=cancel, Enter=accept, Ctrl-b=boot");
329
330         ked->on_exit = on_exit;
331
332         ked->fields = talloc_array(ked, FIELD *, 7);
333
334         ked->fields[0] = ked_setup_field(0, 9, kd->image);
335         ked->fields[1] = ked_setup_field(1, 9, kd->initrd);
336         ked->fields[2] = ked_setup_field(2, 9, kd->args);
337         ked->fields[3] = ked_setup_label(0, 1, "image:");
338         ked->fields[4] = ked_setup_label(1, 1, "initrd:");
339         ked->fields[5] = ked_setup_label(2, 1, "args:");
340         ked->fields[6] = NULL;
341
342         ked->ncf = new_form(ked->fields);
343
344         set_form_win(ked->ncf, ked->scr.main_ncw);
345         set_form_sub(ked->ncf, ked->scr.sub_ncw);
346
347         return ked;
348 }