1 files changed, 40 insertions, 24 deletions

diff --git a/src/screen.c b/src/screen.c
index 9cdac02..5ca2dc2 100644
--- a/src/screen.c
+++ b/src/screen.c
@@ -31,6 +31,7 @@ SLOTS HERE
 #include <menu.h>
 
 #include <stdio.h>
+#include <stdlib.h>
 
 static const char *phrases[] = {
     // By @syxhe on telegram 
@@ -78,8 +79,7 @@ static const char *phrases[] = {
 static const char *menu_choices[] = {
     "Spin",
     "Buy spins",
-    "Quit",
-    NULL
+    "Quit"
 };
 
 static float normalize(float value, float oldmin, float oldmax, float newmin, float newmax) {
@@ -115,16 +115,17 @@ int main() {
             // curses colors made this go away. Also VSCode's terminal theming is fucking with it as well, so
             // use a normal terminal for accurate colors */
         
-        init_rgb_color(CC_RED, 255, 0, 0);
-        init_rgb_color(CC_ORANGE, 255, 128, 0);
-        init_rgb_color(CC_YELLOW, 255, 255, 0);
-        init_rgb_color(CC_GREEN, 0, 255, 0);
-        init_rgb_color(CC_BLUE, 0, 0, 255);
-        init_rgb_color(CC_PURPLE, 255, 0, 255);
-        init_rgb_color(CC_WHITE, 255, 255, 255);
-
-        init_pair(1, CC_WHITE, CURSES_BLUE);
-
+        init_rgb_color(CC_RED,      255,    0,      0);
+        init_rgb_color(CC_ORANGE,   255,    128,    0);
+        init_rgb_color(CC_YELLOW,   255,    255,    0);
+        init_rgb_color(CC_GREEN,    0,      255,    0);
+        init_rgb_color(CC_BLUE,     0,      0,      255);
+        init_rgb_color(CC_PURPLE,   155,    0,      255);
+        init_rgb_color(CC_MAGENTA,  255,    0,      255);
+        init_rgb_color(CC_WHITE,    255,    255,    255);
+
+        init_pair(CCP_BANNER, CC_WHITE, CURSES_BLUE);
+        init_pair(CCP_TESTING, CC_RED, CC_WHITE);
     }
 
     cbreak();               // Disables character buffering
@@ -134,7 +135,8 @@ int main() {
 
     int row = -1, col = -1;
     getmaxyx(stdscr, row, col);
-
+    if(row < 0 || col < 0)
+        error(1, 0, "[VX-GAMBLEGROUND] Couldn't get max terminal size for some reason");
 
     int randomnum = randombytes_uniform(STATIC_ARRSIZE(phrases));
     if(randomnum < 0)
@@ -146,37 +148,43 @@ int main() {
     if(phrase == NULL)
         error(1, errno, "[VX-GAMBLEGROUND] Could not create banner window");
 
-    wbkgd(phrase, COLOR_PAIR(1));
+    if(has_colors())
+        wbkgd(phrase, COLOR_PAIR(1));
+    
     mvwaddstr(phrase, 0, 1, "VX-GAMBLEGROUND: ");
     waddstr(phrase, phrases[randomnum]);
     wrefresh(phrase);
 
-    //* Dummy test for fun. Shows how windows can't be overlapping or things get weird
+    /* Dummy test for fun. Shows how windows can't be overlapping or things get weird (and that row offsets matter)
     WINDOW *test = newwin(row - 1, col, 1, 0);
+    int rand1, rand2;
+
     for(int c = wgetch(test);; c = wgetch(test)) {
-        mvwaddch(test, c % row, c % col, c | (A_BOLD * (randombytes_random() % 2 > 0)));
+        rand1 = randombytes_uniform(row - 1);
+        rand2 = randombytes_uniform(col);
+
+        mvwaddch(test, rand1, rand2, c | (A_BOLD * (randombytes_random() % 2 > 0)));
     }
     //*/
 
-    /* Menu example
-    ITEM **items;
+    //* Menu example (fuck menus got damn)
+    ITEM *items[STATIC_ARRSIZE(menu_choices) + 1];
     MENU *menu;
 
-    items = calloc(STATIC_ARRSIZE(menu_choices) + 1, sizeof(ITEM*));
-    if(items == NULL)
-        error(-1, errno, "Could not allocate space for menu items");
-
     for(long unsigned int i = 0; i < STATIC_ARRSIZE(menu_choices); i++)
         items[i] = new_item(menu_choices[i], menu_choices[i]);
     items[STATIC_ARRSIZE(menu_choices)] = NULL;
 
     menu = new_menu(items);
-    mvprintw(LINES - 2, 0, "F1 to Exit");
+    if(menu == NULL)
+        error(1, errno, "Could not create menu");
+
+    // set_menu_format(menu, 3, col);
     post_menu(menu);
     refresh();
 
     int c;
-    while((c = getch()) != KEY_F(1)) {
+    while((c = getch()) != KEY_F(4)) {
         switch(c) {
             case KEY_DOWN:
                 menu_driver(menu, REQ_DOWN_ITEM);
@@ -186,6 +194,14 @@ int main() {
                 menu_driver(menu, REQ_UP_ITEM);
                 break;
 
+            case KEY_LEFT:
+                menu_driver(menu, REQ_LEFT_ITEM);
+                break;
+
+            case KEY_RIGHT:
+                menu_driver(menu, REQ_RIGHT_ITEM);
+                break;
+
             case KEY_ENTER:
                 break;
         }