#ifndef __GENIE_OTA_H__
#define __GENIE_OTA_H__

//For pingpong OTA
enum
{
    DFU_IMAGE_A = 0,
    DFU_IMAGE_B,
    DFU_IMAGE_TOTAL = DFU_IMAGE_B,
    DFU_IMAGE_INVALID_ID,
};

#define OTA_RECV_BUF_SIZE (CONFIG_AIS_TOTAL_FRAME * 128)
#define OTA_RECV_MAX_ERR_COUNT 10

#define GENIE_OTA_IMAGE_ID_MAGIC (0xA5A5A5)
#define GENIE_OTA_IMAGE_ID_A (GENIE_OTA_IMAGE_ID_MAGIC << 8 | DFU_IMAGE_A)
#define GENIE_OTA_IMAGE_ID_B (GENIE_OTA_IMAGE_ID_MAGIC << 8 | DFU_IMAGE_B)

#define AIS_OTA_AUTH_TIMEOUT 10000    //10s
#define AIS_OTA_DISCONN_TIMEOUT 60000 //60s
#define AIS_OTA_REPORT_TIMEOUT (CONFIG_AIS_TOTAL_FRAME * 400)
#define AIS_OTA_REBOOT_TIMEOUT 3000 //3s

enum
{
    OTA_FLAG_SILENT = 0x02,
    AIS_OTA_VER_REQ = 0x20,      //APP request get version info
    AIS_OTA_VER_RESP = 0x21,     //Response the version info request
    AIS_OTA_FIRMWARE_REQ = 0x22, //APP push firmware info
    AIS_OTA_UPD_RESP = 0x23,     //Response to APP which can update or not and have received firmware size
    AIS_OTA_STATUS = 0x24,       //Response to APP that last frame number and have received firmware size
    AIS_OTA_CHECK_REQ = 0x25,    //APP download finish,then device check firmware
    AIS_OTA_CHECK_RESP = 0x26,   //Response to APP the result of firmeware check
    AIS_OTA_DATA = 0x2f          //This is ota firmware data
};

typedef struct genie_ota_ctx_s
{
    uint8_t err_count;

    uint8_t last_seq;
    uint8_t total_frame;
    uint8_t except_seq;
    uint32_t rx_size;

    uint8_t image_type;
    uint32_t image_ver;
    uint32_t image_size;
    uint16_t image_crc16;
    uint8_t ota_flag;

    uint8_t flash_clean : 1;
    uint8_t ota_ready : 1;
    uint16_t rx_len;
    uint8_t recv_buf[OTA_RECV_BUF_SIZE];
} genie_ota_ctx_t;

typedef struct
{
    uint8_t image_type;
} __attribute__((packed)) ais_ota_ver_req_t;

typedef struct
{
    uint8_t image_type;
    uint32_t ver;
} __attribute__((packed)) ais_ota_ver_resp_t;

typedef struct
{
    uint8_t image_type;
    uint32_t ver;
    uint32_t fw_size;
    uint16_t crc16;
    uint8_t ota_flag;
} __attribute__((packed)) ais_ota_upd_req_t;

typedef struct
{
    uint8_t state;
    uint32_t rx_size;
    uint8_t total_frame;
} __attribute__((packed)) ais_ota_upd_resp_t;

typedef struct
{
    uint8_t last_seq : 4;
    uint8_t total_frame : 4;
    uint32_t rx_size;
} __attribute__((packed)) ais_ota_status_report_t;

typedef struct
{
    uint8_t state;
} __attribute__((packed)) ais_ota_check_req_t;

typedef struct
{
    uint8_t state;
} __attribute__((packed)) ais_ota_check_resp_t;

bool genie_ota_handle_update_request(uint8_t msg_id, ais_ota_upd_req_t *p_ota_req, uint8_t encrypt);

bool genie_ota_handle_version_request(uint8_t msg_id, ais_ota_ver_req_t *p_ver_req, uint8_t encrypt);

bool genie_ota_parse_pdu(ais_pdu_t *p_msg);

bool genie_ota_check_firmware(uint8_t msg_id, ais_ota_check_req_t *p_check_req);

bool genie_ota_is_ready(void);

/**
 * @brief get the ota indication flag from flash.
 * @return the ota indication flag.
 */
bool genie_ota_get_indication(void);

void genie_ota_status_report(void);

bool genie_ota_is_updating(void);

int genie_ota_report_version(void);

int genie_ota_init(void);

#endif