#include <Wire.h>
#include <WiFi.h>
#include "time.h"
#include <Adafruit_MPU6050.h>
#include <Adafruit_Sensor.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include <LiquidCrystal_I2C.h>

const char* ntpServer1 = "pool.ntp.org";
const char* time_zone = "IST-5:30";
String timestamp;

LiquidCrystal_I2C lcd(0x27, 16, 2);
LiquidCrystal_I2C lcd2(0x28, 16, 2);
LiquidCrystal_I2C lcd3(0x29, 16, 2);


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

// Declaration for an SSD1306 display connected to I2C (SDA, SCL pins)
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);
Adafruit_MPU6050 mpu;

void setup() {
  Serial.begin(115200);
  //connect to WiFi

  Serial.print("Connecting to WiFi");
  WiFi.begin("Wokwi-GUEST", "", 6);
  while (WiFi.status() != WL_CONNECTED) {
    delay(100);
    Serial.print(".");
  }
  Serial.println(" Connected!");
  configTzTime(time_zone, ntpServer1);

  // Check if the MPU6050 sensor is detected
  if (!mpu.begin()) {
    Serial.println("Failed to find MPU6050 chip");
    while (1) {
      delay(10);
    }
  }
  Serial.println("MPU6050 Found!");


  // set accelerometer range to +-8G
  mpu.setAccelerometerRange(MPU6050_RANGE_8_G);

  // set gyro range to +- 500 deg/s
  mpu.setGyroRange(MPU6050_RANGE_500_DEG);

  // set filter bandwidth to 21 Hz
  mpu.setFilterBandwidth(MPU6050_BAND_21_HZ);

  // Add a delay for stability
  lcd.init();
  lcd.backlight();
  lcd2.init();
  lcd2.backlight();
  lcd3.init();
  lcd3.backlight();


  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { // Address 0x3D for 128x64
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
}

void loop() {
  lcd3.setCursor(0, 0);
  lcd3.print("Heritage");

  // Get new sensor events with the readings
  sensors_event_t a, g, temp;
  mpu.getEvent(&a, &g, &temp);

  // Print out the acceleration readings in m/s^2
  Serial.print("Acceleration:   X:");
  Serial.print(a.acceleration.x);
  Serial.print(", Y:");
  Serial.print(a.acceleration.y);
  Serial.print(", Z:");
  Serial.print(a.acceleration.z);
  Serial.println(" (m/s^2)");

  // Print out the rotation readings in rad/s
  Serial.print("Rotation:       X:");
  Serial.print(g.gyro.x);
  Serial.print(", Y:");
  Serial.print(g.gyro.y);
  Serial.print(", Z:");
  Serial.print(g.gyro.z);
  Serial.println(" (rad/s)");

  // Print out the temperature reading in degrees Celsius
  Serial.print("Temperature:    ");
  Serial.print(temp.temperature);
  Serial.println(" (degC)");

  // Print out the rotation readings in rad/s
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(WHITE);
  display.setCursor(0, 0);
  display.println("Rotation:");
  display.print("X : ");
  display.println(g.gyro.x);
  display.print("Y : ");
  display.println(g.gyro.y);
  display.print("Z : ");
  display.println(g.gyro.z);
  display.display();

  // Print out the temperature reading in degrees Celsius
  lcd.setCursor(0, 0);
  lcd.print("Temperature:");
  lcd.setCursor(0, 1);
  lcd.print(temp.temperature);
  lcd.println(" (degC)");
  printLocalTime();
  Serial.println("--------------------------------------------------");
  // Add a delay to avoid flooding the serial monitor
  delay(100);
}


void printLocalTime()
{
  struct tm timeinfo;
  if (!getLocalTime(&timeinfo)) {
    Serial.println("No time available (yet)");
    lcd.setCursor(0, 0);
    lcd.print("Please wait...");
    return;
  }
  char printBuffer[100];
  sprintf(printBuffer, "Year %d, Month %d, Day %d, Hour %d, Min %d, Sec %d",
          timeinfo.tm_year + 1900, timeinfo.tm_mon + 1, timeinfo.tm_mday,
          timeinfo.tm_hour, timeinfo.tm_min, timeinfo.tm_sec);
  // Serial.println(printBuffer);
  strftime(printBuffer, 100, "%A, %B %d %Y %H:%M:%S", &timeinfo);
  // Serial.println(printBuffer);

  timestamp = String(timeinfo.tm_year + 1900) + "-" + String(timeinfo.tm_mon + 1) + "-" + String(timeinfo.tm_mday) + ":" + String(timeinfo.tm_hour) + "-" + String(timeinfo.tm_min) + "-" + String(timeinfo.tm_sec);
  // Serial.println(timestamp);

  // lcd.clear();
  lcd2.setCursor(0, 0);
  lcd2.print("Time : " + String(timeinfo.tm_hour) + ":" + String(timeinfo.tm_min) + ":" + String(timeinfo.tm_sec) + "  " + String(timeinfo.tm_mday) + "/"  + String(timeinfo.tm_mon + 1) + "/"  + String((timeinfo.tm_year + 1900) - 2000));
  lcd2.setCursor(0, 1);
  lcd2.print("Date : " + String(timeinfo.tm_mday) + "/"  + String(timeinfo.tm_mon + 1) + "/"  + String((timeinfo.tm_year + 1900)));
}