menu.c

   1
   2 #include <stdlib.h>
   3 #include <stddef.h>
   4 #include <string.h>
   5 #include <stdbool.h>
   6 #include <ncurses.h>
   7
   8 #include "headers/menu.h"
   9 #include "headers/render.h"
  10 #include "headers/utils.h"
  11
  12 #define MENU_PAD_TOP 2
  13 #define MENU_PAD_BOTTOM 1
  14 #define MENU_PAD_LEFT 2
  15 #define MENU_PAD_RIGHT 1
  16
  17 #define MAX_CONTENTS_LENGTH 256
  18
  19 typedef struct MenuItem {
  20     char* contents;
  21     char* description;
  22 } MenuItem;
  23
  24 typedef struct Menu {
  25     char* menu_name;
  26     MenuItem** menu_items;
  27     int menu_length;
  28     int selected_item;
  29     int scroll_offset;
  30     bool focused;
  31     WINDOW* menu_win;
  32     WINDOW* sub_win;
  33     int max_height;
  34     int max_width;
  35 } Menu;
  36
  37 int swap_item(Menu* menu, int src_index, int dest_index);
  38
  39 /* insert mode */
  40 int menu_insert_mode(Menu* menu, int insert_index);
  41
  42 /* prob temp for now */
  43 MenuItem* create_blank_menuitem(void);
  44
  45
  46 MenuItem*
  47 create_menuitem(char* contents)
  48 {
  49     MenuItem* new_menuitem;
  50
  51     new_menuitem = malloc(sizeof(MenuItem));
  52     new_menuitem->contents = contents;
  53
  54     return new_menuitem;
  55 }
  56
  57 MenuItem*
  58 create_blank_menuitem(void)
  59 {
  60     return create_menuitem(strdup(""));
  61 }
  62
  63 Menu*
  64 create_menu(char* menu_name, MenuItem** item_list)
  65 {
  66     Menu* new_menu;
  67
  68     new_menu = malloc(sizeof(Menu));
  69     new_menu->menu_name = menu_name;
  70     new_menu->menu_items = item_list;
  71     new_menu->menu_length = array_length(MenuItem*, item_list);
  72     new_menu->selected_item = 0;
  73     new_menu->scroll_offset = 0;
  74     new_menu->focused = false;
  75     set_menu_win(new_menu, stdscr);
  76
  77     return new_menu;
  78 }
  79
  80 WINDOW*
  81 get_menu_win(Menu* menu)
  82 {
  83     return menu->menu_win;
  84 }
  85
  86 int
  87 set_menu_win(Menu* menu, WINDOW* win)
  88 {
  89     int height, width;
  90
  91     menu->menu_win = win;
  92     getmaxyx(menu->menu_win, height, width);
  93
  94     /* create a subwin (also prob free old subwin?) */
  95     menu->max_height = height-MENU_PAD_TOP-MENU_PAD_BOTTOM;
  96     menu->max_width = width-MENU_PAD_LEFT-MENU_PAD_RIGHT;
  97     menu->sub_win = derwin(
  98             menu->menu_win,
  99             menu->max_height,
 100             menu->max_width,
 101             MENU_PAD_TOP,
 102             MENU_PAD_LEFT
 103     );
 104
 105     return 0;
 106 }
 107
 108 WINDOW*
 109 get_menu_subwin(Menu* menu)
 110 {
 111     return menu->sub_win;
 112 }
 113
 114 MenuItem*
 115 get_menu_item(Menu* menu, int index)
 116 {
 117     if (index < 0 || index >= menu->menu_length) return NULL;
 118
 119     return menu->menu_items[index];
 120 }
 121
 122 int
 123 set_menu_focus(Menu* menu, bool focus)
 124 {
 125     menu->focused = focus;
 126
 127     return 0;
 128 }
 129
 130 int
 131 get_selected_item(Menu* menu)
 132 {
 133     return menu->selected_item;
 134 }
 135
 136 int
 137 set_selected_item(Menu* menu, int selected_item)
 138 {
 139     menu->selected_item = selected_item;
 140
 141     return 0;
 142 }
 143
 144 int
 145 get_menu_length(Menu* menu)
 146 {
 147     return menu->menu_length;
 148 }
 149
 150 char*
 151 get_menu_name(Menu* menu)
 152 {
 153     return menu->menu_name;
 154 }
 155
 156 int
 157 swap_item(Menu* menu, int src_index, int dest_index)
 158 {
 159     ar_swap_item(menu->menu_items, src_index, dest_index);
 160
 161     return 0;
 162 }
 163
 164 int
 165 delete_item(Menu* menu, int index)
 166 {
 167     if (index < 0 || index > menu->menu_length-1) return -1;
 168
 169     for (int i = index; i <= menu->menu_length-1; i++) {
 170         menu->menu_items[i] = menu->menu_items[i+1];
 171     }
 172
 173     menu->menu_items = realloc(menu->menu_items, menu->menu_length*sizeof(MenuItem*));
 174     menu->menu_items[menu->menu_length-1] = 0; // preserve null at end
 175
 176     menu->menu_length -= 1;
 177
 178     /* also move the current selected position if it's last */
 179     if (menu->selected_item > menu->menu_length-1) {
 180         menu->selected_item = menu->menu_length-1;
 181     }
 182
 183     wclear(menu->sub_win);
 184
 185     return 0;
 186 }
 187
 188 int
 189 insert_item(Menu* menu, MenuItem* menuitem, int index)
 190 { // note, this func does not validate index
 191
 192     /* resize array and insert */
 193     menu->menu_items = realloc(menu->menu_items, (menu->menu_length+2)*sizeof(MenuItem*));
 194
 195     for (int i = menu->menu_length; i > index; i--) {
 196         menu->menu_items[i] = menu->menu_items[i-1];
 197     }
 198
 199     menu->menu_items[index] = menuitem;
 200     menu->menu_items[menu->menu_length+1] = 0; // remember null at end
 201     menu->menu_length += 1;
 202
 203     /* move cursor pos */
 204     menu->selected_item = index;
 205
 206     return 0;
 207 }
 208
 209 int
 210 menu_insert_mode(Menu* menu, int insert_index)
 211 {
 212     char temp[MAX_CONTENTS_LENGTH+1]; // remember null
 213     char* new_contents;
 214
 215     /* this is bad */
 216     wclear(menu->sub_win);
 217     render_menu(menu);
 218
 219     curs_on();
 220
 221     /* move cursor to right spot */
 222     ungetstr(menu->menu_items[insert_index]->contents);
 223     mvwgetnstr(menu->sub_win,
 224         insert_index, // account for wrap later too
 225         0,
 226         temp,
 227         MAX_CONTENTS_LENGTH
 228     );
 229     curs_off();
 230
 231     /* copy out */
 232     new_contents = strdup(temp);
 233     menu->menu_items[insert_index]->contents = new_contents;
 234
 235     /* delete if empty - maybe move this to a cleanup stage */
 236     if (strlen(new_contents) == 0) {
 237         delete_item(menu, insert_index);
 238     }
 239
 240     return 0;
 241 }
 242
 243 int
 244 menu_driver(Menu* menu, MenuAction action)
 245 {
 246
 247     switch (action) {
 248         case MENU_UP:
 249             menu->selected_item = menu->selected_item-1 >= 0 ? menu->selected_item-1 : 0;
 250             break;
 251
 252         case MENU_DOWN:
 253             menu->selected_item = menu->selected_item+1 <= menu->menu_length-1 ? menu->selected_item+1 : menu->menu_length-1;
 254             break;
 255
 256         case MENU_TOP:
 257             menu->selected_item = 0;
 258             break;
 259
 260         case MENU_BOTTOM:
 261             menu->selected_item = menu->menu_length-1;
 262             break;
 263
 264         case MENU_MOVE_UP:
 265             if (menu->selected_item <= 0) break;
 266             swap_item(menu, menu->selected_item, menu->selected_item-1);
 267             menu->selected_item -= 1;
 268             break;
 269
 270         case MENU_MOVE_DOWN:
 271             if (menu->selected_item >= menu->menu_length-1) break;
 272             swap_item(menu, menu->selected_item, menu->selected_item+1);
 273             menu->selected_item += 1;
 274             break;
 275
 276         case MENU_DELETE:
 277             delete_item(menu, menu->selected_item);
 278             break;
 279
 280         case MENU_APPEND:
 281             insert_item(menu, create_blank_menuitem(), menu->menu_length);
 282             menu_insert_mode(menu, menu->selected_item);
 283             break;
 284
 285         case MENU_INSERT_ABOVE:
 286             insert_item(menu, create_blank_menuitem(), menu->selected_item);
 287             menu_insert_mode(menu, menu->selected_item);
 288             break;
 289
 290         case MENU_INSERT_BELOW:
 291             insert_item(menu, create_blank_menuitem(), menu->selected_item+1);
 292             menu_insert_mode(menu, menu->selected_item); // inserted item is cur now
 293             break;
 294
 295         case MENU_EDIT:
 296             menu_insert_mode(menu, menu->selected_item);
 297             break;
 298
 299         default: // This is here for debug, disable later
 300             fprintf(stderr, "Invalid menu action");
 301     }
 302
 303     return 0;
 304 }
 305
 306 int
 307 render_menu(Menu* menu)
 308 {
 309     /* draw outer menu (prob dont need this every render) */
 310     int menu_header_color;
 311
 312     /* menu_header_color = */
 313     wattron(menu->menu_win, COLOR_PAIR(
 314         (menu->focused == true) ?
 315         TS_MENU_SELECTED: TS_MENU_NONSELECTED
 316     ));
 317     mvwprintw(menu->menu_win, 0, MENU_PAD_LEFT, menu->menu_name);
 318     wattroff(menu->menu_win, COLOR_PAIR(0));
 319
 320     /* draw inner menu */
 321     int cur_line = 0;
 322     for (int i = 0; i < menu->menu_length-menu->scroll_offset; i++) {
 323
 324         int wrapped_lines;
 325         char* wrapped_text;
 326
 327         /* wrap text by inserting newlines (maxwidth-1 for newline char)*/
 328         wrapped_text = wrap_text(menu->menu_items[i]->contents, menu->max_width-1, &wrapped_lines);
 329
 330         /* color selected item */
 331         wattron(menu->sub_win, COLOR_PAIR(
 332            (i == menu->selected_item && menu->focused == true) ?
 333            TS_SELECTED : TS_NONSELECTED
 334         ));
 335         mvwprintw(menu->sub_win, cur_line, 0, wrapped_text);
 336         wattroff(menu->sub_win, COLOR_PAIR(0));
 337
 338         cur_line += wrapped_lines;
 339
 340         /* display number of items */
 341         /* if (menu->menu_items[i]->) */
 342
 343         free(wrapped_text);
 344
 345     }
 346
 347     wrefresh(menu->sub_win);
 348     wrefresh(menu->menu_win);
 349
 350     return 0;
 351 }
 352
 353 int
 354 free_menu(Menu* menu)
 355 {
 356     return 0;
 357 }
 358