From 431cbdbe7d3ae8c45458dcf914717b0365ecd99a Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sat, 8 Jan 2022 16:31:09 +0800
Subject: ...

---
 include/cru/toml/TomlDocument.hpp | 54 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 include/cru/toml/TomlDocument.hpp

(limited to 'include/cru/toml/TomlDocument.hpp')

diff --git a/include/cru/toml/TomlDocument.hpp b/include/cru/toml/TomlDocument.hpp
new file mode 100644
index 00000000..9b549daa
--- /dev/null
+++ b/include/cru/toml/TomlDocument.hpp
@@ -0,0 +1,54 @@
+#pragma once
+
+#include "cru/common/Base.hpp"
+#include "cru/common/String.hpp"
+
+#include <optional>
+#include <unordered_map>
+
+namespace cru::toml {
+class TomlSection {
+ public:
+  CRU_DEFAULT_CONSTRUCTOR_DESTRUCTOR(TomlSection)
+  CRU_DEFAULT_COPY(TomlSection)
+  CRU_DEFAULT_MOVE(TomlSection)
+
+ public:
+  std::optional<String> GetValue(const String& key) const;
+  void SetValue(const String& key, String value);
+  void DeleteValue(const String& key);
+
+  auto begin() { return values_.begin(); }
+  auto end() { return values_.end(); }
+  auto begin() const { return values_.begin(); }
+  auto end() const { return values_.end(); }
+  auto cbegin() const { return values_.cbegin(); }
+  auto cend() const { return values_.cend(); }
+
+ private:
+  std::unordered_map<String, String> values_;
+};
+
+class TomlDocument {
+ public:
+  CRU_DEFAULT_CONSTRUCTOR_DESTRUCTOR(TomlDocument)
+  CRU_DEFAULT_COPY(TomlDocument)
+  CRU_DEFAULT_MOVE(TomlDocument)
+
+ public:
+  TomlSection* GetSection(const String& name);
+  const TomlSection* GetSection(const String& name) const;
+  void SetSection(const String& name, TomlSection section);
+  void DeleteSection(const String& name);
+
+  auto begin() { return sections_.begin(); }
+  auto end() { return sections_.end(); }
+  auto begin() const { return sections_.begin(); }
+  auto end() const { return sections_.end(); }
+  auto cbegin() const { return sections_.cbegin(); }
+  auto cend() const { return sections_.cend(); }
+
+ private:
+  std::unordered_map<String, TomlSection> sections_;
+};
+}  // namespace cru::toml
-- 
cgit v1.2.3