#include <WiFi.h>
#include <MQTT.h>
#include <ESP32Servo.h>
#include <NusabotSimpleTimer.h>
#include "DHTesp.h"

WiFiClient net;
MQTTClient client;
Servo servo1;
Servo servo2;
NusabotSimpleTimer timer;
DHTesp dhtSensor;

const char ssid[] = "Wokwi-GUEST";
const char pass[] = "";

const int pinRed = 2;
const int pinGreen = 4;
const int pinBlue = 16;
const int pinLED = 12;
const int pinServo1 = 26;
const int pinDHT = 21;

String serial_number = "87654321";

void setup() {
  pinMode(pinRed, OUTPUT);
  pinMode(pinGreen, OUTPUT);
  pinMode(pinBlue, OUTPUT);
  pinMode(pinLED, OUTPUT);
  servo1.attach(pinServo1, 500, 2400);
  dhtSensor.setup(pinDHT, DHTesp::DHT22);

  WiFi.begin(ssid, pass);
  client.begin("hamsur.cloud.shiftr.io", net);

  client.onMessage(subscribe);
  timer.setInterval(2000, publishDHT);

  connect();
}

void loop() {
  client.loop();

  timer.run();
  if(!client.connected()){
    connect();
  }

  delay(10);
}

void subscribe(String &topic, String &data){
  if(topic == "kelasiot/"+ serial_number +"/led"){
    if(data == "nyala"){
      digitalWrite(pinLED, 1);
    } else if(data == "mati"){
      digitalWrite(pinLED, 0);
    }
  }
  if (topic == "kelasiot/" + serial_number + "/servo") {
      int posServo = data.toInt();
      servo1.write(posServo);  // Servo 1
  } 
}

void publishDHT(){
  TempAndHumidity  data = dhtSensor.getTempAndHumidity();

  client.publish("kelasiot/"+ serial_number +"/suhu", String(data.temperature, 2), true, 1);
  client.publish("kelasiot/"+ serial_number +"/kelembapan", String(data.humidity, 1), true, 1);
}

void rgb(bool red, bool green, bool blue){
  digitalWrite(pinRed, red);
  digitalWrite(pinGreen, green);
  digitalWrite(pinBlue, blue);
}

void connect(){
  rgb(1,0,0); // Merah
  while(WiFi.status() != WL_CONNECTED){
    delay(500);
  }
  rgb(0,1,0); // Hijau

  client.setWill("kelasiot/status/87654321", "offline", true, 1);
  while(!client.connect("efbf3290", "hamsur", "oexmWlZWdMCuh7fL")){
    delay(500);
  }
  rgb(0,0,1); // Biru
  client.publish("kelasiot/status/87654321", "online", true, 1);
  client.subscribe("kelasiot/#", 1);  // Subscribe menggunakan QoS 1
}