#include <WiFi.h>
#include "ThingsBoard.h"
#include "DHTesp.h"
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define CURRENT_FIRMWARE_TITLE    "TEST"
#define CURRENT_FIRMWARE_VERSION  "1.0.0"

#define WIFI_SSID           "Wokwi-GUEST"
#define WIFI_PASSWORD       ""

#define TOKEN               "LF9RzroXvmT193x30SOy"
#define THINGSBOARD_SERVER  "thingsboard.cloud"

//OLED
#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

const int DHT_PIN = 15;

//Kelembaban tanah
const int Pin_Potensio = 33;
const int LED = 32;
const int buzzerPin = 2;
float dataPotensio;

DHTesp dhtSensor;

WiFiClient espClient;
ThingsBoard tb(espClient);
int status = WL_IDLE_STATUS;

void InitWiFi()
{
  Serial.println("Connecting to AP ...");
  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("Connected to AP");
}

void reconnect() {
  status = WiFi.status();
  if ( status != WL_CONNECTED) {
    WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
    while (WiFi.status() != WL_CONNECTED) {
      delay(500);
      Serial.print(".");
    }
    Serial.println("Connected to AP");
  }
}

const float BETA = 3950; // should match the Beta Coefficient of the thermistor

void setup() {

  Serial.begin(115200);
  Serial.println();

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

  //InitWiFi();
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);

  pinMode(Pin_Potensio, INPUT);
  pinMode(LED, OUTPUT);
  pinMode(buzzerPin, OUTPUT);
}

void loop() {
   delay(1000);

  if (WiFi.status() != WL_CONNECTED) {
    reconnect();
  }

  if (!tb.connected()) {
    // Connect to the ThingsBoard
    Serial.print("Connecting to: ");
    Serial.print(THINGSBOARD_SERVER);
    Serial.print(" with token ");
    Serial.println(TOKEN);
    if (!tb.connect(THINGSBOARD_SERVER, TOKEN)) {
      Serial.println("Failed to connect");
      return;
    }
  }

  Serial.println("Sending data...");

  TempAndHumidity  data = dhtSensor.getTempAndHumidity();

  //potensio
  dataPotensio = analogRead(Pin_Potensio);
  int kelembaban = map(dataPotensio, 0, 4095, 0, 100);
  
  //Jika suhu <26, dan <55 POMPA ON
  //jika suhu terlalu rendah, kelembaban kurang
  if (data.temperature <26 && kelembaban <55)
    {
      const char* statusSuhu = "Suhu Terlalu Rendah!";
      const char* statusKelembaban = "Kelembaban Kurang!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA ON");
      display.display(); 

      digitalWrite(LED, HIGH);
      const char* statusPompa = "ON";
      tb.sendTelemetryData("Status_Pompa", statusPompa);

      //buzzer ON
      tone(buzzerPin,150, 1000);
      delay(2000);
    } 

  //Jika suhu <26, kelembaban >55 & <60 POMPA OFF
  //jika suhu terlalu rendah, kelembaban cukup
  if (data.temperature <26 && kelembaban >= 55 && kelembaban <= 60)
    {
      const char* statusSuhu = "Suhu terlalu Rendah!";
      const char* statusKelembaban = "Kelembaban Cukup";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    } 

  //Jika suhu <26, kelembaban >60 POMPA OFF
  //jika suhu terlalu rendah, kelembaban terlalu tinggi
  if (data.temperature < 26 && kelembaban > 60)
    {
      const char* statusSuhu = "Suhu terlalu Rendah!";
      const char* statusKelembaban = "Terlalu Lembab!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    } 

  //Jika suhu >26 & <36, kelembaban <55 POMPA ON
  //jika suhu optimal, kelembaban kurang
  if (data.temperature >= 26 && data.temperature <= 36 && kelembaban < 55)
    {
      const char* statusSuhu = "Suhu Optimal";
      const char* statusKelembaban = "Kelembaban Kurang!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA ON");
      display.display(); 

      digitalWrite(LED, HIGH);
      const char* statusPompa = "ON";
      tb.sendTelemetryData("Status_Pompa", statusPompa);

      //buzzer ON
      tone(buzzerPin,150, 1000);
      delay(2000);
    } 

  //Jika suhu >26 & <36, kelembaban >55 & <60 POMPA OFF
  //jika suhu optimal, kelembaban cukup
  if (data.temperature >= 26 && data.temperature <= 36 && kelembaban >= 55 && kelembaban <= 60)
    {
      const char* statusSuhu = "Suhu Optimal";
      const char* statusKelembaban = "Kelembaban Cukup!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    } 

  //Jika suhu >26 & <36, kelembaban >60 POMPA OFF
  //jika suhu optimal, kelembaban tinggi
  if (data.temperature >= 26 && data.temperature <= 36 && kelembaban >60)
    {
      const char* statusSuhu = "Suhu Optimal";
      const char* statusKelembaban = "Terlalu Lembab!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    }

  //Jika suhu >36, kelembaban <55 POMPA ON
  //jika suhu tinggi, kelembaban kurang
  if (data.temperature > 36 && kelembaban < 55)
    {
      const char* statusSuhu = "Suhu Terlalu Tinggi";
      const char* statusKelembaban = "Kelembaban Kurang!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA ON");
      display.display(); 

      digitalWrite(LED, HIGH);
      const char* statusPompa = "ON";
      tb.sendTelemetryData("Status_Pompa", statusPompa);

      //buzzer ON
      tone(buzzerPin,150, 1000);
      delay(2000);
    }

  //Jika suhu >36, kelembaban >55 & <60 POMPA OFF
  //jika suhu tinggi, kelembaban cukup
  if (data.temperature > 36 && kelembaban >= 55 && kelembaban <= 60)
    {
      const char* statusSuhu = "Suhu Terlalu Tinggi";
      const char* statusKelembaban = "Kelembaban Cukup!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    }

  //Jika suhu >36, kelembaban >60 POMPA OFF
  //jika suhu tinggi, kelembaban tinggi
  if (data.temperature > 36 && kelembaban > 60)
    {
      const char* statusSuhu = "Suhu Terlalu Tinggi";
      const char* statusKelembaban = "Terlalu Lembab!";
      tb.sendTelemetryData("Status_Suhu", statusSuhu);
      tb.sendTelemetryData("Status_Kelembaban", statusKelembaban);

      //Tampilkan pada OLED
      display.clearDisplay();
      display.setTextSize(1,1);
      display.setTextColor(WHITE);
      display.setCursor(0,0);
      // Tampilkan
      display.print("Suhu: ");
      display.print(data.temperature);
      display.println(" ^C");
      display.println(statusSuhu);
      display.print("Kelembaban: ");
      display.print(kelembaban);
      display.println("%");
      display.println(statusKelembaban);
      display.drawFastHLine(0, 34, SSD1306_LCDWIDTH, WHITE);
      display.setCursor(0,42);
      display.setTextSize(2);
      display.println("POMPA OFF");
      display.display(); 

      digitalWrite(LED, LOW);
      const char* statusPompa = "OFF";
      tb.sendTelemetryData("Status_Pompa", statusPompa);
    }

  //kirim data ke thingsboard
  tb.sendTelemetryFloat("Suhu", data.temperature);
  tb.sendTelemetryFloat("Kelembaban", kelembaban);

  //const char* statusLED = "ON";
  //tb.sendTelemetryData("Status_LED", statusLED);

  Serial.print("Suhu: ");
  Serial.print(data.temperature);
  Serial.print(" ℃ ");

  Serial.print("Kelembaban Tanah: ");
  Serial.print(kelembaban);
  Serial.println("%");

  tb.sendTelemetryFloat("lat", -7.0054689953047955);
  tb.sendTelemetryFloat("long", 107.88379928841887);

  tb.loop();
}