Add ability to delete an uobject

2 files changed, 37 insertions, 0 deletions

diff --git a/db/uobject.cpp b/db/uobject.cpp
index 087c482..59a5b0b 100644
--- a/db/uobject.cpp
+++ b/db/uobject.cpp
@@ -1,5 +1,8 @@
 #include "uobject.h"
 
+using bsoncxx::builder::basic::kvp;
+using bsoncxx::builder::basic::make_document;
+
 std::string uobject::retrieveAll(std::string collection, struct json_object* jsonObjects) {
 
     auto conn = mongo.get_connection();
@@ -69,3 +72,33 @@ auto conn = mongo.get_connection();
 
     return json_object_to_json_string_ext(jsonObject, JSON_C_TO_STRING_SPACED | JSON_C_TO_STRING_PRETTY);
 }
+
+std::string uobject::remove(std::string collection, std::string oid, struct json_object* jsonObject) {
+    auto conn = mongo.get_connection();
+
+    auto coll = (*conn)[config["mongo_db"]][collection];
+
+    auto result = coll.delete_one(make_document(kvp("_id", bsoncxx::oid(oid))));
+
+
+    if (result->deleted_count() > 0) {
+        /* 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", {});
+    }
+    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 e08429f..3e0fd1e 100644
--- a/db/uobject.h
+++ b/db/uobject.h
@@ -2,6 +2,9 @@
 #define Uobject_H_
 
 #include <bsoncxx/builder/stream/document.hpp>
+#include <bsoncxx/builder/basic/kvp.hpp>
+#include <bsoncxx/oid.hpp>
+
 #include <bsoncxx/json.hpp>
 
 #include <mongocxx/client.hpp>
@@ -19,6 +22,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);
+    std::string remove(std::string collection, std::string oid, struct json_object* jsonObjects);
 }
 
 #endif