CUI add empty item
[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 int ked_move_cursor(struct ked *ked, int req)
52 {
53         int result;
54
55         wchgat(ked->scr.sub_ncw, 1, ked_attr_field_selected, 0, 0);
56         result = form_driver(ked->ncf, req);
57         wchgat(ked->scr.sub_ncw, 1, ked->attr_cursor, 0, 0);
58         wrefresh(ked->scr.main_ncw);
59         return result;
60 }
61
62 /**
63  * ked_insert_mode_set - Set the insert mode.
64  */
65
66 static void ked_insert_mode_set(struct ked *ked, int req)
67 {
68         switch (req) {
69         case REQ_INS_MODE:
70                 ked->attr_cursor = ked_attr_cursor_ins;
71                 break;
72         case REQ_OVL_MODE:
73                 ked->attr_cursor = ked_attr_cursor_ovl;
74                 break;
75         default:
76                 assert(0 && "bad req");
77                 break;
78         }
79         ked_move_cursor(ked, req);
80 }
81
82 /**
83  * ked_insert_mode_tog - Toggle the insert mode.
84  */
85
86 static void ked_insert_mode_tog(struct ked *ked)
87 {
88         if (ked->attr_cursor == ked_attr_cursor_ins)
89                 ked_insert_mode_set(ked, REQ_OVL_MODE);
90         else
91                 ked_insert_mode_set(ked, REQ_INS_MODE);
92 }
93
94 /**
95  * ked_move_field - Move selected field, setting correct attributes.
96  * @req: An ncurses request to send to form_driver().
97  */
98
99 static int ked_move_field(struct ked *ked, int req)
100 {
101         int result;
102
103         set_field_back(current_field(ked->ncf), ked_attr_field_normal);
104         result = form_driver(ked->ncf, req);
105         set_field_back(current_field(ked->ncf), ked_attr_field_selected);
106         ked_move_cursor(ked, REQ_END_FIELD);
107         return result;
108 }
109
110 static int ked_post(struct nc_scr *scr)
111 {
112         struct ked *ked = ked_from_scr(scr);
113
114         post_form(ked->ncf);
115
116         nc_scr_frame_draw(scr);
117         ked_move_field(ked, REQ_FIRST_FIELD);
118         ked_move_field(ked, REQ_END_FIELD);
119         ked_insert_mode_set(ked, REQ_INS_MODE);
120
121         redrawwin(ked->scr.main_ncw);
122         wrefresh(ked->scr.main_ncw);
123
124         return 0;
125 }
126
127 static int ked_unpost(struct nc_scr *scr)
128 {
129         return unpost_form(ked_from_scr(scr)->ncf);
130 }
131
132 static void ked_resize(struct nc_scr *scr)
133 {
134         /* FIXME: forms can't be resized, need to recreate here */
135         ked_unpost(scr);
136         ked_post(scr);
137 }
138
139 /**
140  * ked_chomp - Eat leading and trailing WS.
141  */
142
143 static char *ked_chomp(char *s)
144 {
145         char *start;
146         char *end;
147         char *const s_end = s + strlen(s);
148
149         for (; s < s_end; s++)
150                 if (*s != ' ' && *s != '\t')
151                         break;
152
153         start = end = s;
154
155         for (; s < s_end; s++)
156                 if (*s != ' ' && *s != '\t')
157                         end = s;
158         *(end + 1) = 0;
159         return start;
160 }
161
162 static struct pb_kexec_data *ked_prepare_data(struct ked *ked)
163 {
164         struct pb_kexec_data *kd;
165         char *s;
166
167         kd = talloc(ked, struct pb_kexec_data);
168
169         if (!kd)
170                 return NULL;
171
172         s = ked_chomp(field_buffer(ked->fields[0], 0));
173         kd->image = *s ? talloc_strdup(kd, s) : NULL;
174
175         s = ked_chomp(field_buffer(ked->fields[1], 0));
176         kd->initrd = *s ? talloc_strdup(kd, s) : NULL;
177
178         s = ked_chomp(field_buffer(ked->fields[2], 0));
179         kd->args = *s ? talloc_strdup(kd, s) : NULL;
180
181         return kd;
182 }
183
184 /**
185  * ked_process_key - Process a user keystroke.
186  *
187  * Called from the cui via the scr:process_key method.
188  */
189
190 static void ked_process_key(struct nc_scr *scr)
191 {
192         struct ked *ked = ked_from_scr(scr);
193         struct pb_kexec_data *kd;
194
195         while (1) {
196                 int c = getch();
197
198                 if (c == ERR)
199                         return;
200
201                 /* DBGS("%d (%o)\n", c, c); */
202
203                 switch (c) {
204                 default:
205                         ked_move_cursor(ked, c);
206                         break;
207
208                 /* hot keys */
209                 case 27: /* ESC */
210                         ked->on_exit(ked, ked_cancel, NULL);
211                         nc_flush_keys();
212                         return;
213                 case '\n':
214                 case '\r':
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                 /* insert mode */
222                 case KEY_IC:
223                         ked_insert_mode_tog(ked);
224                         break;
225
226                 /* form nav */
227                 case KEY_PPAGE:
228                         ked_move_field(ked, REQ_FIRST_FIELD);
229                         break;
230                 case KEY_NPAGE:
231                         ked_move_field(ked, REQ_LAST_FIELD);
232                         break;
233                 case KEY_DOWN:
234                         ked_move_field(ked, REQ_NEXT_FIELD);
235                         break;
236                 case KEY_UP:
237                         ked_move_field(ked, REQ_PREV_FIELD);
238                         break;
239
240                 /* field nav */
241                 case KEY_HOME:
242                         ked_move_cursor(ked, REQ_BEG_FIELD);
243                         break;
244                 case KEY_END:
245                         ked_move_cursor(ked, REQ_END_FIELD);
246                         break;
247                 case KEY_LEFT:
248                         ked_move_cursor(ked, REQ_LEFT_CHAR);
249                         break;
250                 case KEY_RIGHT:
251                         ked_move_cursor(ked, REQ_RIGHT_CHAR);
252                         break;
253                 case KEY_BACKSPACE:
254                         if (ked_move_cursor(ked, REQ_LEFT_CHAR) == E_OK)
255                                 ked_move_cursor(ked, REQ_DEL_CHAR);
256                         break;
257                 case KEY_DC:
258                         ked_move_cursor(ked, REQ_DEL_CHAR);
259                         break;
260                 }
261         }
262 }
263
264 /**
265  * ked_destructor - The talloc destructor for a ked.
266  */
267
268 static int ked_destructor(void *arg)
269 {
270         struct ked *ked = ked_from_arg(arg);
271         FIELD **f;
272
273         for (f = ked->fields; *f; f++)
274                 free_field(*f);
275
276         free_form(ked->ncf);
277         ked->scr.sig = pb_removed_sig;
278
279         return 0;
280 }
281
282 static FIELD *ked_setup_field(unsigned int y, unsigned int x, char *str)
283 {
284         FIELD *f;
285
286         f = new_field(1, COLS - 1 - x, y, x, 0, 0);
287         field_opts_off(f, O_STATIC | O_WRAP);
288         set_max_field(f, 256);
289         set_field_buffer(f, 0, str);
290         set_field_status(f, 0);
291         return f;
292 }
293
294 static FIELD *ked_setup_label(unsigned int y, unsigned int x, char *str)
295 {
296         FIELD *f;
297
298         f = new_field(1, strlen(str), y, x, 0, 0);
299         field_opts_off(f, O_ACTIVE);
300         set_field_buffer(f, 0, str);
301         return f;
302 }
303
304 struct ked *ked_init(void *ui_ctx, const struct pb_kexec_data *kd,
305         void (*on_exit)(struct ked *, enum ked_result, struct pb_kexec_data *))
306 {
307         struct ked *ked;
308
309         pb_log("%s: image:  '%s'\n", __func__, kd->image);
310         pb_log("%s: initrd: '%s'\n", __func__, kd->initrd);
311         pb_log("%s: args:   '%s'\n", __func__, kd->args);
312
313         assert(on_exit);
314
315         ked = talloc_zero(ui_ctx, struct ked);
316
317         if (!ked)
318                 return NULL;
319
320         talloc_set_destructor(ked, ked_destructor);
321
322         nc_scr_init(&ked->scr, pb_ked_sig, 0, ui_ctx, ked_process_key,
323                 ked_post, ked_unpost, ked_resize);
324
325         ked->scr.frame.title = talloc_strdup(ked, "Petitboot Option Editor");
326         ked->scr.frame.help = talloc_strdup(ked,
327                 "ESC=cancel, Enter=accept");
328
329         ked->on_exit = on_exit;
330
331         ked->fields = talloc_array(ked, FIELD *, 7);
332
333         ked->fields[0] = ked_setup_field(0, 9, kd->image);
334         ked->fields[1] = ked_setup_field(1, 9, kd->initrd);
335         ked->fields[2] = ked_setup_field(2, 9, kd->args);
336         ked->fields[3] = ked_setup_label(0, 1, "image:");
337         ked->fields[4] = ked_setup_label(1, 1, "initrd:");
338         ked->fields[5] = ked_setup_label(2, 1, "args:");
339         ked->fields[6] = NULL;
340
341         ked->ncf = new_form(ked->fields);
342
343         set_form_win(ked->ncf, ked->scr.main_ncw);
344         set_form_sub(ked->ncf, ked->scr.sub_ncw);
345
346         return ked;
347 }