/*
 * ESPRESSIF MIT License
 *
 * Copyright (c) 2015 <ESPRESSIF SYSTEMS (SHANGHAI) PTE LTD>
 *
 * Permission is hereby granted for use on ESPRESSIF SYSTEMS ESP8266 only, in which case,
 * it is free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the Software is furnished
 * to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or
 * substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
 * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 */
#ifndef OTA_H_
#define OTA_H_
#include "esp_common.h"
#include <stdint.h>

#define OTA_HTTP                    0
#define OTA_HTTPS                   1
#define OTA_SERVER_TYPE             OTA_HTTP

/*IMPORTANT: the following configuration maybe need to be modified*/
/***********************************************************************************************************************/
// for local OTA
#define LOCAL_OTA_SERVER_IP       "192.168.111.104"     // local OTA server ip
#define LOCAL_OTA_SERVER_PORT     3344                  // local OTA server port


// for aliyun OTA
#if (OTA_SERVER_TYPE == OTA_HTTP)
#define REMOTE_OTA_SERVER_PORT    80
#define OTA_REMOTE_GET_FORMAT               "GET %s HTTP/1.1\r\n" "Host: %s\r\n" "Accept-Encoding: gzip, deflate, br\r\n" "Accept-Language: zh-CN,zh;q=0.9\r\n" "Connection: keep-alive\r\n" "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8\r\n" "User-Agent: esp8266-rtos-sdk/1.0 esp8266\r\n\r\n" 
#else
#define REMOTE_OTA_SERVER_PORT    443
#define OTA_REMOTE_GET_FORMAT               /*"GET %s HTTP/1.1\r\n" "Host: %s\r\n"*/ "Accept-Encoding: gzip, deflate, br\r\n" "Accept-Language: zh-CN,zh;q=0.9\r\n" "Connection: keep-alive\r\n" "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8\r\n" "User-Agent: esp8266-rtos-sdk/1.0 esp8266\r\n\r\n" 
#endif
#define MSG_PUB_MAX_SIZE          512            // the max length of publish topic payload
#define MAX_OTA_PATH_LEN          1024           // the max length of ota path in http get request
#define MAX_VERSION_LEN           24             // the max length of firmware version, which will push to aliyun
#define MAX_HOST_NAME_LEN         64             // to save hostname which to do ota
#define MAX_MD5_LEN               32

#define OTA_DEVICE_WIFI           0
#define OTA_DEVICE_MCU            1
#define NE_DEVICE_APCONFIG			2
#define NE_DEVICE_NORMAL			3

#define OTA_DEBUG_ENABLE          1
#if OTA_DEBUG_ENABLE
#define OTA_LOG_EX                  LOG_RAW
#else
#define OTA_LOG_EX     
#endif

#define OTA_DEVICE_MAX_SECTORS      65
#define OTA_CONTROL_DEV             OTA_DEVICE_WIFI     


// for OTA
#define OTA_TIMEOUT_WIFI    120000  // timeout: 120000 ms,  for WIFI OTA
#define OTA_TIMEOUT_MCU     600000  // timeout: 600000 ms,  for MCU OTA
/***********************************************************************************************************************/

// for remote ota, save struct to flash and do ota after system restart
typedef struct {
    uint8_t ota_flag;       // 1: prepare to do ota task, others: mqtt task
    uint8_t ota_device;       //0: WIFI, 1: MCU
    uint32_t bin_size;      // bin size
    uint32_t seg_size;      // segment size
    uint32_t lastest_version_int;
    char latest_version[MAX_VERSION_LEN];// latest version to ota, and report
    char ota_path[MAX_OTA_PATH_LEN];
    char hostname[MAX_HOST_NAME_LEN];
    char md5[MAX_MD5_LEN];
    ip_addr_t target_ip;    // remote ota ip
    uint32_t port;          // remote ota port
} ota_info_t;


/*Please Keep the following configuration if you have no very deep understanding of ESP8266-OTA */
#define SPI_FLASH_SEC_SIZE      4096

// for local OTA
#define USER_BIN1               0x00
#define USER_BIN2               0x01

// for OTA
#define UPGRADE_FLAG_IDLE       0x00
#define UPGRADE_FLAG_START      0x01
#define UPGRADE_FLAG_FINISH     0x02

#define UPGRADE_FW_BIN1         0x00
#define UPGRADE_FW_BIN2         0x01

void system_upgrade_init();
bool system_upgrade(uint8 *data, uint32 len);
void system_upgrade_deinit(void);

void upgrade_recycle(void);

void local_ota_task(void *pvParameter);
void remote_ota_task(void *pvParameter);
void remote_ota_upgrade_task_MCU(void *pvParameter);
void remote_ota_upgrade_task_WIFI(void *pvParameter);

#endif