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

#define RELAY_PIN 15 // Pin D15 digunakan untuk mengontrol relay
#define BUTTON_PIN_1 19 // Tombol 1 (misalnya, pin D19)
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET     -1
#define SCREEN_ADDRESS 0x3C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

const char* ssid = "Wokwi-GUEST";
const char* password = "";

WiFiUDP ntpUDP;
NTPClient timeClient(ntpUDP);

bool isRelayOn = false;
unsigned long lastRelayOnTime = 0;
unsigned long lastRelayOffTime = 0;

void setup() {
  Serial.begin(115200);
  pinMode(RELAY_PIN, OUTPUT);
  digitalWrite(RELAY_PIN, LOW);
  
  pinMode(BUTTON_PIN_1, INPUT_PULLUP); // Aktifkan input pull-up pada tombol 1
  
  if (!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;); // Jangan lanjutkan, lakukan loop selamanya
  }
  
  // Koneksi WiFi
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
    Serial.println("Connecting to WiFi...");
  }

  // Inisialisasi NTP
  timeClient.begin();
  timeClient.setTimeOffset(0);
  timeClient.forceUpdate(); // Memperbarui waktu dari server NTP
}

void loop() {
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);

  display.setCursor(0, 0);
  display.println("Status Lampu:");

  if (digitalRead(BUTTON_PIN_1) == LOW) {
    digitalWrite(RELAY_PIN, HIGH); // Nyalakan relay
    if (!isRelayOn) {
      lastRelayOnTime = timeClient.getEpochTime();
    }
    isRelayOn = true;
    display.setCursor(0, 12);
    display.println("Menyala...");
  } else {
    digitalWrite(RELAY_PIN, LOW); // Matikan relay
    if (isRelayOn) {
      lastRelayOffTime = timeClient.getEpochTime();
    }
    isRelayOn = false;
    display.setCursor(0, 12);
    display.println("Mati...");
  }

  if (isRelayOn) {
    display.setCursor(0, 36);
    display.print("Lampu menyala sejak: ");
    display.print(getFormattedTime(lastRelayOnTime));
  } else {
    display.setCursor(0, 36);
    display.print("Lampu dimatikan pada: ");
    display.print(getFormattedTime(lastRelayOffTime));
  }

  display.display();
}

String getFormattedTime(unsigned long epochTime) {
  struct tm timeinfo;
  time_t epoch = (time_t)epochTime;
  localtime_r(&epoch, &timeinfo);

  String timeString = "";
  if (timeinfo.tm_hour < 10) {
    timeString += "0";
  }
  timeString += String(timeinfo.tm_hour) + ":";
  if (timeinfo.tm_min < 10) {
    timeString += "0";
  }
  timeString += String(timeinfo.tm_min) + ":";
  if (timeinfo.tm_sec < 10) {
    timeString += "0";
  }
  timeString += String(timeinfo.tm_sec);

  return timeString;
}