2 * Copyright (c) 2011 Nathan Whitehorn
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35 #include "diskeditor.h"
38 print_partedit_item(WINDOW *partitions, struct partedit_item *items,
39 int item, int nscroll, int selected)
41 chtype attr = A_NORMAL;
43 int y = item - nscroll + 1;
45 wattrset(partitions, selected ? item_selected_attr : item_attr);
46 wmove(partitions, y, MARGIN + items[item].indentation*2);
47 dlg_print_text(partitions, items[item].name, 10, &attr);
48 wmove(partitions, y, 17);
49 wattrset(partitions, item_attr);
51 humanize_number(sizetext, 7, items[item].size, "B", HN_AUTOSCALE,
53 dlg_print_text(partitions, sizetext, 8, &attr);
54 wmove(partitions, y, 25);
55 dlg_print_text(partitions, items[item].type, 15, &attr);
56 wmove(partitions, y, 40);
57 if (items[item].mountpoint != NULL)
58 dlg_print_text(partitions, items[item].mountpoint, 8, &attr);
62 diskeditor_show(const char *title, const char *cprompt,
63 struct partedit_item *items, int nitems, int *selected, int *nscroll)
65 WINDOW *dialog, *partitions;
67 const char *buttons[] =
68 { "Create", "Delete", "Modify", "Revert", "Auto", "Finish", NULL };
69 const char *help_text[] = {
70 "Add a new partition", "Delete selected partition or partitions",
71 "Change partition type or mountpoint",
72 "Revert changes to disk setup", "Use guided partitioning tool",
73 "Exit partitioner (will ask whether to save changes)", NULL };
76 int height, width, min_width;
77 int partlist_height, partlist_width;
80 int cur_button = 5, cur_part = 0;
81 int result = DLG_EXIT_UNKNOWN;
83 static DLG_KEYS_BINDING binding[] = {
85 DLG_KEYS_DATA( DLGK_ENTER, ' ' ),
86 DLG_KEYS_DATA( DLGK_ITEM_NEXT, KEY_DOWN ),
87 DLG_KEYS_DATA( DLGK_ITEM_PREV, KEY_UP ),
88 DLG_KEYS_DATA( DLGK_FIELD_NEXT, KEY_RIGHT ),
89 DLG_KEYS_DATA( DLGK_FIELD_NEXT, TAB ),
90 DLG_KEYS_DATA( DLGK_FIELD_PREV, KEY_BTAB ),
91 DLG_KEYS_DATA( DLGK_FIELD_PREV, KEY_LEFT ),
97 static DLG_KEYS_BINDING binding2[] = {
100 DLG_KEYS_DATA( DLGK_FIELD_NEXT, TAB ),
101 DLG_KEYS_DATA( DLGK_FIELD_PREV, KEY_BTAB ),
102 DLG_KEYS_DATA( DLGK_ITEM_NEXT, CHR_NEXT ),
103 DLG_KEYS_DATA( DLGK_ITEM_NEXT, KEY_DOWN ),
104 DLG_KEYS_DATA( DLGK_ITEM_NEXT, KEY_NEXT ),
105 DLG_KEYS_DATA( DLGK_ITEM_PREV, CHR_PREVIOUS ),
106 DLG_KEYS_DATA( DLGK_ITEM_PREV, KEY_PREVIOUS ),
107 DLG_KEYS_DATA( DLGK_ITEM_PREV, KEY_UP ),
108 DLG_KEYS_DATA( DLGK_PAGE_NEXT, KEY_NPAGE ),
109 DLG_KEYS_DATA( DLGK_PAGE_PREV, KEY_PPAGE ),
114 * Set up editor window.
116 prompt = dlg_strclone(cprompt);
120 partlist_height = 10;
121 dlg_tab_correct_str(prompt);
122 dlg_button_layout(buttons, &min_width);
123 dlg_auto_size(title, prompt, &height, &width, 2, min_width);
124 height += partlist_height;
125 partlist_width = width - 2*MARGIN;
126 dlg_print_size(height, width);
127 dlg_ctl_size(height, width);
129 x = dlg_box_x_ordinate(width);
130 y = dlg_box_y_ordinate(height);
132 dialog = dlg_new_window(height, width, y, x);
133 dlg_register_window(dialog, "diskeditorbox", binding);
134 dlg_register_buttons(dialog, "diskeditorbox", buttons);
136 dlg_draw_box(dialog, 0, 0, height, width, dialog_attr, border_attr);
137 dlg_draw_bottom_box(dialog);
138 dlg_draw_title(dialog, title);
139 wattrset(dialog, dialog_attr);
141 /* Partition list sub-window */
142 partitions = dlg_sub_window(dialog, partlist_height, partlist_width,
144 dlg_register_window(partitions, "partlist", binding2);
145 dlg_register_buttons(partitions, "partlist", buttons);
146 wattrset(partitions, menubox_attr);
148 dlg_item_help(help_text[cur_button]);
149 dlg_draw_buttons(dialog, height - 2*MARGIN, 0, buttons,
150 cur_button, FALSE, width);
151 dlg_print_autowrap(dialog, prompt, height, width);
153 if (selected != NULL)
154 cur_part = *selected;
156 cur_scroll = *nscroll;
157 if (cur_part - cur_scroll >= partlist_height - 2 ||
158 cur_part - cur_scroll < 0)
159 cur_scroll = cur_part;
162 dlg_draw_box(dialog, 3, 1, partlist_height, partlist_width,
163 menubox_border_attr, menubox_attr);
164 for (i = cur_scroll; i < MIN(cur_scroll + partlist_height - 2, nitems);
166 print_partedit_item(partitions, items, i, cur_scroll,
168 if (nitems > partlist_height - 2)
169 dlg_draw_arrows(partitions, cur_scroll > 0,
170 nitems > cur_scroll + partlist_height - 2,
171 partlist_width - 5, 0, partlist_height - 1);
172 wrefresh(partitions);
174 while (result == DLG_EXIT_UNKNOWN) {
175 key = dlg_mouse_wgetch(dialog, &fkey);
176 if ((i = dlg_char_to_button(key, buttons)) >= 0) {
178 dlg_item_help(help_text[cur_button]);
179 dlg_draw_buttons(dialog, height - 2*MARGIN, 0, buttons,
180 cur_button, FALSE, width);
188 case DLGK_FIELD_NEXT:
189 cur_button = dlg_next_button(buttons, cur_button);
192 dlg_item_help(help_text[cur_button]);
193 dlg_draw_buttons(dialog, height - 2*MARGIN, 0, buttons,
194 cur_button, FALSE, width);
196 case DLGK_FIELD_PREV:
197 cur_button = dlg_prev_button(buttons, cur_button);
200 dlg_item_help(help_text[cur_button]);
201 dlg_draw_buttons(dialog, height - 2*MARGIN, 0, buttons,
202 cur_button, FALSE, width);
205 if (cur_part == nitems - 1)
206 break; /* End of list */
208 /* Deselect old item */
209 print_partedit_item(partitions, items, cur_part,
211 /* Select new item */
213 if (cur_part - cur_scroll >= partlist_height - 2) {
214 cur_scroll = cur_part;
217 print_partedit_item(partitions, items, cur_part,
219 wrefresh(partitions);
223 break; /* Start of list */
225 /* Deselect old item */
226 print_partedit_item(partitions, items, cur_part,
228 /* Select new item */
230 if (cur_part - cur_scroll < 0) {
231 cur_scroll = cur_part;
234 print_partedit_item(partitions, items, cur_part,
236 wrefresh(partitions);
239 cur_scroll += (partlist_height - 2);
240 if (cur_scroll + partlist_height - 2 >= nitems)
241 cur_scroll = nitems - (partlist_height - 2);
244 if (cur_part < cur_scroll)
245 cur_part = cur_scroll;
248 cur_scroll -= (partlist_height - 2);
251 if (cur_part >= cur_scroll + partlist_height - 2)
252 cur_part = cur_scroll;
254 case DLGK_PAGE_FIRST:
256 cur_part = cur_scroll;
259 cur_scroll = nitems - (partlist_height - 2);
262 cur_part = cur_scroll;
267 if (is_DLGK_MOUSE(key)) {
268 cur_button = key - M_EVENT;
269 dlg_item_help(help_text[cur_button]);
270 dlg_draw_buttons(dialog, height - 2*MARGIN, 0,
271 buttons, cur_button, FALSE, width);
279 if (selected != NULL)
280 *selected = cur_part;
282 *nscroll = cur_scroll;
284 dlg_del_window(partitions);
285 dlg_del_window(dialog);
286 dlg_mouse_free_regions();