From a0fdaaaa7d653504cb86fb82800066595e8ce243 Mon Sep 17 00:00:00 2001
From: ThePhD <phdofthehouse@gmail.com>
Date: Fri, 25 Mar 2016 10:39:54 -0400
Subject: [PATCH] Full-on optional support.

---
 sol/stack_check.hpp     |  8 ++++++++
 sol/stack_check_get.hpp | 15 +++++++++++++++
 sol/stack_core.hpp      | 18 ++++++++++++++++--
 sol/types.hpp           |  7 +++++++
 tests.cpp               | 14 ++++++++++++++
 5 files changed, 60 insertions(+), 2 deletions(-)

diff --git a/sol/stack_check.hpp b/sol/stack_check.hpp
index 8a798310..60bc9c7d 100644
--- a/sol/stack_check.hpp
+++ b/sol/stack_check.hpp
@@ -282,6 +282,14 @@ struct checker<std::pair<A, B>, type::poly, C> {
         return stack::check<A>(L, index, handler) && stack::check<B>(L, index + 1, handler);
     }
 };
+
+template<typename T, typename C>
+struct checker<optional<T>, type::poly, C> {
+    template <typename Handler>
+    static bool check(lua_State* L, int index, Handler&& handler) {
+        return stack::check<T>(L, index, std::forward<Handler>(handler));
+    }
+};
 } // stack
 } // sol
 
diff --git a/sol/stack_check_get.hpp b/sol/stack_check_get.hpp
index efc93305..30cae43e 100644
--- a/sol/stack_check_get.hpp
+++ b/sol/stack_check_get.hpp
@@ -42,6 +42,14 @@ struct check_getter {
     }
 };
 
+template <typename T>
+struct check_getter<optional<T>> {
+    template <typename Handler>
+    static decltype(auto) get(lua_State* L, int index, Handler&&) {
+        return check_get<T>(L, index, no_panic);
+    }
+};
+
 template <typename T>
 struct check_getter<T, std::enable_if_t<std::is_integral<T>::value && !std::is_same<T, bool>::value>> {
     template <typename Handler>
@@ -69,6 +77,13 @@ struct check_getter<T, std::enable_if_t<std::is_floating_point<T>::value>> {
         return static_cast<T>(value);
     }
 };
+
+template <typename T>
+struct getter<optional<T>> {
+    static decltype(auto) get(lua_State* L, int index) {
+        return check_get<T>(L, index);
+    }
+};
 } // stack
 } // sol
 
diff --git a/sol/stack_core.hpp b/sol/stack_core.hpp
index d4c580a2..e7a468eb 100644
--- a/sol/stack_core.hpp
+++ b/sol/stack_core.hpp
@@ -158,8 +158,10 @@ inline decltype(auto) check_get(lua_State* L, int index = -1) {
     return check_get<T>(L, index, handler);
 }
 
-template<typename T>
-inline decltype(auto) get(lua_State* L, int index = -1) {
+namespace stack_detail {
+
+template <typename T>
+inline decltype(auto) tagged_get(types<T>, lua_State* L, int index = -1) {
 #ifdef SOL_CHECK_ARGUMENTS
     auto op = check_get<T>(L, index, type_panic);
     typedef typename meta::Unqualified<decltype(op)>::value_type U;
@@ -169,6 +171,18 @@ inline decltype(auto) get(lua_State* L, int index = -1) {
 #endif
 }
 
+template <typename T>
+inline decltype(auto) tagged_get(types<optional<T>>, lua_State* L, int index = -1) {
+    return stack_detail::unchecked_get<optional<T>>(L, index);
+}
+
+} // stack_detail
+
+template<typename T>
+inline decltype(auto) get(lua_State* L, int index = -1) {
+    return stack_detail::tagged_get(types<T>(), L, index);
+}
+
 template<typename T>
 inline decltype(auto) pop(lua_State* L) {
     return popper<meta::Unqualified<T>>{}.pop(L);
diff --git a/sol/types.hpp b/sol/types.hpp
index 271fdf34..baef5dcb 100644
--- a/sol/types.hpp
+++ b/sol/types.hpp
@@ -24,6 +24,7 @@
 
 #include "compatibility.hpp"
 #include "traits.hpp"
+#include "optional.hpp"
 #include <string>
 
 namespace sol {
@@ -274,6 +275,9 @@ struct lua_type_of<protected_function> : std::integral_constant<type, type::func
 template <typename Signature>
 struct lua_type_of<std::function<Signature>> : std::integral_constant<type, type::function>{};
 
+template <typename T>
+struct lua_type_of<optional<T>> : std::integral_constant<type, type::poly>{};
+
 template <typename T>
 struct lua_type_of<T*> : std::integral_constant<type, type::userdata> {};
 
@@ -303,6 +307,9 @@ struct is_proxy_primitive : is_lua_primitive<T> { };
 template <typename T>
 struct is_proxy_primitive<std::reference_wrapper<T>> : std::true_type { };
 
+template <typename T>
+struct is_proxy_primitive<optional<T>> : std::true_type {};
+
 template <typename... Args>
 struct is_proxy_primitive<std::tuple<Args...>> : std::true_type { };
 
diff --git a/tests.cpp b/tests.cpp
index 7155ed2e..8f95ffc5 100644
--- a/tests.cpp
+++ b/tests.cpp
@@ -673,6 +673,20 @@ TEST_CASE("tables/operator[]-valid", "Test if proxies on tables can lazily evalu
     REQUIRE_FALSE(isFullScreen);
 }
 
+TEST_CASE("tables/operator[]-optional", "Test if proxies on tables can lazily evaluate validity") {
+    sol::state lua;
+    
+    sol::optional<int> test1 = lua["no_exist_yet"];
+    bool present = (bool)test1;
+    REQUIRE_FALSE(present);
+
+    lua["no_exist_yet"] = 262;
+    sol::optional<int> test2 = lua["no_exist_yet"];
+    present = (bool)test2;
+    REQUIRE(present);
+    REQUIRE(test2.value() == 262);
+}
+
 TEST_CASE("tables/usertype", "Show that we can create classes from usertype and use them") {
 
     sol::state lua;