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