#ifndef OTA_UPDATE_H
#define OTA_UPDATE_H

#include "esp_err.h"
#include <stdint.h>
#include <sys/types.h>
#include "esp_now.h"
#include "message_structs.h"

#define UPDATE_BUFFER_SIZE 4096
#define UPDATE_PAYLOAD_SIZE 200
#define UPDATE_MAX_SEQUENZES (UPDATE_BUFFER_SIZE / UPDATE_PAYLOAD_SIZE)

typedef enum {
    OTA_COMMAND_PREPARE,
    OTA_COMMAND_CHUNK,
    OTA_COMMAND_REQUEST_STATUS,
    OTA_COMMAND_COMMIT,
    OTA_COMMAND_FINISH,
    OTA_COMMAND_MASTER_READY
} ota_command_t;

typedef struct {
    ota_command_t command;
    uint8_t mac_addr[ESP_NOW_ETH_ALEN];
    union {
        OTA_CHUNK_Payload ota_chunk_payload;
        OTA_REQUEST_BLOCK_STATUS_Payload ota_request_block_status_payload;
        OTA_COMMIT_BLOCK_Payload ota_commit_block_payload;
    } payload;
} ota_task_queue_message_t;


extern uint32_t g_uart_firmware_total_size;

void init_ota();
void ota_task(void *pvParameter);

enum OTA_UPDATE_STATES {
  IDEL,
  START_REQUESTED,
  WAITING_FOR_PAYLOAD,
  WRITING_OTA_TO_PARTITION,

};

int prepare_ota_update();
esp_err_t write_ota_update(uint32_t write_len, const uint8_t *payload);
esp_err_t end_ota_update();

void slave_Prep_Upgrade_Callback(const esp_now_recv_info_t *esp_now_info,
                                 const uint8_t *data, int data_len);
void slave_ota_chunk_callback(const esp_now_recv_info_t *esp_now_info,
                              const uint8_t *data, int data_len);
void slave_request_block_status_callback(
    const esp_now_recv_info_t *esp_now_info, const uint8_t *data,
    int data_len);
void slave_commit_block_callback(const esp_now_recv_info_t *esp_now_info,
                                 const uint8_t *data, int data_len);
void slave_finish_update_callback(const esp_now_recv_info_t *esp_now_info,
                                  const uint8_t *data, int data_len);
void slave_master_ready_to_send_chunks_callback(const esp_now_recv_info_t *esp_now_info,
                                                const uint8_t *data, int data_len);

void start_uart_update(uint8_t msgid, const uint8_t *payload,
                       size_t payload_len, uint8_t *send_payload_buffer,
                       size_t send_payload_buffer_size, uint8_t *send_buffer,
                       size_t send_buffer_size);
void payload_uart_update(uint8_t msgid, const uint8_t *payload_data_from_uart,
                         size_t total_payload_len_from_uart, uint8_t *send_payload_buffer,
                         size_t send_payload_buffer_size, uint8_t *send_buffer,
                         size_t send_buffer_size);
void end_uart_update(uint8_t msgid, const uint8_t *payload, size_t payload_len,
                        uint8_t *send_payload_buffer,
                        size_t send_payload_buffer_size, uint8_t *send_buffer,
                        size_t send_buffer_size);

#endif