#include "TinyGPSPlus.h"
#include "DHT.h"
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>
#include "FS.h"
#include "SD.h"
#include "SPI.h"
#include "Arduino.h"
#include "math.h"

/*
Uncomment and set up if you want to use custom pins for the SPI communication
#define REASSIGN_PINS
int sck = -1;
int miso = -1;
int mosi = -1;
int cs = -1;
*/

// Inicia I2C en pines default ESP32 (21 SDA, 22 SCL)
#define SCREEN_WIDTH 128  // Ancho píxeles
#define SCREEN_HEIGHT 64  // Alto píxeles
#define OLED_RESET -1     // Reset pin (no usado)

// Pines para UART2 (Serial2)
#define RX_PIN 16  // RX del ESP32 conectado a TX del GPS
#define TX_PIN 17  // TX del ESP32 conectado a RX del GPS
#define GPS_BAUD 115200

#define DHTPIN 4     // Digital pin connected to the DHT sensor
#define DHTTYPE DHT22 

#define BUTTON_PIN 27  // pin para pulsador

//definicion de tiempos de pulsacion
#define PULASCION_LARGA_MS 2000
#define DURACION_WATCHDOG_MS 10000

#define MEASUREMENT_INTERVAL_S 1    //separación entre mediciones (s) 

#define DEG2RAD (M_PI/180.0)


// Objeto TinyGPS++
TinyGPSPlus gps;
HardwareSerial gpsSerial(2); // Usar UART2
DHT dht(DHTPIN, DHTTYPE);
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

//valores de los sensores

struct SensorData {
  double latitude = 0.0;
  double longitude = 0.0;
  float altura = 0.0;
  String tiempo = "";
  float temperature = 0.0;
  float humidity = 0.0;
  float pressure = 0.0;
};

SensorData latestData;
SensorData datosAntiguos;

SemaphoreHandle_t dataMutex; // Mutex para proteger el acceso a latestData
SemaphoreHandle_t buttonSemaphore; // Semáforo para la tarea del botón

bool grabando = false; //inicia apagado
TaskHandle_t medicionesHandle = NULL; //para suspend/resume
int pantallaEstado_grab = -1; //maquina de estados cuando se graba ruta
int pantallaEstado_menu = -1; //maquina de estados cuando no se esta grabando ruta
float distancia_total = 0.0;
volatile unsigned long ignore_isr_until = 0; //para debounce

char filename[13];

void OLED_print(const String& line1, const String& line2) {
  display.clearDisplay();
  display.setTextSize(2);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);
  display.println(line1);
  display.println(line2);
  display.display();
}

void DHT_test() {
  dht.begin();
  float h = dht.readHumidity();
  float t = dht.readTemperature();
  if (isnan(h) || isnan(t)) {
    Serial.println("Failed to read from DHT sensor!");
    OLED_print("DHT22", "Error");
    delay(5000);
    DHT_test(); // Reintentar
  }
  else OLED_print("DHT22", "Correcto");
}

void OLED_test() {  //pantallazo a blanco y luego iniciando
  // Inicia I2C en pines default ESP32 (21 SDA, 22 SCL)
  Wire.begin();
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {  // Dirección común: 0x3C
    Serial.println(F("Error: OLED no encontrado!"));
    for (;;);  // Para siempre
  }
  display.clearDisplay();
  display.fillScreen(SSD1306_WHITE); // Pantalla blanca
  delay(500);
  display.display();
  display.clearDisplay();
  display.setTextSize(2);         // Tamaño texto
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0, 0);        // Posición
  display.println("Iniciando...");
  display.display();              // Muestra
  delay(1000);
}


void SD_test(){
if (!SD.begin()) {
    OLED_print("SD Card", "Error\nInserte");
    while (!SD.begin());
    OLED_print("SD Card", "Insertada");
  } else {
    OLED_print("SD Card", "Correcto");
  }
  uint8_t cardType = SD.cardType();
  if (cardType == CARD_NONE) {
    OLED_print("SD Card", "No detectada");
    while (cardType == CARD_NONE) {
      delay(1000);
      cardType = SD.cardType();
    }
    OLED_print("SD Card", "Detectada");
  }
  uint8_t cardSize = SD.cardSize() / (1024 * 1024);
}

void GPS_test_wait() {
  // Iniciar Serial2 para GPS
  gpsSerial.begin(GPS_BAUD, SERIAL_8N1, RX_PIN, TX_PIN);
  while (((gpsSerial.available() > 0) && gps.location.isValid()) && (gps.speed.age() < 2000)) {
    gps.encode(gpsSerial.read());
    delay(100);
    OLED_print("GPS", "Esperando.");
    delay(100);
    OLED_print("GPS", "Esperando..");
    delay(100);
    OLED_print("GPS", "Esperando...");
  }
  OLED_print("GPS", "Encontrado");
}

float calcular_delta_dist(float lat1, float  long1, float lat2, float long2){
  float R = 6371.0; // Radio de la Tierra en km
  float delta_lat = (lat2 - lat1) * DEG2RAD;
  float delta_long = (long2 - long1) * DEG2RAD;
  lat1 = lat1 * DEG2RAD;
  lat2 = lat2 * DEG2RAD;
  float a = sin(delta_lat/2)*sin(delta_lat/2)+cos(lat1)*cos(lat2)*sin(delta_long/2)*sin(delta_long/2);
  float c = 2 * atan2(sqrt(a),sqrt(1-a));
  return R * c;  //En km
}

void task_mediciones(void *pvParameters) {
  TickType_t xLastWakeTime = xTaskGetTickCount();
  while(1) {
    // se leen los valores antes de utilizar el semaphore    
    while (gpsSerial.available() > 0) {
      gps.encode(gpsSerial.read());
    }

    float new_latitude = gps.location.lat();
    float new_longitude = gps.location.lng();
    float new_altitude = gps.altitude.meters();
    String new_fecha = String(gps.date.year())+"-"+String(gps.date.month())+"-"+
                   String(gps.date.day())+"T"+String(gps.time.hour())+":"+
                   String(gps.time.minute())+":"+String(gps.time.second())+"."+
                   String(gps.time.centisecond());
    float new_temp = dht.readTemperature();
    float new_hum = dht.readHumidity();
    float new_press = 0.0; // Placeholder, no hay sensor de presión 
    
    if (gps.location.isValid() && datosAntiguos.latitude != 0.0) {
      distancia_total += calcular_delta_dist(datosAntiguos.latitude, datosAntiguos.longitude, new_latitude, new_longitude);
    }

    if (xSemaphoreTake(dataMutex, portMAX_DELAY) == pdTRUE) {
      latestData.latitude = new_latitude;
      latestData.longitude = new_longitude;
      latestData.altura = new_altitude;
      latestData.tiempo = new_fecha;
      latestData.temperature = new_temp;
      latestData.humidity = new_hum;
      latestData.pressure = new_press;
      datosAntiguos = latestData;
      xSemaphoreGive(dataMutex);
    }

    File file = SD.open(filename, FILE_APPEND);
    if (file) {
      //Crear la string para escribir en el archivo
      String frase = '\t\t\t<trkpt> lat="' + String(datosAntiguos.latitude,6) + 
                   '" lon="' + String(datosAntiguos.longitude,6) + '">\n\t\t\t\t<ele>'+
                   String(datosAntiguos.altura) + '</ele>\n\t\t\t\t<time>'+ 
                   datosAntiguos.tiempo+'</time>\n\t\t\t\t<extensions>\n\t\t\t\t\t<gpxtpx:TrackPointExtension>\n\t\t\t\t\t\t<gpxtpx:atemp>'+
                   String(datosAntiguos.temperature)+'</gpxtpx:atemp>\n\t\t\t\t\t</gpxtpx:TrackPointExtension>\n\t\t\t\t\t<custom:humidity>'+
                   String(datosAntiguos.humidity)+'</custom:humidity>\n\t\t\t\t\t<custom:pressure>'+String(datosAntiguos.pressure)+
                   '</custom:pressure>\n\t\t\t\t</extensions>\n\t\t\t</trkpt>';
      
      // Escribir datos en el archivo
      file.println(frase);
      file.close();
    }

    vTaskDelayUntil(&xLastWakeTime, pdMS_TO_TICKS(MEASUREMENT_INTERVAL_S*1000)); // Espera x*1000 milisegundos
  }
}

void crear_archivo(){
  int num = 1;
  sprintf(filename, "/data%03d.gpx", num);
  while (SD.exists(filename)) {
    num++;
    sprintf(filename, "/data%03d.gpx", num);
  }

  File file = SD.open(filename, FILE_WRITE);
  if (file) {
    file.println('<?xml version="1.0" encoding="UTF-8"?>\n<gpx creator="ESP32 GPS LOGGER" version="1.1"\n\txmlns="http://www.topografix.com/GPX/1/1"\n\t
      xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"\n\txmlns:gpxtpx="http://www.garmin.com/xmlschemas/TrackPointExtension/v2"\n\t
      xmlns:gpxdata="http://www.cluetrust.com/XML/GPXDATA/1/0"\n\txsi:schemaLocation="http://www.topografix.com/GPX/1/1 http://www.topografix.com/GPX/1/1/gpx.xsd">\n\t<trk>\n\t\t
      <name>Rutita</name>\n\t\t<type>hiking</type>\n\t\t<trkseg>');
    file.close();
  } else {
    OLED_print("Error","creando archivo");
  }
}

void IRAM_ATTR isr_button() {
  unsigned long now = millis();
  if (now < ignore_isr_until) {
    return; // Ignorar interrupción si está dentro del período de debounce
  }
  static unsigned long lastInterrupt = 0;
  
  if ((now - lastInterrupt) > 300 ){ // debounce de 300 ms 
    BaseType_t xHigherPriorityTaskWoken = pdFALSE;
    xSemaphoreGiveFromISR(buttonSemaphore, &xHigherPriorityTaskWoken); 
    lastInterrupt = now;
    if (xHigherPriorityTaskWoken) {
      portYIELD_FROM_ISR();
    }
  }
  
}

void drawProgressBar(int x, int y, int w, int h, unsigned long progress, unsigned long total) {
  display.drawRect(x, y, w, h, SSD1306_WHITE); // Dibuja el borde
  int filledWidth = (progress * w) / total;
  display.fillRect(x + 1, y + 1, filledWidth - 2, h - 2, SSD1306_WHITE); // Dibuja la barra llena
  display.display();
}

void task_ui(void *pvParameters){
  unsigned long pressTime = 0;
  unsigned long lastActivity = millis();
  bool pantallaOn = true; //comprobar el estado inicial, no se cual sera
  bool processingButton = false;

  while(1){
    if (xSemaphoreTake(buttonSemaphore, pdMS_TO_TICKS(200)) == pdTRUE){ //button pressed
      if (processingButton) continue; //evita reentradas
      processingButton = true;

      pressTime = millis();
      lastActivity = millis(); //reset watchdog

      if (!pantallaOn){
        display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
        pantallaOn = true;
      }

      bool timed_out = false;
      unsigned long checkTime = millis();
      while (digitalRead(BUTTON_PIN) == LOW){
        vTaskDelay(pdMS_TO_TICKS(10));
        checkTime = millis();
        if ((checkTime - pressTime) > DURACION_WATCHDOG_MS){ //10s timeout para evitar bloqueos
          timed_out = true;
          break;
        }
        drawProgressBar(0, SCREEN_HEIGHT - 10, SCREEN_WIDTH, 8, checkTime - pressTime, PULASCION_LARGA_MS);
      }

      ignore_isr_until = millis() + 50; //ignorar nuevas interrupciones durante 500 ms

      unsigned long duration = checkTime - pressTime;

      if (timed_out){
        OLED_print("Apagando","pantalla");
        display.ssd1306_command(SSD1306_DISPLAYOFF); //se apaga la
        pantallaOn = false;
      } else {
        if (grabando){
          if (duration >= PULASCION_LARGA_MS){
            grabando = false;
            vTaskSuspend(medicionesHandle);
            OLED_print("Ruta","pausada");
          } else {
            pantallaEstado_grab = (pantallaEstado_grab + 1) % 5; //cicla entre 0-4
            SensorData currentData;
            if(xSemaphoreTake(dataMutex, portMAX_DELAY) == pdTRUE){
              currentData = latestData;
              xSemaphoreGive(dataMutex);
            }
            switch (pantallaEstado_grab){
              case 0:
                OLED_print("Posicion",String(currentData.longitude) + "," + String(currentData.latitude));
                break;
              case 1:
                OLED_print("Distancia",String(distancia_total)+"km");
                break;
              case 2:
                OLED_print("Altitud",String(gps.altitude.meters(), 1)+"m");
                break;
              case 3:
                OLED_print("Temp/Hum",String(currentData.temperature,1)+"C/"+String(currentData.humidity,1)+"%");
                break;  
              case 4:
                OLED_print("Velocidad",String(gps.speed.kmph())+"km/h");
                break;
            }
          }
        } else {
          if (duration >= PULASCION_LARGA_MS){
            switch (pantallaEstado_menu){
              case 0:
                //activar la ruta y crear el archivo
                crear_archivo();
                vTaskResume(medicionesHandle);
                break;
              case 1:
                //cerrar el archivo y cambiar el valor de 'filename'
              case 2:
                //implementacion blutuch
            }
          } else {
            pantallaEstado_menu = (pantallaEstado_menu + 1) % 3;
            switch (pantallaEstado_menu){
              case 0:
                if (SD.exists(filename)) OLED_print("Reanudar","ruta");
                else OLED_print("Iniciar","ruta");
                break;
              case 1:
                if (SD.exists(filename)) {
                  OLED_print("Finalizar","ruta");
                  break;
                }
                pantallaEstado_menu += 1;
              case 2:
                if (!SD.exists(filename)) { 
                  OLED_print("Conexion","blutuch 'WIP'");
                  break;
                }
                pantallaEstado_menu += 1;
            }
          }
        }










































        if (duration >= PULASCION_LARGA_MS){
          //pulsacion larga: cabia entre grabacion y no grabacion de datos
          grabando = grabando ? false : true; //toggle
          if (grabando){
            vTaskResume(medicionesHandle);
            //Mostrar que empieza la grabación
            OLED_print("Ruta","iniciada");
          } else {
            vTaskSuspend(medicionesHandle);
            //Mostrar que se pausa/finaliza la grabación
            OLED_print("Ruta","pausada");
          }
        } else {
          //Pulsacion corta + grabando datos, cicla datos
          if (grabando) {
            pantallaEstado_grab = (pantallaEstado_grab + 1) % 5; //cicla entre 0-4
            SensorData currentData;
            if(xSemaphoreTake(dataMutex, portMAX_DELAY) == pdTRUE){
              currentData = latestData;
              xSemaphoreGive(dataMutex);
            }
            switch (pantallaEstado_grab){
              case 0:
                OLED_print("Posicion",String(currentData.longitude) + "," + String(currentData.latitude));
                break;
              case 1:
                OLED_print("Distancia",String(distancia_total)+"km");
                break;
              case 2:
                OLED_print("Altitud",String(gps.altitude.meters(), 1)+"m");
                break;
              case 3:
                OLED_print("Temp/Hum",String(currentData.temperature,1)+"C/"+String(currentData.humidity,1)+"%");
                break;  
              case 4:
                OLED_print("Velocidad",String(gps.speed.kmph())+"km/h");
                break;
            } 
          } else {
            pantallaEstado_menu = (pantallaEstado_menu + 1) % 3;
            OLED_print("Ruta","pausada");
          }      
        }

        
      }
      lastActivity = millis();  //reset watchdog
      processingButton = false;
      vTaskDelay(pdMS_TO_TICKS(100)); //pequeño delay para no busy waiting
    }
    //check watchdog fuera del boton
    if (pantallaOn && ((millis() - lastActivity) > DURACION_WATCHDOG_MS)){
          display.ssd1306_command(SSD1306_DISPLAYOFF); //se apaga la pantalla
          pantallaOn = false;
    }
  }
}


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

  pinMode(BUTTON_PIN, INPUT_PULLUP);
  attachInterrupt(digitalPinToInterrupt(BUTTON_PIN), isr_button, FALLING);
  
  buttonSemaphore = xSemaphoreCreateBinary();
  dataMutex = xSemaphoreCreateMutex();

  // OLED check
  OLED_test();
  delay(1000);
  // DHT check
  DHT_test();
  delay(1000);
  // SD Card check
  SD_test();
  delay(1000);
  // GPS check
  GPS_test_wait();
  delay(2000);

  // Crear tarea para mediciones
  xTaskCreatePinnedToCore(
    task_mediciones,     // Función de la tarea
    "Mediciones",        // Nombre de la tarea
    8192,                // Tamaño del stack
    NULL,     // Parámetro de la tarea
    10,                   // Prioridad de la tarea
    &medicionesHandle,                // Handle de la tarea
    0                    // Núcleo donde se ejecuta
  );

  xTaskCreatePinnedToCore(
    task_ui,             // Función de la tarea
    "UI",                // Nombre de la tarea
    8192,                // Tamaño del stack
    NULL,                // Parámetro de la tarea
    5,                   // Prioridad de la tarea
    NULL,                // Handle de la tarea
    1                    // Núcleo donde se ejecuta
  );

  vTaskSuspend(medicionesHandle); //inicia suspendida

}



void loop() {

  vTaskDelay(pdMS_TO_TICKS(1000)); // Espera 1 segundo
}