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

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

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

const int DHT_PIN = 15;
const int TRIG_PIN = 13;
const int ECHO_PIN = 12;

DHTesp dhtSensor;
WiFiClient espClient;
ThingsBoard tb(espClient);

// Connect to Wi-Fi
void connectWiFi() {
  Serial.println("Connecting to WiFi...");
  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println();
  Serial.println("Connected to WiFi");
}

// Connect to ThingsBoard
void connectThingsBoard() {
  if (!tb.connected()) {
    Serial.print("Connecting to ThingsBoard...");
    if (!tb.connect(THINGSBOARD_SERVER, TOKEN)) {
      Serial.println("Failed to connect");
      return;
    }
    Serial.println("Connected to ThingsBoard");
  }
}

// Read distance from sensor and calculate water level
float readWaterLevel() {
  digitalWrite(TRIG_PIN, LOW);
  delayMicroseconds(2);
  digitalWrite(TRIG_PIN, HIGH);
  delayMicroseconds(10);
  digitalWrite(TRIG_PIN, LOW);

  int duration = pulseIn(ECHO_PIN, HIGH);
  float distance = duration * 0.034 / 2;
  float waterLevel = 400 - distance;
  return waterLevel;
}

void setup() {
  Serial.begin(115200);
  
  // Connect to Wi-Fi
  connectWiFi();
  
  // Initialize DHT sensor
  dhtSensor.setup(DHT_PIN, DHTesp::DHT22);
  
  // Set pin modes for ultrasonic sensor
  pinMode(TRIG_PIN, OUTPUT);
  pinMode(ECHO_PIN, INPUT);
}

void loop() {
  delay(2000);

  // Check Wi-Fi connection
  if (WiFi.status() != WL_CONNECTED) {
    connectWiFi();
  }

  // Check ThingsBoard connection
  connectThingsBoard();

  // Read sensor data
  float waterLevel = readWaterLevel();
  TempAndHumidity data = dhtSensor.getTempAndHumidity();

  Serial.println("Sending data...");
  
  // Send telemetry to ThingsBoard
  tb.sendTelemetryFloat("Suhu", data.temperature);
  tb.sendTelemetryFloat("Kelembaban", data.humidity);
  tb.sendTelemetryFloat("Ketinggian_Air", waterLevel);

  // Print sensor readings
  Serial.print("Suhu: ");
  Serial.print(data.temperature);
  Serial.print(" Kelembaban: ");
  Serial.print(data.humidity);
  Serial.print(" Ketinggian Air: ");
  Serial.print(waterLevel);
  Serial.println(" CM");

  // Maintain ThingsBoard connection
  tb.loop();
}