#include "bme280.h"
#include "pico/stdlib.h"
#include "pms5003.h"
#include "pico/cyw43_arch.h"
#include "tcp_client.h"
#include <hardware/gpio.h>
#include <hardware/i2c.h>
#include <hardware/uart.h>
#include <pico/time.h>
#include <pico/types.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <hardware/watchdog.h>

// 5 sec loop is for testing
// #define LOOP_INTERVAL_MS 5000
// #define LOOP_INTERVAL_MS 10000
#define LOOP_INTERVAL_MS 30000

/**
 * Balcony Weather Station Node 1
 * record sensor data and send it to home assistant every 5 minutes
 * Sensors: BME280, PMS5003
 */

// COMMS LED
absolute_time_t comms_led_off_time;
int16_t comms_led_blink_count = 0;
bool comms_led_state = false;
void comms_led_init() {
  gpio_init(16);
  gpio_set_dir(16, GPIO_OUT);
}
void comms_led_blink() {
  printf("COMMS LED BLINK COUNT: %d\n", comms_led_blink_count);
  comms_led_blink_count++;
  comms_led_state = true;
  comms_led_off_time = make_timeout_time_ms(1000);
  gpio_put(16, comms_led_state);
}
void comms_led_update() {
    if (time_reached(comms_led_off_time)) {
      comms_led_state = false;
      gpio_put(16, comms_led_state);
    }
}

// Callback to reset the Pico after 24 hours
static bool cb_24h(__unused struct repeating_timer *t) {
    printf("Restarting Pico after 24 hours...\n");
    watchdog_enable(1, 1); // 1 ms timeout, reset on next loop
    while (1) { tight_loop_contents(); } // Wait for reset
    return false; // Not reached
}

static tcp_client_config tcp_config;
static bool wifi_connected = false;

// static pms5003_config pms_config;
// static pms5003_reading current_pms5003_reading;

static int8_t readings_index = 0;
static bme280_config bem_config;
static bme280_reading current_bem280_reading;

//  Initialize WiFi connection
bool wifi_init(void) {
    if (cyw43_arch_init()) {
        printf("Failed to initialize CYW43\n");
        return false;
    }

    cyw43_arch_enable_sta_mode();
    printf("Connecting to WiFi '%s'...\n", WIFI_SSID);

    if (cyw43_arch_wifi_connect_timeout_ms(WIFI_SSID, WIFI_PASSWORD,
                                           CYW43_AUTH_WPA2_AES_PSK, 30000)) {
        printf("Failed to connect to WiFi\n");
        return false;
    }

    printf("WiFi connected successfully\n");
    return true;
}

/**
 * Callback function called every 30 seconds
 */
static bool cb_30(__unused struct repeating_timer *t) {
  comms_led_blink();
  printf("cb_30: %d\n", readings_index);
  // if (readings_index >= 10) {
    // printf("Preparing data to send\n");
    // readings_index = 0;

    // Calculate average BME280 reading
    // printf("Tempature: %.2f\n", current_bem280_reading.temperature);
    // printf("Pressure: %.2f\n", current_bem280_reading.pressure);
    // printf("Humidity: %.2f\n", current_bem280_reading.humidity);

    // printf("PM1: %.2f\n", current_pms5003_reading.pm1);
    // printf("PM2.5: %.2f\n", current_pms5003_reading.pm2_5);
    // printf("PM10: %.2f\n", current_pms5003_reading.pm10);

    // Send to backend server if WiFi is connected

  // Read BME280
  printf("Making BME280 Reading\n");
  current_bem280_reading = bme280_read(&bem_config);


  // if (readings_index == 2) {
    // printf("Warming up PMSS5003\n");
    // pms5003_warmup(&pms_config);
  // }

  // if (readings_index == 4) {
    // printf("Starting reads on PMSS5003\n");
    // pms5003_start_reading(&pms_config);
  // }

  // if (readings_index == 6) {
    // printf("Finished reading PMSS5003\n");
    // current_pms5003_reading = pms5003_finish_reading(&pms_config);
  // }

  readings_index++;
  if (wifi_connected) {
      char msg[256];
      snprintf(msg, sizeof(msg),
               "M001,%.2f,%.2f,%2f\n",
               current_bem280_reading.temperature,
               current_bem280_reading.pressure,
               current_bem280_reading.humidity);
      printf("Sending data to backend server...\n");
      printf("MSG: %s", msg);
      bool success = tcp_client_send_message(&tcp_config, msg);
      if (success) {
          printf("✓ Data sent successfully\n");
          comms_led_blink();
      } else {
          printf("✗ Failed to send data\n");
      }
  } else {
      printf("WiFi not connected, skipping send\n");
  }

  return true;
};

int main() {
  stdio_init_all();
  watchdog_enable(60000, 1);
  sleep_ms(2000);  // Give time for USB serial

  // Initialize communication LED
  comms_led_init();

  // Initialize WiFi
  printf("Initializing WiFi...\n");
  if (!wifi_init()) {
      printf("WiFi initialization failed!\n");
      panic("WiFi initialization failed!");
  } else {
      wifi_connected = true;
      // Initialize TCP client
      if (!tcp_client_init(&tcp_config, BACKEND_SERVER_IP,
                          BACKEND_SERVER_PORT, 10000)) {
          printf("TCP client initialization failed\n");
          panic("TCP client initialization failed!");
          wifi_connected = false;
      } else {
          printf("TCP client ready: %s:%d\n", BACKEND_SERVER_IP, BACKEND_SERVER_PORT);
      }
  }

  // Setup BME280
  bme280_init(&bem_config, i2c1, 14, 15);

  // Setup PMS5003
  // pms5003_init(&pms_config, uart1, 20, 21, 18, 19);

  struct repeating_timer timer_30;
  add_repeating_timer_ms(LOOP_INTERVAL_MS, cb_30, NULL, &timer_30);
  struct repeating_timer timer_24h;
  add_repeating_timer_ms(86400000, cb_24h, NULL, &timer_24h);
  while (true) {
    cyw43_arch_poll();
    cyw43_arch_wait_for_work_until(make_timeout_time_ms(1000));
    comms_led_update();
    watchdog_update();
    tight_loop_contents();
  }
}