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

const char* ssid = "Wokwi-GUEST";
const char* password = "";
const char* mqtt_server = "io.adafruit.com";
const char* mqtt_user = "ayushsingh658";
const char* mqtt_password = "aio_txoj23sFekq2lTyojAzKg9a8ab9P";

WiFiClient espClient;
PubSubClient client(espClient);

void setup_wifi() {
  delay(10);
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
  }
}

void callback(char* topic, byte* payload, unsigned int length) {
  String command = "";
  for (int i = 0; i < length; i++) {
    command += (char)payload[i];
  }
  if (command == "turn on relay 1") {
    digitalWrite(5, HIGH);
  } else if (command == "turn off relay 1") {
    digitalWrite(5, LOW);
  } else if (command == "turn on relay 2") {
    digitalWrite(4, HIGH);
  } else if (command == "turn off relay 2") {
    digitalWrite(4, LOW);
  } else if (command == "turn on relay 3") {
    digitalWrite(3, HIGH);
  } else if (command == "turn off relay 3") {
    digitalWrite(3, LOW);
  } else if (command == "turn on relay 4") {
    digitalWrite(2, HIGH);
  } else if (command == "turn off relay 4") {
    digitalWrite(2, LOW);
  }
}

void reconnect() {
  while (!client.connected()) {
    String clientId = "ESP32Client-";
    clientId += String(random(0xffff), HEX);
    if (client.connect(clientId.c_str(), mqtt_user, mqtt_password)) {
      client.subscribe("your-mqtt-feed");
    }
    delay(1000);
  }
}

void setup() {
  pinMode(2, OUTPUT);
  pinMode(3, OUTPUT);
  pinMode(4, OUTPUT);
  pinMode(5, OUTPUT);
  setup_wifi();
  client.setServer(mqtt_server, 1883);
  client.setCallback(callback);
}

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