#include <stdio.h>
#include "JZsdkLib.h"

#include "OpusFile.h"

static int g_OpusFile_SaveLen = 0;
static FILE *g_OpusFile_SaveFp = NULL;
static int g_OpusFile_SaveFlag = JZ_FLAGCODE_OFF; //opus存储标志位

#define OPUS_FILE_DIR "/root/sdcard/opus"
#define OPUS_TEMP_SAVE_DIR "/root/test_audio.opus"
static unsigned char opus_SaveFile[128];

static T_JZTaskHandle s_OpusSaveThread = NULL;

/******************
 *  
 * 
 *  opus文件 存储显示线程
 * 
 * 
 * **********************/
static void *OpusFile_SaveOpusPrint(void *arg)
{
    while (g_OpusFile_SaveFlag == JZ_FLAGCODE_ON)
    {
        int num = 0;

        JZSDK_LOG_INFO("opus存储功能开启中,当前已存入 %d 字节",g_OpusFile_SaveLen);

        //适时延迟
        while (num < 100 && g_OpusFile_SaveFlag == JZ_FLAGCODE_ON)
        {
            delayMs(10);
            num++;
        }
    }
}

/******************
 * 
 * 
 *  开始存储opus文件
 *  
 * 
 * **********************/
T_JZsdkReturnCode OpusFile_SaveOpusFileStart(unsigned char *FileName)
{
    T_JZsdkOsalHandler *OsalHandler = JZsdk_Platform_GetOsalHandler();

    if (g_OpusFile_SaveFlag == JZ_FLAGCODE_ON)
    {
        JZSDK_LOG_ERROR("opus文件保存功能已经开启");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }
    
    g_OpusFile_SaveLen = 0;

    memset(opus_SaveFile, 0, sizeof(opus_SaveFile));
    sprintf(opus_SaveFile, "%s/%s",OPUS_FILE_DIR, FileName);

    JZSDK_LOG_INFO("上传的opus文件存储路径为:%s", opus_SaveFile);

    if (g_OpusFile_SaveFp != NULL)
    {
        JZSDK_LOG_ERROR("要保存的opus文件已经打开,请先关闭保存功能");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }
    
    g_OpusFile_SaveFp = fopen(OPUS_TEMP_SAVE_DIR, "wb");
    if (g_OpusFile_SaveFp == NULL)
    {
        JZSDK_LOG_ERROR("打开opus文件失败");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    g_OpusFile_SaveFlag = JZ_FLAGCODE_ON;

    //创建存储显示线程
    OsalHandler->TaskCreate("OpusSavePrint", OpusFile_SaveOpusPrint, 8192,NULL, &s_OpusSaveThread);

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

/******************
 *      
 *  
 *  存储opus流中
 * 
 * 
 * **********************/
T_JZsdkReturnCode OpusFile_SaveOpusFileStream(unsigned char *Stream, int StreamLen)
{
    if (g_OpusFile_SaveFlag == JZ_FLAGCODE_OFF)
    {
        JZSDK_LOG_ERROR("没有打开opus文件保存功能");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    if (g_OpusFile_SaveFp == NULL)
    {
        JZSDK_LOG_ERROR("要保存的opus文件未打开");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    fseek(g_OpusFile_SaveFp, g_OpusFile_SaveLen, SEEK_SET);
    int writeLen = fwrite(Stream, 1, StreamLen, g_OpusFile_SaveFp);
    if (writeLen != StreamLen)
    {
        JZSDK_LOG_ERROR("opus文件写入失败");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    fflush(g_OpusFile_SaveFp);
    g_OpusFile_SaveLen += StreamLen;

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}

/******************
 *  
 *  
 *  结束存储opus文件
 *  
 *  checkflag是否需要校正 或校正类型
 *  checkcode 校正码
 *  saveflag 是否需要保存
 * 
 * **********************/
T_JZsdkReturnCode OpusFile_SaveOpusFileEnd(int CheckFlag, unsigned char *CheckCode, int SaveFlag)
{
    T_JZsdkReturnCode ret = JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
    unsigned char cmd[128];

    //标志位判定
    if (g_OpusFile_SaveFlag == JZ_FLAGCODE_OFF)
    {
        JZSDK_LOG_ERROR("没有打开opus文件保存功能");
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    //文件指针判定
    if (g_OpusFile_SaveFp == NULL)
    {
        JZSDK_LOG_ERROR("要保存的opus文件未打开");
        g_OpusFile_SaveFlag = JZ_FLAGCODE_OFF;
        return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
    }

    //关闭存储的文件
    fflush(g_OpusFile_SaveFp);
    fclose(g_OpusFile_SaveFp);
    g_OpusFile_SaveFp = NULL;
    
    JZSDK_LOG_INFO("已存储opus文件, 其长度为:%d", g_OpusFile_SaveLen);

/***********************************************************
 * 
 *    校验部分
 *   
 * 
 * *********************************************************/
    if (CheckFlag != JZ_FLAGCODE_OFF)
    {
        switch (CheckFlag)
        {
            case JZSDK_CHECK_SUM_TYPE_MD5:
            {
                JZSDK_LOG_INFO("opus保存使用md5校验码");
                //1、获取文件的md5校验码
                ret = JZsdk_CheckSum(JZSDK_CHECK_SUM_TYPE_MD5, OPUS_TEMP_SAVE_DIR, CheckCode, 32);
            }
            break;

            case JZSDK_CHECK_SUM_TYPE_SRC32:
            {
                JZSDK_LOG_INFO("opus保存使用src32校验码");
                //1、获取文件的src32校验码
                ret = JZsdk_CheckSum(JZSDK_CHECK_SUM_TYPE_SRC32, OPUS_TEMP_SAVE_DIR, CheckCode, 4);
            }
            break;
            
            default:
            {
                JZSDK_LOG_ERROR("opus保存校验码类型错误");
                ret = JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
            }
            break;
        }

        if (ret != JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS)
        {
            JZSDK_LOG_ERROR("opus保存校验失败");
            memset(cmd, 0, sizeof(cmd));
            sprintf(cmd, "rm -rf %s", OPUS_TEMP_SAVE_DIR);
            JZsdk_RunSystemCmd(cmd);  
            g_OpusFile_SaveFlag = JZ_FLAGCODE_OFF;

            return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
        }
    }
    else
    {
        JZSDK_LOG_INFO("opus保存无需校验码");
    }

/***********************************************************
 * 
 *    额外存储部分
 *   
 * 
 * *********************************************************/
    if (SaveFlag != JZ_FLAGCODE_OFF)
    {
        //将临时的opus文件移动到指定的目录
        memset(cmd, 0, sizeof(cmd));
        sprintf(cmd, "mv %s %s", OPUS_TEMP_SAVE_DIR, opus_SaveFile);
        ret = JZsdk_RunSystemCmd(cmd);
        if (ret != JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS)
        {
            JZSDK_LOG_ERROR("opus保存文件移动失败");
            memset(cmd, 0, sizeof(cmd));
            sprintf(cmd, "rm -rf %s", OPUS_TEMP_SAVE_DIR);
            JZsdk_RunSystemCmd(cmd);  
            g_OpusFile_SaveFlag = JZ_FLAGCODE_OFF;

            return JZ_ERROR_SYSTEM_MODULE_CODE_FAILURE;
        }

        //移动成功
        JZsdk_Sync(); //同步文件系统
        JZSDK_LOG_INFO("opus保存成功,刷新列表");
        OpusFile_FlushList(); //刷新opus文件列表
    }
/********************************************************************
 * 
 * 
 *  如果不进行额存储，则进行一次opus解码
 * 
 * 
/********************************************************************/
    else
    {
        OpusFile_DecodeOpus(opus_SaveFile);
    }

    JZSDK_LOG_INFO("opus_save_end");

    g_OpusFile_SaveFlag = JZ_FLAGCODE_OFF;

    return JZ_ERROR_SYSTEM_MODULE_CODE_SUCCESS;
}