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

//define OLED parameters
#define SCREEN_WIDTH 128 // OLED display width, in pixels
#define SCREEN_HEIGHT 64 // OLED display height, in pixels
#define OLED_RESET -1 // Reset pin
#define SCREEN_ADDRESS 0x3C   //I2C address of OLED Display
#define BUZZER 5
#define LED_1 15
#define LED_2 14
#define CANCEL 34
#define UP 33
#define DOWN 35
#define OK 32
#define DHT 12


//Declaring objects
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);  //creating object for OLED Display
DHTesp dhtSensor;

//Global Variables

int days = 0;
int hours = 0;
int minutes = 0;
int seconds = 0;

unsigned long timeNow = 0;
unsigned long timeLast = 0;

bool alarm_enabled = true;
int n_alarms = 2;
int alarm_hours[] = {0, 1};
int alarm_minutes[] = {10, 20};
bool alarm_triggered[] = {false, false};

int n_notes = 8;
int C = 262;
int D = 294;
int E = 330;
int F = 349;
int G = 392;
int A = 440;
int B = 494;
int C_H = 523;
int notes[] = {C, D, E, F, G, A, B, C_H};

int current_mode = 0;
int max_modes = 4;
String modes[] = {"1 - Set Time", "2 - Set Alarm 1", "3 - Set Alarm 2", "4 - Remove Alarm"};


void setup(){
  Serial.begin(9600);  //initialize serial monitor

  pinMode(BUZZER, OUTPUT);
  pinMode(LED_1, OUTPUT);
  pinMode(LED_2, OUTPUT);
  pinMode(CANCEL, INPUT);
  pinMode(UP, INPUT);
  pinMode(DOWN, INPUT);
  pinMode(OK, INPUT);

  dhtSensor.setup(DHT, DHTesp::DHT22);


  // initialize the OLED object
  if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {  //SSD1306_SWITCHCAPVCC = generate display voltage from 3.3V internally
    Serial.println(F("SSD1306 allocation failed"));
    for(;;); // Don't proceed, loop forever
  }

  
  display.display(); //turn on OLED Display
  delay(2000);

  
  display.clearDisplay();  //clear the buffer

   
  print_line("Welcome to Medibox!", 10, 20, 2); // String, cursor column, cursor row, , text size, 
  display.clearDisplay();
  

  

}


void loop(){
  update_time_with_check_alarm();

  if (digitalRead(OK) == LOW) {
    delay(200);
    go_to_menu();
  }
  
  check_temp();

}



//Display a custom message
void print_line(String text,int column, int row, int text_size ){
  display.setTextSize(text_size);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(column, row);   //bring cursor to origin
  display.println(text);

  display.display();

}


//function to display the current time DD:HH:MM:SS
void print_time_now(void) {
  display.clearDisplay();
  print_line(String(days), 0, 0, 2);
  print_line(":", 20, 0, 2);
  print_line(String(hours), 30, 0, 2);
  print_line(":", 50, 0, 2);
  print_line(String(minutes), 60, 0, 2);
  print_line(":", 80, 0, 2);
  print_line(String(seconds), 90, 0, 2);
}


// function to update the current time
void update_time(){
  timeNow = millis()/1000;  //seconds passed from bootup
  seconds = timeNow - timeLast;

  if(seconds >= 60){
    minutes += 1; //minutes = minutes+1
    timeLast += 60;
  }
  
  if(minutes == 60){
    hours += 1;
    minutes = 0;
  }

  if(hours == 24){
    days += 1;
    hours = 0;
  }

}




// ring an alarm
void ring_alarm() {
  //show message on display
  display.clearDisplay();
  print_line("Medicine  Time", 0, 0, 2);

  // light up LED 1
  digitalWrite(LED_1, HIGH);
  
  bool break_happened = false;
  while( break_happened == false && digitalRead(CANCEL) == HIGH){
    for (int i = 0; i < n_notes; i++){
      if(digitalRead(CANCEL)== LOW){
        delay(200);
        break_happened = true;
        break;
      }
      tone(BUZZER, notes[i]);
      delay(500);
      noTone(BUZZER);
      delay(2);
    }
  }

  digitalWrite(LED_1, LOW);
  display.clearDisplay();



}








void update_time_with_check_alarm(void) {
  update_time();
  print_time_now();

  //check for alarms
  if (alarm_enabled == true) {
    //iterating through all alarms
    for (int i=0; i < n_alarms ; i++) {
      if (alarm_triggered[i] == false && alarm_hours[i] == hours && alarm_minutes[i] == minutes) {
        ring_alarm();    //call the ringing fuction
        alarm_triggered[i] = true;
      }
    }
  }
}


//function to wait for button press in the menu
int wait_for_button_press() {
  while (true) {
    if (digitalRead(UP) == LOW) {
      delay(200);
      return UP;
    }

    else if (digitalRead(DOWN) == LOW) {
      delay(200);
      return DOWN;
    }

    else if (digitalRead(CANCEL) == LOW) {
      delay(200);
      return CANCEL;
    }

    else if (digitalRead(OK) == LOW) {
      delay(200);
      return OK;
    }

    update_time();
  }
}





//function to navigate through the menu
void go_to_menu() {
  while (digitalRead(CANCEL) == HIGH) {
    display.clearDisplay();
    print_line(modes [current_mode], 0, 0, 2);

    int pressed = wait_for_button_press();

    if (pressed == UP) {
      current_mode += 1;
      current_mode %= max_modes;
      delay(200);
    }

    else if (pressed == DOWN) {
      current_mode -= 1;
      if (current_mode < 0) {
        current_mode = max_modes - 1;
      }
      delay(200);
    }

    else if (pressed == OK) {
      delay(200);
      run_mode(current_mode);
    }

    else if (pressed == CANCEL) {
      delay(200);
      break;
    }
  }
}


void set_time(){
  
  int temp_hour = hours;
  while (true) {
    display.clearDisplay();
    print_line("Enter Hour" + String(temp_hour), 0, 0, 2);

    int pressed = wait_for_button_press();

    if (pressed == UP) {
      delay(200);
      temp_hour++;
      temp_hour %= 24;
      
    }

    else if (pressed == DOWN) {
      delay(200);
      temp_hour--;
      if(temp_hour < 0) {
        temp_hour = 23;
      }
    }

    else if (pressed == OK) {
      delay(200);
      hours = temp_hour;
      break;
    }

    else if (pressed == CANCEL) {
      delay(200);
      break;
    }
  }

  int temp_minute =minutes;
  while (true) {
    display.clearDisplay();
    print_line("Enter Minute" + String(temp_minute), 0, 0, 2);

    int pressed = wait_for_button_press();

    if (pressed == UP) {
      delay(200);
      temp_minute++;
      temp_minute %= 60;
      
    }

    else if (pressed == DOWN) {
      delay(200);
      temp_minute--;
      if(temp_minute < 0) {
        temp_minute = 59;
      }
    }

    else if (pressed == OK) {
      delay(200);
      minutes = temp_minute;
      break;
    }

    else if (pressed == CANCEL) {
      delay(200);
      break;
    }
  }

  display.clearDisplay();
  print_line("Time Set!", 0, 0, 2);
  delay(1000);
}


void set_alarm(int alarm){

  int temp_hour = alarm_hours[alarm];
  while (true) {
    display.clearDisplay();
    print_line("Enter Hour" + String(temp_hour), 0, 0, 2);

    int pressed = wait_for_button_press();

    if (pressed == UP) {
      delay(200);
      temp_hour++;
      temp_hour %= 24;
      
    }

    else if (pressed == DOWN) {
      delay(200);
      temp_hour--;
      if(temp_hour < 0) {
        temp_hour = 23;
      }
    }

    else if (pressed == OK) {
      delay(200);
      alarm_hours[alarm] = temp_hour;
      break;
    }

    else if (pressed == CANCEL) {
      delay(200);
      break;
    }
  }

  int temp_minute =alarm_minutes[alarm];
  while (true) {
    display.clearDisplay();
    print_line("Enter Minute" + String(temp_minute), 0, 0, 2);

    int pressed = wait_for_button_press();

    if (pressed == UP) {
      delay(200);
      temp_minute++;
      temp_minute %= 60;
      
    }

    else if (pressed == DOWN) {
      delay(200);
      temp_minute--;
      if(temp_minute < 0) {
        temp_minute = 59;
      }
    }

    else if (pressed == OK) {
      delay(200);
      alarm_minutes[alarm] = temp_minute;
      break;
    }

    else if (pressed == CANCEL) {
      delay(200);
      break;
    }
  }
  display.clearDisplay();
  print_line("Alarm is Set!", 0, 0, 2);
  delay(1000);
}




void run_mode(int mode){
  if (mode == 0){
    set_time();
  }
  else if (mode == 1 || mode == 2) {
    set_alarm(mode - 1);

  }
  else if (mode == 3) {
    alarm_enabled = false;
    display.clearDisplay();
    print_line("Alarms Disabled!", 0, 0, 2);
    delay(1000);
  }
}


void check_temp(void) {
  TempAndHumidity data = dhtSensor.getTempAndHumidity();
  bool all_good = true;
  if (data.temperature > 35) {
    all_good = false;
    digitalWrite(LED_2, HIGH);
    display.clearDisplay();
    print_line("TEMP HIGH", 0, 40, 1);
  }

  else if (data.temperature < 25) {
    all_good = false;
    digitalWrite(LED_2, HIGH);
    print_line("TEMP LOW", 0, 40, 1);
  }

  if (data.humidity > 85) {
    all_good = false;
    digitalWrite(LED_2, HIGH);
    display.clearDisplay();
    print_line("HUMD HIGH", 0, 50, 1);
  }

  else if (data.humidity < 20) {
    all_good = false;
    digitalWrite(LED_2, HIGH);
    display.clearDisplay();
    print_line("HUMD LOW", 0, 50, 1);
  }

  if (all_good) {
    digitalWrite(LED_2, LOW);
  }
}