Split addUobject function

2 files changed, 56 insertions, 0 deletions

diff --git a/db/uobject.cpp b/db/uobject.cpp
index 769631e..087c482 100644
--- a/db/uobject.cpp
+++ b/db/uobject.cpp
@@ -14,3 +14,58 @@ std::string uobject::retrieveAll(std::string collection, struct json_object* jso
 
     return json_object_to_json_string_ext(jsonObjects, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY);
 }
+
+std::string uobject::add(std::string collection, struct json_object* jsonObject, const char * body) {
+auto conn = mongo.get_connection();
+
+    auto coll = (*conn)[config["mongo_db"]][collection];
+
+    auto document = bsoncxx::from_json(body);
+
+    auto result = coll.insert_one(document.view());
+
+    std::string oid = result->inserted_id().get_oid().value.to_string();
+
+
+    int stringlen = 0;
+    stringlen = strlen( body );
+    struct json_tokener *tok = json_tokener_new();
+    enum json_tokener_error jerr;
+
+    json_object *json_datas = NULL;
+
+    do {
+        json_datas = json_tokener_parse_ex(tok, body, stringlen);
+    } while ((jerr = json_tokener_get_error(tok)) == json_tokener_continue);
+
+    if (jerr != json_tokener_success)
+    {
+        fprintf(stderr, "Error: %s\n", json_tokener_error_desc(jerr));
+    }
+
+    if (json_object_get_type(json_datas) == json_type_object) {
+        /* Create an object with the following template:
+            * {
+            *  "_id": {
+            *      "$oid": "5d484d371ec4865f767d8424"
+            *  },
+            *  "datas": {
+            *      [...]
+            *  }
+            * }
+        */
+
+        auto json_id = json_object_new_object();
+        auto json_oid = json_object_new_string(oid.c_str());
+
+        json_object_object_add(json_id, "$oid", json_oid);
+        json_object_object_add(jsonObject, "_id", json_id);
+        json_object_object_add(jsonObject, "datas", json_datas);
+    } else {
+        std::cout << typeid(json_datas).name() << std::endl;
+        std::cout << "json_datas type: " << json_type_to_name(json_object_get_type(json_datas)) << std::endl;
+    }
+    json_tokener_reset(tok);
+
+    return json_object_to_json_string_ext(jsonObject, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY);
+}
diff --git a/db/uobject.h b/db/uobject.h
index b4c2ef0..e08429f 100644
--- a/db/uobject.h
+++ b/db/uobject.h
@@ -18,6 +18,7 @@
 
 namespace uobject {
     std::string retrieveAll(std::string collection, struct json_object* jsonObjects);
+    std::string add(std::string collection, struct json_object* jsonObjects, const char * body);
 }
 
 #endif