#include <Wire.h>
#include <WiFi.h>
#include <PubSubClient.h>
#include <LiquidCrystal_I2C.h>

const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqtt_server = "broker.hivemq.com";
const int mqtt_port = 1883;

WiFiClient espClient;
PubSubClient client(espClient);

const int SDA_PIN = 21; // Pin SDA
const int SCL_PIN = 22; // Pin SCL

LiquidCrystal_I2C lcd(0x27, 20, 4); // Alamat I2C, kolom, baris, pin SDA, pin SCL

String tegangan, arus, daya, debitAir;

void callback(char* topic, byte* message, unsigned int length) {
  Serial.print("Pesan diterima dari topik: ");
  Serial.println(topic);

  String receivedMessage = "";
  for (int i = 0; i < length; i++) {
    receivedMessage += (char)message[i];
  }
  Serial.print("Pesan: ");
  Serial.println(receivedMessage);

  if (strcmp(topic, "/UAS/Tgn") == 0) {
    tegangan = receivedMessage + " V";
  } else if (strcmp(topic, "/UAS/Ars") == 0) {
    arus = receivedMessage + " A";
  } else if (strcmp(topic, "/UAS/dya") == 0) {
    daya = receivedMessage + " W";
  } else if (strcmp(topic, "/UAS/dbt") == 0) {
    debitAir = receivedMessage + " L/s";
  }

  lcd.clear();
  lcd.setCursor(0, 0);
  lcd.print("Tegangan: ");
  lcd.print(tegangan);

  lcd.setCursor(0, 1);
  lcd.print("Arus: ");
  lcd.print(arus);

  lcd.setCursor(0, 2);
  lcd.print("Daya: ");
  lcd.print(daya);

  lcd.setCursor(0, 3);
  lcd.print("Debit Air: ");
  lcd.print(debitAir);
}

void reconnect() {
  while (!client.connected()) {
    Serial.println("Menghubungkan ke MQTT...");
    if (client.connect("ESP32-Client")) {
      Serial.println("Terhubung ke broker MQTT");
      client.subscribe("/UAS/Tgn");
      client.subscribe("/UAS/Ars");
      client.subscribe("/UAS/dya");
      client.subscribe("/UAS/dbt");
    } else {
      Serial.print("Gagal terhubung ke MQTT, status = ");
      Serial.println(client.state());
      delay(5000);
    }
  }
}

void setup() {
  Serial.begin(9600);
  lcd.init();
  lcd.backlight();

  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
    Serial.println("Koneksi WiFi...");
  }
  Serial.println("Terhubung ke WiFi");

  client.setServer(mqtt_server, mqtt_port);
  client.setCallback(callback);

  if (client.connect("ESP32-Client")) {
    Serial.println("Terhubung ke broker MQTT");
    client.subscribe("/UAS/Tgn");
    client.subscribe("/UAS/Ars");
    client.subscribe("/UAS/dya");
    client.subscribe("/UAS/dbt");
  } else {
    Serial.print("Gagal terhubung ke MQTT, status = ");
    Serial.println(client.state());
  }
}

void loop() {
  if (!client.connected()) {
    reconnect();
  }
  client.loop();
}