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

float pressure;
float leakage1;
float leakage2;
int buzzer;
int LED1;
int LED2;
int LED3;
int PRESSURE_THRESHOLD;
int LEAKAGE_THRESHOLD;

void callback (char* subscribetopic,byte* payload,unsigned int payloadLength);

#define ORG "h6ary6"
#define DEVICE_TYPE "abcd"
#define DEVICE_ID "1234"
#define TOKEN "12345678"
#define buzzer_pin 18
#define LED1_pin 2
#define LED2_pin 4
#define LED3_pin 5
String data3;


char server[] = ORG ".messaging.internetofthings.ibmcloud.com";
char publishTopic[] = "iot-2/evt/Data/fmt/json";
char subscribetopic[] = "iot-2/cmd/test/fmt/String";
char authMethod[] = "use-token-auth";
char token[] = TOKEN;
char clientId[] = "d" ORG ":" DEVICE_TYPE ":" DEVICE_ID;

WiFiClient wifiClient;
PubSubClient client(server, 1883, callback, wifiClient);

void setup()
{
  Serial.begin(115200);
  delay(10);
  Serial.println();
  wificonnect();
  mqttconnect();
}

void loop()
{
  pressure=analogRead(24);
  leakage1=analogRead(35);
  leakage2=analogRead(32);
  pinMode(buzzer,OUTPUT);
  pinMode(LED1,OUTPUT);
  pinMode(LED2,OUTPUT);
  pinMode(LED3,OUTPUT);

  Serial.print("Pressure: ");
  Serial.println(pressure);
  Serial.print("Leakage 1: ");
  Serial.println(leakage1);
  Serial.print("Leakage 2: ");
  Serial.println(leakage2);
  delay(1000);

  publishData(pressure,leakage1,leakage2,buzzer,LED1,LED2,LED3,PRESSURE_THRESHOLD,LEAKAGE_THRESHOLD);
  delay(1000);

  if (!client.loop())
  {
    mqttconnect();
  }
}

void publishData(float pressure,float leakage1,float leakage2,int buzzer,int LED1,int LED2,int LED3,int PRESSURE_THRESHOLD,int LEAKAGE_THRESHOLD)
{
  mqttconnect();

  String payload = "{\"pressure\":";
  payload += pressure;
  payload +="," "\"leakage1\":";
  payload += leakage1;
  payload += "," "\"leakage2\":";
  payload += leakage2;
  payload += "}";
 
  PRESSURE_THRESHOLD=1500;
  LEAKAGE_THRESHOLD=1000;

  if (pressure < PRESSURE_THRESHOLD)
  {
    tone(buzzer,2000,500);
    digitalWrite(LED1,HIGH);
    digitalWrite(LED2,LOW);
    digitalWrite(LED3,LOW);
  }
  else if (leakage1 > LEAKAGE_THRESHOLD)
  {
    tone(buzzer,2000,500);
    digitalWrite(LED1,LOW);
    digitalWrite(LED2,HIGH);
    digitalWrite(LED3,LOW); 
  }
  else if (leakage2 > LEAKAGE_THRESHOLD)
  {
     tone(buzzer,2000,500);
    digitalWrite(LED1,LOW);
    digitalWrite(LED2,LOW);
    digitalWrite(LED3,HIGH);
  }  
  Serial.print("Sending payload: ");
  Serial.println(payload);

  if (client.publish(publishTopic,(char*) payload.c_str()))
  {
    Serial.println("Publish ok");
  }
  else
  {
    Serial.println("Publish failed");
  }
}

void mqttconnect()
{
  if (!client.connected())
  {
    Serial.print("Reconnecting client to ");
    Serial.println(server);
    while (!!!client.connect(clientId, authMethod, token))
    {
      Serial.print(".");
      delay(500);
    }
    initManagedDevice();
    Serial.println();
  }
}

void wificonnect()
{
  Serial.println();
  Serial.print("Connecting to ");

  WiFi.begin("Wokwi-GUEST", "", 6);
  while (WiFi.status() != WL_CONNECTED) 
  {
    delay(500);
    Serial.print(".");
  }
  Serial.println("");
  Serial.println("WiFi connected");
  Serial.println("IP address: ");
  Serial.println(WiFi.localIP());
}

void initManagedDevice()
{
  if (client.subscribe(subscribetopic)) 
  {
    Serial.println((subscribetopic));
    Serial.println("subscribe to cmd OK");
  } 
  else 
  {
    Serial.println("subscribe to cmd FAILED");
  }
}

void callback(char* subscribetopic, byte* payload, unsigned int payloadLength) 
{
  
  Serial.print("callback invoked for topic: ");
  Serial.println(subscribetopic);
 
  for (int i = 0; i < payloadLength; i++) 
  {
    data3 += (char)payload[i];
  }
  
  Serial.println("data: "+ data3); 
  
data3="";
}