#ifdef SUPPORT_MS_EXTENSIONS
#pragma bss_seg(".testbox_update.data.bss")
#pragma data_seg(".testbox_update.data")
#pragma const_seg(".testbox_update.text.const")
#pragma code_seg(".testbox_update.text")
#endif

#include "update_loader_download.h"
#include "update.h"
#include "btctrler_task.h"
#include "app_config.h"
#include "clock.h"

#if TCFG_TEST_BOX_WIRELESS_EN

#if defined(CONFIG_SPP_AND_LE_CASE_ENABLE) || defined(CONFIG_HID_CASE_ENABLE)
#include "lib_profile_cfg.h"
#else
#include "bt_profile_cfg.h"
#endif

#if (RCSP_MODE == RCSP_MODE_WATCH)
#include "rcsp_task.h"
#endif

#define LOG_TAG "[TEST-UPDATE]"
#define LOG_INFO_ENABLE
#define LOG_ERROR_ENABLE
#include "debug.h"

extern u8 check_le_conn_disconnet_flag(void);
extern void ble_app_disconnect(void);
extern void ll_hci_destory(void);
extern void ram_protect_close(void);
extern void update_close_hw(void *filter_name);
extern void btctrler_testbox_update_msg_handle_register(void (*handle)(int));
extern void __bt_updata_reset_bt_bredrexm_addr(void);
extern int __bt_updata_save_connection_info(void);
extern const update_op_api_t lmp_ch_update_op;
extern const update_op_api_t ble_ll_ch_update_op;
static u8 ble_update_ready_jump_flag = 0;
extern void latch_reset(void);

static void testbox_bt_classic_update_private_param_fill(UPDATA_PARM *p)
{

}

static void testbox_bt_classic_update_before_jump_handle(int type)
{
    if (CONFIG_UPDATE_ENABLE && CONFIG_UPDATE_BT_LMP_EN) {
#if TCFG_USER_TWS_ENABLE || TCFG_USER_BLE_ENABLE
        log_info("close ble hw\n");
        ll_hci_destory();
#endif
        y_printf("\n >>>[test]:func = %s,line= %d\n", __FUNCTION__, __LINE__);

        update_close_hw("bredr");
        if (__bt_updata_save_connection_info()) {
            log_error("bt save conn info fail!\n");
            return;
        }

        ram_protect_close();
        //note:last func will not return;

#if (defined CONFIG_CPU_BR36 || defined CONFIG_CPU_BR27 || defined CONFIG_CPU_BR28)
        __bt_updata_reset_bt_bredrexm_addr();       //仅36、27、28使用跳转，后续CPU升级都直接reset
#else
        cpu_reset();
#endif
    }
}

static void testbox_bt_classic_update_state_cbk(int type, u32 state, void *priv)
{
    update_ret_code_t *ret_code = (update_ret_code_t *)priv;

    if (ret_code) {
        log_info("state:%x err:%x\n", ret_code->stu, ret_code->err_code);
    }

    switch (state) {
    case UPDATE_CH_EXIT:
        if (UPDATE_DUAL_BANK_IS_SUPPORT()) {
            if ((0 == ret_code->stu) && (0 == ret_code->err_code)) {
                log_info("bt update succ\n");
                update_result_set(UPDATA_SUCC);
            } else {
                log_info("bt update fail\n");
                update_result_set(UPDATA_DEV_ERR);
            }
        } else {
            if ((0 == ret_code->stu) && (0 == ret_code->err_code)) {
                //update_mode_api(BT_UPDATA);
                update_mode_api_v2(BT_UPDATA,
                                   testbox_bt_classic_update_private_param_fill,
                                   testbox_bt_classic_update_before_jump_handle);
            }
        }
        break;
    }
}

u8 ble_update_get_ready_jump_flag(void)
{
    return ble_update_ready_jump_flag;
}

static void testbox_ble_update_private_param_fill(UPDATA_PARM *p)
{
    u8 addr[6];
    extern int le_controller_get_mac(void *addr);
    if (BT_MODULES_IS_SUPPORT(BT_MODULE_LE) && CONFIG_UPDATE_ENABLE && CONFIG_UPDATE_BLE_TEST_EN) {
        le_controller_get_mac(addr);
        //memcpy(p->parm_priv, addr, 6);
        update_param_priv_fill(p, addr, sizeof(addr));
        puts("ble addr:\n");
        put_buf(p->parm_priv, 6);
    }
}

static void testbox_ble_update_before_jump_handle(int type)
{
    if (BT_MODULES_IS_SUPPORT(BT_MODULE_LE) && CONFIG_UPDATE_ENABLE && CONFIG_UPDATE_BLE_TEST_EN) {
#if (CPU_CORE_NUM == 1)
        local_irq_enable(); //解决ll_hci_destory内调用mutex_pend接口失败问题
#endif
        ll_hci_destory();
#if (CPU_CORE_NUM == 1)
        local_irq_disable();
#endif
    }


#if CONFIG_UPDATE_JUMP_TO_MASK
    y_printf(">>>[test]:latch reset update\n");
    latch_reset();
#else
    cpu_reset();
#endif
}

static void testbox_ble_update_state_cbk(int type, u32 state, void *priv)
{
    update_ret_code_t *ret_code = (update_ret_code_t *)priv;

    if (ret_code) {
        printf("ret_code->stu:%d err_code:%d\n", ret_code->stu, ret_code->err_code);
    }

    switch (state) {
    case UPDATE_CH_EXIT:
        if (UPDATE_DUAL_BANK_IS_SUPPORT()) {
            if ((0 == ret_code->stu) && (0 == ret_code->err_code)) {
                log_info("bt update succ\n");
                update_result_set(UPDATA_SUCC);
            } else {
                log_info("bt update fail\n");
                update_result_set(UPDATA_DEV_ERR);
            }
        } else {
            if ((0 == ret_code->stu) && (0 == ret_code->err_code)) {
                /* #if TCFG_USER_BLE_ENABLE && (TCFG_BLE_DEMO_SELECT != DEF_BLE_DEMO_NULL) \ */
                /* &&(TCFG_BLE_DEMO_SELECT != DEF_BLE_DEMO_ADV || defined(CONFIG_MESH_CASE_ENABLE))\ */
                /* && (TCFG_BLE_DEMO_SELECT != DEF_BLE_DEMO_CLIENT) */
#if TCFG_USER_BLE_ENABLE

                ble_update_ready_jump_flag = 1;
                /* ble_app_disconnect(); */
#if (BT_AI_SEL_PROTOCOL & LE_AUDIO_CIS_RX_EN)
                extern void le_hci_disconnect_all_connections(void);
                extern void bt_le_audio_adv_enable(u8 enable);
                le_hci_disconnect_all_connections();
                bt_le_audio_adv_enable(0);
#endif
                /* #if TCFG_BT_BLE_ADV_ENABLE */
                extern void ble_module_enable(u8 en);
                ble_module_enable(0);
                /* #endif */
                u8 cnt = 0;
                while (!check_le_conn_disconnet_flag()) {
                    log_info("wait discon\n");
                    os_time_dly(2);
                    if (cnt++ > 5) {
                        break;
                    }
                }

                //update_mode_api(BLE_TEST_UPDATA);
                update_mode_api_v2(BLE_TEST_UPDATA,
                                   testbox_ble_update_private_param_fill,
                                   testbox_ble_update_before_jump_handle);
#endif
            }
        }
        break;
    }
}

void testbox_update_msg_handle(int msg)
{
    log_info("msg:%x\n", msg);

    switch (msg) {
    case MSG_BT_UPDATE_LOADER_DOWNLOAD_START:
        if (CONFIG_UPDATE_ENABLE && CONFIG_UPDATE_BT_LMP_EN) {
// #if (RCSP_MODE == RCSP_MODE_WATCH)
//             app_rcsp_task_prepare(0, RCSP_TASK_ACTION_WATCH_TRANSFER, 0);
// #endif
            update_mode_info_t info = {
                .type = BT_UPDATA,
                .state_cbk = testbox_bt_classic_update_state_cbk,
                .p_op_api = &lmp_ch_update_op,
                .task_en = 1,
            };
            app_active_update_task_init(&info);
        }
        break;

#if TCFG_USER_BLE_ENABLE
    case MSG_BLE_TEST_OTA_LOADER_DOWNLOAD_START:
        if (CONFIG_UPDATE_ENABLE && CONFIG_UPDATE_BLE_TEST_EN) {
            update_mode_info_t info = {
                .type = BLE_TEST_UPDATA,
                .state_cbk =  testbox_ble_update_state_cbk,
                .p_op_api = &ble_ll_ch_update_op,
                .task_en = 1,
            };
            app_active_update_task_init(&info);
        }
        break;
#endif

    default:
        log_error("not support update msg:%x\n", msg);
        break;
    }

}

void testbox_update_init(void)
{
    if (CONFIG_UPDATE_ENABLE && (CONFIG_UPDATE_BLE_TEST_EN || CONFIG_UPDATE_BT_LMP_EN)) {
        log_info("testbox msg handle reg:%x\n", testbox_update_msg_handle);
        btctrler_testbox_update_msg_handle_register(testbox_update_msg_handle);
    }
}

#else // TCFG_TEST_BOX_WIRELESS_EN

void testbox_update_init(void)
{

}

u8 ble_update_get_ready_jump_flag(void)
{
    return 0;
}

#endif // TCFG_TEST_BOX_WIRELESS_EN