#include "string.h"
#include "vad_main.h"
#include "ifly_dec_file.h"
#include "sparkdesk_main.h"
#include "authentication.h"
#include "websocket_define.h"
#include "ifly_socket.h"
#include "app_config.h"
#include "app_task.h"
#include "system/timer.h"
#include "app_main.h"
#include "init.h"
#include "key_event_deal.h"
#include "device/device.h"
#include "app_power_manage.h"
#include "btstack/avctp_user.h"
#include "asm/charge.h"
#include "cJSON.h"
#include "audio_config.h"
#include "third_party_profile/interface/app_protocol_common.h"
#include "circular_buf.h"
#include "ui/ui_api.h"
#include "ifly_common.h"

#if TCFG_IFLYTEK_ENABLE

#define LOG_TAG_CONST       NET_IFLY
#define LOG_TAG             "[IFLY_VAD]"
#define LOG_ERROR_ENABLE
#define LOG_DEBUG_ENABLE
#define LOG_INFO_ENABLE
#define LOG_CLI_ENABLE
#include "debug.h"


#ifdef TCFG_IFLYTEK_APP_ID
#define APP_ID       				TCFG_IFLYTEK_APP_ID
#else
#define APP_ID       				"123"
#endif

#if TCFG_ENC_SPEEX_ENABLE
#define AI_AUDIO_CODING_TYPE    	AUDIO_CODING_SPEEX // 编码格式
#else
#error "ONLY SUPPORT SPEEX"
#endif

#define AI_AUDIO_CODING_SR          16000 // 采样率。和audio_mic_enc_open()函数中的对应

#define PCM_OUT_BUF_LEN				(AI_AUDIO_CODING_SR*2/1000 * 30)

#define SPEEX_SIZE   				42

#define AUDIO_LEN    				168
#define BASE63_AUDIO_LEN    		256

#define STATUS_FIRST_FRAME      	0
#define STATUS_CONTINUE_FRAME   	1
#define STATUS_LAST_FRAME       	2
#define STATUS_NED_FRAME       		3

#define HEART_BEAT_REQ              "client ping"    // 服务器下发的心跳保持请求

typedef enum {
    IFLY_VAD_STATUS_NULL = 0,
    IFLY_VAD_STATUS_START,	// 启动
    IFLY_VAD_STATUS_PCM_START,	// 音频启动发数
    IFLY_VAD_STATUS_SENDING,	// 音频数据发数中
    IFLY_VAD_STATUS_SEND_END,	// 音频数据发数完毕
    IFLY_VAD_STATUS_RECV,		// 有接受到数据
    IFLY_VAD_STATUS_RECV_END,	// 接受完成
    IFLY_VAD_STATUS_RECV_ERROR,	// 接受错误
    IFLY_VAD_STATUS_EXIT,		// 已经退出
} ifly_vad_status;

struct vad_info_t {
    u8 force_stop; 			// 强制结束
    u8 recv_finish;         // 接收结束
    u8 frame_status;
    char *pcm_out_buf;
    cbuffer_t pcm_cbuf;
    ifly_vad_status status;
    ifly_vad_param *param;
};

static struct vad_info_t vad_info;
static struct ifly_websocket_struct vad_socket;


#define AI_AUDIO_SAVE_TEST          0
#if AI_AUDIO_SAVE_TEST
static FILE *save_file = NULL;
#endif

extern int mbedtls_base64_encode(unsigned char *dst, size_t dlen, size_t *olen, const unsigned char *src, size_t slen);

//录音编码模块
static u16 vad_audio_send_data(u8 *voice_buf, u16 voice_len)
{
#if AI_AUDIO_SAVE_TEST
    if (save_file) {
        int wlen = fwrite(save_file, voice_buf, voice_len);
        if (wlen != voice_len) {
            log_error("save file err: %d, %d\n", wlen, voice_len);
        }
    }
#endif
    //上传数据到服务器
    int wlen = cbuf_write(&vad_info.pcm_cbuf, voice_buf, voice_len);
    if (wlen != voice_len) {
        log_error("pcm out err: %d, %d\n", wlen, voice_len);
    }
    return 0;
}

static int vad_audio_stop(int cancel)
{
    if (!ai_mic_is_busy()) {
        log_error("ai_mic_is_null \n\n");
        return true;
    }
    ai_mic_rec_close();

#if AI_AUDIO_SAVE_TEST
    if (save_file) {
        fclose(save_file);
        save_file = NULL;
    }
#endif
    return true;
}

static int vad_audio_start(void)
{
    if (ai_mic_is_busy()) {
        log_error("ai_mic_is_busy \n\n");
        return false;
    }
#if AI_AUDIO_SAVE_TEST
    if (save_file) {
        fclose(save_file);
        save_file = NULL;
    }
    save_file = fopen("storage/sd0/C/sf.bin", "w+");
    if (!save_file) {
        log_error("fopen err \n\n");
    }
#endif

    mic_rec_pram_init(AI_AUDIO_CODING_TYPE, 0, vad_audio_send_data, 4, 1024);

    ai_mic_rec_start();

    return true;
}

static void ifly_vad_recv_cb(u8 *j_str, u32 len, u8 type)
{
    log_info("recv:%s\n", j_str);

    if (vad_info.force_stop) {
        return;
    }
    if (vad_info.status >= IFLY_VAD_STATUS_RECV_END) {
        return;
    }
    if (!strcmp((char *)j_str, HEART_BEAT_REQ)) { // 保持心跳请求，不解析
        return;
    }
    cJSON *cjson_root = cJSON_Parse((char *)j_str);  //json解析错误
    if (cjson_root == NULL) {
        log_error("cjson error...\r\n");
        if (vad_info.status <= IFLY_VAD_STATUS_RECV_END) {
            vad_info.status = IFLY_VAD_STATUS_RECV_ERROR;
        }
        vad_info.param->event_cb(IFLY_VAD_EVT_NETWORK_RECV_ERROR, vad_info.param);
        return;
    }
    vad_info.status = IFLY_VAD_STATUS_RECV;

    cJSON *cjson_code = cJSON_GetObjectItem(cjson_root, "code");
    if (cjson_code && cjson_code->valueint != 0) {
        log_error("code error...%d\n", cjson_code->valueint);
        if (vad_info.status <= IFLY_VAD_STATUS_RECV_END) {
            vad_info.status = IFLY_VAD_STATUS_RECV_ERROR;
        }
        vad_info.param->event_cb(IFLY_VAD_EVT_NETWORK_RECV_ERROR, vad_info.param);
        return;
    }

    cJSON *cjson_data = cJSON_GetObjectItem(cjson_root, "data");
    cJSON *cjson_status = cJSON_GetObjectItem(cjson_data, "status");
    cJSON *cjson_result = cJSON_GetObjectItem(cjson_data, "result");
    cJSON *cjson_ws = cJSON_GetObjectItem(cjson_result, "ws");

    int arr_size = cJSON_GetArraySize(cjson_ws);
    cJSON *arr_item = cjson_ws->child;
    u32 vad_res_len = strlen(vad_info.param->vad_res);
    for (int i = 0; i < arr_size; i++) {
        cJSON *cjson_cw = cJSON_GetObjectItem(arr_item, "cw");
        int arr_size_cw = cJSON_GetArraySize(cjson_cw);
        cJSON *arr_item_cw = cjson_cw->child;
        for (int j = 0; j < arr_size_cw; j++) {
            cJSON *cjson_w = cJSON_GetObjectItem(arr_item_cw, "w");
            char *cjson_str = cJSON_Print(cjson_w);
            u32 json_len = strlen(cjson_str);
            if ((vad_res_len + json_len + 1) > vad_info.param->vad_res_len) {
                log_error("len error\n");
            } else {
                strcpy(&vad_info.param->vad_res[vad_res_len], cjson_str);
                vad_res_len += json_len;
            }
            arr_item_cw = arr_item_cw->next;
            cJSON_free(cjson_str);
        }

        arr_item = arr_item->next;
    }

    int res_len = strlen(vad_info.param->vad_res);

    str_remove_quote(vad_info.param->vad_res, res_len);

    log_info("final res:%s\n", vad_info.param->vad_res);

    if (cjson_status && cjson_status->valueint == STATUS_LAST_FRAME) {
        vad_info.status = IFLY_VAD_STATUS_RECV_END;
        vad_info.recv_finish = 1;
        vad_info.param->event_cb(IFLY_VAD_EVT_RECV_OK, vad_info.param);
    }

    cJSON_Delete(cjson_root);
}


//语音听写数据模块
char *ifly_vad_format_audio_data(void)
{
    char *data_str = NULL;
    cJSON *cjson_test = NULL;
    cJSON *cjson_common = NULL;
    cJSON *cjson_business = NULL;
    cJSON *cjson_data = NULL;

    int out_len = 0;
    char *buf = net_iflytek_malloc(BASE63_AUDIO_LEN);
    char *audio_data = net_iflytek_malloc(AUDIO_LEN);
    ASSERT(buf);
    ASSERT(audio_data);

    int rlen = cbuf_read(&vad_info.pcm_cbuf, audio_data, AUDIO_LEN);
    if (rlen != AUDIO_LEN) {
        net_iflytek_free(buf);
        net_iflytek_free(audio_data);
        return NULL;
    }

    if (vad_info.frame_status == STATUS_FIRST_FRAME) {

        mbedtls_base64_encode((unsigned char *)buf, BASE63_AUDIO_LEN, (size_t *)&out_len, (unsigned char *)audio_data, AUDIO_LEN);

        //编码第一帧
        cjson_test = cJSON_CreateObject();
        cjson_common = cJSON_CreateObject();
        cjson_business = cJSON_CreateObject();
        cjson_data = cJSON_CreateObject();

        cJSON_AddStringToObject(cjson_common, "app_id", APP_ID);
        cJSON_AddItemToObject(cjson_test, "common", cjson_common);

        cJSON_AddStringToObject(cjson_business, "language", "zh_cn");
        cJSON_AddStringToObject(cjson_business, "domain", "iat");
        cJSON_AddStringToObject(cjson_business, "accent", "mandarin");
        cJSON_AddNumberToObject(cjson_business, "speex_size", SPEEX_SIZE);
        cJSON_AddItemToObject(cjson_test, "business", cjson_business);

        cJSON_AddNumberToObject(cjson_data, "status", 0);
        cJSON_AddStringToObject(cjson_data, "format", "audio/L16;rate=16000");
        cJSON_AddStringToObject(cjson_data, "encoding", "speex-wb");
        cJSON_AddStringToObject(cjson_data, "audio", buf);
        cJSON_AddItemToObject(cjson_test, "data", cjson_data);

        data_str = cJSON_Print(cjson_test);

        cJSON_Delete(cjson_test);

        vad_info.frame_status = STATUS_CONTINUE_FRAME;

    } else if (vad_info.frame_status == STATUS_CONTINUE_FRAME) {

        mbedtls_base64_encode((unsigned char *)buf, BASE63_AUDIO_LEN, (size_t *)&out_len, (unsigned char *)audio_data, AUDIO_LEN);

        //编码
        cjson_test = cJSON_CreateObject();
        cjson_data = cJSON_CreateObject();

        cJSON_AddNumberToObject(cjson_data, "status", 1);
        cJSON_AddStringToObject(cjson_data, "format", "audio/L16;rate=16000");
        cJSON_AddStringToObject(cjson_data, "encoding", "speex-wb");
        cJSON_AddStringToObject(cjson_data, "audio", buf);
        cJSON_AddItemToObject(cjson_test, "data", cjson_data);

        data_str = cJSON_Print(cjson_test);

        cJSON_Delete(cjson_test);

    } else {
        //编码最后一帧
        mbedtls_base64_encode((unsigned char *)buf, BASE63_AUDIO_LEN, (size_t *)&out_len, (unsigned char *)audio_data, AUDIO_LEN);

        cjson_test = cJSON_CreateObject();
        cjson_data = cJSON_CreateObject();

        cJSON_AddNumberToObject(cjson_data, "status", 2);
        cJSON_AddStringToObject(cjson_data, "format", "audio/L16;rate=16000");
        cJSON_AddStringToObject(cjson_data, "encoding", "speex-wb");
        cJSON_AddStringToObject(cjson_data, "audio", buf);
        cJSON_AddItemToObject(cjson_test, "data", cjson_data);

        data_str = cJSON_Print(cjson_test);

        cJSON_Delete(cjson_test);

        if (vad_info.status < IFLY_VAD_STATUS_SEND_END) {
            vad_info.status = IFLY_VAD_STATUS_SEND_END;
            vad_audio_stop(1);
            cbuf_clear(&vad_info.pcm_cbuf);
        }
    }

    net_iflytek_free(buf);
    net_iflytek_free(audio_data);

    return data_str;
}

static bool ifly_vad_get_send(u8 **buf, u32 *len)
{
    if (vad_info.force_stop) {
        log_info("vad task kill!\n");
        return false;
    }
    if (vad_info.status >= IFLY_VAD_STATUS_RECV_END) {
        log_info("recv end vad task kill!\n");
        return false;
    }
    if (vad_info.status < IFLY_VAD_STATUS_PCM_START) {
        vad_info.status = IFLY_VAD_STATUS_PCM_START;
        vad_audio_start();
        vad_info.param->event_cb(IFLY_VAD_EVT_AUDIO_START, vad_info.param);
    }
    if (cbuf_get_data_len(&vad_info.pcm_cbuf) >= AUDIO_LEN) {
        char *input_src_json = ifly_vad_format_audio_data();
        if (input_src_json == NULL) {
            log_error("get json err \n");
            return false;
        }
        *buf = (u8 *)input_src_json;
        *len = strlen(input_src_json);
        return true;
    }
    os_time_dly(2);
    return true;
}

static int ifly_vad_event_cb(ifly_socket_event_enum evt, void *param)
{
    switch (evt) {
    case IFLY_SOCKET_EVT_SEND_OK:
        cJSON_free(param);
        break;
    case IFLY_SOCKET_EVT_SEND_ERROR:
        cJSON_free(param);
        break;
    case IFLY_SOCKET_EVT_INIT_OK:
        break;
    case IFLY_SOCKET_EVT_INIT_ERROR:
    case IFLY_SOCKET_EVT_HANSHACK_ERROR:
    case IFLY_SOCKET_EVT_ACCIDENT_END:
    case IFLY_SOCKET_EVT_END:
    case IFLY_SOCKET_EVT_FORCE_END:
        vad_audio_stop(1);
        if ((evt != IFLY_SOCKET_EVT_END) && (evt != IFLY_SOCKET_EVT_FORCE_END)) {
            if (!vad_info.recv_finish) {
                vad_info.param->event_cb(IFLY_VAD_EVT_NETWORK_FAIL, vad_info.param);
            }
        }
        break;
    case IFLY_SOCKET_EVT_EXIT:
        vad_info.status = IFLY_VAD_STATUS_EXIT;
        vad_info.param->event_cb(IFLY_VAD_EVT_EXIT, vad_info.param);
        if (vad_socket.auth) {
            net_iflytek_free(vad_socket.auth);
            vad_socket.auth = NULL;
        }
        if (vad_info.pcm_out_buf) {
            net_iflytek_free(vad_info.pcm_out_buf);
            vad_info.pcm_out_buf = NULL;
        }
        break;
    default:
        break;
    }
    return 0;
}

bool ifly_vad_start(ifly_vad_param *param)
{
    memset(&vad_info, 0, sizeof(struct vad_info_t));
    memset(&vad_socket, 0, sizeof(struct ifly_websocket_struct));

#if TCFG_IFLYTEK_USE_PSRAM
    cJSON_Hooks hooks;
    hooks.malloc_fn = net_iflytek_malloc;
    hooks.free_fn = net_iflytek_free;
    cJSON_InitHooks(&hooks);
#endif

    vad_info.pcm_out_buf = net_iflytek_malloc(PCM_OUT_BUF_LEN);
    ASSERT(vad_info.pcm_out_buf);
    cbuf_init(&vad_info.pcm_cbuf, vad_info.pcm_out_buf, PCM_OUT_BUF_LEN);

    vad_info.param = param;
    vad_socket.auth = (u8 *)ifly_authentication("wss://iat-api.xfyun.cn/v2/iat",
                      "iat-api.xfyun.cn",
                      "GET /v2/iat HTTP/1.1", 20);
    if (!vad_socket.auth) {
        net_iflytek_free(vad_info.pcm_out_buf);
        vad_info.pcm_out_buf = NULL;
        vad_info.param->event_cb(IFLY_VAD_EVT_NETWORK_FAIL, vad_info.param);
        return false;
    }
    vad_socket.task_name = "ifly_vad";
    vad_socket.socket_mode = WEBSOCKET_MODE;
    vad_socket.recv_cb = ifly_vad_recv_cb;
    vad_socket.get_send = ifly_vad_get_send;
    vad_socket.event_cb = ifly_vad_event_cb;

    vad_info.status = IFLY_VAD_STATUS_START;

    //创建链接
    bool ret = ifly_websocket_client_create(&vad_socket);
    if (ret == false) {
        vad_info.status = IFLY_VAD_STATUS_NULL;
        net_iflytek_free(vad_socket.auth);
        vad_socket.auth = NULL;
        net_iflytek_free(vad_info.pcm_out_buf);
        vad_info.pcm_out_buf = NULL;
    }
    return ret;
}

void ifly_vad_stop(u8 force_stop, u32 to_ms)
{
    log_info("vad stop!\n");
    ifly_vad_audio_stop();
    vad_info.force_stop = force_stop;
    while (vad_socket.auth) { // 结束时auth会自动释放
        os_time_dly(1);
        if (to_ms <= 10) {
            break;
        }
        to_ms -= 10;
    }
    if (to_ms < 1000) {
        to_ms = 1000;
    }
    vad_info.force_stop = 1;
    ifly_websocket_client_release(&vad_socket, to_ms);
}

void ifly_vad_audio_stop(void)
{
    vad_info.frame_status = STATUS_LAST_FRAME; // 停止语音发送
}

bool ifly_vad_is_work()
{
    if ((vad_info.status != IFLY_VAD_STATUS_NULL) && (vad_info.status != IFLY_VAD_STATUS_EXIT)) {
        return true;
    }
    return false;
}
#endif