#include <WiFi.h>
#include <MQTT.h>

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

WiFiClient net;
MQTTClient client;


#define ledRed 26
#define ledYellow 27
#define ledGreen 14
#define pot 32

int bacaPot;

unsigned long lastMillis = 0;

void connect(){
  // Menghubungkan ke WiFi
  Serial.print("Cek WiFi");
  while(WiFi.status() != WL_CONNECTED){
    Serial.print(".");
    delay(1000);
  }
  Serial.println("");
  Serial.println("Sukses Terhubung ke WiFi");

  // Menghubungkan ke Broker
  Serial.print("Cek Broker");
  while(!client.connect("")){ // Client ID boleh dikosongkan
    Serial.print(".");
    delay(1000);
  }
  Serial.println("");
  Serial.println("Sukses Terhubung ke Server");
  
  client.subscribe("labskite/iot/#", 1); // subcribe dengan QoS 1
}

void subscribeData(String &topic, String &data){
  Serial.print("Topic: ");
  Serial.println(topic);
  Serial.print("Data: ");
  Serial.println(data);

  if(topic == "labskite/iot/led1"){  
    digitalWrite(ledRed, data.toInt());   //control led 1
     if (data == "1"){
        client.publish("labskite/iot/led1/status", "ON", true, 1);
     }
  }

  if(topic == "labskite/iot/led2"){
    digitalWrite(ledYellow, data.toInt());   //control led 2
    if (data == "1"){
      client.publish("labskite/iot/led2/status", "ON",true, 1);
    }
  }
  if(topic == "labskite/iot/led3"){
    digitalWrite(ledYellow, data.toInt());   //control led 2
    if (data == "1"){
      client.publish("labskite/iot/led3/status", "ON",true, 1);
    }
  }
  if(topic == "labskite/iot/led3"){
    digitalWrite(ledGreen, data.toInt());   //control led 3
    if (data == "1"){
      client.publish("labskite/iot/led3/status", "ON", true, 1);
    }
  }  
}

void publish(){
  int bacaPot = analogRead(pot);
  Serial.print("Topic: ");
  Serial.println(bacaPot);
  client.publish("labskite/iot/pot1", String(bacaPot), true, 1); // publish variabel pot ke topic labskite/iot/pot1 dengan QoS 1
  if (bacaPot == 0){
      digitalWrite(ledRed, LOW);
      digitalWrite(ledYellow, LOW);
      digitalWrite(ledGreen, LOW);
    }
    else if ((bacaPot > 0) && (bacaPot < 1000)){
      digitalWrite(ledRed, HIGH);
      digitalWrite(ledYellow, LOW);
      digitalWrite(ledGreen, LOW);
      client.publish("labskite/iot/led1", "ON", true, 1);
      client.publish("labskite/iot/led2", "OFF", true, 1);
      client.publish("labskite/iot/led3", "OFF", true, 1);
    }
    else if ((bacaPot >= 1000) && (bacaPot <= 2500)){
      digitalWrite(ledRed, LOW);
      digitalWrite(ledYellow, HIGH);
      digitalWrite(ledGreen, LOW);
      client.publish("labskite/iot/led1", "OFF", true, 1);
      client.publish("labskite/iot/led2", "ON", true, 1);
      client.publish("labskite/iot/led3", "OFF", true, 1);
    }
    else if (bacaPot >= 2500){
      digitalWrite(ledRed, LOW);
      digitalWrite(ledYellow, LOW);
      digitalWrite(ledGreen, HIGH);
      client.publish("labskite/iot/led1", "OFF", true, 1);
      client.publish("labskite/iot/led2", "OFF", true, 1);
      client.publish("labskite/iot/led3", "ON", true, 1);
    }

}

void setup() {
  pinMode(ledRed,OUTPUT); 
  pinMode(ledYellow,OUTPUT);
  pinMode(ledGreen,OUTPUT);
  pinMode(pot, INPUT);
  
  Serial.begin(9600);
  WiFi.begin(ssid, pass);

  client.begin("broker.emqx.io", net);
  client.onMessage(subscribeData); 
  connect();
}

void loop() {
  client.loop();
  // publish a message roughly every second.
 if (millis() - lastMillis > 1000) {
    lastMillis = millis();
    publish();
 }
    
  delay(10);
}