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

// WiFi credentials
const char* ssid = "POCO M3";
const char* password = "B4ru1236";

// MQTT broker
const char* mqtt_server = "broker.emqx.io"; // Ganti dengan broker lokal jika diperlukan
const int mqtt_port = 1883;

// Topics
const char* relay_topic = "home/relay";

// Relay pins
const int relay1 = 23;
const int relay2 = 22;
const int relay3 = 21;
const int relay4 = 19;

// MQTT client
WiFiClient espClient;
PubSubClient client(espClient);

// Callback function
void callback(char* topic, byte* payload, unsigned int length) {
  payload[length] = '\0';
  String message = String((char*)payload);

  // Control relay
  if (message == "ON1") digitalWrite(relay1, LOW); // LOW to activate
  if (message == "OFF1") digitalWrite(relay1, HIGH);
  if (message == "ON2") digitalWrite(relay2, LOW);
  if (message == "OFF2") digitalWrite(relay2, HIGH);
  if (message == "ON3") digitalWrite(relay3, LOW);
  if (message == "OFF3") digitalWrite(relay3, HIGH);
  if (message == "ON4") digitalWrite(relay4, LOW);
  if (message == "OFF4") digitalWrite(relay4, HIGH);
}

void setup_wifi() {
  delay(10);
  Serial.println("Connecting to WiFi...");
  WiFi.begin(ssid, password);

  while (WiFi.status() != WL_CONNECTED) {
    delay(1000);
    Serial.println("Connecting...");
  }
  Serial.println("Connected to WiFi.");
}

void reconnect() {
  while (!client.connected()) {
    Serial.println("Connecting to MQTT...");
    if (client.connect("ESP32Client")) {
      Serial.println("Connected to MQTT.");
      client.subscribe(relay_topic);
    } else {
      Serial.print("Failed, rc=");
      Serial.print(client.state());
      Serial.println(" retrying in 5 seconds...");
      delay(5000);
    }
  }
}

void setup() {
  Serial.begin(115200);

  pinMode(relay1, OUTPUT);
  pinMode(relay2, OUTPUT);
  pinMode(relay3, OUTPUT);
  pinMode(relay4, OUTPUT);

  digitalWrite(relay1, HIGH);
  digitalWrite(relay2, HIGH);
  digitalWrite(relay3, HIGH);
  digitalWrite(relay4, HIGH);

  setup_wifi();
  client.setServer(mqtt_server, mqtt_port);
  client.setCallback(callback);
}

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