#include <PubSubClient.h>
#include <WiFi.h>
#define Push_Button_pin_1 17
#define Push_Button_pin_2 16
#define Push_Button_pin_3 4


const char* ssid = "Wokwi-GUEST";
const char* password = "";
WiFiClient wifiClient;
const char* mqttServer = "broker.hivemq.com";
const int mqttPort = 1883;
const char* mqttUser = "";
const char* mqttPassword = "";
const char* topic_light1 = "ITKPS/Lab/bellging/light_g";
const char* topic_light2 = "ITKPS/Lab/bellging/light_y";
const char* topic_speaker = "ITKPS/Lab/bellging/speaker";


PubSubClient mqttClient(wifiClient);
int pushState1 = 0;
int pushState2 = 0;
int pushState3 = 0;


void setup_wifi() {
  Serial.println();
  Serial.print("Connecting to ");
  Serial.println(ssid);
  WiFi.begin(ssid, password);
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println();
  Serial.print("WiFi connected on ");
  Serial.print("IP address: ");
  Serial.println(WiFi.localIP());
}


void reconnectMQTT() {
  while (!mqttClient.connected()) {
    Serial.print("Attempting MQTT connection...");
    String clientId = "ESP32Client-" + String(random(0xffff), HEX);
    if (mqttClient.connect(clientId.c_str())) {
      Serial.println(" connected");
    } else {
      Serial.print("failed, rc=");
      Serial.print(mqttClient.state());
      Serial.println(" try again in 5 seconds");
      delay(5000);
    }
  }
}


void setup() {
  Serial.begin(115200);
  setup_wifi();
  mqttClient.setServer(mqttServer, mqttPort);
  pinMode(Push_Button_pin_1, INPUT_PULLUP);
  pinMode(Push_Button_pin_2, INPUT_PULLUP);
  pinMode(Push_Button_pin_3, INPUT_PULLUP);
}


void loop() {
  if (!mqttClient.connected()) {
    reconnectMQTT();
  }
  mqttClient.loop();


  int newPushState1 = digitalRead(Push_Button_pin_1);
  int newPushState2 = digitalRead(Push_Button_pin_2);
  int newPushState3 = digitalRead(Push_Button_pin_3);
 
  if (newPushState1 != pushState1) {
    pushState1 = newPushState1;
   
    String light1Message = String(pushState1);
    mqttClient.publish(topic_light1, light1Message.c_str());
  }
  else if (newPushState2 != pushState2) {
    pushState2 = newPushState2;
   
    String light2Message = String(pushState2);
    mqttClient.publish(topic_light2, light2Message.c_str());
  }
  else if (newPushState3 != pushState3) {
    pushState3 = newPushState3;
   
    String speakerMessage = String(pushState3);
    mqttClient.publish(topic_speaker, speakerMessage.c_str());
  }
  delay(1000);
}