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

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

// MQTT Broker
const char* mqtt_server = "test.mosquitto.org";
const int mqtt_port = 1883;

// MQTT topics for each LED
const char* greenTopicA = "device/mano/data/traffic/green/A";
const char* yellowTopicA = "device/mano/data/traffic/yellow/A";
const char* redTopicA = "device/mano/data/traffic/red/A";
const char* greenTopicB = "device/mano/data/traffic/green/B";
const char* yellowTopicB = "device/mano/data/traffic/yellow/B";
const char* redTopicB = "device/mano/data/traffic/red/B";
const char* greenTopicC = "device/mano/data/traffic/green/C";
const char* yellowTopicC = "device/mano/data/traffic/yellow/C";
const char* redTopicC = "device/mano/data/traffic/red/C";
const char* greenTopicD = "device/mano/data/traffic/green/D";
const char* yellowTopicD = "device/mano/data/traffic/yellow/D";
const char* redTopicD = "device/mano/data/traffic/red/D";

// Define the pins for controlling the traffic lights
const int pinGreenA = 2;
const int pinYellowA = 4;
const int pinRedA = 16;
const int pinGreenB = 17;
const int pinYellowB = 5;
const int pinRedB = 18;
const int pinGreenC = 19;
const int pinYellowC = 21;
const int pinRedC = 22;
const int pinGreenD = 23;
const int pinYellowD = 15;
const int pinRedD = 13;

// Define the durations for each light phase in milliseconds
const unsigned long greenDuration = 10000; // 10 seconds
const unsigned long yellowDuration = 1500; // 1.5 seconds
const unsigned long redDuration = 10000;   // 10 seconds
const unsigned long zebra_crossDuration = 5000; // 5 seconds
// MQTT keep alive interval in seconds
const int keepAliveInterval = 60; // 1 minute
// WiFi client
WiFiClient espClient;

// MQTT client
PubSubClient client(espClient);

// Function prototypes
void setLights(int pinGreen, int pinYellow, int pinRed, bool green, bool yellow, bool red);
void connectToWiFi();
void connectToMQTT();
void callback(char* topic, byte* payload, unsigned int length);

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

  // Initialize pins as outputs
  pinMode(pinGreenA, OUTPUT);
  pinMode(pinYellowA, OUTPUT);
  pinMode(pinRedA, OUTPUT);
  pinMode(pinGreenB, OUTPUT);
  pinMode(pinYellowB, OUTPUT);
  pinMode(pinRedB, OUTPUT);
  pinMode(pinGreenC, OUTPUT);
  pinMode(pinYellowC, OUTPUT);
  pinMode(pinRedC, OUTPUT);
  pinMode(pinGreenD, OUTPUT);
  pinMode(pinYellowD, OUTPUT);
  pinMode(pinRedD, OUTPUT);
// Copyright
  Serial.println("Coded by Manomoy Maity :)");
  // Connect to WiFi
  connectToWiFi();

  // Connect to MQTT
  connectToMQTT();
}

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

  // Phase 1: A and C green, B and D red
  setLights(pinGreenA, pinYellowA, pinRedA, true, false, false);
  setLights(pinGreenB, pinYellowB, pinRedB, false, false, true);
  setLights(pinGreenC, pinYellowC, pinRedC, true, false, false);
  setLights(pinGreenD, pinYellowD, pinRedD, false, false, true);
  client.publish(greenTopicA,"on");
  client.publish(greenTopicB,"off");
  client.publish(greenTopicC,"on");
  client.publish(greenTopicD,"off");

  client.publish(redTopicA,"off");
  client.publish(redTopicB,"on");
  client.publish(redTopicC,"off");
  client.publish(redTopicD,"on");

  client.publish(yellowTopicA,"off");
  client.publish(yellowTopicB,"off");
  client.publish(yellowTopicC,"off");
  client.publish(yellowTopicD,"off");

  delay(greenDuration);

  // Phase 2: A and C yellow, B and D red
  setLights(pinGreenA, pinYellowA, pinRedA, false, true, false);
  setLights(pinGreenB, pinYellowB, pinRedB, false, false, true);
  setLights(pinGreenC, pinYellowC, pinRedC, false, true, false);
  setLights(pinGreenD, pinYellowD, pinRedD, false, false, true);
  client.publish(greenTopicA,"off");
  client.publish(greenTopicB,"off");
  client.publish(greenTopicC,"off");
  client.publish(greenTopicD,"off");

  client.publish(redTopicA,"off");
  client.publish(redTopicB,"on");
  client.publish(redTopicC,"off");
  client.publish(redTopicD,"on");

  client.publish(yellowTopicA,"on");
  client.publish(yellowTopicB,"off");
  client.publish(yellowTopicC,"on");
  client.publish(yellowTopicD,"off");

  delay(yellowDuration);

  // Phase 3: A and C red, B and D green
  setLights(pinGreenA, pinYellowA, pinRedA, false, false, true);
  setLights(pinGreenB, pinYellowB, pinRedB, true, false, false);
  setLights(pinGreenC, pinYellowC, pinRedC, false, false, true);
  setLights(pinGreenD, pinYellowD, pinRedD, true, false, false);
  client.publish(greenTopicA,"off");
  client.publish(greenTopicB,"on");
  client.publish(greenTopicC,"off");
  client.publish(greenTopicD,"on");

  client.publish(redTopicA,"on");
  client.publish(redTopicB,"off");
  client.publish(redTopicC,"on");
  client.publish(redTopicD,"off");

  client.publish(yellowTopicA,"off");
  client.publish(yellowTopicB,"off");
  client.publish(yellowTopicC,"off");
  client.publish(yellowTopicD,"off");

  delay(redDuration);

  // Phase 4: A and C red, B and D yellow
  setLights(pinGreenA, pinYellowA, pinRedA, false, false, true);
  setLights(pinGreenB, pinYellowB, pinRedB, false, true, false);
  setLights(pinGreenC, pinYellowC, pinRedC, false, false, true);
  setLights(pinGreenD, pinYellowD, pinRedD, false, true, false);
  client.publish(greenTopicA,"off");
  client.publish(greenTopicB,"off");
  client.publish(greenTopicC,"off");
  client.publish(greenTopicD,"off");

  client.publish(redTopicA,"on");
  client.publish(redTopicB,"off");
  client.publish(redTopicC,"on");
  client.publish(redTopicD,"off");

  client.publish(yellowTopicA,"off");
  client.publish(yellowTopicB,"on");
  client.publish(yellowTopicC,"off");
  client.publish(yellowTopicD,"on");

  delay(yellowDuration);

  // Phase 5: All red
  setLights(pinGreenA, pinYellowA, pinRedA, false, false, true);
  setLights(pinGreenB, pinYellowB, pinRedB, false, false, true);
  setLights(pinGreenC, pinYellowC, pinRedC, false, false, true);
  setLights(pinGreenD, pinYellowD, pinRedD, false, false, true);
  client.publish(greenTopicA,"off");
  client.publish(greenTopicB,"off");
  client.publish(greenTopicC,"off");
  client.publish(greenTopicD,"off");

  client.publish(redTopicA,"on");
  client.publish(redTopicB,"on");
  client.publish(redTopicC,"on");
  client.publish(redTopicD,"on");

  client.publish(yellowTopicA,"off");
  client.publish(yellowTopicB,"off");
  client.publish(yellowTopicC,"off");
  client.publish(yellowTopicD,"off");

  delay(zebra_crossDuration);

}

void setLights(int pinGreen, int pinYellow, int pinRed, bool green, bool yellow, bool red) {
  digitalWrite(pinGreen, green);
  digitalWrite(pinYellow, yellow);
  digitalWrite(pinRed, red);
}

void connectToWiFi() {
  WiFi.begin(ssid, password);
  Serial.print("Connecting to WiFi");
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("WiFi connected");
}

void connectToMQTT() {
  client.setServer(mqtt_server, mqtt_port);
  client.setCallback(callback);
  client.setKeepAlive(keepAliveInterval); // Set MQTT keep-alive interval

  while (!client.connected()) {
    Serial.print("Connecting to MQTT...");
    if (client.connect("espClient")) {
      Serial.println("connected");
    } else {
      Serial.print("failed, rc=");
      Serial.print(client.state());
      Serial.println(" try again in 5 seconds");
      delay(5000);
    }
  }
}

void callback(char* topic, byte* payload, unsigned int length) {
  // Handle MQTT messages if needed
}