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

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

WiFiClient net;
MQTTClient client;
const int pinRed = 2;
const int pinGreen = 4;
const int pinBlue = 16;
const int pinRelay = 33;

void connect() {
  rgb(1,0,0);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
  }
  rgb(0,1,0);

  while (!client.connect("clientid-led")) {
    delay(1000);
  }
  rgb(0,0,1);

  client.subscribe("kelasiot/lampu", 1);  // Subscribe to the specific topic
}

void subscribe(String &topic, String &payload) {
  if(topic == "kelasiot/lampu") {
    if(payload == "nyala") {
      digitalWrite(pinRelay, 0);  
    } else if(payload == "mati") {
      digitalWrite(pinRelay, 1);   
    }
  }
}

void setup() {
  pinMode(pinRed, OUTPUT);
  pinMode(pinGreen, OUTPUT);
  pinMode(pinBlue, OUTPUT);
  pinMode(pinRelay, OUTPUT);

  WiFi.begin(ssid, pass);
  client.begin("broker.emqx.io", net);
  client.onMessage(subscribe);

  connect();
}

void loop() {
  client.loop();
  delay(10);  

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

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