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