...

1 files changed, 57 insertions, 0 deletions

diff --git a/include/cru/toml/TomlDocument.h b/include/cru/toml/TomlDocument.h
new file mode 100644
index 00000000..6da5ad74
--- /dev/null
+++ b/include/cru/toml/TomlDocument.h
@@ -0,0 +1,57 @@
+#pragma once
+
+#include "Base.h"
+
+#include "cru/common/Base.h"
+#include "cru/common/String.h"
+
+#include <optional>
+#include <unordered_map>
+
+namespace cru::toml {
+class CRU_TOML_API 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 CRU_TOML_API TomlDocument {
+ public:
+  CRU_DEFAULT_CONSTRUCTOR_DESTRUCTOR(TomlDocument)
+  CRU_DEFAULT_COPY(TomlDocument)
+  CRU_DEFAULT_MOVE(TomlDocument)
+
+ public:
+  TomlSection* GetSection(const String& name);
+  TomlSection* GetSectionOrCreate(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