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