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

#define buzzerPin 14
unsigned long lastMsg = 0;
bool activate = false;

// Wifi
const char* ssid = "Wokwi-GUEST";
const char* password = "";


// MQTT Broker
const char* mqtt_server = "broker.mqttdashboard.com";
const char* clientID = "andres";
const int mqtt_port = 1883;
const char* mqtt_username = "";
const char* mqtt_password = "";
const char* topic = "alarmaiot";


WiFiClient espClient;
PubSubClient client(espClient);

void connect_wifi(){
  delay(10);
  WiFi.begin( ssid, password);

  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.println("Connecting to WiFi...");
  }
  Serial.println("");
  Serial.println("WiFi connected");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());
}

void activatealarm(){
  tone(buzzerPin, 150); // Generate a tone with the given volume
  delay(1000);
  noTone(buzzerPin); // Turn off the buzzer if sound is disabled
  delay(1000);
}

void callback(char* topic, byte* payload, unsigned int length) {
  Serial.print("Message arrived [");
  Serial.print(topic);
  Serial.print("] ");

  for (int i = 0; i < length; i++) {
    Serial.print((char)payload[i]);
  }
  Serial.println();

  // Switch on the LED if an 1 was received as first character
  if ( char(payload[0]) == '1' ) {
    activate = true;
  } else if( (char)payload[0] == '0' ){
     activate = false;
  } else {
    Serial.println("I dont understand :(" );
  }
}

void reconnect() {
  while(!client.connected()) {
    if (client.connect(clientID)) {
      Serial.println("\nMQTT connected");
      client.subscribe(topic);
      Serial.println("Topic Subscribed");
    }
    else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      delay(5000);  // wait 5sec and retry
    }
  }
}


void setup() {
  pinMode(buzzerPin, OUTPUT); // Set the buzzer pin as an output
  Serial.begin(115200);
  connect_wifi();

  client.setServer( mqtt_server, 1883 ); // Setting MQTT server
  client.setCallback( callback ); 
}


void loop() {
  if (!client.connected()) { // If client is not connected
    reconnect(); // Try to reconnect
  }

  client.loop();
  unsigned long now = millis();

  if (now - lastMsg > 2000) { //perintah publish data
    lastMsg = now;
    Serial.println("Listening...");
  }

  if( activate ){
    activatealarm();
  }
}