#include "WeWorkFinanceSdk_C.h"
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <string>
using std::string;

// 私钥内容，示例
static const char* priKey = 
"-----BEGIN RSA PRIVATE KEY-----\n"
"MIIEogIBAAKCAQEAngIe+L+lJFQOWu+FtgWzTYcG+bQTANvgatpjM0CswTLR5oY+\n"
"IvpQ9X0EID3AEC0PyaatOgSvmvWCUQunZtA3lhdkdsJG1Nxe1FOOI+Q1qlgW/ykT\n"
"JkuJJv8qnd9HFdpkTkkAB3CxW5zGoYYtBlnp3G3mUhGXZ/6dIWQ4Wln4lJlUlxSe\n"
"Qv4YvDOGN2dWbIG79nL9YNKyk9yYgLop2O7XJGXgqCuV5CUz4jdxB5XCH3SzqFgJ\n"
"A36JguG5jD7ovFh7cL+yp1iTyF/+1woDARXXpIj+W6K3Skoo5cpt0BboPYu0EuCz\n"
"YQtfn+hUfkcS6kltnwSXUBFzrSLhjCkf5tY/LwIDAQABAoIBADxB8M/KPk9hYuwZ\n"
"BjQJbny1btqRauHb+HSnfkdQddFj8RppvY6XSYKlP4qQIgvJM4kicFN87/83e5Au\n"
"0pNp+pjZqO2FX3/T3rI4fy9+21xwJaz/XBcxnUdq7jSwrqpN1Azq2AD+YBTg7/sv\n"
"cUJHzeDLZJeNr+Dv/7p5PuhuAVehm3g+b0GwuWK2HieXAlRR240vn+dCZoq8ti6Y\n"
"jmcZcJ9E6urY/L3d5bqByN2SGSbgJav6VAQYf8vfOU+SeAXFunGUWasuS3tvQlMO\n"
"eEPW37oNAsKHkKDQcKAcPo59fwPdbPKE/5N/wKPsbN8nVYTqdzociQdZnNpG8DK2\n"
"SdE5ZzkCgYEA4S3mNaklMA0vNR20lSum68j4L0r57Q7pN0ATmItBhF5ihEbtLlAg\n"
"OsLsxQZ2XIzevsnvUvRKDi7pcu3u2VqlX/89aWOcTOyEyKGHCUPkkOr4+3m2cGtQ\n"
"WF676TtNcfj4yOyJ5gMmtld84yUwrHjHgvtP/L73MftHzssQFj0Qk20CgYEAs6Ka\n"
"kXLAHtKwB5R/knXmkCYnfRX/TqjhTVO3Hr2PvqojAmiuWmWp9YsXtbYDFpxQPOp0\n"
"Y1qosC1qW+pZCXgSlsjsdhJa3t2hhsF2SAWoy2QpAsuHWizHq9T8XFb14U/4PjBz\n"
"B9P1mf5c84Iox3tUnwKuBsNt8FOXwJRPnddsH4sCgYB2zpLR6SaXzb0El2G5+sgd\n"
"S3G28iYfb2iZIBO2vnKY9nugMxPdBaWbm1D1O02j0yeLHlxJNl8EULgXTXZh87Oh\n"
"jLfILShAZeJBQQhQWSP1WiR3+mXJtrUzivg2PHXGFoic6kBvvWXlew+HgQ3HUgpi\n"
"D1rKja87Dct7TiR0t09RmQKBgBspvWG9Dlwa089xnxDQjzrA6ju59c85Lkh5Hjmh\n"
"xHi6VWzp02OL9/VPQRiTG0hx/pgXg936OYPaejjH6QrLg1QVsdvwzpjgvqoX2wDt\n"
"CicnUCRCDoeOOes726VC6Q9W/ly5CsvmOz87/WY9ClmPxq38w7/2EUvD+DCAvMFL\n"
"b+gdAoGAUiY1xefNe4qnbQzJXpPUUcOGiJcl7r9HwrZx8tDwGgBg+/5zbwi/0TfH\n"
"YrIj86g+JHaMEQCcKko8H0KdQzzp1vNECNxOGvz8ckOrO+NsLzHOpZ4jhojV/vFg\n"
"FNl6PhOdUozqTYwrnszoO67a/jKS/rWQh5Ta2Y2K5sUNprk8Z1w=\n"
"-----END RSA PRIVATE KEY-----\n";

// rsa 解密函数，需企业自行实现
int RsaDecryptChatData(const string &encryptdata,const string &priKey,string &aeskey);

int main(int argc, char *argv[])
{
	int ret = 0;
	if (argc < 2)
	{
		printf("./sdktools 1 seq limit proxy passwd timeout\n");
		printf("./sdktools 2 fileid savefile proxy passwd timeout\n");
		printf("./sdktools 3 encrypt_key encrypt_msg\n");
		return -1;
	}

	// new sdk api
    WeWorkFinanceSdk_t* sdk = NewSdk();

	// init api
    ret = Init(sdk, "wwd08coe7d775abaaa","zJ6k0naVVQ--gt9PUSSEvs03zW_nlDVmjAkPOTAfrew");
    if (ret != 0) {

       printf("init sdk err ret:%d\n", ret);
	   return ret;
    }

    int type =  strtoul(argv[1], NULL, 10);
	if (type == 1)
	{
		uint64_t iSeq = strtoul(argv[2], NULL, 10);
		uint64_t iLimit = strtoul(argv[3], NULL, 10);
		uint64_t timeout = strtoul(argv[6], NULL, 10);        
        Slice_t* chatDatas = NewSlice();

		// getchatdata api
        ret = GetChatData(sdk, iSeq, iLimit, argv[4],argv[5], timeout, chatDatas);
        if (ret != 0) {
            printf("GetChatData err ret:%d\n", ret);
            return -1;
        }
        printf("GetChatData len:%d data:%s\n", chatDatas->len, GetContentFromSlice(chatDatas));
        FreeSlice(chatDatas);
	}
	else if (type == 2)
	{
		std::string index = "";
		uint64_t timeout = strtoul(argv[6], NULL, 10);        
		while(true){
			printf("index:%s\n",index.c_str());
			MediaData_t* media = NewMediaData();

			// getmediadata api
			ret = GetMediaData(sdk, index.c_str(), argv[2],argv[4],argv[5], timeout, media);

			if (ret != 0) {
				printf("GetMediaData err ret:%d\n", ret);
				FreeMediaData(media);
				return -1;
			}
			printf("content size:%d isfin:%d outindex:%s\n",GetDataLen(media),IsMediaDataFinish(media),GetOutIndexBuf(media));

			char file[200];
			snprintf(file, sizeof(file), "%s%s", "/home/qspace/upload/", argv[3]);
			FILE *fp = fopen(file, "ab+");
			printf("filename:%s \n", file);
			if (NULL == fp) {
				printf("open file err\n");
				return -1;
			}

			fwrite(GetData(media),GetDataLen(media), 1, fp);
			fclose(fp);

			if(IsMediaDataFinish(media)==1)
			{
				FreeMediaData(media);
				break;
			}
			else{
				// assign index value if not finish
				index.assign(GetOutIndexBuf(media));

				// need free media 
				FreeMediaData(media);
			}
		}
    } 	else if (type == 3) {
        Slice_t* msg = NewSlice();
        std::string encrypt_key;

		// rsa decrypt code .need do it by yourself
        int ret = RsaDecryptChatData(argv[2], priKey, encrypt_key);
        printf("Rsadecrypt key :%s ret :%d\n" ,encrypt_key.c_str(), ret);

		// decryptdata api
        ret = DecryptData(encrypt_key.c_str(), argv[3], msg);
        printf("chatdata :%s ret :%d\n" ,GetContentFromSlice(msg), ret);
		FreeSlice(msg);
    } else if(type == 4){
		Slice_t* msg = NewSlice();
		ret = DecryptData(argv[2], argv[3], msg);
		printf("chatdata :%s ret :%d\n" ,GetContentFromSlice(msg), ret);
	}
    DestroySdk(sdk);
	return ret;
}