#include "driver/gpio.h"
#include "driver/uart.h"
#include "esp_log.h"
#include "esp_phy_init.h"
#include "esp_rom_gpio.h"
#include "esp_timer.h"
#include "esp_wifi.h"
#include "hal/uart_types.h"
#include "nvs_flash.h"

#include "main.h"
#include "uart_handler.h"
#include "communication_handler.h"

static const char *TAG = "ALOX - MAIN";

void app_main(void) {
  esp_err_t ret = nvs_flash_init();
  if (ret == ESP_ERR_NVS_NO_FREE_PAGES ||
      ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
    ESP_ERROR_CHECK(nvs_flash_erase());
    ret = nvs_flash_init();
  }
  ESP_ERROR_CHECK(ret);

  // GPIO-Pin für Moduserkennung
  gpio_reset_pin(MASTER_MODE_PIN);
  gpio_set_direction(MASTER_MODE_PIN, GPIO_MODE_INPUT);
  bool isMaster = (gpio_get_level(MASTER_MODE_PIN) == 0);

  // ESP-NOW Initialisieren
  ESP_ERROR_CHECK(esp_netif_init());
  ESP_ERROR_CHECK(esp_event_loop_create_default());
  wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
  ESP_ERROR_CHECK(esp_wifi_init(&cfg));
  wifi_config_t wifi_config = {
      .sta =
          {
              .channel = 1, // Kanal 1, stelle sicher, dass alle Geräte
                            // denselben Kanal verwenden
          },
  };
  ESP_ERROR_CHECK(esp_wifi_set_config(WIFI_IF_STA, &wifi_config));
  ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));
  ESP_ERROR_CHECK(esp_wifi_start());
  ESP_ERROR_CHECK(esp_now_init());

  if (isMaster) {
    ESP_ERROR_CHECK(esp_now_register_recv_cb(master_receive_callback));
  } else {
    ESP_ERROR_CHECK(esp_now_register_recv_cb(client_receive_callback));
  }

  init_com();

  // Tasks starten basierend auf Master/Client
  if (isMaster) {
    ESP_LOGI(TAG, "Started in Mastermode");
    add_peer(broadcast_address);
    xTaskCreate(master_broadcast_task, "MasterBroadcast", 4096, NULL, 1, NULL);
    xTaskCreate(master_ping_task, "MasterPing", 4096, NULL, 1, NULL);
    xTaskCreate(client_monitor_task, "MonitorClientTask", 4096, NULL, 1, NULL);
    init_uart();
  } else {
    ESP_LOGI(TAG, "Started in Slavemode");
    xTaskCreate(client_data_sending_task, "ClientDataSending", 4096, NULL, 1,
                NULL);
  }
}