2023-10-06 21:10:59 -04:00

90 lines
3.0 KiB
C++

#include "cjson/cJSON.h"
#include "pocketpy/pocketpy.h"
namespace pkpy{
static Dict convert_cjson_to_dict(const cJSON * const item, VM* vm)
{
Dict output(vm);
List list;
if ((item == NULL))
{
return NULL;
}
if (((item->type) & 0xFF) == cJSON_Object){
cJSON *child = item->child;
while(child != NULL){
const cJSON *child_value = cJSON_GetObjectItemCaseSensitive(item, child->string);
if (cJSON_IsString(child_value))
{
output.set(VAR(Str(child->string)), VAR(Str(child_value->valuestring)));
}
else if (cJSON_IsNumber(child_value)){
output.set(VAR(Str(child->string)), VAR(child_value->valueint));
}
else if (cJSON_IsBool(child_value)){
output.set(VAR(Str(child->string)), VAR(child_value->valueint == 0?false:true));
}
else if (cJSON_IsNull(child_value)){
output.set(VAR(Str(child->string)), VAR(vm->None)); //Todo: Covert to python None
}
else if (cJSON_IsArray(child_value)){
cJSON *array_child = child_value->child;
while(array_child != NULL){
if (cJSON_IsString(array_child))
{
list.push_back(VAR(Str(array_child->valuestring)));
}
else if (cJSON_IsNumber(array_child)){
list.push_back(VAR(array_child->valueint));
}
else if (cJSON_IsBool(array_child)){
list.push_back(VAR(array_child->valueint == 0?false:true));
}
else if (cJSON_IsNull(array_child)){
list.push_back(VAR(vm->None));
}
array_child = array_child->next;
}
output.set(VAR(Str(child->string)), VAR(list));
}
else if (cJSON_IsObject(child_value)){
Dict child_object = convert_cjson_to_dict(child_value, vm);
output.set(VAR(Str(child->string)), VAR(child_object));
}
child = child->next;
}
}
return output;
}
inline void add_module_cjson(VM* vm){
PyObject* mod = vm->new_module("cjson");
vm->bind_func<1>(mod, "loads", [](VM* vm, ArgsView args) {
const Str& expr = CAST(Str&, args[0]);
cJSON *json = cJSON_Parse(expr.c_str());
Dict output = convert_cjson_to_dict(json, vm);
return VAR(output);
});
vm->bind_func<1>(mod, "dumps", [](VM* vm, ArgsView args) {
const Dict& dict = CAST(Dict&, args[0]);
//TODO: Convert dict to cJSON object
char* out = "This will be created from the dictionary";
return VAR(Str(out));
});
}
} // namespace pkpy