#include <WiFi.h>
#include <WiFiClientSecure.h>
#include <UniversalTelegramBot.h>

// Wifi network station credentials
#define WIFI_SSID "Wokwi-GUEST"
#define WIFI_PASSWORD ""
// Telegram BOT Token (Get from Botfather)
#define BOT_TOKEN "6941741151:AAH6GOhN0WnUQlAHAc8P09kO-8bLXDV_7d4"

const unsigned long BOT_MTBS = 1000; // mean time between scan messages

WiFiClientSecure secured_client;
UniversalTelegramBot bot(BOT_TOKEN, secured_client);
unsigned long bot_lasttime; // last time messages' scan has been done

const int ledPin_red = 5;
const int ledPin_yellow = 16;
const int ledPin_green = 17;
int greenStatus = 0;
int redStatus =0;
int  yellowStatus =0;

void handleNewMessages(int numNewMessages)
{
  Serial.print("handleNewMessages ");
  Serial.println(numNewMessages);

  for (int i = 0; i < numNewMessages; i++)
  {
    String chat_id = bot.messages[i].chat_id;
    String text = bot.messages[i].text;

    String from_name = bot.messages[i].from_name;
    if (from_name == "")
      from_name = "Guest";

    if (text == "/redon")
    {
      digitalWrite(ledPin_red, HIGH); // turn the LED on (HIGH is the voltage level)
      redStatus = 1;
      bot.sendMessage(chat_id, "RED is ON", "");
    }

    if (text == "/redoff")
    {
      redStatus = 0;
      digitalWrite(ledPin_red, LOW); // turn the LED off (LOW is the voltage level)
      bot.sendMessage(chat_id, "RED is OFF", "");
    }
//
    if (text == "/yellowon")
    {
      digitalWrite(ledPin_yellow, HIGH); // turn the LED on (HIGH is the voltage level)
      yellowStatus = 1;
      bot.sendMessage(chat_id, "Yellow is ON", "");
    }

    if (text == "//yellowoff")
    {
      yellowStatus = 0;
      digitalWrite(ledPin_yellow, LOW); // turn the LED off (LOW is the voltage level)
      bot.sendMessage(chat_id, "Yellow is OFF", "");
    }
//2green
    if (text == "/greenon")
    {
      digitalWrite(ledPin_green, HIGH); // turn the LED on (HIGH is the voltage level)
      greenStatus = 1;
      bot.sendMessage(chat_id, "Green is ON", "");
    }

    if (text == "/greenoff")
    {
      greenStatus = 0;
      digitalWrite(ledPin_green, LOW); // turn the LED off (LOW is the voltage level)
      bot.sendMessage(chat_id, "Green OFF", "");
    }

    if (text == "/status")
    {
      if (redStatus==1)
      {
        bot.sendMessage(chat_id, "Red is ON", "");
      }
      if (redStatus==0)
      {
        bot.sendMessage(chat_id, "Red is OFF", "");
      }
    
    if (yellowStatus==1)
      {
        bot.sendMessage(chat_id, "yellow is ON", "");
      }
     if (yellowStatus==0)
      {
        bot.sendMessage(chat_id, "Yellow is OFF", "");
      }
       if (greenStatus==1)
      {
        bot.sendMessage(chat_id, "Green is ON", "");
      }
     if (greenStatus==0)
      {
        bot.sendMessage(chat_id, "Green is OFF", "");
      }
    }

    if (text == "/start")
    {
      String welcome = "Welcome to lab2 DFN40312, " + from_name + ".\n";
      welcome += "CODES BY NASRUL\n\n";
      welcome += "/redon : to switch the Led RED ON\n";
      welcome += "/redoff : to switch the Led RED OFF\n";
      welcome += "/yellowon : to switch the Led YELLOW ON\n";
      welcome += "/yellowoff : to switch the Led YELLOW OFF\n";
      welcome += "/greenon : to switch the Led GREEN ON\n";
      welcome += "/greenoff : to switch the Led GREEN OFF\n";
      welcome += "/status : Returns current status of LED\n";
      bot.sendMessage(chat_id, welcome, "Markdown");
    }
  }
}


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

  pinMode(ledPin_red, OUTPUT); // initialize digital ledPin as an output.
  pinMode(ledPin_green, OUTPUT); // initialize digital ledPin as an output.
  pinMode(ledPin_yellow, OUTPUT); // initialize digital ledPin as an output.
  //delay(10);
 // digitalWrite(ledPin_red, HIGH); // initialize pin as off (active LOW)

  // attempt to connect to Wifi network:
  Serial.print("Connecting to Wifi SSID ");
  Serial.print(WIFI_SSID);
  WiFi.begin(WIFI_SSID, WIFI_PASSWORD);
  secured_client.setCACert(TELEGRAM_CERTIFICATE_ROOT); // Add root certificate for api.telegram.org
  while (WiFi.status() != WL_CONNECTED)
  {
    Serial.print(".");
    delay(500);
  }
  Serial.print("\nWiFi connected. IP address: ");
  Serial.println(WiFi.localIP());

  Serial.print("Retrieving time: ");
  configTime(0, 0, "pool.ntp.org"); // get UTC time via NTP
  time_t now = time(nullptr);
  while (now < 24 * 3600)
  {
    Serial.print(".");
    delay(100);
    now = time(nullptr);
  }
  Serial.println(now);
}

void loop()
{
  if (millis() - bot_lasttime > BOT_MTBS)
  {
    int numNewMessages = bot.getUpdates(bot.last_message_received + 1);

    while (numNewMessages)
    {
      Serial.println("got response");
      handleNewMessages(numNewMessages);
      numNewMessages = bot.getUpdates(bot.last_message_received + 1);
    }

    bot_lasttime = millis();
  }
}