#include <Adafruit_SSD1306.h>
#include <Wire.h>
#include <WiFi.h>
#include <NTPClient.h>
#include <PubSubClient.h>
#include "DHTesp.h" // sensor temperatura
// #define RELAY_PIN    17 // Relay com sensor temp
#define MOISTURE_PIN 23 // ESP32 pin GPIO36 (ADC0) that connects to AOUT pin of moisture sensor
#define THRESHOLD 2800 

#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 = "";
const char* thingsboardServer = "thingsboard.cloud";
const char* accessToken = "1y1n43amby32bn7hxxvh";
const int DHT_PIN = 5;
WiFiUDP ntpUDP;
NTPClient timeClient(ntpUDP);

WiFiClient wifiClient;
PubSubClient client(wifiClient);

bool isRelayOn = false;
unsigned long lastRelayOnTime = 0;
unsigned long lastRelayOffTime = 0;
//sensor temperatura
DHTesp dhtSensor;
void setup() {
  Serial.begin(115200);
  pinMode(RELAY_PIN, OUTPUT);
  digitalWrite(RELAY_PIN, LOW);

  
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  
  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

  // Setup MQTT
  client.setServer(thingsboardServer, 1883);
  while (!client.connected()) {
    Serial.println("Connecting to ThingsBoard...");
    if (client.connect("ESP32_Device", accessToken, "")) {
      Serial.println("Connected to ThingsBoard");
    } else {
      Serial.print("Failed to connect, rc=");
      Serial.print(client.state());
      delay(2000);
    }
  }


  
}

void loop() {
  if (!client.connected()) {
    reconnect();
  }
  client.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();
      publishRelayStatus(true);
    }
    isRelayOn = true;
    display.setCursor(0, 12);
    display.println("Menyala...");
  } else {
    digitalWrite(RELAY_PIN, LOW); // Matikan relay
    if (isRelayOn) {
      lastRelayOffTime = timeClient.getEpochTime();
      publishRelayStatus(false);
    }
    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));
  }
  Sensor_temp();
  display.display();
}
//Tempo Amostrado
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;
}
//reconexao
void reconnect() {
  while (!client.connected()) {
    Serial.print("Attempting MQTT connection...");
    if (client.connect("ESP32_Device", accessToken, "")) {
      Serial.println("connected");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      delay(5000);
    }
  }
}
//imprime quando o botao foi acionado
void publishRelayStatus(bool relayStatus) {
  String payload = "{";
  payload += "\"relayStatus\":";
  payload += relayStatus ? "true" : "false";
  payload += "}";
  
  if (client.publish("v1/devices/me/telemetry", (char*) payload.c_str())) {
    Serial.println("Telemetry published successfully");
  } else {
    Serial.println("Telemetry publish failed");
  }
}

void Sensor_temp(){
  
  TempAndHumidity  data = dhtSensor.getTempAndHumidity();
  Serial.println("Temp: " + String(data.temperature, 2) + "°C");
  Serial.println("Humidity: " + String(data.humidity, 1) + "%");
  Serial.println("---");
  // Wait for a new reading from the sensor (DHT22 has ~0.5Hz sample rate)

 /* if (TempAndHumidity) {
    Serial.print("The soil moisture is DRY => activate pump");
    digitalWrite(RELAY_PIN, HIGH);
  } else {
    Serial.print("The soil moisture is WET=> deactivate the pump");
    digitalWrite(RELAY_PIN, LOW);
  }*/
   if (isRelayOn) {
    display.setCursor(0, 36);
    display.print("Temperatura : ");
    display.print(String(data.temperature) + "°C");
    display.print("Humidity: " + String(data.humidity) + "%");
  } else {
    display.setCursor(0, 36);
    // display.print("Lampu dimatikan pada: ");
    // display.print(getFormattedTime(lastRelayOffTime));
  }
 /**
   ESP32 + DHT22 Example for Wokwi
   
   https://wokwi.com/arduino/projects/322410731508073042
*/




  Serial.print(" (");
  Serial.print(String(data.temperature) + "°C");
  Serial.println(")");

  delay(1000);
}