From 90267c610530d8be1c14288fc742b73d3473050f Mon Sep 17 00:00:00 2001
From: Kae <80987908+Novaenia@users.noreply.github.com>
Date: Tue, 15 Aug 2023 14:20:56 +1000
Subject: message.setHandler now accepts a Json config in place of the name

---
 source/game/scripting/StarLuaComponents.hpp | 49 +++++++++++++++++++++++------
 1 file changed, 40 insertions(+), 9 deletions(-)

(limited to 'source/game/scripting/StarLuaComponents.hpp')

diff --git a/source/game/scripting/StarLuaComponents.hpp b/source/game/scripting/StarLuaComponents.hpp
index 7e30c39..2098829 100644
--- a/source/game/scripting/StarLuaComponents.hpp
+++ b/source/game/scripting/StarLuaComponents.hpp
@@ -173,7 +173,14 @@ protected:
   virtual void contextShutdown() override;
 
 private:
-  StringMap<LuaFunction> m_messageHandlers;
+  struct MessageHandler {
+    Maybe<LuaFunction> function;
+    String name;
+    bool passName = true;
+    bool localOnly = false;
+  };
+
+  StringMap<MessageHandler> m_messageHandlers;
 };
 
 template <typename Ret, typename... V>
@@ -304,13 +311,26 @@ void LuaWorldComponent<Base>::uninit() {
 template <typename Base>
 LuaMessageHandlingComponent<Base>::LuaMessageHandlingComponent() {
   LuaCallbacks scriptCallbacks;
-  scriptCallbacks.registerCallback("setHandler",
-      [this](String message, Maybe<LuaFunction> handler) {
-        if (handler)
-          m_messageHandlers.set(move(message), handler.take());
-        else
-          m_messageHandlers.remove(message);
-      });
+  scriptCallbacks.registerCallback("setHandler", [this](Variant<String, Json> message, Maybe<LuaFunction> handler) {
+      MessageHandler handlerInfo = {};
+
+      if (Json* config = message.ptr<Json>()) {
+        handlerInfo.passName = config->getBool("passName", false);
+        handlerInfo.localOnly = config->getBool("localOnly", false);
+        handlerInfo.name = config->getString("name");
+      } else {
+        handlerInfo.passName = true;
+        handlerInfo.localOnly = false;
+        handlerInfo.name = message.get<String>();
+      }
+
+      if (handler) {
+        handlerInfo.function.emplace(handler.take());
+        m_messageHandlers.set(handlerInfo.name, handlerInfo);
+      }
+      else
+        m_messageHandlers.remove(handlerInfo.name);
+    });
 
   Base::addCallbacks("message", move(scriptCallbacks));
 }
@@ -323,7 +343,18 @@ Maybe<Json> LuaMessageHandlingComponent<Base>::handleMessage(
 
   if (auto handler = m_messageHandlers.ptr(message)) {
     try {
-      return handler->template invoke<Json>(message, localMessage, luaUnpack(args));
+      if (handler->localOnly) {
+        if (!localMessage)
+          return {};
+        else if (handler->passName)
+          return handler->function->template invoke<Json>(message, luaUnpack(args));
+        else
+          return handler->function->template invoke<Json>(luaUnpack(args));
+      }
+      else if (handler->passName)
+        return handler->function->template invoke<Json>(message, localMessage, luaUnpack(args));
+      else
+        return handler->function->template invoke<Json>(localMessage, luaUnpack(args));
     } catch (LuaException const& e) {
       Logger::error(
           "Exception while invoking lua message handler for message '{}'. {}", message, outputException(e, true));
-- 
cgit v1.2.3