// 应用入口函数
void app_main(void)
{
    // 初始化 FLASH
    ESP_ERROR_CHECK(nvs_flash_init());
    // 初始化 NETIF
    ESP_ERROR_CHECK(esp_netif_init());
    // 创建缺省的事件循环
    ESP_ERROR_CHECK(esp_event_loop_create_default());

    /* This helper function configures Wi-Fi or Ethernet, as selected in menuconfig.
     * Read "Establishing Wi-Fi or Ethernet Connection" section in
     * examples/protocols/README.md for more information about this function.
     */
    ESP_ERROR_CHECK(example_connect());
    // 创建 SDDC task
    xTaskCreate(esp_sddc_task, "sddc_task", ESP_SDDC_TASK_STACK_SIZE, NULL, ESP_SDDC_TASK_PRIO, NULL);
}

// SDDC 工作任务
static void esp_sddc_task(void *arg)
{
    sddc_t *sddc;
    char *data;
    uint8_t mac[6];
    char ip[sizeof("255.255.255.255")];
    tcpip_adapter_ip_info_t ip_info = { 0 };

    // 创建 SDDC 协议对象
    sddc = sddc_create(SDDC_CFG_PORT);

    // 设置事件响应函数
    sddc_set_on_message(sddc, esp8266_on_message);              // 设置接收消息请求时的回调函数
    sddc_set_on_message_ack(sddc, esp8266_on_message_ack);      // 设置接收消息确认时的回调函数
    sddc_set_on_message_lost(sddc, esp8266_on_message_lost);    // 设置丢失消息时的回调函数
    sddc_set_on_invite(sddc, esp8266_on_invite);                // 设置接受邀请请求时的回调函数
    sddc_set_on_invite_end(sddc, esp8266_on_invite_end);        // 设置发送邀请后的回调函数
    sddc_set_on_update(sddc, esp8266_on_update);                // 设置接收更新请求时的回调函数
    sddc_set_on_edgeros_lost(sddc, esp8266_on_edgeros_lost);    // 设置 EdgerOS 断连时的回调函数

    // 设置设备密码
#if SDDC_CFG_SECURITY_EN > 0
    sddc_set_token(sddc, "1234567890");
#endif

    // 创建并设置 Report 报文数据
    data = esp_report_data_create();
    sddc_set_report_data(sddc, data, strlen(data));

    // 创建并设置 Invite 报文数据
    data = esp_invite_data_create();
    sddc_set_invite_data(sddc, data, strlen(data));

    // 获取网卡 mac 地址
    esp_wifi_get_mac(ESP_IF_WIFI_STA, mac);

    // 使用网卡 mac 地址设置设备唯一标识 UID
    sddc_set_uid(sddc, mac);

    // 获取并打印 IP 地址
    tcpip_adapter_get_ip_info(TCPIP_ADAPTER_IF_STA, &ip_info);

    inet_ntoa_r(ip_info.ip, ip, sizeof(ip));

    sddc_printf("IP addr: %s\n", ip);
    // 创建 Wi-Fi 事件组
    wifi_event_group = xEventGroupCreate();
    // 创建 FLASH 按键检测 task
    xTaskCreate(esp_flash_key_task, "flash_key_task",  ESP_KEY_TASK_STACK_SIZE, sddc, ESP_KEY_TASK_PRIO, NULL);

    // 运行 SDDC 协议循环
    while (1) {
        sddc_printf("SDDC running...\n");

        sddc_run(sddc);

        sddc_printf("SDDC quit!\n");
    }

    // 销毁 SDDC 协议对象
    sddc_destroy(sddc);
    // 删除 SmartConfig 任务
    vTaskDelete(NULL);
}

// FLASH 按键检测任务
static void esp_flash_key_task(void *arg)
{
    sddc_t *sddc = arg;
    gpio_config_t io_conf;
    int i = 0;
    
    (void)sddc;
    // 配置 GPIO 禁能中断
    io_conf.intr_type = GPIO_INTR_DISABLE;
    // 配置 GPIO 为输入模式
    io_conf.mode = GPIO_MODE_INPUT;
    // 配置 GPIO 引脚
    io_conf.pin_bit_mask = 1ULL << GPIO_INPUT_IO_SMARTCOFNIG;
    // 禁能下拉
    io_conf.pull_down_en = 0;
    // 使能上拉
    io_conf.pull_up_en = 1;
    // 配置 GPIO
    gpio_config(&io_conf);

    while (1) {
        vTaskDelay(1000 / portTICK_RATE_MS);
        // 检测 FLASH 按键按下
        if (!gpio_get_level(GPIO_INPUT_IO_SMARTCOFNIG)) {
            i++;
            // FLASH 按键按下超过三秒
            if (i > 3) {
                i = 0;

                sddc_printf("Start SmartConfig....\n");

                esp_wifi_disconnect();
                xEventGroupClearBits(wifi_event_group, CONNECTED_BIT | ESPTOUCH_DONE_BIT);
                // 注册 Wi-Fi 断开连接回调函数
                ESP_ERROR_CHECK(esp_event_handler_register(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &on_wifi_disconnect, NULL));
                // 注册 Wi-Fi 获取 IP 回调函数
                ESP_ERROR_CHECK(esp_event_handler_register(IP_EVENT, IP_EVENT_STA_GOT_IP, &on_got_ip, NULL));
                // 设置 SmartConfig 类型
                ESP_ERROR_CHECK( esp_smartconfig_set_type(SC_TYPE_ESPTOUCH_AIRKISS) );
                // 开始 SmartConfig
                ESP_ERROR_CHECK( esp_smartconfig_start(sc_callback) );

                while (1) {
                    // 等待连接上 Wi-Fi AP 和 ESPTOUCH 结束
                    EventBits_t uxBits = xEventGroupWaitBits(wifi_event_group, CONNECTED_BIT | ESPTOUCH_DONE_BIT,
                                                             true, false, portMAX_DELAY);
                    if (uxBits & CONNECTED_BIT) {
                        // 连接上 Wi-Fi AP
                        ESP_LOGI(TAG, "Wi-Fi Connected to AP");
                    }
                    if (uxBits & ESPTOUCH_DONE_BIT) {
                        // ESPTOUCH 结束
                        ESP_LOGI(TAG, "SmartConfig over");
                        // 停止 SmartConfig
                        esp_smartconfig_stop();
                        break;
                    }
                }
                // 取消注册 Wi-Fi 断开连接回调函数
                ESP_ERROR_CHECK(esp_event_handler_unregister(WIFI_EVENT, WIFI_EVENT_STA_DISCONNECTED, &on_wifi_disconnect));
                // 取消注册 Wi-Fi 获取 IP 回调函数
                ESP_ERROR_CHECK(esp_event_handler_unregister(IP_EVENT, IP_EVENT_STA_GOT_IP, &on_got_ip));
            }
        } else {
            i = 0;
        }
    }
}