#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include "MenuLib.h"

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire);

const char* menuItems[] = {
  "Start",
  "Settings",
  "Info",
  "Shutdown",
  "Extra 1",
  "Extra 2",
  "Extra 3",
  "Extra 4",
  "Extra 5"
};
const char* menu2Items[] = {
  "1",
  "2"
};

int analogButtonHandler() {
  int val = analogRead(A0);
  if (val < 100) return 1;   // Up
  if (val < 300) return 2;   // Down
  if (val < 600) return 3;   // Select
  if (val < 900) return 4;   // Back
  return 0;
}

void setup() {
  Serial.begin(9600);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();

  SimpleMenu::begin(4, 5, 3, 6);          // Up, Down, Back, Select
  SimpleMenu::setDisplay(display);
  SimpleMenu::setTextSize(1);             // 8px text height
  SimpleMenu::setDebounce(300);           // Optional debounce override
SimpleMenu::setInputHandler(analogButtonHandler);
}

void loop() {
  int result = SIMPLE_MENU_SHOW("Main Menu", menuItems, 2); // Start at "Info"

  if (result == -1) {
    Serial.println("Back selected");
    delay(500);
    return;
  }

  Serial.print("You selected: ");
  Serial.println(menuItems[result]);
if (result == 2){
  Serial.println(menu2Items[SIMPLE_MENU_SHOW("sub Menu", menu2Items, 0)]);
}
  delay(1000);
}