feat: add ProxiedCommandSender in type caster

EndstoneMC · Feb 10, 2025 · 747637a · 747637a
1 parent df567c9
commit 747637a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/include/endstone/command/command_sender.h b/include/endstone/command/command_sender.h
@@ -28,6 +28,7 @@ class ConsoleCommandSender;
 class Mob;
 class Server;
 class Player;
+class ProxiedCommandSender;
 
 /**
  * @brief Represents a command sender.
@@ -80,6 +81,16 @@ class CommandSender : public Permissible {
         return nullptr;
     }
 
+    /**
+     * @brief Gets a CommandSender as ProxiedCommandSender
+     *
+     * @return ProxiedCommandSender, nullptr if not a ProxiedCommandSender
+     */
+    [[nodiscard]] virtual ProxiedCommandSender *asProxiedCommandSender() const
+    {
+        return nullptr;
+    }
+
     /**
      * @brief Sends this sender a message
      *

diff --git a/include/endstone/command/proxied_command_sender.h b/include/endstone/command/proxied_command_sender.h
@@ -23,6 +23,11 @@ namespace endstone {
  */
 class ProxiedCommandSender : public CommandSender {
 public:
+    [[nodiscard]] ProxiedCommandSender *asProxiedCommandSender() const override
+    {
+        return const_cast<ProxiedCommandSender *>(this);
+    }
+
     /**
      * @brief Returns the CommandSender which triggered this proxied command.
      *

diff --git a/src/endstone/python/poly.h b/src/endstone/python/poly.h
@@ -63,6 +63,10 @@ struct polymorphic_type_hook<endstone::CommandSender> {
             type = &typeid(endstone::ConsoleCommandSender);
             return console;
         }
+        if (const auto *proxied = src->asProxiedCommandSender()) {
+            type = &typeid(endstone::ProxiedCommandSender);
+            return proxied;
+        }
         type = &typeid(endstone::CommandSender);
         return src;
     }