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

// MQTT broker
const char* mqtt_server = "broker.hivemq.com"; // Adresa brokeru
const int mqtt_port = 1883;                   // Port brokeru
const char* mqtt_topic = "esp32/test";  // Téma, kam budeme odosielať dáta

WiFiClient espClient;
PubSubClient client(espClient);

#define red 5
#define green 18
#define blue 19

const int pins_led[3] = {red, green, blue};
int ulozene[3];

// Callback funkcia na spracovanie prichádzajúcich správ
void callback(char* topic, byte* payload, unsigned int length) {
  Serial.print("Prijatá správa na tému: ");
  Serial.println(topic);

  // Konverzia payload na string
  String message = "";
  for (unsigned int i = 0; i < length; i++) {
    message += (char)payload[i];
  }
  Serial.print("Správa: ");
  Serial.println(message);

  // Na základe prichádzajúcej správy môžeme vykonať nejakú akciu, napríklad rozsvietiť LED
  if (message == "LED_ON") {
    for (int i = 0; i < 3; i++) {
      digitalWrite(pins_led[i], HIGH);
    }
  } else if (message == "LED_OFF") {
    for (int i = 0; i < 3; i++) {
      digitalWrite(pins_led[i], LOW);
    }
  }
}

void setup() {

  WiFi.mode(WIFI_STA);

  Serial.begin(9600);
  WiFi.begin("Wokwi-GUEST", "", 6);
  while (WiFi.status() != WL_CONNECTED) 
  {
    delay(250);
    Serial.print(".");
  }
  Serial.print("WIFI uspesne pripojena: ");
  Serial.println(WiFi.localIP());

  client.setServer("broker.hivemq.com", 1883);
  client.setCallback(callback);
  Serial.begin(9600);

  // Nastavenie LED
  pinMode(red, OUTPUT);
  pinMode(green, OUTPUT);
  pinMode(blue, OUTPUT);

  digitalWrite(red, LOW);
  digitalWrite(green, LOW);
  digitalWrite(blue, LOW);

}

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

  // Generovanie náhodných stavov pre LED
  for (int i = 0; i <= 2; i++) {
    int ran = random(0, 2); // Náhodne vyber 0 alebo 1
    digitalWrite(pins_led[i], ran);
    ulozene[i] = ran;       // Ulož hodnotu do poľa
  }

  // Poslanie dát na MQTT server
  String message = "[";
  for (int i = 0; i <= 2; i++) {
    message += String(ulozene[i]);
    if (i < 2) message += ",";
  }
  message += "]";

  Serial.print("Odosielam na MQTT: ");
  Serial.println(message);

  client.publish(mqtt_topic, message.c_str()); // Poslanie na MQTT server

  delay(5000); // Čakaj 5 sekúnd
}

// Funkcia na opätovné pripojenie k MQTT serveru
void reconnect() {
  while (!client.connected()) {
    Serial.print("Pripájam sa na MQTT broker...");
    if (client.connect("ESP32_Client")) {
      Serial.println("Pripojené!");
      // Po pripojení sa prihlásime na tému
      client.subscribe("esp32/test");  // Prihlásenie na tému pre prichádzajúce správy
    } else {
      Serial.print("Zlyhalo, kód: ");
      Serial.print(client.state());
      Serial.println(". Skúšam znova o 5 sekúnd.");
      delay(5000);
    }
  }
}