#include <Arduino.h>
#include <ezButton.h>
#include <U8g2lib.h>

// Setup u8g2 display
U8G2_SSD1306_128X64_NONAME_1_SW_I2C u8g2(U8G2_R0, 0, 1); 


// Setup ezButton
ezButton on_off(9);

// ICONS
// Disconnected Bluetooth
static const unsigned char image_Ble_disconnected_15x15_bits[] U8X8_PROGMEM = {0xe0,0x03,0x18,0x0c,0x84,0x10,0x82,0x21,0x92,0x22,0xa1,0x44,0xc1,0x42,0x81,0x41,0xc1,0x42,0xa1,0x44,0x92,0x02,0x82,0x21,0x84,0x10,0x18,0x0c,0xe0,0x03};
// Connected Bluetooth
static const unsigned char image_Ble_connected_15x15_bits[] U8X8_PROGMEM = {0xe0,0x03,0xf8,0x0f,0x7c,0x1f,0x7e,0x3e,0x6e,0x3d,0x5f,0x7b,0x3f,0x7d,0x7f,0x7e,0x3f,0x7d,0x5f,0x7b,0x6e,0x3d,0x7e,0x3e,0x7c,0x1f,0xf8,0x0f,0xe0,0x03};
// Battery Empty u8g2.drawXBMP( 2, 3, 26, 11, image_battery_empty_bits);
static const unsigned char image_battery_empty_bits[] U8X8_PROGMEM = {0xfe,0xff,0x7f,0x00,0x01,0x00,0x80,0x00,0x01,0x00,0x80,0x00,0x01,0x00,0x80,0x03,0x01,0x00,0x80,0x02,0x01,0x00,0x80,0x02,0x01,0x00,0x80,0x02,0x01,0x00,0x80,0x03,0x01,0x00,0x80,0x00,0x01,0x00,0x80,0x00,0xfe,0xff,0x7f,0x00};
// Battery <10% u8g2.drawXBMP( 82, 6, 2, 7, image_low_low_bits);
static const unsigned char image_low_low_bits[] U8X8_PROGMEM = {0x03,0x03,0x03,0x03,0x03,0x03,0x03};
// Battery <25% u8g2.drawXBMP( 4, 5, 4, 7, image__25_bits);
static const unsigned char image__25_bits[] U8X8_PROGMEM = {0x0f,0x0f,0x0f,0x0f,0x0f,0x0f,0x0f};
// Battery <50% u8g2.drawXBMP( 4, 5, 9, 7, image__50_bits);
static const unsigned char image__50_bits[] U8X8_PROGMEM = {0xef,0x01,0xef,0x01,0xef,0x01,0xef,0x01,0xef,0x01,0xef,0x01,0xef,0x01};
// Battery <75% u8g2.drawXBMP( 4, 5, 15, 7, image__75_bits);
static const unsigned char image__75_bits[] U8X8_PROGMEM = {0xef,0x79,0xef,0x79,0xef,0x79,0xef,0x79,0xef,0x79,0xef,0x79,0xef,0x79};
// Battery 100% u8g2.drawXBMP( 4, 5, 20, 7, image__100_bits);
static const unsigned char image__100_bits[] U8X8_PROGMEM = {0xef,0x79,0x0f,0xef,0x79,0x0f,0xef,0x79,0x0f,0xef,0x79,0x0f,0xef,0x79,0x0f,0xef,0x79,0x0f,0xef,0x79,0x0f};
// Loading bar u8g2.drawXBMP( 21, 57, 86, 4, image_loading_bar_empty_bits);
static const unsigned char image_loading_bar_empty_bits[] U8X8_PROGMEM = {0xfe,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0x1f,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x20,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x20,0xfe,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0x1f};


// Start up state
bool enabled = false;

// how often to fire an 'event' in milliseconds
const long interval = 5000;

// Time of last action
unsigned long previousMillis = 0;

// Arrow key identifier
byte arrow_key_id;
byte page_id;
char *arrow_key[4] = {"ARROW UP", "ARROW DOWN", "ARROW LEFT", "ARROW RIGHT"};
char *page_updown[2] = {"PAGE UP", "PAGE DOWN"};
char *alt_tabbed = "ALT TAB";
char *last_action;
int last_x;
int last_y;

void random_arrow_key(){
  int rand = random(4);

  if (rand == 0) {
    arrow_key_id = 0;
	} else if (rand == 1) {
    arrow_key_id = 1;
	} else if (rand == 2) {
    arrow_key_id = 2;
	} else if (rand == 3) {
    arrow_key_id = 3;
	}

  last_action = arrow_key[arrow_key_id];
}

void random_page_up_down(){
  int rand = random(1);

  if (rand == 1){
    page_id = 1;
  } else {
    page_id = 2;
  }

  last_action = page_updown[page_id];
}

void random_alt_tab(){
  last_action = alt_tabbed;
}

void setup() {
  Serial.begin(115200);
  Serial.println("Starting BLE Keyboard");
  Serial.println("Starting BLE Mouse");

  on_off.setDebounceTime(50);
  u8g2.begin();
}

void loop() {

  on_off.loop();

  if(on_off.isReleased()){
    enabled = !enabled;
  }

  unsigned long currentMillis = millis();

  u8g2.firstPage();
  do {
    u8g2.setFont(u8g2_font_profont17_tf);
    if (enabled){
      u8g2.setBitmapMode(1);
      u8g2.drawXBMP( 111, 2, 15, 15, image_Ble_connected_15x15_bits);
    } else {
      u8g2.setBitmapMode(1);
      u8g2.drawXBMP( 111, 2, 15, 15, image_Ble_disconnected_15x15_bits);
    }   

    if (!enabled){
        u8g2.drawStr(28, 40, "DISABLED");
    } else {
      if (currentMillis - previousMillis >= interval){

        previousMillis = millis();

        if (random(2) == 1){
          random_arrow_key();
        } else if (random(3) == 1){
          random_alt_tab();
        } else if (random(2) == 1){
          random_page_up_down();
        }        
      }

      u8g2.drawStr(28, 40, last_action);
    }

    u8g2.drawXBMP( 21, 46, 86, 4, image_loading_bar_empty_bits);
    u8g2.setFont(u8g2_font_squeezed_b6_tr);
    u8g2.drawStr(32, 59, "next action...");
  } while ( u8g2.nextPage() );
}