#include <Arduino.h>
#ifdef ESP32
  #include <WiFi.h>
#else
  #include <ESP8266WiFi.h>
#endif
#include <time.h>
#include <SPI.h>
#include <Wire.h>
#include <Adafruit_SH110X.h>
#include <Adafruit_GFX.h>
#include <Fonts/FreeSans9pt7b.h>
#include <Fonts/FreeSansBold9pt7b.h>

#define i2c_Address 0x3d
#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 64 // OLED display height, in pixels
#define OLED_RESET -1   //   QT-PY / XIAO
Adafruit_SH1106G display = Adafruit_SH1106G(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);


// 'scrollbar_background', 8x64px
const unsigned char bitmap_scrollbar_background [] PROGMEM = {
  0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 
  0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 
  0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 
  0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x02, 0x00, 0x00
};

// 'item_sel_outline', 128x21px
const unsigned char bitmap_item_sel_outline [] PROGMEM = {
  0x1f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xc0, 
  0x20, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x20, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x30, 
  0x3f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xe0, 
  0x1f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xc0
};

void setup();
void screenA();
void screenB();
void screenC();
void screenD();
void screenE();
void screenF();
void screenG();
void screenH();
void loop();


// ------------------ end generated bitmaps from image2cpp ---------------------------------

const int NUM_ITEMS = 8; // number of items in the list and also the number of screenshots and screenshots with QR codes (other screens)
const int MAX_ITEM_LENGTH = 20; // maximum characters for the item name

char menu_items [NUM_ITEMS] [MAX_ITEM_LENGTH] = {  // array with item names
  { "Screen A" }, 
  { "Screen B" }, 
  { "Screen C" },
  { "Screen D" },
  { "Screen E" },
  { "Screen F" },
  { "Screen G" },
  { "Screen H" }

 };
// note - when changing the order of items above, make sure the other arrays referencing bitmaps
// also have the same order, for example array "bitmap_icons" for icons, and other arrays for screenshots and QR codes

#define BUTTON_UP_PIN 12 // pin for UP button 
#define BUTTON_SELECT_PIN 8 // pin for SELECT button
#define BUTTON_DOWN_PIN 4 // pin for DOWN button

int button_up_clicked = 0; // only perform action when button is clicked, and wait until another press
int button_select_clicked = 0; // same as above
int button_down_clicked = 0; // same as above

int item_selected = 0; // which item in the menu is selected

int item_sel_previous; // previous item - used in the menu screen to draw the item before the selected one
int item_sel_next; // next item - used in the menu screen to draw next item after the selected one

int current_screen = 0;   // 0 = clock, 1 = menu, 2 = screenshot

void setup() {
Serial.begin(115200);
Wire.begin(); 
display.begin(i2c_Address, true); 
delay(2000);
display.clearDisplay();

   // define pins for buttons
  // INPUT_PULLUP means the button is HIGH when not pressed, and LOW when pressed
  // since it´s connected between some pin and GND
  pinMode(BUTTON_UP_PIN, INPUT_PULLUP); // up button
  pinMode(BUTTON_SELECT_PIN, INPUT_PULLUP); // select button
  pinMode(BUTTON_DOWN_PIN, INPUT_PULLUP); // down button
  
}

void screenA(){
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen A");
    }  

void screenB(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen B");
  display.display();
    }  

void screenC(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen C");
  display.display();
    }  
void screenD(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen D");
  display.display();
    } 
void screenE(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen E");
  display.display();
    } 
void screenF(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen F");
  display.display();
    } 
void screenG(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen G");
  display.display();
    } 
void screenH(){  
  display.clearDisplay(); 
  display.setTextColor(SH110X_WHITE);
  display.setTextSize(1);
  display.setCursor(30,32);
  display.print("Screen H");
  display.display();
    }  

void loop() {

  if (current_screen == 0) { // MENU SCREEN

      // up and down buttons only work for the menu screen
      if ((digitalRead(BUTTON_UP_PIN) == LOW) && (button_up_clicked == 0)) { // up button clicked - jump to previous menu item
        item_selected = item_selected - 1; // select previous item
        button_up_clicked = 1; // set button to clicked to only perform the action once
        if (item_selected < 0) { // if first item was selected, jump to last item
          item_selected = NUM_ITEMS-1;
        }
      }
      else if ((digitalRead(BUTTON_DOWN_PIN) == LOW) && (button_down_clicked == 0)) { // down button clicked - jump to next menu item
        item_selected = item_selected + 1; // select next item
        button_down_clicked = 1; // set button to clicked to only perform the action once
        if (item_selected >= NUM_ITEMS) { // last item was selected, jump to first menu item
          item_selected = 0;
          }
      } 

      if ((digitalRead(BUTTON_UP_PIN) == HIGH) && (button_up_clicked == 1)) { // unclick 
        button_up_clicked = 0;
      }
      if ((digitalRead(BUTTON_DOWN_PIN) == HIGH) && (button_down_clicked == 1)) { // unclick
        button_down_clicked = 0;
      }

  }


  if ((digitalRead(BUTTON_SELECT_PIN) == LOW) && (button_select_clicked == 0)) { // select button clicked, jump between screens
     button_select_clicked = 1; // set button to clicked to only perform the action once
     if (current_screen == 0) {current_screen = 1;} // menu items screen --> screenshots screen
     else if (current_screen == 1) {current_screen = 0;} // screenshots scrinteen --> qr codes screen
     else {current_screen = 0;} // qr codes screen --> menu items screen
     Serial.print("Current Screen");
     Serial.println(current_screen);
  }
  if ((digitalRead(BUTTON_SELECT_PIN) == HIGH) && (button_select_clicked == 1)) { // unclick 
    button_select_clicked = 0;
  }

  // set correct values for the previous and next items
  item_sel_previous = item_selected - 1;
  if (item_sel_previous < 0) {item_sel_previous = NUM_ITEMS - 1;} // previous item would be below first = make it the last
  item_sel_next = item_selected + 1;  
  if (item_sel_next >= NUM_ITEMS) {item_sel_next = 0;} // next item would be after last = make it the first

    
//display.clearDisplay();

    if (current_screen == 0) { // MENU SCREEN
       display.clearDisplay();
      // selected item background
      display.drawXBitmap(0, 22, bitmap_item_sel_outline, 128, 21, SH110X_WHITE);

      // draw previous item as icon + label
      display.setTextColor(SH110X_WHITE);
      display.setTextSize(1);
      display.setCursor(25,15);
      display.setFont(&FreeSans9pt7b); 
      display.println(menu_items[item_sel_previous]); 
             

      // draw selected item as icon + label in bold font
      display.setTextSize(1); 
      display.setCursor(25, 15+20+2);
      display.setFont(&FreeSansBold9pt7b);
      display.println(menu_items[item_selected]);   
         

      // draw next item as icon + label
      display.setTextSize(1); 
      display.setCursor(25, 15+20+20+2+2);
      display.setFont(&FreeSans9pt7b); 
      display.println(menu_items[item_sel_next]);   
    

      // draw scrollbar background
      display.drawXBitmap(128-8, 0, bitmap_scrollbar_background, 8, 64, SH110X_WHITE);

      // draw scrollbar handle
      display.drawRect(125, 64/NUM_ITEMS * item_selected, 3, 64/NUM_ITEMS, SH110X_WHITE); 
      
      display.display();
              

    } 
    else if (current_screen == 1) { // SCREENSHOTS SCREEN
        //u8g2.drawStr(0, 0, bitmap_screenshots[item_selected]);
        Serial.print("Item Selected");
        Serial.println (item_selected);
        if (item_selected == 0){
          screenA();
        }
        else if(item_selected == 1){
          screenB();
        }
        else if(item_selected == 2){
          screenC(); 
          }
        else if(item_selected == 3){
          screenD(); 
          }
        else if(item_selected == 4){
          screenE(); 
          } 
        else if(item_selected == 5){
          screenF(); 
          } 
        else if(item_selected == 6){
          screenG(); 
          } 
         else if(item_selected == 7){
          screenH(); 
          }          
        //else if (current_screen == 2) { // QR SCREEN
          //screenB();

        //}

        
    }
}