/*
   Network related functions
*/

#include "net_functions.h"

#include <string.h>

#include "driver/gpio.h"
#include "esp_event.h"
#include "esp_log.h"
#include "esp_netif.h"
#include "esp_sntp.h"
#include "esp_system.h"
#include "esp_wifi.h"
#include "freertos/FreeRTOS.h"
#include "freertos/event_groups.h"
#include "freertos/task.h"
#include "mdns.h"
#include "netdb.h"
#include "wifi_interface.h"

static const char *TAG = "NETF";

extern EventGroupHandle_t s_wifi_event_group;

static const char *if_str[] = {"STA", "AP", "ETH", "MAX"};
static const char *ip_protocol_str[] = {"V4", "V6", "MAX"};

void net_mdns_register(const char *clientname) {
  ESP_LOGI(TAG, "Setup mdns");
  ESP_ERROR_CHECK(mdns_init());
  ESP_ERROR_CHECK(mdns_hostname_set(clientname));
  ESP_ERROR_CHECK(mdns_instance_name_set("ESP32 SNAPcast client OTA"));
  ESP_ERROR_CHECK(mdns_service_add(NULL, "_http", "_tcp", 8032, NULL, 0));
}

void mdns_print_results(mdns_result_t *results) {
  mdns_result_t *r = results;
  mdns_ip_addr_t *a = NULL;
  int i = 1, t;
  while (r) {
    //      printf ("%d: Interface: %s, Type: %s\n", i++, if_str[r->tcpip_if],
    //              ip_protocol_str[r->ip_protocol]);
    printf("%d: Type: %s\n", i++, ip_protocol_str[r->ip_protocol]);
    if (r->instance_name) {
      printf("  PTR : %s\n", r->instance_name);
    }
    if (r->hostname) {
      printf("  SRV : %s.local:%u\n", r->hostname, r->port);
    }
    if (r->txt_count) {
      printf("  TXT : [%u] ", r->txt_count);
      for (t = 0; t < r->txt_count; t++) {
        printf("%s=%s; ", r->txt[t].key, r->txt[t].value);
      }
      printf("\n");
    }
    a = r->addr;
    while (a) {
      if (a->addr.type == IPADDR_TYPE_V6) {
        printf("  AAAA: " IPV6STR "\n", IPV62STR(a->addr.u_addr.ip6));
      } else {
        printf("  A   : " IPSTR "\n", IP2STR(&(a->addr.u_addr.ip4)));
      }
      a = a->next;
    }
    r = r->next;
  }
}

uint32_t find_mdns_service(const char *service_name, const char *proto) {
  ESP_LOGI(TAG, "Query PTR: %s.%s.local", service_name, proto);

  mdns_result_t *r = NULL;
  esp_err_t err = mdns_query_ptr(service_name, proto, 3000, 20, &r);
  if (err) {
    ESP_LOGE(TAG, "Query Failed");
    return -1;
  }
  if (!r) {
    ESP_LOGW(TAG, "No results found!");
    return -1;
  }

  if (r->instance_name) {
    printf("  PTR : %s\n", r->instance_name);
  }
  if (r->hostname) {
    printf("  SRV : %s.local:%u\n", r->hostname, r->port);
    return r->port;
  }
  mdns_query_results_free(r);
  return 0;
}

/**
 *
 */
void sntp_cb(struct timeval *tv) {
  struct tm timeinfo = {0};
  time_t now = tv->tv_sec;
  localtime_r(&now, &timeinfo);
  char strftime_buf[64];
  strftime(strftime_buf, sizeof(strftime_buf), "%c", &timeinfo);
  ESP_LOGI(TAG, "sntp_cb called :%s", strftime_buf);
}

void set_time_from_sntp() {
  xEventGroupWaitBits(s_wifi_event_group, WIFI_CONNECTED_BIT, false, true,
                      portMAX_DELAY);
  // ESP_LOGI(TAG, "clock %");
  ESP_LOGI(TAG, "Initializing SNTP");
  sntp_setoperatingmode(SNTP_OPMODE_POLL);
  sntp_setservername(0, CONFIG_SNTP_SERVER);
  sntp_init();
  // sntp_set_time_sync_notification_cb(sntp_cb);
  setenv("TZ", SNTP_TIMEZONE, 1);
  tzset();

  time_t now = 0;
  struct tm timeinfo = {0};
  int retry = 0;
  const int retry_count = 10;
  while (timeinfo.tm_year < (2016 - 1900) && ++retry < retry_count) {
    ESP_LOGI(TAG, "Waiting for system time to be set... (%d/%d)", retry,
             retry_count);
    vTaskDelay(2000 / portTICK_PERIOD_MS);
    time(&now);
    localtime_r(&now, &timeinfo);
  }
  char strftime_buf[64];

  strftime(strftime_buf, sizeof(strftime_buf), "%c", &timeinfo);
  ESP_LOGI(TAG, "The current date/time in UTC is: %s", strftime_buf);
}