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

//WiFi Setup
#define WIFI_SSID           "Wokwi-GUEST"
#define WIFI_PASSWORD       ""
WiFiClient espClient;
ThingsBoard tb(espClient);
int status = WL_IDLE_STATUS;

//ThingsBoard Setup
#define CURRENT_FIRMWARE_TITLE    "TEST"
#define CURRENT_FIRMWARE_VERSION  "1.0.0"
#define TOKEN               "Cc8N4IAfcrhFId8eM2IH" //Copy token device dari ThingsBoard
#define THINGSBOARD_SERVER  "thingsboard.cloud"

//DHT Setup
const int DHT_PIN = 13;
DHTesp dhtSensor;

//Define Ultra Sonic
#define ECHO_PIN 12
#define TRIG_PIN 14

//Define Sensor Cahaya
#define SensorCahaya 26

float X=400/397.63;

// Define saklar Pompa, Motor Kincir Air dan Aerator
#define Kincir 15 // Kincir Air (LED Biru)
#define PompaKeluar 4 // Pompa Air Keluar Tambak (LED Ungu)
#define PompaMasuk 5 // Pompa Air Masuk Tambak (LED Hijau)
#define Aerator 27 // Aerator (Led Kuning)



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");
  }
}

void setup() {
  Serial.begin(115200);
  Serial.println();
  InitWiFi();
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  pinMode( Kincir, OUTPUT);
  pinMode( PompaKeluar, OUTPUT);
  pinMode( PompaMasuk, OUTPUT);
  pinMode( Aerator, OUTPUT);
  pinMode( SensorCahaya, INPUT);
  digitalWrite(Kincir, LOW);
  digitalWrite(PompaKeluar, LOW);
  digitalWrite(PompaMasuk, LOW);
  digitalWrite(Aerator, LOW);
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);
}

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;
    }
  }
  
  //Generate ultrasonic signal
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);
  
  //Read ultrasonic signal
  int duration = pulseIn(ECHO_PIN, HIGH);
  float distance = duration * 0.034 * X / 2;
  float tinggi = 200 - distance ;

  Serial.println("Sending data...");
  TempAndHumidity  data = dhtSensor.getTempAndHumidity();
  tb.sendTelemetryFloat("Suhu Air Tambak", data.temperature);
  tb.sendTelemetryFloat("Tinggi Air Tambak", tinggi);

  //Pengontrol Aerator Tambak
  if (digitalRead(SensorCahaya) == LOW) {
    Serial.println("Cahaya Terang");
    digitalWrite(Aerator, LOW);
    tb.sendTelemetryString("Aerator", "OFF");
  } else {
    Serial.println("Cahaya Gelap");
    digitalWrite(Aerator, HIGH);
    tb.sendTelemetryString("Aerator", "ON");
  }

  // Penstabil Suhu Air, sekaligus menjaga ketinggian air kolam
  if (data.temperature < 26 )
  {
    digitalWrite(Kincir, HIGH);
    tb.sendTelemetryString("Kincir Air", "ON");
    // Utk mencegah defisit air, kita batasi utk pengeluaran debit air di tinggi air minimal 125cm
    if(tinggi > 125)
    {
    digitalWrite(PompaKeluar, HIGH);
    digitalWrite(PompaMasuk, LOW);
    tb.sendTelemetryString("Pompa Keluar", "ON");
    tb.sendTelemetryString("Pompa Masuk", "OFF");
    }
    else
    {
    digitalWrite(PompaKeluar, LOW);
    digitalWrite(PompaMasuk, LOW);
    tb.sendTelemetryString("Pompa Keluar", "OFF");
    tb.sendTelemetryString("Pompa Masuk", "OFF");
    }
    Serial.print("Suhu : ");
    Serial.print(data.temperature);
    Serial.println(" Terlalu Dingin");
    tb.sendTelemetryString("Kondisi Suhu Air", "Terlalu Dingin");
    Serial.print("Tinggi Air : ");
    Serial.print(tinggi);
    Serial.println(" CM");
  }

  else if (data.temperature >= 26 && data.temperature <= 34)
  {
    digitalWrite(Kincir, LOW);
    tb.sendTelemetryString("Kincir Air", "OFF"); 
    if(tinggi < 110) // Menjaga ketinggain air tambak di 110cm
    {
    digitalWrite(PompaMasuk, HIGH);
    digitalWrite(PompaKeluar, LOW);
    tb.sendTelemetryString("Pompa Keluar", "OFF");
    tb.sendTelemetryString("Pompa Masuk", "ON");
    }
    else
    {
    digitalWrite(PompaMasuk, LOW);
    digitalWrite(PompaKeluar, LOW);
    tb.sendTelemetryString("Pompa Keluar", "OFF");
    tb.sendTelemetryString("Pompa Masuk", "OFF");
    }
    Serial.print("Suhu : ");
    Serial.print(data.temperature);
    Serial.println(" Normal");
    tb.sendTelemetryString("Kondisi Suhu Air", "Normal");
    Serial.print("Tinggi Air : ");
    Serial.print(tinggi);
    Serial.println(" CM");
  }

  else
  {
    digitalWrite(Kincir, LOW);
    digitalWrite(PompaKeluar, HIGH);
    digitalWrite(PompaMasuk, HIGH);
    tb.sendTelemetryString("Kincir Air", "OFF");
    tb.sendTelemetryString("Pompa Keluar", "ON");
    tb.sendTelemetryString("Pompa Masuk", "ON");
    Serial.print("Suhu : ");
    Serial.print(data.temperature);
    Serial.println(" Terlalu Panas");
    tb.sendTelemetryString("Kondisi Suhu Air", "Terlalu Panas");
    Serial.print("Tinggi Air : ");
    Serial.print(tinggi);
    Serial.println(" CM");
  }

  tb.loop();
}