1 files changed, 122 insertions, 0 deletions
diff --git a/internal/dynamic_depth/container.cc b/internal/dynamic_depth/container.cc
new file mode 100644
index 0000000..1e3b9b6
--- /dev/null
+++ b/internal/dynamic_depth/container.cc
@@ -0,0 +1,122 @@
+#include "dynamic_depth/container.h"
+
+#include "android-base/logging.h"
+#include "dynamic_depth/const.h"
+
+using photos_editing_formats::xml::Deserializer;
+using photos_editing_formats::xml::Serializer;
+
+namespace photos_editing_formats {
+namespace dynamic_depth {
+
+constexpr char kNamespaceHref[] =
+    "http://ns.google.com/photos/dd/1.0/container/";
+constexpr char kDirectory[] = "Directory";
+
+// Private constructor.
+Container::Container() {}
+
+void Container::GetNamespaces(
+    std::unordered_map<string, string>* ns_name_href_map) {
+  if (ns_name_href_map == nullptr || items_.empty()) {
+    LOG(ERROR) << "Namespace list is null or item list is empty";
+    return;
+  }
+  ns_name_href_map->emplace(DynamicDepthConst::Container(), kNamespaceHref);
+  items_[0]->GetNamespaces(ns_name_href_map);
+}
+
+std::unique_ptr<Container> Container::FromItems(
+    std::vector<std::unique_ptr<Item>>* items) {
+  if (items == nullptr || items->empty()) {
+    LOG(ERROR) << "Item list is empty";
+    return nullptr;
+  }
+
+  std::unique_ptr<Container> container(new Container());
+  container->items_ = std::move(*items);
+  // Purge item elements that are null.
+  container->items_.erase(
+      std::remove_if(
+          container->items_.begin(), container->items_.end(),
+          [](const std::unique_ptr<Item>& item) { return item == nullptr; }),
+      container->items_.end());
+  if (container->items_.empty()) {
+    LOG(ERROR) << "No non-null elements in items";
+    return nullptr;
+  }
+
+  return container;
+}
+
+std::unique_ptr<Container> Container::FromDeserializer(
+    const Deserializer& parent_deserializer) {
+  std::unique_ptr<Container> container(new Container());
+  int i = 0;
+  std::unique_ptr<Deserializer> deserializer =
+      parent_deserializer.CreateDeserializerFromListElementAt(
+          DynamicDepthConst::Namespace(DynamicDepthConst::Container()),
+          DynamicDepthConst::Container(), 0);
+  while (deserializer) {
+    std::unique_ptr<Item> item = Item::FromDeserializer(*deserializer);
+    if (item == nullptr) {
+      LOG(ERROR) << "Unable to deserialize a item";
+      return nullptr;
+    }
+    container->items_.emplace_back(std::move(item));
+    deserializer = parent_deserializer.CreateDeserializerFromListElementAt(
+        DynamicDepthConst::Namespace(DynamicDepthConst::Container()),
+        DynamicDepthConst::Container(), ++i);
+  }
+
+  if (container->items_.empty()) {
+    return nullptr;
+  }
+  return container;
+}
+
+const std::vector<const Item*> Container::GetItems() const {
+  std::vector<const Item*> items;
+  for (const auto& item : items_) {
+    items.push_back(item.get());
+  }
+  return items;
+}
+
+bool Container::Serialize(Serializer* serializer) const {
+  if (items_.empty()) {
+    LOG(ERROR) << "Item list is empty";
+    return false;
+  }
+
+  std::unique_ptr<Serializer> container_serializer =
+      serializer->CreateSerializer(
+          DynamicDepthConst::Namespace(DynamicDepthConst::Container()),
+          DynamicDepthConst::Container());
+  std::unique_ptr<Serializer> directory_serializer =
+      container_serializer->CreateListSerializer(DynamicDepthConst::Container(),
+                                                 kDirectory);
+  if (directory_serializer == nullptr) {
+    // Error is logged in Serializer.
+    return false;
+  }
+
+  for (int i = 0; i < items_.size(); i++) {
+    std::unique_ptr<Serializer> item_serializer =
+        directory_serializer->CreateItemSerializer(
+            DynamicDepthConst::Namespace(DynamicDepthConst::Item()),
+            DynamicDepthConst::Item());
+    if (item_serializer == nullptr) {
+      LOG(ERROR) << "Could not create a list item serializer for Item";
+      return false;
+    }
+    if (!items_[i]->Serialize(item_serializer.get())) {
+      LOG(ERROR) << "Could not serialize item " << i;
+      return false;
+    }
+  }
+  return true;
+}
+
+}  // namespace dynamic_depth
+}  // namespace photos_editing_formats