#include "snowboy/snowboy-detect-c-wrapper.h"
#include <stdlib.h>
#include <stdio.h>
#include "record.h"
#include "stt.h"
#include <string.h>
#include <stdbool.h>
#include "config.h"
#include "http.h"
#include "tts.h"  // 添加这个头文件包含

char* app_id = "0a9a80f0-2e03-435a-81a3-e2b402cc8f3a";

// 创建会话
char* create_conversation(char* authtoken) {
    char* url = "https://qianfan.baidubce.com/v2/app/conversation";
    char* auth;
    asprintf(&auth, "Authorization: Bearer %s", authtoken);
    struct curl_slist* headers = NULL;
    headers = curl_slist_append(headers, "Content-Type: application/json;charset=utf-8");
    headers = curl_slist_append(headers, auth);

    cJSON* obj = cJSON_CreateObject();
    cJSON_AddStringToObject(obj, "app_id", app_id);
    char* json = cJSON_Print(obj);

    size_t size = strlen(json);
    char* response = http_post(url, headers, json, &size);
    cJSON_Delete(obj);
    free(json);
    free(auth);
    curl_slist_free_all(headers);

    if (!response) {
        return NULL;
    }

    obj = cJSON_ParseWithLength(response, size);
    free(response);

    if (!obj) {
        fprintf(stderr, "解析 JSON 失败: [%s]\n", cJSON_GetErrorPtr());
        return NULL;
    }

    cJSON* conversation_id = cJSON_GetObjectItem(obj, "conversation_id");
    if (!conversation_id) {
        fprintf(stderr, "conversation_id 字段不存在\n");
        cJSON_Delete(obj);
        return NULL;
    }

    char* retval = strdup(conversation_id->valuestring);
    cJSON_Delete(obj);
    return retval;
}

// 调用对话API
char* chat(char* authtoken, char* conv_id, char* query) {
    char* url = "https://qianfan.baidubce.com/v2/app/conversation/runs";
    char* auth;
    asprintf(&auth, "Authorization: Bearer %s", authtoken);
    struct curl_slist* headers = NULL;
    headers = curl_slist_append(headers, "Content-Type: application/json;charset=utf-8");
    headers = curl_slist_append(headers, auth);

    cJSON* obj = cJSON_CreateObject();
    cJSON_AddStringToObject(obj, "app_id", app_id);
    cJSON_AddStringToObject(obj, "conversation_id", conv_id);
    cJSON_AddStringToObject(obj, "query", query);
    cJSON_AddBoolToObject(obj, "stream", false);
    char* json = cJSON_Print(obj);

    size_t size = strlen(json);
    char* response = http_post(url, headers, json, &size);
    cJSON_Delete(obj);
    free(json);
    curl_slist_free_all(headers);
    free(auth);

    if (!response) {
        return NULL;
    }

    obj = cJSON_ParseWithLength(response, size);
    free(response);

    if (!obj) {
        fprintf(stderr, "解析 JSON 失败: [%s]\n", cJSON_GetErrorPtr());
        return NULL;
    }

    cJSON* answer = cJSON_GetObjectItem(obj, "answer");
    if (!answer) {
        fprintf(stderr, "answer 字段不存在\n");
        puts(cJSON_Print(obj));
        cJSON_Delete(obj);
        return NULL;
    }

    char* retval = strdup(answer->valuestring);
    cJSON_Delete(obj);
    return retval;
}

int main() {
    // 创建snowboy检测器
    SnowboyDetect* detector = SnowboyDetectConstructor("common.res", "model.pmdl");
    if (!detector) {
        return EXIT_FAILURE;
    }
    SnowboyDetectSetSensitivity(detector, "0.6");
    SnowboyDetectSetAudioGain(detector, 1);

    int bits = SnowboyDetectBitsPerSample(detector);
    int channels = SnowboyDetectNumChannels(detector);
    int rate = SnowboyDetectSampleRate(detector);

    printf("采样深度: %d\n", bits);
    printf("声道数量: %d\n", channels);
    printf("采样频率: %d\n", rate);

    snd_pcm_uframes_t period = 999;
    snd_pcm_t* capture = record_open("hw:0,1", SND_PCM_FORMAT_S16_LE, channels, rate, &period);
    if (!capture) {
        return EXIT_FAILURE;
    }

    char* buffer = malloc(snd_pcm_frames_to_bytes(capture, period));
    if (!buffer) {
        perror("malloc");
        record_close(capture);
        SnowboyDetectDestructor(detector);
        return EXIT_FAILURE;
    }

    int recording = 0;
    int silence = 0;
    FILE* memstream = NULL;
    char* audio = NULL;
    size_t audio_size = 0;

    // 读取配置文件中的平台密钥
    cJSON* config = read_config("config.json");
    if (!config) {
        return EXIT_FAILURE;
    }

    cJSON* authtoken = cJSON_GetObjectItem(config, "authtoken");
    if (!authtoken) {
        fprintf(stderr, "配置文件错误: 找不到 'authtoken' 字段\n");
        cJSON_Delete(config);
        return EXIT_FAILURE;
    }

    printf("%s\n", authtoken->valuestring);

    // 创建会话
    char* conv_id = create_conversation(authtoken->valuestring);
    if (!conv_id) {
        fprintf(stderr, "创建会话失败\n");
        return EXIT_FAILURE;
    }
    printf("conv_id: %s\n", conv_id);

    while (1) {
        snd_pcm_sframes_t frames = snd_pcm_readi(capture, buffer, period);
        if (frames < 0) {
            fprintf(stderr, "Error from read: %s\n", snd_strerror(frames));
            snd_pcm_recover(capture, frames, 0);
            continue;
        }

        int status = SnowboyDetectRunDetection(detector, (int16_t*)buffer, snd_pcm_frames_to_bytes(capture, frames) / sizeof(int16_t), 0);
        if (status > 0) {
            printf("检测到唤醒词，开始录音\n");
            recording = 1;
            memstream = open_memstream(&audio, &audio_size);
            if (!memstream) {
                perror("open_memstream");
                continue;
            }
        }

        if (recording) {
            if (status == -2) {
                silence++;
            }

            if (status == 0) {
                silence = 0;
            }

            if (silence > 50) {
                printf("停止录音\n");
                recording = 0;
                silence = 0;
                if (memstream) {
                    fclose(memstream);
                    memstream = NULL;
                }

                if (audio_size) {
                    snd_pcm_drop(capture);
                    char* text = speech_to_text(audio, audio_size);
                    if (text) {
                        puts(text);

                        // 调用chat函数并打印对话反馈
                        char* response = chat(authtoken->valuestring, conv_id, text);
                        if (response) {
                            printf("< %s\n", response);

                            // 调用text_to_speech函数输出语音
                            text_to_speech(response);

                            free(response);
                        }
                    }
                    snd_pcm_prepare(capture);
                }
            }

            if (memstream) {
                fwrite(buffer, 1, snd_pcm_frames_to_bytes(capture, frames), memstream);
            }
        }
    }

    free(buffer);
    record_close(capture);
    SnowboyDetectDestructor(detector);
    free(conv_id);

    return EXIT_SUCCESS;
}