include/services/lua/module.h

   1 #ifndef AWFUL_LUA_MODULE_H
   2 #define AWFUL_LUA_MODULE_H
   3
   4 namespace awful { namespace LuaImpl
   5 {
   6         struct ModuleVisitor
   7         {
   8                 ModuleVisitor( lua_State* pLS_ ) :
   9                         pLS( pLS_ ),
  10                         bAtGlobalScope( true )
  11                 {
  12                 }
  13
  14                 // Called for each nested namespace until the namespace we're interested in.
  15                 // Create all intermediate namespace tables as needed.
  16                 // Also build the module name in the form module.submodule.subsubmodule
  17                 // to be able to set the package.loaded properly.
  18                 void QualifiedNameComponent( const char* pName_ )
  19                 {
  20                         if( bAtGlobalScope )
  21                         {
  22                                 FullModuleName = pName_;
  23                                 lua_getglobal( pLS, pName_ );
  24                         }
  25                         else
  26                         {
  27                                 FullModuleName += '.';
  28                                 FullModuleName += pName_;
  29                                 lua_getfield( pLS, -1, pName_ );
  30                         }
  31
  32                         if( lua_isnil( pLS, -1 ) )
  33                         {
  34                                 lua_pop( pLS, 1 );
  35                                 lua_newtable( pLS );
  36                                 lua_pushvalue( pLS, -1 );
  37
  38                                 if( bAtGlobalScope )
  39                                 {
  40                                         lua_setglobal( pLS, pName_ );
  41                                 }
  42                                 else
  43                                 {
  44                                         lua_setfield( pLS, -3, pName_ );
  45                                         lua_remove( pLS, -3 );
  46                                 }
  47                         }
  48
  49                         bAtGlobalScope = false;
  50                 }
  51
  52                 template< class C > void Class()
  53                 {
  54                         ClassStuff< C >::Setup();
  55                 }
  56
  57                 template< class C > void Struct()
  58                 {
  59                         StructStuff< C >::Setup();
  60                 }
  61
  62                 lua_State* pLS;
  63                 bool bAtGlobalScope;
  64                 std::string FullModuleName;
  65         };
  66 }}
  67
  68 #endif