#include<Wire.h>
#include<Adafruit_GFX.h>
#include<Adafruit_SSD1306.h>
#include <DHTesp.h>
#include <WiFi.h>
#include <NTPClient.h>
#include <WiFiUdp.h>


//define OLED parameters
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET -1
#define SCREEN_ADDRESS 0X3C



// push buttons
#define PB_CANCEL 34
#define PB_OK 32
#define PB_UP 33
#define PB_DOWN 35
#define DHTPIN 12

#define NTP_SERVER     "pool.ntp.org"
#define UTC_OFFSET     19800
#define UTC_OFFSET_DST 0

//buzzer
#define BUZZER 5
#define LED_1 15

//Declare Object
Adafruit_SSD1306 display(SCREEN_WIDTH,SCREEN_HEIGHT,&Wire,OLED_RESET);
DHTesp dhtSensor;
 

// Global variables
int days = 0;
int hours = 0;
int minutes = 0;
int seconds = 0;

unsigned long time_Now=0;
unsigned long time_Last=0;

bool alarm_enabled = true;
int n_alarms = 3;
int alarm_hours[]={0,1,1};
int alarm_minutes[]={1,10,12};
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 D_H = 523;
int notes[] = {C,D,E,F,G,A,B,D_H};

//This for menu

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

//for setting the time
const long utcOffsetInSeconds = 19800;
WiFiUDP ntpUDP;
NTPClient timeClient(ntpUDP, "pool.ntp.org", utcOffsetInSeconds);





void setup(){

  pinMode(BUZZER, OUTPUT);
  pinMode(LED_1, OUTPUT);
  pinMode(PB_CANCEL, INPUT);
  pinMode(PB_OK, INPUT);
  pinMode(PB_UP, INPUT);
  pinMode(PB_DOWN, INPUT);

  //DHT sensor
  dhtSensor.setup(DHTPIN, DHTesp::DHT22);


  //OLED display initialization
  Serial.begin(9600);
  if (! display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {

    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
  //Display turn on
  display.display();
  delay(2000);

  //setting time through wifi

  WiFi.begin("Wokwi-GUEST", "", 6);
  while (WiFi.status() != WL_CONNECTED) {
    delay(250);
    display.clearDisplay();
    print_line("Connecting to wifi...", 0, 0, 2);
  }
  display.clearDisplay();
  print_line("Connected to wifi...", 0, 0, 2);

  configTime(utcOffsetInSeconds, 0, "pool.ntp.org", "time.nist.gov");
  timeClient.begin();
  display.clearDisplay();
  print_line("Welcome to Medibox",10,20,1);
  delay(1000);
  display.clearDisplay();


}



void loop(){
  //For time
  timeClient.update();

  //Checking for alarms
  update_time_with_check_alarm();

  //Go for setting
  if(digitalRead(PB_OK)==LOW){
    delay(200);
    go_to_menu();
  }

  check_temp();

  
  
}

void print_line(String text, int column, int row,int text_size){


  display.setTextSize(text_size);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(column,row);
  display.println(text);
  display.display();
 
}

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);

}

void update_time(){
  
  struct tm timeinfo;
  getLocalTime(&timeinfo);

  char timeHour[3];
  strftime(timeHour,3,"%H",&timeinfo);
  hours = atoi(timeHour);

  char timeMinute[3];
  strftime(timeMinute,3,"%M",&timeinfo);
  minutes = atoi(timeMinute);

  char timeSecond[3];
  strftime(timeSecond,3,"%S",&timeinfo);
  seconds = atoi(timeSecond);

  char timeDay[3];
  strftime(timeDay,3,"%d",&timeinfo);
  days = atoi(timeDay);
}


void ring_alarm(){
  display.clearDisplay();
  print_line("MEDICINE TIME!",0,0,2);

  digitalWrite(LED_1, HIGH);

  bool break_happened = false;

 
  // ring the BUzzer
  while(break_happened == false && digitalRead(PB_CANCEL)== HIGH){
    for(int i=0; i<n_notes; i++){
      if (digitalRead(PB_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();

  if(alarm_enabled ==true){
    for(int i=0;i<n_alarms;i++){
      if(alarm_triggered[i]==false && alarm_hours[i]==hours && alarm_minutes[i]==minutes){
        ring_alarm();
        alarm_triggered[i]=true;
      }
    }
  }
}


int wait_for_button_press(){
  while(true){
    if(digitalRead(PB_UP)==LOW){
      delay(200);
      return PB_UP;
    }

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

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

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

    update_time();
  }
}
void go_to_menu(){

  while(digitalRead(PB_CANCEL)==HIGH){
    display.clearDisplay();
    print_line(modes[current_mode],0,0,2);

    int pressed = wait_for_button_press();
    if (pressed == PB_UP){
      delay(200);
      current_mode +=1;
      current_mode= current_mode % max_modes;
    }

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

    else if (pressed == PB_OK){
      delay(200);
      //Serial.println(current_mode);
      run_mode(current_mode);
    }

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


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 == PB_UP){
      delay(200);
      temp_hour +=1;
      temp_hour = temp_hour%24;
    }

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

    else if (pressed == PB_OK){
      delay(200);
      //Serial.println(current_mode);
      alarm_hours[alarm] = temp_hour;
      break;
    }

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

  }

  int temp_minutes =alarm_minutes[alarm] ;
  while (true){

    display.clearDisplay();
    print_line("Enter minutes:" + String(temp_minutes),0,0,2);

    int pressed = wait_for_button_press();
    if (pressed == PB_UP){
      delay(200);
      temp_minutes +=1;
      temp_minutes = temp_minutes % 60;
    }

    else if (pressed == PB_DOWN){
      delay(200);
      temp_minutes -= 1;
      if (temp_minutes < 0){
        temp_minutes = 59;
      }
    }

    else if (pressed == PB_OK){
      delay(200);
      //Serial.println(current_mode);
      alarm_minutes[alarm] = temp_minutes;
      break;
    }

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

  display.clearDisplay();
  print_line("Alarm is set",0,0,2);
  delay(1000);

}


void run_mode(int mode){

  if (mode == 0 || mode == 1|| mode == 2){
    set_alarm(mode);
  }

  else if (mode ==3){
    alarm_enabled = false;
  }
}

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

  else if (data.temperature < 25){
    display.clearDisplay();
    print_line("TEMP LOW",0,40,1);
  }

  if (data.humidity > 40){
    display.clearDisplay();
    print_line("HUMIDITY HIGH",0,50,1);
  }

  else if (data.humidity < 20){
    display.clearDisplay();
    print_line("HUMIDITY LOW",0,50,1);
  }
}