sol2/sol/table_core.hpp

// The MIT License (MIT)

// Copyright (c) 2013-2015 Danny Y., Rapptz

// Permission is hereby granted, free of charge, to any person obtaining a copy of
// this software and associated documentation files (the "Software"), to deal in
// the Software without restriction, including without limitation the rights to
// use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
// the Software, and to permit persons to whom the Software is furnished to do so,
// subject to the following conditions:

// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
// FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
// COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
// IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
// CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

#ifndef SOL_TABLE_CORE_HPP
#define SOL_TABLE_CORE_HPP

#include "proxy.hpp"
#include "stack.hpp"
#include "function_types.hpp"
#include "usertype.hpp"

namespace sol {
namespace detail {
    struct global_overload_tag { } const global_overload;
} // detail

template <bool top_level>
class table_core : public reference {
    friend class state;
    template<typename T, typename Key, EnableIf<Not<std::is_arithmetic<Unqualified<Key>>>, Bool<top_level>> = 0>
    stack::get_return<T> single_get( Key&& key ) const {
        lua_getglobal( lua_state( ), &key[ 0 ] );
        stack::get_return<T> result = stack::pop<T>( lua_state( ) );
        return result;
    }

    template<typename T, typename Key, DisableIf<Not<std::is_arithmetic<Unqualified<Key>>>, Bool<top_level>> = 0>
    stack::get_return<T> single_get( Key&& key ) const {
        push( );
        stack::push( lua_state( ), std::forward<Key>( key ) );
        lua_gettable( lua_state( ), -2 );
        stack::get_return<T> result = stack::pop<T>( lua_state( ) );
        pop( );
        return result;
    }

    template<typename Keys, typename... Ret, std::size_t... I>
    stack::get_return<Ret...> tuple_get( types<Ret...>, indices<I...>, Keys&& keys ) const {
        return stack::get_return<Ret...>( single_get<Ret>( std::get<I>( keys ) )... );
    }

    template<typename Keys, typename Ret, std::size_t I>
    stack::get_return<Ret> tuple_get( types<Ret>, indices<I>, Keys&& keys ) const {
        return single_get<Ret>( std::get<I>( keys ) );
    }

#if SOL_LUA_VERSION < 502
    table_core( detail::global_overload_tag, const table_core<false>& reg ) noexcept : reference( reg.lua_state(), LUA_GLOBALSINDEX ) { }
#else
    table_core( detail::global_overload_tag, const table& reg ) noexcept : reference( reg.get<table>( LUA_RIDX_GLOBALS ) ) { }
#endif
public:
    table_core( ) noexcept : reference( ) { }
    table_core( const table_core<true>& global ) : reference( global ) { }
    table_core( lua_State* L, int index = -1 ) : reference( L, index ) {
        type_assert( L, index, type::table );
    }

    template<typename... Ret, typename... Keys>
    stack::get_return<Ret...> get( Keys&&... keys ) const {
        return tuple_get( types<Ret...>( ), build_indices<sizeof...( Ret )>( ), std::tie( keys... ) );
    }

    template<typename T, typename U>
    table_core& set( T&& key, U&& value ) {
        if ( top_level ) {
            stack::push( lua_state( ), std::forward<U>( value ) );
            lua_setglobal( lua_state( ), &key[0] );
        }
        else {
            push( );
            stack::push( lua_state( ), std::forward<T>( key ) );
            stack::push( lua_state( ), std::forward<U>( value ) );
            lua_settable( lua_state( ), -3 );
            pop( );
        }
        return *this;
    }

    template<typename T>
    SOL_DEPRECATED table_core& set_userdata( usertype<T>& user ) {
        return set_usertype( user );
    }

    template<typename Key, typename T>
    SOL_DEPRECATED table_core& set_userdata( Key&& key, usertype<T>& user ) {
        return set_usertype( std::forward<Key>( key ), user );
    }

    template<typename T>
    table_core& set_usertype( usertype<T>& user ) {
        return set_usertype( usertype_traits<T>::name, user );
    }

    template<typename Key, typename T>
    table_core& set_usertype( Key&& key, usertype<T>& user ) {
        if ( top_level ) {
            stack::push( lua_state( ), user );
            lua_setglobal( lua_state( ), &key[ 0 ] );
            pop( );
        }
        else {
            push( );
            stack::push( lua_state( ), std::forward<Key>( key ) );
            stack::push( lua_state( ), user );
            lua_settable( lua_state( ), -3 );
            pop( );
        }
        return *this;
    }

    template<typename Fx>
    void for_each( Fx&& fx ) const {
        push( );
        stack::push( lua_state( ), nil );
        while ( lua_next( this->lua_state( ), -2 ) ) {
            sol::object key( lua_state( ), -2 );
            sol::object value( lua_state( ), -1 );
            fx( key, value );
            lua_pop( lua_state( ), 1 );
        }
        pop( );
    }

    size_t size( ) const {
        push( );
        size_t result = lua_rawlen( lua_state( ), -1 );
        pop( );
        return result;
    }

    template<typename T>
    proxy<table_core, T> operator[]( T&& key ) {
        return proxy<table_core, T>( *this, std::forward<T>( key ) );
    }

    template<typename T>
    proxy<const table_core, T> operator[]( T&& key ) const {
        return proxy<const table_core, T>( *this, std::forward<T>( key ) );
    }

    void pop( int n = 1 ) const noexcept {
        lua_pop( lua_state( ), n );
    }

    template<typename... Args, typename R, typename Key>
    table_core& set_function( Key&& key, R fun_ptr( Args... ) ) {
        set_resolved_function( std::forward<Key>( key ), fun_ptr );
        return *this;
    }

    template<typename Sig, typename Key>
    table_core& set_function( Key&& key, Sig* fun_ptr ) {
        set_resolved_function( std::forward<Key>( key ), fun_ptr );
        return *this;
    }

    template<typename... Args, typename R, typename C, typename T, typename Key>
    table_core& set_function( Key&& key, R( C::*mem_ptr )( Args... ), T&& obj ) {
        set_resolved_function( std::forward<Key>( key ), mem_ptr, std::forward<T>( obj ) );
        return *this;
    }

    template<typename Sig, typename C, typename T, typename Key>
    table_core& set_function( Key&& key, Sig C::* mem_ptr, T&& obj ) {
        set_resolved_function( std::forward<Key>( key ), mem_ptr, std::forward<T>( obj ) );
        return *this;
    }

    template<typename... Sig, typename Fx, typename Key>
    table_core& set_function( Key&& key, Fx&& fx ) {
        set_fx( types<Sig...>( ), std::forward<Key>( key ), std::forward<Fx>( fx ) );
        return *this;
    }

private:
    template<typename R, typename... Args, typename Fx, typename Key, typename = typename std::result_of<Fx( Args... )>::type>
    void set_fx( types<R( Args... )>, Key&& key, Fx&& fx ) {
        set_resolved_function<R( Args... )>( std::forward<Key>( key ), std::forward<Fx>( fx ) );
    }

    template<typename Fx, typename Key>
    void set_fx( types<>, Key&& key, Fx&& fx ) {
        typedef Unqualified<Unwrap<Fx>> fx_t;
        typedef decltype( &fx_t::operator() ) Sig;
        set_fx( types<function_signature_t<Sig>>( ), std::forward<Key>( key ), std::forward<Fx>( fx ) );
    }

    template<typename... Sig, typename... Args, typename Key, EnableIf<Not<std::is_arithmetic<Unqualified<Key>>>, Bool<top_level>> = 0>
    void set_resolved_function( Key&& key, Args&&... args ) {
        stack::push<function_sig_t<Sig...>>( lua_state( ), std::forward<Args>( args )... );
        lua_setglobal( lua_state( ), &key[ 0 ] );
    }

    template<typename... Sig, typename... Args, typename Key, DisableIf<Not<std::is_arithmetic<Unqualified<Key>>>, Bool<top_level>> = 0>
    void set_resolved_function( Key&& key, Args&&... args ) {
        push( );
        int tabletarget = lua_gettop( lua_state( ) );
        stack::push<function_sig_t<Sig...>>( lua_state( ), std::forward<Args>( args )... );
        lua_setfield( lua_state( ), tabletarget, &key[ 0 ] );
        pop( );
    }
};
} // sol

#endif // SOL_TABLE_CORE_HPP