#include <LiquidCrystal_I2C.h>
#include <Arduino_FreeRTOS.h>
#include <task.h>
#include <queue.h>

LiquidCrystal_I2C lcd(0x27, 16, 2);

const int ledPin1 = 11; // the number of the first LED pin

// Struct to store LED settings information
struct LEDSettings {
  int blinkCount;
};

int settingDelay = 0;

// Task handles
TaskHandle_t TaskAHandle;
TaskHandle_t TaskBHandle;

// Queues
QueueHandle_t xQueue1;
QueueHandle_t xQueue2;

void TaskA(void *pvParameters);
void TaskB(void *pvParameters);

void setup() {
  // initialize the LED pins as outputs:
  pinMode(ledPin1, OUTPUT);
  Serial.begin(9600);
  lcd.init();
  lcd.backlight();
  Serial.println("Program started");

  // Create Queues
  xQueue1 = xQueueCreate(5, sizeof(int));
  xQueue2 = xQueueCreate(5, sizeof(char));

  // Check if Queues were created successfully
  if (xQueue1 == NULL || xQueue2 == NULL) {
    Serial.println("Failed to create queues");
    while (1)
      ;
  }

  // Create Task A
  xTaskCreate(TaskA, "TaskA", 1000, NULL, 1, &TaskAHandle);

  // Create Task B
  xTaskCreate(TaskB, "TaskB", 1000, NULL, 2, &TaskBHandle);

  // Start FreeRTOS scheduler
  vTaskStartScheduler();
}

void loop() {
  // Empty, as the tasks will run in FreeRTOS scheduler
}

void TaskA(void *pvParameters) {
  int intResult;
  char charResult;

  while (1) {
    Serial.println("Task A");
    // Receive message from Serial
    if (Serial.available() > 0) {
      String input = Serial.readStringUntil('\n');

      // Check if input is an integer
      intResult = input.toInt();
      if (!isnan(intResult)) {
        // Send the integer value to Queue 1
        xQueueSend(xQueue1, &intResult, portMAX_DELAY);
        Serial.println("Received integer: " + String(intResult));
      }

      // Check if input is a char/string
      charResult = input[0];
      if (charResult != '\0' && isAlpha(charResult)) {
        // Send the char value to Queue 2
        xQueueSend(xQueue2, &charResult, portMAX_DELAY);
        Serial.println("Received char: " + String(charResult));
      }
    }
    vTaskDelay(100 / portTICK_PERIOD_MS); // Delay to avoid busy-waiting
  }
}

void TaskB(void *pvParameters) {
  LEDSettings settings;
  settings.blinkCount = 0;

  while (1) {
    Serial.println("Task B");

    // Set text to be displayed on LCD
    lcd.clear();
    lcd.setCursor(0, 0);
    lcd.print("Set Delay " + String(settingDelay));

    // Receive value from Queue 1
    int receivedInt;
    if (xQueueReceive(xQueue1, &receivedInt, 0) == pdTRUE) {
      // Update LED delay with received integer
      settings.blinkCount = receivedInt;
      Serial.println("Updated blinkCount with integer: " + String(receivedInt));
    }

    // Receive value from Queue 2
    char receivedChar;
    if (xQueueReceive(xQueue2, &receivedChar, 0) == pdTRUE) {
      // Update char variable with received char
      // Here, you can perform any specific action with the received char
      Serial.println("Updated char variable with: " + String(receivedChar));
    }

    // Blink LED
    digitalWrite(ledPin1, HIGH);
    vTaskDelay(settings.blinkCount / portTICK_PERIOD_MS);
    digitalWrite(ledPin1, LOW);
    vTaskDelay(settings.blinkCount / portTICK_PERIOD_MS);

    // Send a message to Queue 2 every 100 LED blinks
    settings.blinkCount++;
    if (settings.blinkCount == 100) {
      xQueueSend(xQueue2, &settings, portMAX_DELAY);
      Serial.println("Blinked 100 times");
      settings.blinkCount = 0;
    }
  }
}