From 9b2d71f6d88d9defa49d463535060829ba628fbd Mon Sep 17 00:00:00 2001
From: Eduardo Bart <edub4rt@gmail.com>
Date: Mon, 30 May 2011 22:55:34 -0300
Subject: [PATCH] protocol via script

---
 data/modules/mainmenu/entergamewindow.yml |  1 +
 data/modules/mainmenu/menustate.lua       | 37 +++++++++++++++++++++
 data/modules/messagebox/messagebox.lua    | 37 +++++++++++++++++----
 src/framework/net/protocol.h              |  7 +++-
 src/framework/script/luafunctions.cpp     | 39 +++++++++++++++++++++++
 src/framework/script/luafunctions.h       |  4 +++
 src/framework/script/scriptable.cpp       |  6 ++--
 src/framework/script/scriptable.h         |  2 +-
 src/framework/ui/uibutton.cpp             |  2 +-
 src/framework/ui/uielement.cpp            |  2 +-
 src/framework/ui/uitextedit.h             |  2 +-
 src/protocollogin.cpp                     |  7 ++--
 src/protocollogin.h                       |  2 ++
 13 files changed, 132 insertions(+), 16 deletions(-)

diff --git a/data/modules/mainmenu/entergamewindow.yml b/data/modules/mainmenu/entergamewindow.yml
index 228d342b..5a20fcd8 100644
--- a/data/modules/mainmenu/entergamewindow.yml
+++ b/data/modules/mainmenu/entergamewindow.yml
@@ -43,6 +43,7 @@ window#enterGameWindow
     anchors.bottom: parent.bottom
     margin.bottom: 10
     margin.right: 66
+    onClick: enterGame_onOkClicked()
 
   button#cancelButton
     text: Cancel
diff --git a/data/modules/mainmenu/menustate.lua b/data/modules/mainmenu/menustate.lua
index fb8929c1..017ae584 100644
--- a/data/modules/mainmenu/menustate.lua
+++ b/data/modules/mainmenu/menustate.lua
@@ -12,6 +12,43 @@ function onApplicationClose()
     App.exit()
 end
 
+function enterGame_onOkClicked()
+    local enterGameWindow = UI.getRootContainer():child("enterGameWindow")
+    enterGameWindow.visible = false
+
+    local loadMessageBox = messageBox("Please wait", "Connecting..")
+    loadMessageBox.onDestroy = function()
+            --TODO: cancel protocol
+            enterGameWindow.visible = true
+            protocolLogin = nil
+        end
+
+    protocolLogin = ProtocolLogin.new()
+    protocolLogin.onError = function(error)
+            loadMessageBox.onDestroy = nil
+            loadMessageBox:destroy()
+            local msgBox = messageBox("Login Error", error)
+            msgBox.onDestroy = function()
+                    enterGameWindow.visible = true
+                end
+            protocolLogin = nil
+        end
+    protocolLogin.onMotd = function(motd)
+            loadMessageBox.onDestroy = nil
+            loadMessageBox:destroy()
+            local msgBox = messageBox("Message of the day", motd)
+            msgBox.onDestroy = function()
+                    enterGameWindow.visible = true
+                end
+            protocolLogin = nil
+        end
+
+    local account = enterGameWindow:child("accountNameTextEdit").text
+    local password = enterGameWindow:child("passwordTextEdit").text
+
+    protocolLogin:login(account, password)
+end
+
 -- here is where everything starts
 if not initialStateLoaded then
     onEnterMenuState()
diff --git a/data/modules/messagebox/messagebox.lua b/data/modules/messagebox/messagebox.lua
index f4f67e14..01386656 100644
--- a/data/modules/messagebox/messagebox.lua
+++ b/data/modules/messagebox/messagebox.lua
@@ -1,11 +1,34 @@
-function autoDestroyParent()
-    self.parent:destroy()
+MessageBox = {}
+MessageBox.__index = MessageBox
+
+function MessageBox.create(title, text)
+    local msgBox = {}
+    setmetatable(msgBox, MessageBox)
+
+    local window = UI.load("messagebox.yml")
+    window.locked = true
+    window.title = title
+    window:child("textLabel").text = text
+    window:child("okButton").onClick = function()
+            self.parent:destroy()
+        end
+    window.onDestroy = function()
+            if msgBox.onDestroy then
+                msgBox.onDestroy()
+            end
+        end
+
+    msgBox.window = window
+    return msgBox
+end
+
+function MessageBox:destroy()
+    if self.window then
+        self.window:destroy()
+        self.window = nil
+    end
 end
 
 function messageBox(title, text)
-    local msgBox = UI.load("messagebox.yml")
-    msgBox.locked = true
-    msgBox.title = title
-    msgBox:child("textLabel").text = text
-    msgBox:child("okButton").onClick = autoDestroyParent
+    return MessageBox.create(title, text)
 end
diff --git a/src/framework/net/protocol.h b/src/framework/net/protocol.h
index 9801e89c..46155a70 100644
--- a/src/framework/net/protocol.h
+++ b/src/framework/net/protocol.h
@@ -28,6 +28,7 @@
 #include <net/connection.h>
 #include <net/inputmessage.h>
 #include <net/outputmessage.h>
+#include <script/scriptable.h>
 
 #define CIPSOFT_PUBLIC_RSA "1321277432058722840622950990822933849527763264961655079678763618" \
                            "4334395343554449668205332383339435179772895415509701210392836078" \
@@ -37,7 +38,7 @@
 
 //#define RSA "109120132967399429278860960508995541528237502902798129123468757937266291492576446330739696001110603907230888610072655818825358503429057592827629436413108566029093628212635953836686562675849720620786279431090218017681061521755056710823876476444260558147179707119674283982419152118103759076030616683978566631413"
 
-class Protocol
+class Protocol : public Scriptable
 {
 public:
     Protocol();
@@ -48,6 +49,8 @@ public:
     virtual void onRecv(InputMessage *inputMessage);
     virtual void onError(const boost::system::error_code& error);
 
+    virtual const char *getScriptableName() const { return "Protocol"; }
+
 protected:
     uint32 m_xteaKey[4];
     bool m_xteaEncryptionEnabled;
@@ -60,4 +63,6 @@ private:
     ConnectionPtr m_connection;
 };
 
+typedef boost::shared_ptr<Protocol> ProtocolPtr;
+
 #endif
diff --git a/src/framework/script/luafunctions.cpp b/src/framework/script/luafunctions.cpp
index f666d25c..672de75a 100644
--- a/src/framework/script/luafunctions.cpp
+++ b/src/framework/script/luafunctions.cpp
@@ -29,6 +29,7 @@
 #include <core/resources.h>
 #include <ui/ui.h>
 #include <core/dispatcher.h>
+#include "../../protocollogin.h"
 
 void registerLuaFunctions()
 {
@@ -65,10 +66,27 @@ void registerLuaFunctions()
 
     // UITextEdit
     g_lua.registerClass("UITextEdit", "UIElement");
+    g_lua.registerMemberField("text", &lua_UITextEdit_getText, &lua_UITextEdit_setText);
 
     // UIWindow
     g_lua.registerClass("UIWindow", "UIContainer");
     g_lua.registerMemberField("title", &lua_UIWindow_getTitle, &lua_UIWindow_setTitle);
+
+    // Protocol
+    g_lua.registerClass("ProtocolLogin");
+    g_lua.registerMemberFunction("new", []{
+        ProtocolLoginPtr protocolLogin(new ProtocolLogin);
+        g_lua.pushClassInstance(protocolLogin);
+        return 1;
+    });
+    g_lua.registerMemberFunction("login", []{
+        std::string accountPassword = g_lua.popString();
+        std::string accountName = g_lua.popString();
+        if(ProtocolLoginPtr protocolLogin = boost::dynamic_pointer_cast<ProtocolLogin>(g_lua.popClassInstance()))
+            protocolLogin->login(accountName, accountPassword);
+        return 0;
+    });
+
 }
 
 
@@ -274,6 +292,27 @@ int lua_UILabel_getText()
     return 1;
 }
 
+
+////////////////////////////////////////////////////////////////////////////////
+// UILabel
+
+int lua_UITextEdit_setText()
+{
+    std::string text = g_lua.popString();
+    if(UITextEditPtr textEdit = boost::dynamic_pointer_cast<UITextEdit>(g_lua.popClassInstance()))
+        textEdit->setText(text);
+    return 1;
+}
+
+int lua_UITextEdit_getText()
+{
+    if(UITextEditPtr textEdit = boost::dynamic_pointer_cast<UITextEdit>(g_lua.popClassInstance()))
+        g_lua.pushString(textEdit->getText());
+    else
+        g_lua.pushNil();
+    return 1;
+}
+
 ////////////////////////////////////////////////////////////////////////////////
 // UIWindow
 
diff --git a/src/framework/script/luafunctions.h b/src/framework/script/luafunctions.h
index da6aec0d..2e026278 100644
--- a/src/framework/script/luafunctions.h
+++ b/src/framework/script/luafunctions.h
@@ -61,6 +61,10 @@ int lua_UIContainer_getChildren();
 int lua_UILabel_setText();
 int lua_UILabel_getText();
 
+// UITextEdit
+int lua_UITextEdit_setText();
+int lua_UITextEdit_getText();
+
 // UIButton
 void lua_UIButton_onClick();
 
diff --git a/src/framework/script/scriptable.cpp b/src/framework/script/scriptable.cpp
index 75575ae6..26389d91 100644
--- a/src/framework/script/scriptable.cpp
+++ b/src/framework/script/scriptable.cpp
@@ -44,7 +44,7 @@ void Scriptable::releaseLuaTableRef()
     }
 }
 
-void Scriptable::callLuaTableField(const std::string& field)
+void Scriptable::callLuaTableField(const std::string& field, int numArgs)
 {
     // set self
     g_lua.pushClassInstance(shared_from_this());
@@ -55,9 +55,11 @@ void Scriptable::callLuaTableField(const std::string& field)
 
     // call it if its a function
     if(g_lua.isFunction()) {
-        g_lua.callFunction();
+        g_lua.insert(-numArgs-1);
+        g_lua.callFunction(numArgs);
     // if its an array call each element
     } else if(g_lua.isTable()) {
+        //TODO: call here with arguments
         g_lua.pushNil();
         while(g_lua.next()) {
             // call it if its a function
diff --git a/src/framework/script/scriptable.h b/src/framework/script/scriptable.h
index fd1eada5..bbad58c3 100644
--- a/src/framework/script/scriptable.h
+++ b/src/framework/script/scriptable.h
@@ -36,7 +36,7 @@ public:
 
     int getLuaTableRef();
     void releaseLuaTableRef();
-    void callLuaTableField(const std::string& field);
+    void callLuaTableField(const std::string& field, int numArgs = 0);
 
 private:
     int m_luaTableRef;
diff --git a/src/framework/ui/uibutton.cpp b/src/framework/ui/uibutton.cpp
index 12ebfa59..61bcaac4 100644
--- a/src/framework/ui/uibutton.cpp
+++ b/src/framework/ui/uibutton.cpp
@@ -34,7 +34,7 @@ void UIButton::onInputEvent(const InputEvent& event)
     } else if(event.type == EV_MOUSE_LUP && m_state == ButtonDown) {
         m_state = ButtonUp;
         if(getRect().contains(event.mousePos)) {
-            g_dispatcher.addTask(boost::bind(&Scriptable::callLuaTableField, shared_from_this(), "onClick"));
+            g_dispatcher.addTask(boost::bind(&Scriptable::callLuaTableField, shared_from_this(), "onClick", 0));
         }
     } else if(event.type == EV_MOUSE_MOVE && m_state != ButtonDown) {
         if(isMouseOver())
diff --git a/src/framework/ui/uielement.cpp b/src/framework/ui/uielement.cpp
index 8f09a4e9..262ed784 100644
--- a/src/framework/ui/uielement.cpp
+++ b/src/framework/ui/uielement.cpp
@@ -115,7 +115,7 @@ void UIElement::setSkin(const UIElementSkinPtr& skin)
 
 void UIElement::onLoad()
 {
-    g_dispatcher.addTask(boost::bind(&Scriptable::callLuaTableField, shared_from_this(), "onLoad"));
+    g_dispatcher.addTask(boost::bind(&Scriptable::callLuaTableField, shared_from_this(), "onLoad", 0));
 }
 
 void UIElement::render()
diff --git a/src/framework/ui/uitextedit.h b/src/framework/ui/uitextedit.h
index 959fa703..8b70e619 100644
--- a/src/framework/ui/uitextedit.h
+++ b/src/framework/ui/uitextedit.h
@@ -40,7 +40,7 @@ public:
     void onRectUpdate();
     void onFocusChange();
 
-    void setText(const std::string& text);
+    void setText(const std::string& text) { m_textArea.setText(text); }
     std::string getText() const { return m_textArea.getText(); }
     TextArea& getTextArea() { return m_textArea; }
 
diff --git a/src/protocollogin.cpp b/src/protocollogin.cpp
index 503dee67..72749ade 100644
--- a/src/protocollogin.cpp
+++ b/src/protocollogin.cpp
@@ -24,6 +24,7 @@
 #include "protocollogin.h"
 #include <net/outputmessage.h>
 #include <util/rsa.h>
+#include <script/luascript.h>
 
 ProtocolLogin::ProtocolLogin()
 {
@@ -131,13 +132,15 @@ void ProtocolLogin::onRecv(InputMessage *inputMessage)
 void ProtocolLogin::parseError(InputMessage *inputMessage)
 {
     std::string error = inputMessage->getString();
-    logError(error.c_str());
+    g_lua.pushString(error);
+    callLuaTableField("onError", 1);
 }
 
 void ProtocolLogin::parseMOTD(InputMessage *inputMessage)
 {
     std::string motd = inputMessage->getString();
-    logError(motd.c_str());
+    g_lua.pushString(motd);
+    callLuaTableField("onMotd", 1);
 }
 
 void ProtocolLogin::parseCharacterList(InputMessage *inputMessage)
diff --git a/src/protocollogin.h b/src/protocollogin.h
index 596fffd2..0a7b6342 100644
--- a/src/protocollogin.h
+++ b/src/protocollogin.h
@@ -39,6 +39,8 @@ public:
     void sendPacket();
     void onRecv(InputMessage *inputMessage);
 
+    const char *getScriptableName() const { return "ProtocolLogin"; }
+
 private:
     void parseError(InputMessage *inputMessage);
     void parseMOTD(InputMessage *inputMessage);