/*
 * Proyecto ESP32: Construir un dispositivo de chat LoRa seguro y de largo alcance Heltec WiFi LoRa 32
 * 
 * Archivo: Heltec_ESP32_LoRa_V3_chat_TX_RX.ino
 * escrito el 29 de junio de 2025 por Ahmad Shamshiri www.Roboajx.com
 * Por favor, mira el video tutorial en: https://youtu.be/IsOSvrF60J4
 * Página de recursos para este video: https://robojax.com/RJT642
 * 
 * =====================================================================
 * DESCRIPCIÓN DEL CÓDIGO ARDUINO: TRANSCEIVER DE TECLADO LoRa SEGURO (sistema de chat)
 * =====================================================================
 * 
 * COMPONENTES DE HARDWARE:
 * -------------------
 * - Controlador principal: Heltec WiFi LoRa 32 V3
 * - Caja: Estuche Meshnology N33 con batería de 3000mAh
 * - Entrada: Encoder rotativo con botón
 * - Retroalimentación: Pantalla OLED integrada
 * 
 * FUNCIONALIDAD DEL SISTEMA:
 * -------------------
 * NUEVOS CONTROLES:
 * - Perilla de rotación: Seleccionar carácter.
 * - Presión corta (clic): Agregar carácter seleccionado.
 * - Presión larga (mantener): Insertar un espacio (como una barra espaciadora).
 * - Doble clic: Retroceso.
 * 
 * [2] TRANSMISIÓN WIRELESS SEGURA:
 * - Todos los mensajes encriptados antes de la transmisión LoRa
 * - Comando de posición inicial transmitido como paquete seguro especial
 * - Utiliza la banda LoRa de 433MHz/915MHZ para una comunicación confiable
 * 
 * [3] GESTIÓN DE ENERGÍA:
 * - Optimizados para operación con batería (3000mAh)
 * - Modos de bajo consumo entre transmisiones
 * 
 * PARA INSTRUCCIONES COMPLETAS DE CONFIGURACIÓN:
 * Por favor, mira el video tutorial en: https://youtu.be/IsOSvrF60J4
 * Página de recursos para este video: https://robojax.com/RJT642
 * =====================================================================
 * 
 * RENUNCIA:
 * Este código se proporciona "TAL CUAL" sin garantía de ningún tipo. El autor
 * no será responsable por daños que surjan del uso de este código.
 * 
 * LICENCIA:
 * Esta obra está licenciada bajo la Licencia Pública General GNU v3.0
 * Permisos más allá del alcance de esta licencia pueden estar disponibles en Robojax.com
 * 
 * TÉRMINOS DE COMPARTICIÓN:
 * Eres libre de compartir, copiar y modificar este código para fines no comerciales
 * SIEMPRE QUE:
 * 1. Mantengas este bloque de comentarios intacto con el código original
 * 2. Incluyas el enlace original de Robojax.com
 * 3. Mantengas el enlace del tutorial de YouTube (si aplica)
 * 4. Indiques claramente cualquier modificación realizada
 * 
 * /
 * // Bibliotecas centrales de Heltec y display
 */
#include <Wire.h>
#include "HT_SSD1306Wire.h"
#include "WiFi.h"

 // Bibliotecas de LoRa y seguridad
#include "LoRaWan_APP.h"
#include "mbedtls/aes.h"
#include <cstring>

 // La biblioteca de codificador rotatorio aprobada por el usuario
#include "AiEsp32RotaryEncoder.h"

 // --- DEFINICIONES Y CONFIGURACIÓN DE PIN ---
static SSD1306Wire  display(0x3c, 500000, SDA_OLED, SCL_OLED, GEOMETRY_128_64, RST_OLED);

#define ROTARY_ENCODER_A_PIN 5 // pín A del codificador rotatorio
#define ROTARY_ENCODER_B_PIN 6 // pin B del codificador rotatorio
#define ROTARY_ENCODER_BUTTON_PIN 4 // pin de conmutación del codificador rotativo

#define RF_FREQUENCY 915432000 // frecuencia de operación de LoRa
#define TX_OUTPUT_POWER 20 // la potencia de salida debe estar entre 2 y 21 en dBm.
const char *userKey = "wahJ8uy$gt~_3r"; // clave de seguridad para comunicación privada

#define ROTARY_ENCODER_VCC_PIN -1
#define ROTARY_ENCODER_STEPS 4

 // --- PARÁMETROS LORA ---
#define LORA_BANDWIDTH 0
#define LORA_SPREADING_FACTOR 7
#define LORA_CODINGRATE 1
#define LORA_PREAMBLE_LENGTH 8
#define LORA_SYMBOL_TIMEOUT 0
#define LORA_FIX_LENGTH_PAYLOAD_ON false
#define LORA_IQ_INVERSION_ON false
#define BUFFER_SIZE 256

#define USER_BUTTON_PIN 0
 // --- VARIABLES Y ESTADO GLOBAL ---
enum AppState { COMPOSING, READING };
AppState currentState = COMPOSING;
String messageToCompose = "";
String lastReceivedMessage = "No messages yet.";
bool hasUnreadMessage = false;
volatile bool newPacketAvailable = false;
char rxpacket[BUFFER_SIZE];
AiEsp32RotaryEncoder rotaryEncoder = AiEsp32RotaryEncoder(ROTARY_ENCODER_A_PIN, ROTARY_ENCODER_B_PIN, ROTARY_ENCODER_BUTTON_PIN, ROTARY_ENCODER_VCC_PIN, ROTARY_ENCODER_STEPS);
const char* charset = " ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789.,!?>";
char selectedChar;
static RadioEvents_t RadioEvents;
bool lora_idle = true;

mbedtls_aes_context aes;
unsigned long shortPressAfterMiliseconds = 50;
unsigned long longPressAfterMiliseconds = 1000;
unsigned long doubleClickTimeout = 400;
static unsigned long buttonDownTime = 0;
static unsigned long buttonUpTime = 0;
static bool buttonWasDown = false;
static bool doubleClickWaiting = false;

 // --- PROTOTIPOS DE FUNCIÓN ---
void OnTxDone(void);
void OnTxTimeout(void);
void OnRxDone(uint8_t *payload, uint16_t size, int16_t rssi, int8_t snr);
void encryptAES(uint8_t *data, const char *key);
void decryptAES(uint8_t *data, const char *key);
void processKey(const char *userKey, uint8_t *processedKey, size_t keySize);
void updateDisplay();
void sendData();
void VextON(void);
void processButton();

 // --- ROUTINAS DE SERVICIO DE INTERRUPCIÓN ---
void IRAM_ATTR readEncoderISR() {
  rotaryEncoder.readEncoder_ISR();
}

 // --- ACCIONES DEL MANIPULADOR DE BOTONES ---
void on_button_short_click() {
  if (currentState != COMPOSING) return;
  if (selectedChar == '>') {
      sendData();
  } else {
    messageToCompose += selectedChar;
  }
  updateDisplay();
}
void on_button_long_click() {
  if (currentState != COMPOSING) return;
  messageToCompose += " ";
  updateDisplay();
}
void on_button_double_click() {
  if (currentState != COMPOSING) return;
  if (messageToCompose.length() > 0) {
    messageToCompose.remove(messageToCompose.length() - 1);
  }
  updateDisplay();
}

 // --- CONFIGURACIÓN ---
void setup() {
  Serial.begin(115200);
  Serial.println("LoRa Keyboard Transceiver - Definitive Version");

  VextON();
  delay(100);
  Mcu.begin(HELTEC_BOARD,SLOW_CLK_TPYE);
  WiFi.mode(WIFI_OFF);
  btStop();

  display.init();
  display.setFont(ArialMT_Plain_10);

  rotaryEncoder.begin();
  rotaryEncoder.setup(readEncoderISR);
  rotaryEncoder.setBoundaries(0, strlen(charset) - 1, true);
  rotaryEncoder.disableAcceleration();

  pinMode(USER_BUTTON_PIN, INPUT_PULLUP);

  RadioEvents.TxDone = OnTxDone;
  RadioEvents.TxTimeout = OnTxTimeout;
  RadioEvents.RxDone = OnRxDone;
  Radio.Init(&RadioEvents);
  Radio.SetChannel(RF_FREQUENCY);

  Radio.SetRxConfig(MODEM_LORA, LORA_BANDWIDTH, LORA_SPREADING_FACTOR,
                   LORA_CODINGRATE, 0, LORA_PREAMBLE_LENGTH,
                   LORA_SYMBOL_TIMEOUT, LORA_FIX_LENGTH_PAYLOAD_ON,
                   0, true, 0, 0, LORA_IQ_INVERSION_ON, true);

  updateDisplay();
}

 // --- BUCLE PRINCIPAL ---
void loop() {
  if(lora_idle) {
    lora_idle = false;
    Serial.println("Entering RX mode...");
    Radio.Rx(0);
  }
  Radio.IrqProcess();

  if (newPacketAvailable) {
    newPacketAvailable = false;
    hasUnreadMessage = true;
    lastReceivedMessage = String(rxpacket);
    Serial.print("Received Decrypted Message: ");
    Serial.println(lastReceivedMessage);
    updateDisplay();
  }

  if (currentState == COMPOSING) {
    if (rotaryEncoder.encoderChanged()) {
      updateDisplay();
    }
    processButton();
  }

  if (digitalRead(USER_BUTTON_PIN) == LOW) {
    delay(50);
    if (digitalRead(USER_BUTTON_PIN) == LOW) {
      if (currentState == COMPOSING) {
        if (lastReceivedMessage != "No messages yet.") {
          currentState = READING;
          hasUnreadMessage = false;
          updateDisplay();
        }
      } else {
        currentState = COMPOSING;
        updateDisplay();
      }
      while (digitalRead(USER_BUTTON_PIN) == LOW);
    }
  }

  delay(10);
}

 // --- CONTROLADORES Y AYUDANTES DE LORA ---
void OnTxDone(void) {
  Serial.println("TX done.");
  Radio.Sleep();
  lora_idle = true;
}
void OnTxTimeout(void) {
  Serial.println("TX Timeout.");
  Radio.Sleep();
  lora_idle = true;
}
void OnRxDone(uint8_t *payload, uint16_t size, int16_t rssi, int8_t snr) {
    Radio.Sleep();
    memset(rxpacket, 0, sizeof(rxpacket));
    memcpy(rxpacket, payload, size > BUFFER_SIZE ? BUFFER_SIZE : size);
    decryptAES((uint8_t*)rxpacket, userKey);
    newPacketAvailable = true;
    lora_idle = true;
}

void sendData() {
  if (messageToCompose.length() > 0) {
    lora_idle = false;

    Radio.SetTxConfig(MODEM_LORA, TX_OUTPUT_POWER, 0, LORA_BANDWIDTH,
                   LORA_SPREADING_FACTOR, LORA_CODINGRATE,
                   LORA_PREAMBLE_LENGTH, LORA_FIX_LENGTH_PAYLOAD_ON,
                   true, 0, 0, LORA_IQ_INVERSION_ON, 3000);

    uint8_t data[BUFFER_SIZE];
    memset(data, 0, BUFFER_SIZE);
    strncpy((char*)data, messageToCompose.c_str(), BUFFER_SIZE - 1);

    encryptAES(data, userKey);

    Serial.print("Sending message: ");
    Serial.println(messageToCompose);

    Radio.Send(data, 16);

    messageToCompose = "";
  }
}

 // --- LÓGICA DE DISPLAY Y BOTONES ---
void updateDisplay() {
  display.clear();
  display.setTextAlignment(TEXT_ALIGN_LEFT);

  if (currentState == COMPOSING) {
    selectedChar = charset[rotaryEncoder.readEncoder()];
    display.setFont(ArialMT_Plain_10);

 // --- ESTA ES LA SOLUCIÓN para la notificación ---
    String msgLabel;
    if (hasUnreadMessage) {
        msgLabel = ">> NEW MSG <<"; // Haz que la notificación sea muy obvia.
    } else {
        msgLabel = "Msg:";
    }
    display.drawStringMaxWidth(0, 0, 128, msgLabel + " " + messageToCompose);

    display.setFont(ArialMT_Plain_16);
    int16_t labelWidth = display.getStringWidth("Select: ");
    display.drawString(0, 35, "Select: ");
    display.setFont(ArialMT_Plain_24);
    display.drawString(labelWidth, 32, String(selectedChar));
  } else { // estadoActual == LECTURA
    display.setFont(ArialMT_Plain_16);
    display.drawString(0, 0, "Received:");
    display.setFont(ArialMT_Plain_10);
    display.drawStringMaxWidth(0, 18, 128, lastReceivedMessage);
  }

  display.display();
}

void processButton() {
    if (doubleClickWaiting && (millis() - buttonUpTime > doubleClickTimeout)) {
        doubleClickWaiting = false;
        on_button_short_click();
    }
    bool isButtonDown = rotaryEncoder.isEncoderButtonDown();
    if (isButtonDown && !buttonWasDown) {
        buttonDownTime = millis();
        buttonWasDown = true;
    }
    else if (!isButtonDown && buttonWasDown) {
        unsigned long pressDuration = millis() - buttonDownTime;
        if (pressDuration >= longPressAfterMiliseconds) {
            on_button_long_click();
            doubleClickWaiting = false;
        }
        else if (pressDuration >= shortPressAfterMiliseconds) {
            if (doubleClickWaiting) {
                on_button_double_click();
                doubleClickWaiting = false;
            } else {
                doubleClickWaiting = true;
                buttonUpTime = millis();
            }
        }
        buttonWasDown = false;
    }
}

 // --- UTILIDADES ---
void VextON(void) {
  pinMode(Vext,OUTPUT);
  digitalWrite(Vext, LOW);
}
void processKey(const char *userKey, uint8_t *processedKey, size_t keySize) {
    memset(processedKey, 0, keySize);
    size_t len = strlen(userKey);
    if (len > keySize) len = keySize;
    memcpy(processedKey, userKey, len);
}
void encryptAES(uint8_t *data, const char *key) {
    uint8_t processedKey[16];
    processKey(key, processedKey, 16);
    mbedtls_aes_init(&aes);
    mbedtls_aes_setkey_enc(&aes, processedKey, 128);
    mbedtls_aes_crypt_ecb(&aes, MBEDTLS_AES_ENCRYPT, data, data);
    mbedtls_aes_free(&aes);
}
void decryptAES(uint8_t *data, const char *key) {
    uint8_t processedKey[16];
    processKey(key, processedKey, 16);
    mbedtls_aes_init(&aes);
    mbedtls_aes_setkey_dec(&aes, processedKey, 128);
    mbedtls_aes_crypt_ecb(&aes, MBEDTLS_AES_DECRYPT, data, data);
    mbedtls_aes_free(&aes);
}

/*
 * Proyecto ESP32: Construye un dispositivo de chat LoRa seguro y de largo alcance Heltec WiFi LoRa 32
 * 
 * Archivo: Heltec_ESP32_LoRa_V3_chat_TX_RX_buzzer.ino
 * Escrito el 29 de junio de 2025 por Ahmad Shamshiri www.Roboajx.com
 * Vea el videotutorial en: https://youtu.be/IsOSvrF60J4
 * Página de recursos para este video: https://robojax.com/RJT642
 * 
 * =========================================================================
 * DESCRIPCIÓN DEL CÓDIGO ARDUINO: Transceptor de teclado LoRa seguro (chat) Sistema)
 * =======================================================================
 * 
 * COMPONENTES DE HARDWARE:
 * -------------------
 * - Controlador principal: Heltec WiFi LoRa 32 V3
 * - Carcasa: Meshnology N33 con batería de 3000 mAh
 * - Entrada: Codificador rotatorio con pulsador
 * - Sonido: Zumbador activo (conectado al pin 40)
 * - Retroalimentación: Pantalla OLED integrada
 * 
 * FUNCIONALIDAD DEL SISTEMA:
 * -------------------
 * NUEVOS CONTROLES:
 * - Perilla giratoria: Seleccionar carácter.
 * - Pulsación corta (clic): Añadir carácter seleccionado.
 * - Pulsación larga (mantener): Insertar un espacio (como la barra espaciadora).
 * - Doble clic: Retroceso.
 * 
 * [2] TRANSMISIÓN INALÁMBRICA SEGURA:
 * - Todos los mensajes se cifran antes de la transmisión LoRa
 * - El comando de posición inicial se transmite como un paquete seguro especial
 * - Utiliza la banda LoRa de 433 MHz/915 MHz para una comunicación fiable
 * 
 * [3] GESTIÓN DE ENERGÍA:
 * - Optimizado para funcionar con batería (3000 mAh)
 * - Modos de bajo consumo entre transmisiones
 * 
 * PARA INSTRUCCIONES DE INSTALACIÓN COMPLETAS:
 * Vea el videotutorial en: https://youtu.be/IsOSvrF60J4
 * Página de recursos para este video: https://robojax.com/RJT642
 * ==============================================================================
 * 
 * DESCARGO DE RESPONSABILIDAD:
 * Este código se proporciona "TAL CUAL", sin garantía de ningún tipo. El autor no se responsabiliza de ningún daño derivado del uso de este código.
 * 
 * LICENCIA:
 * Esta obra está licenciada bajo la Licencia Pública General GNU v3.0. Permisos adicionales al alcance de esta licencia pueden estar disponibles en Robojax.com.
 * 
 * TÉRMINOS DE COMPARTICIÓN:
 * Puede compartir, copiar y modificar este código con fines no comerciales, siempre y cuando:
 * 1. Mantenga el bloque de comentarios intacto con el código original.
 * 2. Incluya el enlace original a Robojax.com.
 * 3. Conserve el enlace del tutorial de YouTube (si corresponde).
 * 4. Indique claramente cualquier modificación realizada.
 * 
 * /
 * // Bibliotecas de Core Heltec y de visualización.
 */
#include <Wire.h>
#include "HT_SSD1306Wire.h"
#include "WiFi.h"

 // LoRa y bibliotecas de seguridad
#include "LoRaWan_APP.h"
#include "mbedtls/aes.h"
#include <cstring>

 // La biblioteca de codificadores rotatorios aprobada por el usuario
#include "AiEsp32RotaryEncoder.h"

 // --- DEFINICIONES Y CONFIGURACIÓN DE PIN ---
static SSD1306Wire  display(0x3c, 500000, SDA_OLED, SCL_OLED, GEOMETRY_128_64, RST_OLED);

#define ROTARY_ENCODER_A_PIN 5 // pin A del codificador rotatorio
#define ROTARY_ENCODER_B_PIN 6 // pin B del codificador rotatorio
#define ROTARY_ENCODER_BUTTON_PIN 4 // pin de conmutación del codificador rotatorio

#define BUZZER_PIN 40 // El pin del zumbador o relé que se va a activar
const bool DEBUG = false;

#define RF_FREQUENCY 915432000 // frecuencia de funcionamiento de LoRa
#define TX_OUTPUT_POWER 2 // La potencia de salida debe estar entre 2 y 21 dBm.
const char *userKey = "wahJ8uy$gt~_3r"; // clave de seguridad para la comunicación privada

#define ROTARY_ENCODER_VCC_PIN -1
#define ROTARY_ENCODER_STEPS 4

 // --- PARÁMETROS DE LORA ---
#define LORA_BANDWIDTH 0
#define LORA_SPREADING_FACTOR 7
#define LORA_CODINGRATE 1
#define LORA_PREAMBLE_LENGTH 8
#define LORA_SYMBOL_TIMEOUT 0
#define LORA_FIX_LENGTH_PAYLOAD_ON false
#define LORA_IQ_INVERSION_ON false
#define BUFFER_SIZE 256

#define LOOP_DELAY 10
 // Variables de estado del zumbador
unsigned long buzzerStartTime = 0;
bool isBuzzerActive = false;
int beepPhase = 0; // 0=inactivo, 1=primer pitido, 2=pausa, 3=segundo pitido

#define USER_BUTTON_PIN 0
 // --- VARIABLES GLOBALES Y ESTADO ---
enum AppState { COMPOSING, READING };
AppState currentState = COMPOSING;
String messageToCompose = "";
String lastReceivedMessage = "No messages yet.";
bool hasUnreadMessage = false;
volatile bool newPacketAvailable = false;
char rxpacket[BUFFER_SIZE];
AiEsp32RotaryEncoder rotaryEncoder = AiEsp32RotaryEncoder(ROTARY_ENCODER_A_PIN, ROTARY_ENCODER_B_PIN, ROTARY_ENCODER_BUTTON_PIN, ROTARY_ENCODER_VCC_PIN, ROTARY_ENCODER_STEPS);
const char* charset = " ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789.,!?>";
char selectedChar;
static RadioEvents_t RadioEvents;
bool lora_idle = true;

mbedtls_aes_context aes;
unsigned long shortPressAfterMiliseconds = 50;
unsigned long longPressAfterMiliseconds = 1000;
unsigned long doubleClickTimeout = 400;
static unsigned long buttonDownTime = 0;
static unsigned long buttonUpTime = 0;
static bool buttonWasDown = false;
static bool doubleClickWaiting = false;

 // --- PROTOTIPOS DE FUNCIONES ---
void OnTxDone(void);
void OnTxTimeout(void);
void OnRxDone(uint8_t *payload, uint16_t size, int16_t rssi, int8_t snr);
void encryptAES(uint8_t *data, const char *key);
void decryptAES(uint8_t *data, const char *key);
void processKey(const char *userKey, uint8_t *processedKey, size_t keySize);
void updateDisplay();
void sendData();
void VextON(void);
void processButton();
void buzzer();
void updateBuzzer();


 // --- RUTINAS DE SERVICIO DE INTERRUPCIÓN ---
void IRAM_ATTR readEncoderISR() {
  rotaryEncoder.readEncoder_ISR();
}

 // --- ACCIONES DEL MANEJADOR DE BOTONES ---
void on_button_short_click() {
  if (currentState != COMPOSING) return;
  if (selectedChar == '>') {
      sendData();
  } else {
    messageToCompose += selectedChar;
  }
  updateDisplay();
}
void on_button_long_click() {
  if (currentState != COMPOSING) return;
  messageToCompose += " ";
  updateDisplay();
}
void on_button_double_click() {
  if (currentState != COMPOSING) return;
  if (messageToCompose.length() > 0) {
    messageToCompose.remove(messageToCompose.length() - 1);
  }
  updateDisplay();
}

 // --- CONFIGURACIÓN ---
void setup() {
  Serial.begin(115200);
  Serial.println("LoRa Keyboard Transceiver - By Robojax");

  VextON();
  delay(100);
  Mcu.begin(HELTEC_BOARD,SLOW_CLK_TPYE);
  WiFi.mode(WIFI_OFF);
  btStop();

  display.init();
  display.setFont(ArialMT_Plain_10);

  rotaryEncoder.begin();
  rotaryEncoder.setup(readEncoderISR);
  rotaryEncoder.setBoundaries(0, strlen(charset) - 1, true);
  rotaryEncoder.disableAcceleration();

  pinMode(USER_BUTTON_PIN, INPUT_PULLUP);
  pinMode(BUZZER_PIN, OUTPUT);

  RadioEvents.TxDone = OnTxDone;
  RadioEvents.TxTimeout = OnTxTimeout;
  RadioEvents.RxDone = OnRxDone;
  Radio.Init(&RadioEvents);
  Radio.SetChannel(RF_FREQUENCY);

  Radio.SetRxConfig(MODEM_LORA, LORA_BANDWIDTH, LORA_SPREADING_FACTOR,
                   LORA_CODINGRATE, 0, LORA_PREAMBLE_LENGTH,
                   LORA_SYMBOL_TIMEOUT, LORA_FIX_LENGTH_PAYLOAD_ON,
                   0, true, 0, 0, LORA_IQ_INVERSION_ON, true);

  updateDisplay();
}

 // --- BUCLE PRINCIPAL ---
void loop() {
  if(lora_idle) {
    lora_idle = false;
    if(DEBUG)
    {
    Serial.println("Entering RX mode...");
    }

    Radio.Rx(0);
  }
  Radio.IrqProcess();

  if (newPacketAvailable) {
    newPacketAvailable = false;
    hasUnreadMessage = true;
    lastReceivedMessage = String(rxpacket);
    if(DEBUG)
    {
    Serial.print("Received Decrypted Message: ");
    Serial.println(lastReceivedMessage);
    }
    updateDisplay();
    buzzer();
  }

  if (currentState == COMPOSING) {
    if (rotaryEncoder.encoderChanged()) {
      updateDisplay();
    }
    processButton();
  }

  if (digitalRead(USER_BUTTON_PIN) == LOW) {
    delay(50);
    if (digitalRead(USER_BUTTON_PIN) == LOW) {
      if (currentState == COMPOSING) {
        if (lastReceivedMessage != "No messages yet.") {
          currentState = READING;
          hasUnreadMessage = false;
          updateDisplay();
        }
      } else {
        currentState = COMPOSING;
        updateDisplay();
      }
      while (digitalRead(USER_BUTTON_PIN) == LOW);
    }
  }

  updateBuzzer(); // Actualizar el estado del zumbador (sin bloqueo)
  delay(LOOP_DELAY); // Su retraso de bucle de 10 ms
}

 // --- MANEJADORES DE EVENTOS Y AYUDANTES DE LORA ---
void OnTxDone(void) {
    if(DEBUG)
    {
      Serial.println("TX done.");
    }
  Radio.Sleep();
  lora_idle = true;
}
void OnTxTimeout(void) {
    if(DEBUG)
    {
      Serial.println("TX Timeout.");
    }
  Radio.Sleep();
  lora_idle = true;
}
void OnRxDone(uint8_t *payload, uint16_t size, int16_t rssi, int8_t snr) {
    Radio.Sleep();
    memset(rxpacket, 0, sizeof(rxpacket));
    memcpy(rxpacket, payload, size > BUFFER_SIZE ? BUFFER_SIZE : size);
    decryptAES((uint8_t*)rxpacket, userKey);
    newPacketAvailable = true;
    lora_idle = true;
}

void sendData() {
  if (messageToCompose.length() > 0) {
    lora_idle = false;

    Radio.SetTxConfig(MODEM_LORA, TX_OUTPUT_POWER, 0, LORA_BANDWIDTH,
                   LORA_SPREADING_FACTOR, LORA_CODINGRATE,
                   LORA_PREAMBLE_LENGTH, LORA_FIX_LENGTH_PAYLOAD_ON,
                   true, 0, 0, LORA_IQ_INVERSION_ON, 3000);

    uint8_t data[BUFFER_SIZE];
    memset(data, 0, BUFFER_SIZE);
    strncpy((char*)data, messageToCompose.c_str(), BUFFER_SIZE - 1);

    encryptAES(data, userKey);
    if(DEBUG)
    {
    Serial.print("Sending message: ");
    Serial.println(messageToCompose);
    }

    Radio.Send(data, 16);

    messageToCompose = "";
  }
}

 // --- LÓGICA DE PANTALLA Y BOTONES ---
void updateDisplay() {
  display.clear();
  display.setTextAlignment(TEXT_ALIGN_LEFT);

  if (currentState == COMPOSING) {
    selectedChar = charset[rotaryEncoder.readEncoder()];
    display.setFont(ArialMT_Plain_10);

 // --- ESTA ES LA SOLUCIÓN para la notificación ---
    String msgLabel;
    if (hasUnreadMessage) {
        msgLabel = ">> NEW MSG <<"; // Haga que la notificación sea muy obvia
    } else {
        msgLabel = "Msg:";
    }
    display.drawStringMaxWidth(0, 0, 128, msgLabel + " " + messageToCompose);

    display.setFont(ArialMT_Plain_16);
    int16_t labelWidth = display.getStringWidth("Select: ");
    display.drawString(0, 35, "Select: ");
    display.setFont(ArialMT_Plain_24);
    display.drawString(labelWidth, 32, String(selectedChar));
  } else { // estadoActual == LECTURA
    display.setFont(ArialMT_Plain_16);
    display.drawString(0, 0, "Received:");
    display.setFont(ArialMT_Plain_10);
    display.drawStringMaxWidth(0, 18, 128, lastReceivedMessage);
  }

  display.display();
}

void processButton() {
    if (doubleClickWaiting && (millis() - buttonUpTime > doubleClickTimeout)) {
        doubleClickWaiting = false;
        on_button_short_click();
    }
    bool isButtonDown = rotaryEncoder.isEncoderButtonDown();
    if (isButtonDown && !buttonWasDown) {
        buttonDownTime = millis();
        buttonWasDown = true;
    }
    else if (!isButtonDown && buttonWasDown) {
        unsigned long pressDuration = millis() - buttonDownTime;
        if (pressDuration >= longPressAfterMiliseconds) {
            on_button_long_click();
            doubleClickWaiting = false;
        }
        else if (pressDuration >= shortPressAfterMiliseconds) {
            if (doubleClickWaiting) {
                on_button_double_click();
                doubleClickWaiting = false;
            } else {
                doubleClickWaiting = true;
                buttonUpTime = millis();
            }
        }
        buttonWasDown = false;
    }
}


void buzzer()
{
  if (!isBuzzerActive) {
    isBuzzerActive = true;
    buzzerStartTime = millis(); // Iniciar secuencia de timbre
    beepPhase = 1; // Primer pitido
    digitalWrite(BUZZER_PIN, HIGH);
  }
}
void updateBuzzer() {
  if (!isBuzzerActive) return;

  unsigned long currentTime = millis();
  unsigned long elapsed = currentTime - buzzerStartTime;

  switch (beepPhase) {
    case 1: // Primer pitido (300 ms encendido)
      if (elapsed >= 300) {
        digitalWrite(BUZZER_PIN, LOW);
        buzzerStartTime = currentTime;
        beepPhase = 2; // Mover para pausar
      }
      break;

    case 2: // Pausa (200 ms APAGADO)
      if (elapsed >= 200) {
        digitalWrite(BUZZER_PIN, HIGH);
        buzzerStartTime = currentTime;
        beepPhase = 3; // Pasar al segundo pitido
      }
      break;

    case 3: // Segundo pitido (300 ms activado)
      if (elapsed >= 300) {
        digitalWrite(BUZZER_PIN, LOW);
        isBuzzerActive = false; // Reiniciar
        beepPhase = 0;
      }
      break;
  }
}
 // --- UTILIDADES ---
void VextON(void) {
  pinMode(Vext,OUTPUT);
  digitalWrite(Vext, LOW);
}
void processKey(const char *userKey, uint8_t *processedKey, size_t keySize) {
    memset(processedKey, 0, keySize);
    size_t len = strlen(userKey);
    if (len > keySize) len = keySize;
    memcpy(processedKey, userKey, len);
}
void encryptAES(uint8_t *data, const char *key) {
    uint8_t processedKey[16];
    processKey(key, processedKey, 16);
    mbedtls_aes_init(&aes);
    mbedtls_aes_setkey_enc(&aes, processedKey, 128);
    mbedtls_aes_crypt_ecb(&aes, MBEDTLS_AES_ENCRYPT, data, data);
    mbedtls_aes_free(&aes);
}
void decryptAES(uint8_t *data, const char *key) {
    uint8_t processedKey[16];
    processKey(key, processedKey, 16);
    mbedtls_aes_init(&aes);
    mbedtls_aes_setkey_dec(&aes, processedKey, 128);
    mbedtls_aes_crypt_ecb(&aes, MBEDTLS_AES_DECRYPT, data, data);
    mbedtls_aes_free(&aes);
}