From f05c048f6dff9565e4cf2685b3c3b5451d43b406 Mon Sep 17 00:00:00 2001
From: Eduardo Bart <edub4rt@gmail.com>
Date: Mon, 29 Aug 2011 16:35:58 -0300
Subject: [PATCH] straightforward signal and slots system for lua events

---
 modules/console/console.lua              |  2 +-
 modules/core/util.lua                    | 15 ++++++++++++++-
 modules/game/game.lua                    |  3 +++
 src/framework/luascript/luainterface.cpp | 23 +++++++++++++++++------
 src/otclient/core/game.cpp               | 12 +++++++-----
 5 files changed, 42 insertions(+), 13 deletions(-)

diff --git a/modules/console/console.lua b/modules/console/console.lua
index dc88cba4..73eb0d0c 100644
--- a/modules/console/console.lua
+++ b/modules/console/console.lua
@@ -67,7 +67,7 @@ end
 function Console.init()
   consoleWidget = UI.loadAndDisplay("/console/console.otui")
   consoleWidget:hide()
-  consoleWidget.onKeyPress = onKeyPress
+  connect(consoleWidget, { onKeyPress = onKeyPress })
 
   commandLineEdit = consoleWidget:getChildById('commandLineEdit')
   consoleBuffer = consoleWidget:getChildById('consoleBuffer')
diff --git a/modules/core/util.lua b/modules/core/util.lua
index 58d75fb2..bc60c00b 100644
--- a/modules/core/util.lua
+++ b/modules/core/util.lua
@@ -10,4 +10,17 @@ function createEnvironment()
   local env = { }
   setmetatable(env, { __index = _G} )
   return env
-end
\ No newline at end of file
+end
+
+function connect(object, signalsAndSlots)
+  for signal,slot in pairs(signalsAndSlots) do
+    if not object[signal] then
+        object[signal] = slot
+    elseif type(object[signal]) == 'function' then
+        object[signal] = { object[signal], slot }
+    elseif type(signal) == 'table' then
+        table.insert(object[signal], slot)
+    else
+    end
+  end
+end
diff --git a/modules/game/game.lua b/modules/game/game.lua
index 5b2219e9..71759ee5 100644
--- a/modules/game/game.lua
+++ b/modules/game/game.lua
@@ -7,6 +7,9 @@ local function onGameKeyPress(self, keyCode, keyChar, keyboardModifiers)
     if keyCode == KeyG then
       CharacterList.show()
       return true
+    elseif keyCode == KeyQ then
+      Game.logout()
+      return true
     end
   end
   return false
diff --git a/src/framework/luascript/luainterface.cpp b/src/framework/luascript/luainterface.cpp
index 00c4eba8..a5737756 100644
--- a/src/framework/luascript/luainterface.cpp
+++ b/src/framework/luascript/luainterface.cpp
@@ -419,21 +419,32 @@ int LuaInterface::protectedCall(int numArgs, int requestedResults)
         else if(isTable(funcIndex)) {
             // loop through table values
             pushNil();
+            bool done = false;
             while(next(funcIndex-1)) {
                 if(isFunction()) {
                     // repush arguments
                     for(int i=0;i<numArgs;++i)
-                        pushValue(-(numArgs-i)-2);
+                        pushValue(-numArgs-2);
 
                     int rets = safeCall(numArgs);
-                    pop(rets);
-                }
-                // just ignore
-                else {
-                    pop();
+                    if(rets == 1) {
+                        done = popBoolean();
+                        if(done) {
+                            pop();
+                            break;
+                        }
+                    } else if(rets != 0)
+                        throw LuaException("function call didn't return the expected number of results", 0);
+                } else {
+                    throw LuaException("attempt to call a non function", 0);
                 }
             }
             pop(numArgs + 1); // pops the table of function and arguments
+
+            if(requestedResults == -1 || requestedResults == 1) {
+                numRets = 1;
+                pushBoolean(done);
+            }
         }
         // nil values are ignored
         else if(isNil(funcIndex)) {
diff --git a/src/otclient/core/game.cpp b/src/otclient/core/game.cpp
index 1fd1f4b8..b71a659a 100644
--- a/src/otclient/core/game.cpp
+++ b/src/otclient/core/game.cpp
@@ -62,13 +62,15 @@ void Game::processConnectionError(const boost::system::error_code& error)
 {
     // connection errors only have meaning if we still have a protocol
     if(m_protocolGame) {
-        g_lua.callGlobalField("Game", "onConnectionError", error.message());
+        if(error != asio::error::eof)
+            g_lua.callGlobalField("Game", "onConnectionError", error.message());
 
-        if(m_online)
+        if(m_online) {
             processLogout();
-
-        // disconnect isn't needed, we are already disconnected
-        m_protocolGame.reset();
+        } else {
+            m_protocolGame->disconnect();
+            m_protocolGame.reset();
+        }
     }
 }