#include <WiFi.h>
#include <PubSubClient.h>
#define LED 22

// WiFi
const char *ssid = "Wokwi-GUEST"; // Enter your WiFi name
const char *password = "";  // Enter WiFi password

// MQTT Broker
const char *mqtt_broker = "broker.emqx.io";
const char *topic = "test/nyalain/lampu";
const char *mqtt_username = "emqx";
const char *mqtt_password = "public";
const int mqtt_port = 1883;

WiFiClient espClient;
PubSubClient client(espClient);

void turn_on_lamp(){
  digitalWrite(LED, HIGH);
  Serial.println("Turning on lamp");
}

void turn_off_lamp(){
  digitalWrite(LED, LOW);
  Serial.println("Turning off lamp");
}

int parseMessage(String message) {
  message.toLowerCase();  // Convert message to lowercase for case-insensitive comparison
  if (message.indexOf("on") != -1) {
    return 1;  // Lamp on
  } else if (message.indexOf("off") != -1) {
    return 0;  // Lamp off
  } else {
    return -1;  // Invalid message
  }
}

void controlLamp(int state) {
  if (state == 1) {
    turn_on_lamp();
  } else if (state == 0) {
    turn_off_lamp();
    Serial.println("Lamp turned off");
  } else {
    Serial.println("Invalid lamp command");
  }
}

void setup() {
  //setup LED
  pinMode(LED, OUTPUT);
 // Set software serial baud to 115200;
 Serial.begin(115200);
 // connecting to a WiFi network
 WiFi.begin(ssid, password);
 while (WiFi.status() != WL_CONNECTED) {
     delay(500);
     Serial.println("Connecting to WiFi..");
 }
 Serial.println("Connected to the WiFi network");
 //connecting to a mqtt broker
 client.setServer(mqtt_broker, mqtt_port);
 client.setCallback(callback);
 while (!client.connected()) {
     String client_id = "esp32-client-";
     client_id += String(WiFi.macAddress());
     Serial.printf("The client %s connects to the public mqtt broker\n", client_id.c_str());
     if (client.connect(client_id.c_str(), mqtt_username, mqtt_password)) {
         Serial.println("Public emqx mqtt broker connected");
     } else {
         Serial.print("failed with state ");
         Serial.print(client.state());
         delay(2000);
     }
 }
 // publish and subscribe
 client.publish(topic, "Hi EMQX I'm ESP32 ^^");
 client.subscribe(topic);
}

void callback(char *topic, byte *payload, unsigned int length) {
 Serial.print("Message arrived in topic: ");
 Serial.println(topic);
 Serial.print("Message:");
  String message;
  for (int i = 0; i < length; i++) {
    message += (char)payload[i];
  }
  
  controlLamp(parseMessage(message));
 Serial.println("-----------------------");
}



void loop() {
 client.loop();
}