From 50b99a75b9c13b815667d8d0af9a5650de739248 Mon Sep 17 00:00:00 2001
From: Eduardo Bart <edub4rt@gmail.com>
Date: Thu, 21 Apr 2011 19:44:30 -0300
Subject: [PATCH] lua scripting

---
 CMakeLists.txt                                |   8 +-
 data/lib/init.lua                             |   0
 data/menustate.lua                            |  22 --
 .../mainmenu/entergamewindow.yml}             |   3 +
 .../mainmenu/infowindow.yml}                  |   0
 data/modules/mainmenu/init.lua                |   1 -
 data/modules/mainmenu/mainmenu.lua            |  24 +-
 data/modules/mainmenu/mainmenu.yml            |   6 +-
 .../mainmenu/optionswindow.yml}               |   5 +-
 data/ui/mainMenu.yml                          |  37 --
 src/framework/prerequisites.h                 |   1 +
 src/framework/script/luafunctions.cpp         | 112 ++++++
 src/framework/script/luascript.cpp            | 344 ++++++++++++++++++
 src/framework/script/luascript.h              |  95 +++++
 src/framework/script/scriptable.h             |  39 ++
 src/framework/ui/uibutton.h                   |   2 +
 src/framework/ui/uicheckbox.h                 |   5 +-
 src/framework/ui/uicontainer.h                |   2 +
 src/framework/ui/uielement.h                  |   1 +
 src/framework/ui/uilabel.h                    |   2 +
 src/framework/ui/uilayout.h                   |   7 +-
 src/framework/ui/uiloader.cpp                 |  22 +-
 src/framework/ui/uiloader.h                   |   4 +
 src/framework/ui/uitextedit.h                 |   2 +
 src/framework/ui/uiwindow.h                   |   2 +
 src/main.cpp                                  |   4 +-
 src/menustate.cpp                             |  69 +---
 27 files changed, 674 insertions(+), 145 deletions(-)
 delete mode 100644 data/lib/init.lua
 delete mode 100644 data/menustate.lua
 rename data/{ui/enterGameWindow.yml => modules/mainmenu/entergamewindow.yml} (93%)
 rename data/{ui/infoWindow.yml => modules/mainmenu/infowindow.yml} (100%)
 delete mode 100644 data/modules/mainmenu/init.lua
 rename data/{ui/optionsWindow.yml => modules/mainmenu/optionswindow.yml} (96%)
 delete mode 100644 data/ui/mainMenu.yml
 create mode 100644 src/framework/script/luafunctions.cpp
 create mode 100644 src/framework/script/luascript.cpp
 create mode 100644 src/framework/script/luascript.h
 create mode 100644 src/framework/script/scriptable.h

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 7a8bb5a5..1a177991 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -6,7 +6,7 @@ SET(CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/cmake;${CMAKE_MODULE_PATH}")
 # find needed packages
 SET(Boost_USE_STATIC_LIBS ON)
 SET(Boost_USE_MULTITHREADED ON)
-FIND_PACKAGE(Boost COMPONENTS system regex REQUIRED)
+FIND_PACKAGE(Boost COMPONENTS system regex signals REQUIRED)
 FIND_PACKAGE(OpenGL REQUIRED)
 FIND_PACKAGE(Lua51 REQUIRED)
 FIND_PACKAGE(YamlCpp REQUIRED)
@@ -52,7 +52,7 @@ IF(CMAKE_BUILD_TYPE STREQUAL "Debug")
     ADD_DEFINITIONS(-D_DEBUG)
 ENDIF(CMAKE_BUILD_TYPE STREQUAL "Debug")
 
-SET(SOURCES src/framework/script/script.cpp
+SET(SOURCES
 # game sources
     src/main.cpp
     src/menustate.cpp
@@ -68,8 +68,8 @@ SET(SOURCES src/framework/script/script.cpp
     src/framework/core/engine.cpp
 
 # framework script
-    src/framework/script/script.cpp
-    src/framework/script/scriptfunctions.cpp
+    src/framework/script/luascript.cpp
+    src/framework/script/luafunctions.cpp
 
 # framework utilities
     src/framework/util/color.cpp
diff --git a/data/lib/init.lua b/data/lib/init.lua
deleted file mode 100644
index e69de29b..00000000
diff --git a/data/menustate.lua b/data/menustate.lua
deleted file mode 100644
index eb26f26a..00000000
--- a/data/menustate.lua
+++ /dev/null
@@ -1,22 +0,0 @@
-MainMenuController = {}
-enterGameController = {}
-
-function enterGameController.enterGameButton_clicked()
-    createEnterGameWindow()
-end
-
-function mainMenuController.createMainMenu()
-    mainMenuPanel = loadUI("ui/mainMenuPanel.yml")
-    enterGameButton = mainMenuPanel:getElementById("enterGameButton")
-    enterGameButton.setOnClick(enterGameController.enterGameButton_clicked)
-    exitButton = mainMenuPanel:getElementById("exitButton")
-    exitButton:setOnClick(function() exitGame() end)
-end
-
-function onEnter()
-    mainMenuController:createMainMenu()
-end
-
-function onLeave()
-
-end
diff --git a/data/ui/enterGameWindow.yml b/data/modules/mainmenu/entergamewindow.yml
similarity index 93%
rename from data/ui/enterGameWindow.yml
rename to data/modules/mainmenu/entergamewindow.yml
index 8871ec21..0a38b1c8 100644
--- a/data/ui/enterGameWindow.yml
+++ b/data/modules/mainmenu/entergamewindow.yml
@@ -49,6 +49,9 @@ window#enterGameWindow:
     anchors.bottom: parent.bottom
     margin.bottom: 10
     margin.right: 13
+    onClick: |
+      enterGameWindow:destroy()
+      enterGameWindow = nil
     
   textEdit#accountNameTextEdit:
     anchors.right: parent.right
diff --git a/data/ui/infoWindow.yml b/data/modules/mainmenu/infowindow.yml
similarity index 100%
rename from data/ui/infoWindow.yml
rename to data/modules/mainmenu/infowindow.yml
diff --git a/data/modules/mainmenu/init.lua b/data/modules/mainmenu/init.lua
deleted file mode 100644
index 296b7928..00000000
--- a/data/modules/mainmenu/init.lua
+++ /dev/null
@@ -1 +0,0 @@
-require("modules/mainmenu/mainmenu")
\ No newline at end of file
diff --git a/data/modules/mainmenu/mainmenu.lua b/data/modules/mainmenu/mainmenu.lua
index 22150b0a..32529fbf 100644
--- a/data/modules/mainmenu/mainmenu.lua
+++ b/data/modules/mainmenu/mainmenu.lua
@@ -1 +1,23 @@
-loadUI("modules/mainmenu/mainmenu.yml")
\ No newline at end of file
+-- events
+function MainMenu_enterGameClicked()
+    enterGameWindow = loadUI("modules/mainmenu/entergamewindow.yml")
+end
+
+function MainMenu_optionsClicked()
+    optionsWindow = loadUI("modules/mainmenu/optionswindow.yml")
+end
+
+function MainMenu_infoClicked()
+    infoWindow = loadUI("modules/mainmenu/infowindow.yml")
+end
+
+function MainMenu_exitClicked()
+    exitGame()
+end
+
+-- create main menu
+function MainMenu_create()
+    menuPanel = loadUI("modules/mainmenu/mainmenu.yml")
+end
+
+MainMenu_create()
\ No newline at end of file
diff --git a/data/modules/mainmenu/mainmenu.yml b/data/modules/mainmenu/mainmenu.yml
index 1438a708..58b99944 100644
--- a/data/modules/mainmenu/mainmenu.yml
+++ b/data/modules/mainmenu/mainmenu.yml
@@ -11,6 +11,7 @@ panel#mainMenu:
     anchors.top: parent.top
     anchors.horizontalCenter: parent.horizontalCenter
     margin.top: 16
+    onClick: MainMenu_enterGameClicked()
 
   button#accessAccountButton:
     text: Access Account
@@ -23,15 +24,18 @@ panel#mainMenu:
     anchors.top: parent.top
     anchors.horizontalCenter: parent.horizontalCenter
     margin.top: 76
+    onClick: MainMenu_optionsClicked()
 
   button#infoButton:
     text: Info
     anchors.top: parent.top
     anchors.horizontalCenter: parent.horizontalCenter
     margin.top: 106
+    onClick: MainMenu_infoClicked()
 
   button#exitGameButton:
     text: Exit
     anchors.top: parent.top
     anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 136
\ No newline at end of file
+    margin.top: 136
+    onClick: MainMenu_exitClicked()
\ No newline at end of file
diff --git a/data/ui/optionsWindow.yml b/data/modules/mainmenu/optionswindow.yml
similarity index 96%
rename from data/ui/optionsWindow.yml
rename to data/modules/mainmenu/optionswindow.yml
index f580e888..61c4f18c 100644
--- a/data/ui/optionsWindow.yml
+++ b/data/modules/mainmenu/optionswindow.yml
@@ -108,4 +108,7 @@ window#optionsWindow:
     anchors.right: parent.right
     anchors.bottom: parent.bottom
     margin.right: 10
-    margin.bottom: 13
\ No newline at end of file
+    margin.bottom: 13
+    onClick: |
+      optionsWindow:destroy()
+      optionsWindow = nil
\ No newline at end of file
diff --git a/data/ui/mainMenu.yml b/data/ui/mainMenu.yml
deleted file mode 100644
index 1438a708..00000000
--- a/data/ui/mainMenu.yml
+++ /dev/null
@@ -1,37 +0,0 @@
-panel#mainMenu:
-  skin: roundedGridPanel
-  size: [117, 171]
-  anchors.left: parent.left
-  anchors.bottom: parent.bottom
-  margin.left: 60
-  margin.bottom: 70
-  
-  button#enterGameButton:
-    text: Enter Game
-    anchors.top: parent.top
-    anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 16
-
-  button#accessAccountButton:
-    text: Access Account
-    anchors.top: parent.top
-    anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 46
-
-  button#optionsButton:
-    text: Options
-    anchors.top: parent.top
-    anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 76
-
-  button#infoButton:
-    text: Info
-    anchors.top: parent.top
-    anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 106
-
-  button#exitGameButton:
-    text: Exit
-    anchors.top: parent.top
-    anchors.horizontalCenter: parent.horizontalCenter
-    margin.top: 136
\ No newline at end of file
diff --git a/src/framework/prerequisites.h b/src/framework/prerequisites.h
index 0714b2d9..af132389 100644
--- a/src/framework/prerequisites.h
+++ b/src/framework/prerequisites.h
@@ -69,6 +69,7 @@ typedef int8_t int8;
 #include <boost/shared_ptr.hpp>
 #include <boost/scoped_ptr.hpp>
 #include <boost/enable_shared_from_this.hpp>
+#include <boost/signals.hpp>
 #define foreach BOOST_FOREACH
 
 typedef boost::function<void()> Callback;
diff --git a/src/framework/script/luafunctions.cpp b/src/framework/script/luafunctions.cpp
new file mode 100644
index 00000000..165c5fdf
--- /dev/null
+++ b/src/framework/script/luafunctions.cpp
@@ -0,0 +1,112 @@
+/* The MIT License
+ *
+ * Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
+ *
+ * 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.
+ */
+
+
+#include <prerequisites.h>
+#include <script/luascript.h>
+#include <core/engine.h>
+#include <core/resources.h>
+#include <ui/ui.h>
+
+void LuaScript::registerFunctions()
+{
+    registerGlobalFunction("exitGame", &LuaScript::lua_exitGame);
+    registerGlobalFunction("loadUI", &LuaScript::lua_loadUI);
+    registerGlobalFunction("getUIRootContainer", &LuaScript::lua_getUIRootContainer);
+
+    registerClass("UILayout");
+    registerClass("UIElement",   "UILayout");
+    registerClass("UIButton",    "UIElement");
+    registerClass("UITextEdit",  "UIElement");
+    registerClass("UICheckBox",  "UIElement");
+    registerClass("UIContainer", "UIElement");
+    registerClass("UIWindow",    "UIContainer");
+
+    registerMemberFunction("UIElement", "destroy", &LuaScript::lua_UIElement_destroy);
+    registerMemberFunction("UIContainer", "getChildByID", &LuaScript::lua_UIContainer_getChildByID);
+    registerMemberFunction("UIButton", "setOnClick", &LuaScript::lua_UIButton_setOnClick);
+}
+
+int LuaScript::lua_exitGame()
+{
+    g_engine.stop();
+    return 1;
+}
+
+int LuaScript::lua_loadUI()
+{
+    UIContainerPtr parent;
+    if(getStackSize() > 1) {
+        parent = boost::static_pointer_cast<UIContainer>(popClassInstance());
+    } else {
+        parent = UIContainer::getRootContainer();
+    }
+    std::string uiFile = popString();
+    UIElementPtr element = UILoader::loadFile(uiFile.c_str(), parent);
+    pushClassInstance(element);
+    return 1;
+}
+
+int LuaScript::lua_getUIRootContainer()
+{
+    UIContainerPtr rootContainer = UIContainer::getRootContainer();
+    pushClassInstance(rootContainer);
+    return 1;
+}
+
+int LuaScript::lua_UIElement_destroy()
+{
+    UIElementPtr element = boost::static_pointer_cast<UIElement>(popClassInstance());
+    element->destroy();
+    return 1;
+}
+
+int LuaScript::lua_UIButton_setOnClick()
+{
+    lua_insert(L, -2);
+    UIButtonPtr button = boost::static_pointer_cast<UIButton>(popClassInstance());
+    if(button) {
+        int funcRef = popFunction();
+        button->setOnClick([this, funcRef] {
+            pushFunction(funcRef);
+            callFunction();
+        });
+    } else {
+        pop();
+    }
+    return 1;
+}
+
+int LuaScript::lua_UIContainer_getChildByID()
+{
+    std::string id = popString();
+    ScriptablePtr object = popClassInstance();
+    if(object && strcmp("UIContainer", object->getScriptableName()) == 0) {
+        UIContainerPtr container = boost::static_pointer_cast<UIContainer>(object);
+        pushClassInstance(container->getChildById(id));
+    } else {
+        pushNil();
+    }
+
+    return 1;
+}
\ No newline at end of file
diff --git a/src/framework/script/luascript.cpp b/src/framework/script/luascript.cpp
new file mode 100644
index 00000000..397596cd
--- /dev/null
+++ b/src/framework/script/luascript.cpp
@@ -0,0 +1,344 @@
+/* The MIT License
+ *
+ * Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
+ *
+ * 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.
+ */
+
+
+#include <prerequisites.h>
+#include <script/luascript.h>
+#include <core/resources.h>
+
+LuaScript g_lua;
+
+LuaScript::LuaScript()
+{
+    L = luaL_newstate();
+    if(!L)
+        logFatal("could not create lua context");
+
+    // load lua standard libraries
+    luaL_openlibs(L);
+
+    // setup custom package loader
+    setupPackageLoader();
+
+    // register classes and functions
+    registerFunctions();
+}
+
+LuaScript::~LuaScript()
+{
+    lua_close(L);
+}
+
+bool LuaScript::loadFile(const std::string& fileName)
+{
+    if(!g_resources.fileExists(fileName)) {
+        logError("lua script file '%s' doesn't exist", fileName.c_str());
+        return false;
+    }
+    std::string fileContents = g_resources.loadTextFile(fileName);
+    return loadBuffer(fileContents, fileName);
+}
+
+bool LuaScript::loadBuffer(const std::string& text, const std::string& what)
+{
+    // load buffer
+    int ret = luaL_loadbuffer(L, text.c_str(), text.length(), what.c_str());
+    if(ret != 0){
+        logError((std::string("while parsing lua code: ") + popString()).c_str());
+        return false;
+    }
+
+    // check if that is loaded as a function
+    if(lua_isfunction(L, -1) == 0) {
+        logError("lua code not loaded as function");
+        return false;
+    }
+
+    // execute it
+    ret = lua_pcall(L, 0, 0, 0);
+    if(ret != 0){
+        logError((std::string("while loading lua code: ") + popString()).c_str());
+        return false;
+    }
+
+    return true;
+}
+
+int LuaScript::loadBufferAsFunction(const std::string& text, const std::string& what)
+{
+    int ret = luaL_loadstring(L, text.c_str());
+    if(ret != 0){
+        logError((std::string("while parsing lua code: ") + popString()).c_str());
+        return LUA_REFNIL;
+    }
+
+    // check if that is loaded as a function
+    if(lua_isfunction(L, -1) == 0) {
+        logError("lua code not loaded as function");
+        return LUA_REFNIL;
+    }
+
+    return popFunction();
+}
+
+int LuaScript::getStackSize()
+{
+    return lua_gettop(L);
+}
+
+void LuaScript::pop(int n)
+{
+    lua_pop(L, n);
+}
+
+bool LuaScript::popBoolean()
+{
+    bool b = (lua_toboolean(L, -1) != 0);
+    pop();
+    return b;
+}
+
+int32_t LuaScript::popInteger()
+{
+    double d;
+    if(lua_istable(L, -1)) {
+        lua_getfield(L, -1, "__intValue");
+        d = lua_tonumber(L, -1);
+        pop(2);
+    }
+    else{
+        d = lua_tonumber(L, -1);
+        pop(1);
+    }
+    return (int)d;
+}
+
+std::string LuaScript::popString()
+{
+    size_t len;
+    const char* cstr = lua_tolstring(L, -1, &len);
+    std::string str(cstr, len);
+    pop();
+    return str;
+}
+
+void LuaScript::pushNil()
+{
+    lua_pushnil(L);
+}
+
+void LuaScript::pushBoolean(bool b)
+{
+    lua_pushboolean(L, b);
+}
+
+void LuaScript::pushInteger(int32_t i)
+{
+    lua_pushnumber(L, i);
+}
+
+void LuaScript::pushString(const std::string& str)
+{
+    lua_pushstring(L, str.c_str());
+}
+
+void LuaScript::pushUserdata(void* ptr)
+{
+    lua_pushlightuserdata(L, ptr);
+}
+
+void LuaScript::pushClassInstance(const ScriptablePtr& object)
+{
+    if(object->getScriptableName()) {
+        new(lua_newuserdata(L, sizeof(ScriptableWeakPtr))) ScriptableWeakPtr(object);
+        lua_getfield(L, LUA_REGISTRYINDEX, (std::string(object->getScriptableName()) + "_mt").c_str());
+        lua_setmetatable(L, -2);
+    }
+}
+
+ScriptablePtr LuaScript::popClassInstance()
+{
+    if(!lua_isuserdata(L, -1)) {
+        logError("Couldn't pop a class instance, top objet is not a valid type (%s)", luaL_typename(L, -1));
+        lua_pop(L, 1);
+        return ScriptablePtr();
+    }
+
+    ScriptableWeakPtr *objectRef = (ScriptableWeakPtr *)lua_touserdata(L, -1);
+    lua_pop(L, 1);
+    return objectRef->lock();
+}
+
+int LuaScript::popFunction()
+{
+    return luaL_ref(L, LUA_REGISTRYINDEX);
+}
+
+void LuaScript::pushFunction(int functionRef)
+{
+    lua_rawgeti(L, LUA_REGISTRYINDEX, functionRef);
+}
+
+void LuaScript::callFunction(int numArgs, bool itReturnsValue)
+{
+    lua_call(L, numArgs, itReturnsValue ? 1 : 0);
+}
+
+void LuaScript::setupPackageLoader()
+{
+    lua_getfield(L, LUA_GLOBALSINDEX, "package");
+    lua_getfield(L, -1, "loaders");
+    lua_remove(L, -2);
+    lua_pushnil(L);
+    int numLoaders = 0;
+    while(lua_next(L, -2) != 0) {
+        lua_pop(L, 1);
+        numLoaders++;
+    }
+    lua_pushinteger(L, numLoaders + 1);
+    lua_pushcfunction(L, &LuaScript::luaPackageLoader);
+    lua_rawset(L, -3);
+    lua_pop(L, 1);
+}
+
+void LuaScript::registerClass(const std::string& klass, const std::string& baseClass)
+{
+    // klass_mt = {}
+    lua_newtable(L); // klass metatable
+    lua_pushvalue(L, -1); // another reference to the metatable
+    lua_setfield(L, LUA_REGISTRYINDEX, (klass + "_mt").c_str()); // register the metatable at registry index
+
+    // set __gc metamethod, which collects userdata
+    lua_pushcfunction(L, &LuaScript::luaCollectClassInstance);
+    lua_setfield(L, -2, "__gc");
+
+    // set __eq metamethod
+    lua_pushcfunction(L, &LuaScript::luaCompareClassInstances);
+    lua_setfield(L, -2, "__eq");
+
+    // klass = {}
+    lua_newtable(L); // klass table
+    lua_pushvalue(L, -1); // another reference to the table
+    lua_setfield(L, LUA_GLOBALSINDEX, klass.c_str()); // register at globals index
+
+    // klass.className = "klass"
+    lua_pushstring(L, klass.c_str());
+    lua_setfield(L, -2, "className");
+
+    // klass_mt.__index = klass
+    lua_setfield(L, -2, "__index");
+
+    // pop the class metatable
+    lua_pop(L, 1);
+
+    if(baseClass.length() > 0) {
+        // add klass table to the top of the stack
+        lua_getfield(L, LUA_GLOBALSINDEX, klass.c_str());
+
+        // redirect = { __index = baseClass }
+        lua_newtable(L);
+        lua_getfield(L, LUA_GLOBALSINDEX, baseClass.c_str());
+        lua_setfield(L, -2, "__index");
+
+        // setmetatable(klass, redirect)
+        lua_setmetatable(L, -2);
+
+        // pop the derived class table
+        lua_pop(L, 1);
+    }
+}
+
+void LuaScript::registerMemberFunction(const std::string& klass, const std::string& functionName, LuaCFunction function)
+{
+    int functionId = m_functions.size();
+    m_functions.push_back(function);
+
+    // push the class table
+    lua_getfield(L, LUA_GLOBALSINDEX, klass.c_str());
+    // push the function id
+    lua_pushnumber(L, functionId);
+    // store id in the closure
+    lua_pushcclosure(L, &LuaScript::luaFunctionCallback, 1);
+    // store the function at the class field functionName
+    lua_setfield(L, -2, functionName.c_str());
+    // pop the table
+    lua_pop(L, 1);
+}
+
+void LuaScript::registerGlobalFunction(const std::string& functionName, LuaCFunction function)
+{
+    int functionId = m_functions.size();
+    m_functions.push_back(function);
+
+    // push the function id
+    lua_pushnumber(L, functionId);
+    // store id in the closure
+    lua_pushcclosure(L, &LuaScript::luaFunctionCallback, 1);
+    // store the function in the global lua table
+    lua_setfield(L, LUA_GLOBALSINDEX, functionName.c_str());
+}
+
+int LuaScript::luaPackageLoader(lua_State* L)
+{
+    std::string fileName = lua_tostring(L, -1);
+    fileName += ".lua";
+    if(g_resources.fileExists(fileName)) {
+        std::string fileContents = g_resources.loadTextFile(fileName);
+        luaL_loadbuffer(L, fileContents.c_str(), fileContents.length(), fileName.c_str());
+    } else {
+        lua_pushfstring(L, "\n\tcould not load lua script " LUA_QS, fileName.c_str());
+    }
+    return 1;
+}
+
+int LuaScript::luaCollectClassInstance(lua_State* L)
+{
+    ScriptableWeakPtr *objectRef = (ScriptableWeakPtr *)lua_touserdata(L, -1);
+    objectRef->reset();
+    lua_pop(L, 1);
+    return 1;
+}
+
+int LuaScript::luaCompareClassInstances(lua_State* L)
+{
+    if(!lua_isuserdata(L, -1) || !lua_isuserdata(L, -2)) {
+        lua_pop(L, 2);
+        lua_pushboolean(L, 0);
+    } else {
+        ScriptableWeakPtr *objectRef1 = (ScriptableWeakPtr *)lua_touserdata(L, -1);
+        ScriptableWeakPtr *objectRef2 = (ScriptableWeakPtr *)lua_touserdata(L, -2);
+        lua_pop(L, 2);
+        if(objectRef1->lock() == objectRef2->lock())
+            lua_pushboolean(L, 1);
+        else
+            lua_pushboolean(L, 0);
+    }
+    return 1;
+}
+
+int LuaScript::luaFunctionCallback(lua_State* L)
+{
+    int id = lua_tonumber(L, lua_upvalueindex(1));
+    return (g_lua.*(g_lua.m_functions[id]))();
+}
+
diff --git a/src/framework/script/luascript.h b/src/framework/script/luascript.h
new file mode 100644
index 00000000..dc488b97
--- /dev/null
+++ b/src/framework/script/luascript.h
@@ -0,0 +1,95 @@
+/* The MIT License
+ *
+ * Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
+ *
+ * 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 LUASCRIPT_H
+#define LUASCRIPT_H
+
+#include <prerequisites.h>
+#include <script/scriptable.h>
+#include <lua.hpp>
+
+class LuaScript
+{
+public:
+    LuaScript();
+    virtual ~LuaScript();
+
+    bool loadFile(const std::string& fileName);
+    bool loadBuffer(const std::string& text, const std::string& what = "luaBuffer");
+    int loadBufferAsFunction(const std::string& text, const std::string& what = "luaBuffer");
+
+    int getStackSize();
+
+    void pushNil();
+    void pushBoolean(bool b);
+    void pushInteger(int i);
+    void pushString(const std::string& str);
+    void pushUserdata(void* ptr);
+
+    void pop(int n = 1);
+    bool popBoolean();
+    int32_t popInteger();
+    std::string popString();
+
+    int popFunction();
+    void pushFunction(int functionRef);
+    void callFunction(int numArgs = 0, bool itReturnsValue = false);
+
+    void pushClassInstance(const ScriptablePtr& object);
+    ScriptablePtr popClassInstance();
+
+    typedef int (LuaScript::*LuaCFunction)();
+
+    void setupPackageLoader();
+    void registerClass(const std::string& klass, const std::string& baseClass = "");
+    void registerMemberFunction(const std::string& klass, const std::string& functionName, LuaCFunction function);
+    void registerGlobalFunction(const std::string& functionName, LuaCFunction function);
+
+    static int luaFunctionCallback(lua_State* L);
+    static int luaPackageLoader(lua_State* L);
+    static int luaCollectClassInstance(lua_State* L);
+    static int luaCompareClassInstances(lua_State* L);
+
+    void registerFunctions();
+
+    int lua_UIButton_setOnClick();
+
+    int lua_UIElement_destroy();
+
+    // container functions
+    int lua_UIContainer_getChildByID();
+
+    // global functions
+    int lua_exitGame();
+    int lua_loadUI();
+    int lua_getUIRootContainer();
+
+private:
+    std::vector<LuaCFunction> m_functions;
+    lua_State *L;
+};
+
+extern LuaScript g_lua;
+
+#endif // LUASCRIPT_H
diff --git a/src/framework/script/scriptable.h b/src/framework/script/scriptable.h
new file mode 100644
index 00000000..f6275ab6
--- /dev/null
+++ b/src/framework/script/scriptable.h
@@ -0,0 +1,39 @@
+/* The MIT License
+ *
+ * Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
+ *
+ * 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 SCRIPTABLE_H
+#define SCRIPTABLE_H
+
+#include <prerequisites.h>
+
+class Scriptable : public boost::enable_shared_from_this<Scriptable>
+{
+public:
+    virtual const char *getScriptableName() const { return NULL; }
+};
+
+typedef boost::shared_ptr<Scriptable> ScriptablePtr;
+typedef boost::weak_ptr<Scriptable> ScriptableWeakPtr;
+
+#endif
diff --git a/src/framework/ui/uibutton.h b/src/framework/ui/uibutton.h
index 36ceb5ec..86276338 100644
--- a/src/framework/ui/uibutton.h
+++ b/src/framework/ui/uibutton.h
@@ -45,6 +45,8 @@ public:
 
     void setOnClick(const Callback& callback) { m_buttonClickCallback = callback; }
 
+    virtual const char *getScriptableName() const { return "UIButton"; }
+
 private:
     std::string m_text;
     UI::EButtonState m_state;
diff --git a/src/framework/ui/uicheckbox.h b/src/framework/ui/uicheckbox.h
index 4f301d08..59972ca5 100644
--- a/src/framework/ui/uicheckbox.h
+++ b/src/framework/ui/uicheckbox.h
@@ -30,9 +30,10 @@
 
 class UICheckBox : public UIElement
 {
-
 public:
-UICheckBox(UI::EElementType type = UI::Element);
+    UICheckBox(UI::EElementType type = UI::Element);
+
+    virtual const char *getScriptableName() const { return "UICheckBox"; }
 };
 
 #endif // UICHECKBOX_H
diff --git a/src/framework/ui/uicontainer.h b/src/framework/ui/uicontainer.h
index 29d5f5ea..e6e5581f 100644
--- a/src/framework/ui/uicontainer.h
+++ b/src/framework/ui/uicontainer.h
@@ -62,6 +62,8 @@ public:
     virtual UI::EElementType getElementType() const { return UI::Container; }
     UIContainerPtr asUIContainer() { return boost::static_pointer_cast<UIContainer>(shared_from_this()); }
 
+    virtual const char *getScriptableName() const { return "UIContainer"; }
+
     /// Get root container (the container that contains everything)
     static UIContainerPtr& getRootContainer();
 
diff --git a/src/framework/ui/uielement.h b/src/framework/ui/uielement.h
index 757faca7..55314d2d 100644
--- a/src/framework/ui/uielement.h
+++ b/src/framework/ui/uielement.h
@@ -84,6 +84,7 @@ public:
 
     UIElementPtr asUIElement() { return boost::static_pointer_cast<UIElement>(shared_from_this()); }
     virtual UIContainerPtr asUIContainer() { return UIContainerPtr(); }
+    virtual const char *getScriptableName() const { return "UIElement"; }
 
     friend class UIContainer;
 
diff --git a/src/framework/ui/uilabel.h b/src/framework/ui/uilabel.h
index d9eb108f..7a79e9b6 100644
--- a/src/framework/ui/uilabel.h
+++ b/src/framework/ui/uilabel.h
@@ -43,6 +43,8 @@ public:
     void setAlign(int align) { m_align = align; }
     int getAlign() const { return m_align; }
 
+    virtual const char *getScriptableName() const { return "UILabel"; }
+
 private:
     std::string m_text;
     int m_align;
diff --git a/src/framework/ui/uilayout.h b/src/framework/ui/uilayout.h
index 4790d2fc..233098ee 100644
--- a/src/framework/ui/uilayout.h
+++ b/src/framework/ui/uilayout.h
@@ -27,6 +27,7 @@
 
 #include <prerequisites.h>
 #include <ui/uiconstants.h>
+#include <script/scriptable.h>
 
 enum EAnchorType {
     ANCHOR_LEFT = 0,
@@ -62,7 +63,7 @@ private:
     EAnchorType m_anchorType;
 };
 
-class UILayout : public boost::enable_shared_from_this<UILayout>
+class UILayout : public Scriptable
 {
 public:
     UILayout() :
@@ -106,7 +107,9 @@ public:
     void setMarginTop(int margin) { m_marginTop = margin; recalculateLayout(); }
     void setMarginBottom(int margin) { m_marginBottom = margin; recalculateLayout(); }
 
-    UILayoutPtr asUILayout() { return shared_from_this(); }
+    UILayoutPtr asUILayout() { return boost::static_pointer_cast<UILayout>(shared_from_this()); }
+
+    virtual const char *getScriptableName() const { return "UILayout"; }
 
 protected:
     virtual void onLayoutRectChange(const Rect& newRect) { }
diff --git a/src/framework/ui/uiloader.cpp b/src/framework/ui/uiloader.cpp
index 8dfedd68..26214378 100644
--- a/src/framework/ui/uiloader.cpp
+++ b/src/framework/ui/uiloader.cpp
@@ -26,6 +26,7 @@
 #include <core/resources.h>
 #include <ui/ui.h>
 #include <ui/uiloader.h>
+#include <script/luascript.h>
 
 UIElementPtr UILoader::createElementFromId(const std::string& id)
 {
@@ -149,10 +150,8 @@ void UILoader::loadElements(const UIElementPtr& parent, const YAML::Node& node)
 void UILoader::loadElement(const UIElementPtr& element, const YAML::Node& node)
 {
     // load specific element type
-    if(element->getElementType() == UI::Button) {
-        UIButtonPtr button = boost::static_pointer_cast<UIButton>(element);
-        button->setText(node["text"].Read<std::string>());
-    }
+    if(element->getElementType() == UI::Button)
+        loadButton(boost::static_pointer_cast<UIButton>(element), node);
     else if(element->getElementType() == UI::Window) {
         UIWindowPtr window = boost::static_pointer_cast<UIWindow>(element);
         window->setTitle(node["title"].Read<std::string>());
@@ -266,3 +265,18 @@ void UILoader::loadElementAnchor(const UIElementPtr& element, EAnchorType type,
         throw YAML::Exception(node.GetMark(), "anchoring failed, does the relative element really exists?");
     }
 }
+
+void UILoader::loadButton(const UIButtonPtr& button, const YAML::Node& node)
+{
+    button->setText(node["text"].Read<std::string>());
+
+    // set on click event
+    if(node.FindValue("onClick")) {
+        int funcRef = g_lua.loadBufferAsFunction(node["onClick"].Read<std::string>());
+        if(funcRef != LUA_REFNIL) {
+            g_lua.pushClassInstance(button);
+            g_lua.pushFunction(funcRef);
+            g_lua.lua_UIButton_setOnClick();
+        }
+    }
+}
diff --git a/src/framework/ui/uiloader.h b/src/framework/ui/uiloader.h
index e36b7441..eca59ea9 100644
--- a/src/framework/ui/uiloader.h
+++ b/src/framework/ui/uiloader.h
@@ -28,6 +28,7 @@
 #include <prerequisites.h>
 #include <ui/uiconstants.h>
 #include <ui/uicontainer.h>
+#include <ui/uibutton.h>
 
 class UILoader
 {
@@ -50,6 +51,9 @@ private:
 
     /// Load anchor from a YAML node
     static void loadElementAnchor(const UIElementPtr& element, EAnchorType type, const YAML::Node& node);
+
+    // specific elements loading
+    static void loadButton(const UIButtonPtr& button, const YAML::Node& node);
 };
 
 #endif // UILOADER_H
diff --git a/src/framework/ui/uitextedit.h b/src/framework/ui/uitextedit.h
index 15c71c64..3fb8261e 100644
--- a/src/framework/ui/uitextedit.h
+++ b/src/framework/ui/uitextedit.h
@@ -46,6 +46,8 @@ public:
 
     bool isFocusable() const { return true; }
 
+    virtual const char *getScriptableName() const { return "UITextEdit"; }
+
 private:
     TextArea m_textArea;
 };
diff --git a/src/framework/ui/uiwindow.h b/src/framework/ui/uiwindow.h
index 04237e32..653782ba 100644
--- a/src/framework/ui/uiwindow.h
+++ b/src/framework/ui/uiwindow.h
@@ -40,6 +40,8 @@ public:
     void setTitle(const std::string& title) { m_title = title; }
     const std::string& getTitle() const { return m_title; }
 
+    virtual const char *getScriptableName() const { return "UIWindow"; }
+
 private:
     std::string m_title;
     bool m_moving;
diff --git a/src/main.cpp b/src/main.cpp
index c17666f5..8360c926 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -31,7 +31,7 @@
 #include <ui/uiskins.h>
 #include "menustate.h"
 #include "teststate.h"
-#include <script/script.h>
+#include <script/luascript.h>
 
 /// Catches signals so we can exit nicely
 void signal_handler(int sig)
@@ -110,8 +110,6 @@ int main(int argc, const char *argv[])
 
     // init engine
     g_engine.init();
-    g_script.init();
-    g_script.loadFile("lib/init.lua");
     g_uiSkins.load("skins/tibiaskin.yml");
 
     // state scope
diff --git a/src/menustate.cpp b/src/menustate.cpp
index 18f2cd0c..a5b0d5a1 100644
--- a/src/menustate.cpp
+++ b/src/menustate.cpp
@@ -32,29 +32,14 @@
 #include <graphics/fonts.h>
 #include <ui/ui.h>
 #include "menustate.h"
-#include <script/script.h>
+#include <script/luascript.h>
 
 void MenuState::onEnter()
 {
     m_background = g_textures.get("background.png");
     m_background->enableBilinearFilter();
 
-    g_script.loadFile("modules/mainmenu/init.lua");
-    /*
-    UIContainerPtr mainMenuPanel = UILoader::loadFile("ui/mainMenu.yml")->asUIContainer();
-
-    UIButtonPtr button = boost::static_pointer_cast<UIButton>(mainMenuPanel->getChildById("exitGameButton"));
-    button->setOnClick(boost::bind(&MenuState::onClose, this));
-
-    button = boost::static_pointer_cast<UIButton>(mainMenuPanel->getChildById("enterGameButton"));
-    button->setOnClick(boost::bind(&MenuState::enterGameButton_clicked, this));
-
-    button = boost::static_pointer_cast<UIButton>(mainMenuPanel->getChildById("infoButton"));
-    button->setOnClick(boost::bind(&MenuState::infoButton_clicked, this));
-
-    button = boost::static_pointer_cast<UIButton>(mainMenuPanel->getChildById("optionsButton"));
-    button->setOnClick(boost::bind(&MenuState::optionsButton_clicked, this));
-    */
+    g_lua.loadFile("modules/mainmenu/mainmenu.lua");
 }
 
 void MenuState::onLeave()
@@ -92,53 +77,3 @@ void MenuState::render()
     g_graphics.drawTexturedRect(Rect(0, 0, screenSize), m_background, texCoords);
 }
 
-void MenuState::enterGameButton_clicked()
-{
-    UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UILoader::loadFile("ui/enterGameWindow.yml"));
-    window->getParent()->lockElement(window);
-
-    UIButtonPtr button = boost::static_pointer_cast<UIButton>(window->getChildById("cancelButton"));
-    button->setOnClick([] {
-        UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UIContainer::getRootContainer()->getChildById("enterGameWindow"));
-        window->getParent()->unlockElement();
-        window->destroy();
-    });
-
-    button = boost::static_pointer_cast<UIButton>(window->getChildById("okButton"));
-    button->setOnClick(boost::bind(&MenuState::enterGameWindowOkButton_clicked, this));
-}
-
-void MenuState::infoButton_clicked()
-{
-    UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UILoader::loadFile("ui/infoWindow.yml"));
-    window->getParent()->lockElement(window);
-    UIButtonPtr button = boost::static_pointer_cast<UIButton>(window->getChildById("okButton"));
-    button->setOnClick([] {
-        UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UIContainer::getRootContainer()->getChildById("infoWindow"));
-        window->getParent()->unlockElement();
-        window->destroy();
-    });
-}
-
-void MenuState::enterGameWindowOkButton_clicked()
-{
-    UIContainerPtr enterGameWindow = boost::static_pointer_cast<UIContainer>(UIContainer::getRootContainer()->getChildById("enterGameWindow"));
-
-    std::string accountName = boost::static_pointer_cast<UITextEdit>(enterGameWindow->getChildById("accountNameTextEdit"))->getText();
-    std::string password = boost::static_pointer_cast<UITextEdit>(enterGameWindow->getChildById("passwordTextEdit"))->getText();
-
-    //m_protocolLogin = ProtocolLoginPtr(new ProtocolLogin);
-    //m_protocolLogin->login(accountName, password);
-}
-
-void MenuState::optionsButton_clicked()
-{
-    UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UILoader::loadFile("ui/optionsWindow.yml"));
-    window->getParent()->lockElement(window);
-    UIButtonPtr button = boost::static_pointer_cast<UIButton>(window->getChildById("okButton"));
-    button->setOnClick([] {
-        UIWindowPtr window = boost::static_pointer_cast<UIWindow>(UIContainer::getRootContainer()->getChildById("optionsWindow"));
-        window->getParent()->unlockElement();
-        window->destroy();
-    });
-}