#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 64 // OLED display height, in pixels

// Declaration for SSD1306 display connected using I2C
#define OLED_RESET     -1 // Reset pin # (or -1 if sharing Arduino reset pin)
#define SCREEN_ADDRESS 0x3C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

#define BLINK_INTERVAL 1000 // Blink interval in milliseconds

// Button pin
#define BUTTON_PIN 2

// States for eye blinking
enum EyeState {
  OPEN,
  CLOSED
};

EyeState eyeState = OPEN;
unsigned long lastBlinkTime = 500;

void setup() {
  Serial.begin(9600);
  pinMode(BUTTON_PIN, INPUT_PULLUP);

  // Initialize the OLED object
  if (!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;); // Don't proceed, loop forever
  }
  // Clear the buffer.
  display.clearDisplay();
  display.display();
  delay(3000);
}

const unsigned char image_bluetooth_bits[] PROGMEM = {
  0x01,0x00,0x02,0x80,0x02,0x40,0x22,0x20,0x12,0x20,0x0a,
  0x40,0x06,0x80,0x03,0x00,0x06,0x80,0x0a,0x40,0x12,0x20,
  0x22,0x20,0x02,0x40,0x02,0x80,0x01,0x00,0x00,0x00};

const unsigned char image_device_key_retro_bits[] PROGMEM = {
  0x00,0x70,0x00,0x88,0x01,0x24,0x01,0x52,0x01,0x2a,0x00,0x92,
  0x01,0x44,0x02,0xb8,0x05,0x00,0x0a,0x00,0x14,0x00,0x28,0x00,
  0x54,0x00,0xa2,0x00,0x54,0x00,0x08,0x00};


const unsigned char image_bluetooth_connected_bits[] PROGMEM = {
  0x01,0x00,0x02,0x80,0x02,0x40,0x22,0x20,0x12,0x20,0x4a,0x48,0x26,
  0x90,0x33,0x30,0x26,0x90,0x4a,0x48,0x12,0x20,0x22,0x20,0x02,0x40,
  0x02,0x80,0x01,0x00,0x00,0x00};

const unsigned char image_hand_thumbs_up_bits[] PROGMEM = {
  0x00,0x10,0x00,0x28,0x00,0x28,0x00,0x48,0x00,0x50,0x00,0x90,
  0x01,0x3e,0xfa,0x03,0x8c,0x05,0x88,0x03,0x88,0x05,0x88,0x03,
  0x88,0x05,0xa8,0x02,0x8e,0x06,0xf9,0xfc};

const unsigned char PROGMEM image_network_bits[] = {0x00,0x0e,
0x00,0x0a,0x00,0x0a,0x00,0x0a,0x00,0xea,0x00,0xaa,0x00,0xaa,0x00,
0xaa,0x0e,0xaa,0x0a,0xaa,0x0a,0xaa,0x0a,0xaa,0xea,0xaa,0xaa,0xaa,
0xee,0xee,0x00,0x00};

const unsigned char PROGMEM image_notification_bell_bits[] = {0x11,
0x88,0x63,0xc6,0x44,0x22,0x8c,0x11,0x88,0x11,0x10,0x08,0x10,0x08,0x10,
0x08,0x10,0x08,0x20,0x04,0x20,0x04,0x40,0x02,0xff,0xff,0x06,0x60,0x03,
0xc0,0x00,0x00};

const unsigned char PROGMEM image_Pin_arrow_down_bits[] = {0x38,0x38,
0x38,0x38,0x38,0xfe,0x7c,0x38,0x10};

const unsigned char PROGMEM image_Pin_arrow_left_bits[] = {0x10,0x00,
0x30,0x00,0x7f,0x80,0xff,0x80,0x7f,0x80,0x30,0x00,0x10,0x00};

const unsigned char PROGMEM image_Pin_arrow_right_bits[] = {0x04,0x00,
0x06,0x00,0xff,0x00,0xff,0x80,0xff,0x00,0x06,0x00,0x04,0x00};

const unsigned char PROGMEM image_Pin_arrow_up_bits[] = {0x10,0x38,0x7c,
0xfe,0x38,0x38,0x38,0x38,0x38};

void loop() {
  // Check if it's time to blink the eyes
  if (millis() - lastBlinkTime >= BLINK_INTERVAL) {
    if (eyeState == OPEN) {
      // Close eyes
      display.clearDisplay();
// display.drawRect(76, 5, 43, 26, WHITE);
// display.fillCircle(87, 16, 6, WHITE);
// display.drawLine(14, 50, 68, 58, WHITE);
// display.drawLine(69, 58, 117, 48, WHITE);
// display.drawRect(6, 5, 43, 26, WHITE);
// display.drawLine(14, 49, 68, 41, WHITE);
// display.fillCircle(17, 15, 6, WHITE);
// display.drawLine(118, 47, 69, 40, WHITE);
display.drawBitmap(5, 3, image_bluetooth_bits, 14, 16, WHITE);
display.drawBitmap(3, 29, image_device_key_retro_bits, 15, 16,WHITE);
display.drawBitmap(57, 4, image_notification_bell_bits, 16, 16, WHITE);
display.drawBitmap(100, 47, image_Pin_arrow_down_bits, 7, 9,WHITE);
display.drawBitmap(87, 38, image_Pin_arrow_left_bits, 9, 7, WHITE);
display.drawBitmap(112, 38, image_Pin_arrow_right_bits, 9, 7, WHITE);
display.drawBitmap(100, 26, image_Pin_arrow_up_bits, 7, 9, WHITE);

display.display();

      display.display();
      eyeState = CLOSED;
    } else {
      // Open eyes
      display.clearDisplay();
// display.drawRect(76, 5, 43, 26, WHITE);
// display.fillCircle(108, 15, 6, WHITE);
// display.drawLine(25, 47, 68, 58, WHITE);
// display.drawLine(69, 58, 106, 48, WHITE);
// display.drawRect(6, 5, 43, 26, WHITE);
// display.drawLine(24, 47, 65, 37, WHITE);
// display.fillCircle(38, 15, 6, WHITE);
// display.drawLine(106, 47, 66, 36, WHITE);
display.drawBitmap(5, 3, image_bluetooth_connected_bits, 14, 16,WHITE);
display.drawBitmap(30, 29, image_hand_thumbs_up_bits, 16, 16, WHITE);
display.drawBitmap(107, 2, image_network_bits, 15, 16, WHITE);
display.drawBitmap(100, 47, image_Pin_arrow_down_bits, 7, 9,WHITE);
display.drawBitmap(87, 38, image_Pin_arrow_left_bits, 9, 7, WHITE);
display.drawBitmap(112, 38, image_Pin_arrow_right_bits, 9, 7, WHITE);
display.drawBitmap(100, 26, image_Pin_arrow_up_bits, 7, 9, WHITE);
display.drawBitmap(57, 4, image_notification_bell_bits, 16, 16, WHITE);

display.display();
      display.display();
      eyeState = OPEN;
    }
    lastBlinkTime = millis();
  }

  // Check button press to trigger happy face
  if (digitalRead(BUTTON_PIN) == LOW) {
    display.clearDisplay();
    // Draw happy face or any other pattern you want
    // Example: display.drawCircle(64, 32, 30, WHITE);
    // Example: display.fillCircle(48, 32, 10, WHITE);
    // Example: display.fillCircle(80, 32, 10, WHITE);
    // Example: display.drawPixel(58, 40, WHITE);
    // Example: display.drawPixel(70, 40, WHITE);
    // Example: display.drawLine(54, 46, 74, 46, WHITE);
    display.display();
    delay(2000); // Display happy face for 2 seconds
    // After displaying, clear the display to return to normal
    display.clearDisplay();
    lastBlinkTime = millis(); // Reset blink timer to maintain blinking pattern
  }
}