From 3083f6549388141a0d02eba79c31a1385a88e67a Mon Sep 17 00:00:00 2001
From: ToruNiina <niina.toru.68u@gmail.com>
Date: Mon, 11 Dec 2017 12:04:57 +0900
Subject: [PATCH] add get_or

---
 tests/CMakeLists.txt  |  1 +
 tests/test_get_or.cpp | 83 +++++++++++++++++++++++++++++++++++++++++++
 toml/get.hpp          | 15 +++++++-
 3 files changed, 98 insertions(+), 1 deletion(-)
 create mode 100644 tests/test_get_or.cpp

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 09cccef..51ca609 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -5,6 +5,7 @@ set(TEST_NAMES
     test_to_toml
     test_from_toml
     test_get
+    test_get_or
     test_value_operator
     test_datetime
     test_acceptor
diff --git a/tests/test_get_or.cpp b/tests/test_get_or.cpp
new file mode 100644
index 0000000..edb439b
--- /dev/null
+++ b/tests/test_get_or.cpp
@@ -0,0 +1,83 @@
+#define BOOST_TEST_MODULE "test_get"
+#include <boost/test/included/unit_test.hpp>
+#include <toml.hpp>
+#include <map>
+#include <unordered_map>
+#include <list>
+#include <deque>
+#include <array>
+
+
+BOOST_AUTO_TEST_CASE(test_get_or_exist)
+{
+    toml::Boolean  raw_v1(true);
+    toml::Integer  raw_v2(42);
+    toml::Float    raw_v3(3.14);
+    toml::String   raw_v4("hoge");
+    toml::Array    raw_v5{2,7,1,8,2};
+    toml::Table    raw_v6{{"key", 42}};
+
+    toml::value v1(raw_v1);
+    toml::value v2(raw_v2);
+    toml::value v3(raw_v3);
+    toml::value v4(raw_v4);
+    toml::value v5(raw_v5);
+    toml::value v6(raw_v6);
+
+    toml::Table table{
+        {"value1", v1},
+        {"value2", v2},
+        {"value3", v3},
+        {"value4", v4},
+        {"value5", v5},
+        {"value6", v6}
+    };
+
+    toml::Boolean u1 = toml::get_or(table, "value1", raw_v1);
+    toml::Integer u2 = toml::get_or(table, "value2", raw_v2);
+    toml::Float   u3 = toml::get_or(table, "value3", raw_v3);
+    toml::String  u4 = toml::get_or(table, "value4", raw_v4);
+    toml::Array   u5 = toml::get_or(table, "value5", raw_v5);
+    toml::Table   u6 = toml::get_or(table, "value6", raw_v6);
+
+    BOOST_CHECK_EQUAL(u1, raw_v1);
+    BOOST_CHECK_EQUAL(u2, raw_v2);
+    BOOST_CHECK_EQUAL(u3, raw_v3);
+    BOOST_CHECK_EQUAL(u4, raw_v4);
+    BOOST_CHECK_EQUAL(u5.at(0).cast<toml::value_t::Integer>(), raw_v5.at(0).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(1).cast<toml::value_t::Integer>(), raw_v5.at(1).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(2).cast<toml::value_t::Integer>(), raw_v5.at(2).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(3).cast<toml::value_t::Integer>(), raw_v5.at(3).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(4).cast<toml::value_t::Integer>(), raw_v5.at(4).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u6.at("key").cast<toml::value_t::Integer>(), 42);
+}
+
+BOOST_AUTO_TEST_CASE(test_get_or_empty)
+{
+    toml::Boolean  raw_v1(true);
+    toml::Integer  raw_v2(42);
+    toml::Float    raw_v3(3.14);
+    toml::String   raw_v4("hoge");
+    toml::Array    raw_v5{2,7,1,8,2};
+    toml::Table    raw_v6{{"key", 42}};
+
+    toml::Table    table; // empty!
+
+    toml::Boolean  u1 = toml::get_or(table, std::string("value1"), raw_v1);
+    toml::Integer  u2 = toml::get_or(table, std::string("value2"), raw_v2);
+    toml::Float    u3 = toml::get_or(table, std::string("value3"), raw_v3);
+    toml::String   u4 = toml::get_or(table, std::string("value4"), raw_v4);
+    toml::Array    u5 = toml::get_or(table, std::string("value5"), raw_v5);
+    toml::Table    u6 = toml::get_or(table, std::string("value6"), raw_v6);
+
+    BOOST_CHECK_EQUAL(u1, raw_v1);
+    BOOST_CHECK_EQUAL(u2, raw_v2);
+    BOOST_CHECK_EQUAL(u3, raw_v3);
+    BOOST_CHECK_EQUAL(u4, raw_v4);
+    BOOST_CHECK_EQUAL(u5.at(0).cast<toml::value_t::Integer>(), raw_v5.at(0).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(1).cast<toml::value_t::Integer>(), raw_v5.at(1).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(2).cast<toml::value_t::Integer>(), raw_v5.at(2).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(3).cast<toml::value_t::Integer>(), raw_v5.at(3).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u5.at(4).cast<toml::value_t::Integer>(), raw_v5.at(4).cast<toml::value_t::Integer>());
+    BOOST_CHECK_EQUAL(u6.at("key").cast<toml::value_t::Integer>(), 42);
+}
diff --git a/toml/get.hpp b/toml/get.hpp
index ea32216..c532fe5 100644
--- a/toml/get.hpp
+++ b/toml/get.hpp
@@ -14,7 +14,7 @@ inline T get(const toml::value& v)
     return static_cast<T>(v.cast<vT>());
 }
 
-// array case
+// array-like type
 template<typename T, toml::value_t vT = toml::detail::check_type<T>(),
          typename std::enable_if<(vT == toml::value_t::Unknown) &&
          (!toml::detail::is_map<T>::value) &&
@@ -24,6 +24,7 @@ T get(const toml::value& v)
     if(v.type() != value_t::Array)
         throw type_error("get: value type: " + stringize(v.type()) +
                          std::string(" is not argument type: Array"));
+
     const auto& ar = v.cast<value_t::Array>();
     T tmp;
     try
@@ -39,6 +40,7 @@ T get(const toml::value& v)
     return tmp;
 }
 
+// table-like case
 template<typename T, toml::value_t vT = toml::detail::check_type<T>(),
          typename std::enable_if<(vT == toml::value_t::Unknown) &&
          toml::detail::is_map<T>::value, std::nullptr_t>::type = nullptr>
@@ -53,5 +55,16 @@ T get(const toml::value& v)
     return tmp;
 }
 
+//  get_or -----------------------------------------------------------------
+
+template<typename T>
+inline typename std::remove_cv<typename std::remove_reference<T>::type>::type
+get_or(const toml::Table& tab, const toml::key& ky, T&& opt)
+{
+    if(tab.count(ky) == 0) {return std::forward<T>(opt);}
+    return get<typename std::remove_cv<
+        typename std::remove_reference<T>::type>::type>(tab.find(ky)->second);
+}
+
 } // toml
 #endif// TOML11_GET