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

// OLED Display
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET -1
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

// Wi-Fi Credentials
const char* ssid = "Wokwi-GUEST";
const char* password = "";

// NTP Client
WiFiUDP ntpUDP;
NTPClient timeClient(ntpUDP, "pool.ntp.org");

// Pin Definitions
const int voltageSensorPin = 34;
const int currentSensorPin = 35;
const int relayPins[] = {16, 17, 18, 19};
const int buzzerPin = 23;
const int ledPins[] = {25, 26};

// Variables
float voltage = 0;
float current = 0;
float power = 0;
unsigned long lastUpdateTime = 0;

void setup() {
  Serial.begin(115200);

  // Initialize I2C
  Wire.begin(21, 22);

  // Initialize OLED Display
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println("ATS System Initialized");
  display.display(); // Ensure display is updated

  // Initialize Wi-Fi
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("\nWiFi connected");

  // Initialize NTP Client
  timeClient.begin();
  timeClient.setTimeOffset(19800);

  // Initialize Pins
  for (int i = 0; i < 4; i++) {
    pinMode(relayPins[i], OUTPUT);
    digitalWrite(relayPins[i], LOW);
  }
  pinMode(buzzerPin, OUTPUT);
  for (int i = 0; i < 2; i++) {
    pinMode(ledPins[i], OUTPUT);
    digitalWrite(ledPins[i], LOW);
  }
}

void loop() {
  voltage = analogRead(voltageSensorPin) * (3.3 / 4095.0) * 100;
  current = analogRead(currentSensorPin) * (3.3 / 4095.0) * 10;
  power = voltage * current;

  // Display Data on OLED
  display.clearDisplay();
  display.setCursor(0, 0);
  display.print("Voltage: ");
  display.print(voltage);
  display.println(" V");
  display.print("Current: ");
  display.print(current);
  display.println(" A");
  display.print("Power: ");
  display.print(power);
  display.println(" W");
  display.display();

  if (voltage < 50) {
    digitalWrite(relayPins[0], HIGH);
    digitalWrite(relayPins[1], LOW);
    digitalWrite(relayPins[2], LOW);
    digitalWrite(relayPins[3], LOW);
    digitalWrite(ledPins[0], HIGH);
    digitalWrite(buzzerPin, HIGH);
  } else {
    digitalWrite(relayPins[0], LOW);
    digitalWrite(relayPins[1], HIGH);
    digitalWrite(relayPins[2], LOW);
    digitalWrite(relayPins[3], LOW);
    digitalWrite(ledPins[0], LOW);
    digitalWrite(buzzerPin, LOW);
  }

  if (millis() - lastUpdateTime > 10000) {
    sendDataToServer(voltage, current, power);
    lastUpdateTime = millis();
  }

  timeClient.update();
  Serial.println(timeClient.getFormattedTime());

  delay(1000);
}

void sendDataToServer(float voltage, float current, float power) {
  Serial.println("Sending data to server...");
}