#include <string.h>

#include "google/protobuf/descriptor.h"
#include "google/protobuf/util/json_util.h"
#include "google/protobuf/util/type_resolver.h"
#include "google/protobuf/util/type_resolver_util.h"

#include "google/protobuf/compiler/importer.h"
#include "google/protobuf/descriptor.h"
#include "google/protobuf/descriptor.pb.h"
#include "google/protobuf/dynamic_message.h"
#include "google/protobuf/message.h"

#include "3rdparty/bus_client/cbhomeclient.h"
#include "3rdparty/bus_client/message.h"
#include "3rdparty/yyjson/yyjson.h"
#include "proto/x86_64/sysset.pb.h"
#include <vector>

using namespace protomsg;
using namespace std;
using google::protobuf::util::JsonStringToMessage;

const char *CONTENT_TYPE_JSON = "application/json";

bool json_to_proto(const std::string &json, google::protobuf::Message &message) {
    google::protobuf::util::JsonParseOptions options;
    options.ignore_unknown_fields = true;

    return JsonStringToMessage(json, &message, options).ok();
}

bool proto_to_json(const google::protobuf::Message &message, std::string &json) {
    google::protobuf::util::JsonPrintOptions options;
    options.add_whitespace = true;
    options.always_print_primitive_fields = true;
    options.preserve_proto_field_names = true;

    return MessageToJsonString(message, &json, options).ok();
}

char *make_get_request(const char *topic) {
    auto doc = yyjson_mut_doc_new(NULL);
    auto root = yyjson_mut_obj(doc);
    yyjson_mut_obj_add_strn(doc, root, "path", topic, strlen(topic));
    yyjson_mut_obj_add_strn(doc, root, "method", "GET", 3);
    yyjson_mut_obj_add_strn(doc, root, "contentType", CONTENT_TYPE_JSON, strlen(CONTENT_TYPE_JSON));

    size_t jsonl = 0;
    char *json = yyjson_mut_val_write(root, 0, &jsonl);

    yyjson_mut_doc_free(doc);

    return json;
}

char *make_get_request(const char *topic, std::map<string, string> &query_params) {
    yyjson_mut_doc *doc = yyjson_mut_doc_new(NULL);
    yyjson_mut_val *root = yyjson_mut_obj(doc);
    yyjson_mut_obj_add_strn(doc, root, "path", topic, strlen(topic));
    yyjson_mut_obj_add_strn(doc, root, "method", "GET", 3);

    // 填充请求参数
    if (!query_params.empty()) {
        yyjson_mut_val *query_map = yyjson_mut_obj(doc);

        // std::map<string, string> t_map = *query_params;
        std::map<string, string>::iterator iter;
        for (iter = query_params.begin(); iter != query_params.end(); iter++) {
            yyjson_mut_val *_key = yyjson_mut_str(doc, iter->first.c_str());

            yyjson_mut_val *param = yyjson_mut_arr(doc);
            yyjson_mut_val *_val = yyjson_mut_str(doc, iter->second.c_str());
            yyjson_mut_arr_append(param, _val);

            yyjson_mut_obj_add(query_map, _key, param);
        }

        yyjson_mut_val *query_map_key = yyjson_mut_str(doc, "queryMap");
        yyjson_mut_obj_add(root, query_map_key, query_map);
    }

    yyjson_mut_doc_set_root(doc, root);

    size_t jsonl = 0;
    char *json = yyjson_mut_val_write(root, 0, &jsonl);

    yyjson_mut_doc_free(doc);

    return json;
}

extern "C" int bus_dbapi_get_json_data_by_key(const char *json, const char *flag_key, const char *data_key, char **pptr_data, yyjson_type *ptr_type) {
    int ret = -1;
    size_t data_len = 0;
    const char *ptr_value = NULL;
    const char *success_name = flag_key;
    const char *data_name = data_key;
    const char *msg_name = "msg";

    if (NULL == *pptr_data) {
        return -1;
    }

    // Read JSON and get root
    yyjson_doc *doc = yyjson_read(json, strlen(json), 0);
    yyjson_val *root = yyjson_doc_get_root(doc);

    if (NULL != success_name) {
        // Get root["success"]
        yyjson_val *success = yyjson_obj_get(root, success_name);

        int success_val = yyjson_get_bool(success);
        printf("success: %d\n", success_val);
        if (!success_val) {
            yyjson_doc_free(doc);
            return -1;
        }
    }

    // Get root["msg"]
    //yyjson_val *msg = yyjson_obj_get(root, "msg");
    yyjson_val *msg = yyjson_obj_get(root, "msg");
    printf("msg: %s\n", yyjson_get_str(msg));

    // Get root["data"]
    //yyjson_val *msg = yyjson_obj_get(root, "data");
    yyjson_val *data;
    if ( NULL != data_name )
    {
        data = yyjson_obj_get(root, data_name);        
    }
	else
	{
        data = root;     	    
	}


    *ptr_type = yyjson_get_type(data);
    ptr_value = yyjson_val_write(data, YYJSON_WRITE_NOFLAG, &data_len);
    printf("%s: %s\n", data_name, ptr_value);
    printf("len: %lu\n", data_len);

    memcpy(*pptr_data, ptr_value, data_len);
    ret = 0;

    // Free the doc
    yyjson_doc_free(doc);

    // All functions accept NULL input, and return NULL on error.
    return ret;
}

extern "C" int bus_dbapi_get_json_data(const char *json, char **pptr_data, yyjson_type *ptr_type) {
    /*{
      "success": true,
      "msg": "",
      "data": {
          "alarm_ip": "192.168.20.189",
          "alarm_port": 9200,
      }
  }*/

    /*{
"code": 200,
"data": [
  {
      "id": "295F36D6-7489-30AC-3263-335B4660AA79",
      "name": "全天",
      "time_rule":
"[{\"day\":1,\"time_range\":[{\"start\":\"05:31\",\"end\":\"21:07\"}]},{\"day\":2,\"time_range\":[{\"start\":\"01:42\",\"end\":\"23:59\"}]},{\"day\":3,\"time_range\":[{\"start\":\"00:00\",\"end\":\"23:59\"}]},{\"day\":4,\"time_range\":[{\"start\":\"00:00\",\"end\":\"23:59\"}]},{\"day\":5,\"time_range\":[{\"start\":\"00:00\",\"end\":\"23:59\"}]},{\"day\":6,\"time_range\":[{\"start\":\"00:00\",\"end\":\"23:59\"}]},{\"day\":7,\"time_range\":[{\"start\":\"00:00\",\"end\":\"23:59\"}]}]"
  },
  {
      "id": "21FA488C-2858-F366-4587-CE0661BFAD60",
      "name": "SAAS",
      "time_rule":
"[{\"day\":1,\"time_range\":[{\"start\":\"00:00\",\"end\":\"17:04\"}]},{\"day\":2,\"time_range\":[{\"start\":\"00:00\",\"end\":\"17:25\"}]},{\"day\":3,\"time_range\":[{\"start\":\"00:00\",\"end\":\"17:22\"}]},{\"day\":4,\"time_range\":[{\"start\":\"00:00\",\"end\":\"15:33\"}]},{\"day\":5,\"time_range\":[{\"start\":\"00:00\",\"end\":\"17:40\"}]},{\"day\":6,\"time_range\":[{\"start\":\"00:00\",\"end\":\"18:04\"}]},{\"day\":7,\"time_range\":[{\"start\":\"00:00\",\"end\":\"13:51\"}]}]"
  },
  {
      "id": "37B82C8F-3BDB-A247-3683-F045190D2991",
      "name": "116",
      "time_rule":
"[{\"day\":1,\"time_range\":[{\"start\":\"03:28\",\"end\":\"23:59\"}]},{\"day\":2,\"time_range\":[{\"start\":\"04:04\",\"end\":\"23:59\"}]},{\"day\":3,\"time_range\":[{\"start\":\"04:40\",\"end\":\"23:59\"}]},{\"day\":4,\"time_range\":[{\"start\":\"04:46\",\"end\":\"23:59\"}]},{\"day\":5,\"time_range\":[{\"start\":\"04:58\",\"end\":\"23:59\"}]},{\"day\":6,\"time_range\":[{\"start\":\"06:16\",\"end\":\"23:59\"}]},{\"day\":7,\"time_range\":[{\"start\":\"05:52\",\"end\":\"23:59\"}]}]"
  }
],
"msg": "",
"success": true
}*/
    const char *success_name = "success";
    const char *data_name = "data";

    if (NULL == *pptr_data) {
        return -1;
    }

    // All functions accept NULL input, and return NULL on error.
    return bus_dbapi_get_json_data_by_key(json, success_name, data_name, pptr_data, ptr_type);
}

bool bus_dbapi_get_topic_data(void *handle, const char *topic, crepmsg **pptr_repmsg, size_t len) {
    const auto topicl = strlen(topic);
    auto reqData = make_get_request(topic);
    auto reqmsg = make_req_msg(topic, topicl, reqData, strlen(reqData));

    if (bus_client_request(handle, reqmsg, pptr_repmsg)) {
        printf("======>> bus_client_reqest [%s] get [%s]\n", topic, (*pptr_repmsg)->data);
    } else {
        free(reqData);
        return false;
    }

    len = strlen((*pptr_repmsg)->data) + 1;

    free(reqData);
    return true;
}

bool bus_dbapi_get_topic_data(void *handle, const char *topic, crepmsg **pptr_repmsg, size_t len, std::map<string, string> &params) {
    const auto topicl = strlen(topic);
    auto reqData = make_get_request(topic, params);
    auto reqmsg = make_req_msg(topic, topicl, reqData, strlen(reqData));

    if (bus_client_request(handle, reqmsg, pptr_repmsg)) {
        printf("======>> bus_client_reqest [%s] get [%s]\n", topic, (*pptr_repmsg)->data);
    } else {
        free(reqData);
        return false;
    }

    len = strlen((*pptr_repmsg)->data) + 1;

    free(reqData);
    return true;
}

bool bus_dbapi_get(void *handle, const char *topic, google::protobuf::Message &message) {
    yyjson_type type = YYJSON_TYPE_OBJ;
    crepmsg *repmsg = NULL;
    size_t len = 0;
    bool bRet = false;

    bus_dbapi_get_topic_data(handle, topic, &repmsg, len);

    int msglen = strlen(repmsg->data);
    char *msgdata = (char *)malloc(msglen);
    memset(msgdata, 0, msglen);

    if (0 ==
        bus_dbapi_get_json_data((const char *)repmsg->data, &msgdata, &type)) {
        printf("======>> protomsg: %s\n", msgdata);
    }

    std::string jsonString = msgdata;
    bRet = json_to_proto(jsonString, message);
    if (bRet) {
        printf("======>> json_to_proto done\n");
    } else {
        printf("======>> json_to_proto fail\n");
    }

    free(msgdata);
    free_reply_msg(repmsg);

    return bRet;
}

/*****************************************************************************
 函 数 名  : bus_dbapi_get_str
 功能描述  : 获取订阅主题消息字符串，返回的内存调用者需要手工调用free函数释放
 输入参数  : void *handle
             const char* topic
             char **pptr_str
 输出参数  : 无
 返 回 值  :
 调用函数  :
 被调函数  :

 修改历史      :
  1.日    期   : 2023年1月13日
    作    者   : cheliequan
    修改内容   : 新生成函数

*****************************************************************************/
bool bus_dbapi_get_str(void *handle, const char *topic, char **pptr_str) {
    yyjson_type type = YYJSON_TYPE_OBJ;
    crepmsg *repmsg = NULL;
    size_t len = 0;
    int iRet = 0;
    bool bRet = false;

    bus_dbapi_get_topic_data(handle, topic, &repmsg, len);

    int msglen = strlen(repmsg->data);
    char *msgdata = (char *)malloc(msglen);
    memset(msgdata, 0, msglen);

    iRet = bus_dbapi_get_json_data((const char *)repmsg->data, &msgdata, &type);
    if (0 == iRet) {
        printf("======>> protomsg: %s\n", msgdata);
        bRet = true;
    }

    *pptr_str = msgdata;

    free(msgdata);
    free_reply_msg(repmsg);

    return bRet;
}

/*****************************************************************************
 函 数 名  : get_value_by_key
 功能描述  : 获取json字符串中字段为key值的value，pptr_data内存需要用户申请和释放                 
 输入参数  : char *json        
             const char *key   
             char **pptr_data  
 输出参数  : 无
 返 回 值  : 
 调用函数  : 
 被调函数  : 
 
 修改历史      :
  1.日    期   : 2023年1月13日
    作    者   : cheliequan
    修改内容   : 新生成函数

*****************************************************************************/
int get_value_by_key(char *json, const char *key, char **pptr_data) {
    size_t data_len = 0;

    // Read JSON and get root
    yyjson_doc *doc = yyjson_read(json, strlen(json), 0);
    yyjson_val *root = yyjson_doc_get_root(doc);

    // Get root[key]
    yyjson_val *val = yyjson_obj_get(root, key);

    char *val_str = (char *)yyjson_get_str(val);
    printf("val: %s\n", val_str);
    if (NULL == val_str) {
        yyjson_doc_free(doc);
        return -1;
    }

    char *ptr_value = yyjson_val_write(val, YYJSON_WRITE_NOFLAG, &data_len);
    printf("data: %s\n", ptr_value);
    printf("len: %lu\n", data_len);

    *pptr_data = (char *)malloc(data_len);
    memset(*pptr_data, 0, data_len);
    memcpy(*pptr_data, ptr_value, data_len);

    // Free the doc
    yyjson_doc_free(doc);

    return 0;
}