From af0097fb7f139b171f83bca1eb4f67ffd5576bac Mon Sep 17 00:00:00 2001
From: Rapptz <rapptz@gmail.com>
Date: Thu, 12 Dec 2013 19:48:26 -0500
Subject: [PATCH] Add support for setting functions with operator[]

---
 sol/table.hpp  | 7 ++++++-
 sol/traits.hpp | 8 ++++----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/sol/table.hpp b/sol/table.hpp
index 7cefa37c..3ea92e20 100644
--- a/sol/table.hpp
+++ b/sol/table.hpp
@@ -217,10 +217,15 @@ private:
         proxy(table& t, T& key): t(t), key(key) {}
 
         template<typename U>
-        void operator=(U&& other) {
+        DisableIf<Function<Unqualified<U>>> operator=(U&& other) {
             t.set(key, std::forward<U>(other));
         }
 
+        template<typename U>
+        EnableIf<Function<Unqualified<U>>> operator=(U&& other) {
+            t.set_function(key, std::forward<U>(other));
+        }
+
         template<typename U>
         operator U() const {
             return t.get<U>(key);
diff --git a/sol/traits.hpp b/sol/traits.hpp
index f49b0ca5..63f5b9be 100644
--- a/sol/traits.hpp
+++ b/sol/traits.hpp
@@ -34,11 +34,14 @@ using DisableIf = typename std::enable_if<!T::value, R>::type;
 template<typename T>
 using Unqualified = typename std::remove_reference<typename std::remove_cv<T>::type>::type;
 
+template<typename T>
+using Decay = typename std::decay<T>::type;
+
 namespace detail {
 // code borrowed from Gears 
 // https://github.com/Rapptz/Gears/
 template<typename T, typename = void>
-struct is_function_impl : std::is_function<T> {};
+struct is_function_impl : std::is_function<typename std::remove_pointer<T>::type> {};
 
 template<typename T>
 struct is_function_impl<T, EnableIf<std::is_class<Unqualified<T>>>> {
@@ -64,9 +67,6 @@ using Bool = std::integral_constant<bool, B>;
 
 template<typename T>
 struct Function : Bool<detail::is_function_impl<T>::value> {};
-
-template<typename T>
-using Decay = typename std::decay<T>::type;
 } // sol
 
 #endif // SOL_TRAITS_HPP
\ No newline at end of file