#include "app_dhcp.h"

#include "OBSWConfig.h"
#include "lwip/dhcp.h"
#include "networking.h"
#include "stm32h7xx_nucleo.h"
#include "udp_config.h"

#if LWIP_DHCP == 1

uint8_t DHCP_state = DHCP_OFF;
uint32_t DHCPfineTimer = 0;

void handle_dhcp_timeout(struct netif *netif);
void handle_dhcp_start(struct netif *netif);
void handle_dhcp_wait(struct netif *netif, struct dhcp **dhcp);
void handle_dhcp_down(struct netif *netif);

/**
 * @brief  DHCP_Process_Handle
 * @param  None
 * @retval None
 */
void DHCP_Process(struct netif *netif) {
  struct dhcp *dhcp = nullptr;
  switch (DHCP_state) {
    case DHCP_START: {
      handle_dhcp_start(netif);
      break;
    }
    case DHCP_WAIT_ADDRESS: {
      handle_dhcp_wait(netif, &dhcp);
      break;
    }

    case DHCP_LINK_DOWN: {
      handle_dhcp_down(netif);
      break;
    }
    default: {
      break;
    }
  }
}

void handle_dhcp_timeout(struct netif *netif) {
  ip_addr_t ipaddr;
  ip_addr_t netmask;
  ip_addr_t gw;

  DHCP_state = DHCP_TIMEOUT;

  /* Stop DHCP */
  dhcp_stop(netif);

  /* Static address used */
  networking::setLwipAddresses(&ipaddr, &netmask, &gw);
  netif_set_addr(netif, &ipaddr, &netmask, &gw);

  printf("DHCP Timeout\n\r");
  uint8_t iptxt[20];
  sprintf((char *)iptxt, "%s", ip4addr_ntoa(netif_ip4_addr(netif)));
  printf("Assigning static IP address: %s\n", iptxt);

#if defined FSFW_OSAL_FREERTOS
  ETH_HandleTypeDef *handle = getEthernetHandle();
  handle->gState = HAL_ETH_STATE_READY;
#endif

#if OBSW_ETHERNET_TMTC_COMMANDING == 1
#if OBSW_ETHERNET_USE_LED1_LED2 == 1
  BSP_LED_On(LED1);
  BSP_LED_Off(LED2);
#endif
#endif
}

/**
 * @brief  DHCP periodic check
 * @param  netif
 * @retval None
 */
void DHCP_Periodic_Handle(struct netif *netif) {
  /* Fine DHCP periodic process every 500ms */
  if (HAL_GetTick() - DHCPfineTimer >= DHCP_FINE_TIMER_MSECS) {
    DHCPfineTimer = HAL_GetTick();
    /* process DHCP state machine */
    DHCP_Process(netif);
  }
}

void handle_dhcp_start(struct netif *netif) {
  printf("handle_dhcp_start: Looking for DHCP server ...\n\r");
#if OBSW_ETHERNET_TMTC_COMMANDING == 1
#if OBSW_ETHERNET_USE_LED1_LED2 == 1
  BSP_LED_Off(LED1);
  BSP_LED_Off(LED2);
#endif
#endif
  ip_addr_set_zero_ip4(&netif->ip_addr);
  ip_addr_set_zero_ip4(&netif->netmask);
  ip_addr_set_zero_ip4(&netif->gw);
  dhcp_start(netif);
  DHCP_state = DHCP_WAIT_ADDRESS;
}

void handle_dhcp_wait(struct netif *netif, struct dhcp **dhcp) {
  if (dhcp_supplied_address(netif)) {
    DHCP_state = DHCP_ADDRESS_ASSIGNED;
    printf("IP address assigned by a DHCP server: %s\n\r", ip4addr_ntoa(netif_ip4_addr(netif)));
    printf("Listener port: %d\n\r", UDP_SERVER_PORT);
#if OBSW_ETHERNET_TMTC_COMMANDING == 1
#if OBSW_ETHERNET_USE_LED1_LED2 == 1
    BSP_LED_On(LED1);
    BSP_LED_Off(LED2);
#endif
#endif
  } else {
    *dhcp =
        static_cast<struct dhcp *>(netif_get_client_data(netif, LWIP_NETIF_CLIENT_DATA_INDEX_DHCP));

    /* DHCP timeout */
    if ((*dhcp)->tries > MAX_DHCP_TRIES) {
      handle_dhcp_timeout(netif);
    }
  }
}

void handle_dhcp_down(struct netif *netif) {
  static_cast<void>(netif);
  DHCP_state = DHCP_OFF;
#if OBSW_ETHERNET_TMTC_COMMANDING == 1
  printf("DHCP_Process: The network cable is not connected.\n\r");
#if OBSW_ETHERNET_USE_LED1_LED2 == 1
  BSP_LED_Off(LED1);
  BSP_LED_On(LED2);
#endif
#endif

  /* Global boolean to track ethernet connection */
  networking::setEthCableConnected(false);
}

uint8_t get_dhcp_state() { return DHCP_state; }

void set_dhcp_state(uint8_t new_state) { DHCP_state = new_state; }

#endif /* LWIP_DHCP == 1 */