From c4641f2594d0c9e63058a930b571c64d6b6545fc Mon Sep 17 00:00:00 2001
From: Gene Walters <genewalt@amazon.com>
Date: Tue, 18 May 2021 14:13:01 -0700
Subject: [PATCH] ScriptCanvas can now check if an entity net-component is
 authority, autonomous, server, or client

---
 .../Components/MultiplayerComponent.cpp       | 76 +++++++++++++++++++
 1 file changed, 76 insertions(+)

diff --git a/Gems/Multiplayer/Code/Source/Components/MultiplayerComponent.cpp b/Gems/Multiplayer/Code/Source/Components/MultiplayerComponent.cpp
index 8542288b23..ff3b64aa38 100644
--- a/Gems/Multiplayer/Code/Source/Components/MultiplayerComponent.cpp
+++ b/Gems/Multiplayer/Code/Source/Components/MultiplayerComponent.cpp
@@ -24,6 +24,82 @@ namespace Multiplayer
             serializeContext->Class<MultiplayerComponent, AZ::Component>()
                 ->Version(1);
         }
+
+        AZ::BehaviorContext* behaviorContext = azrtti_cast<AZ::BehaviorContext*>(context);
+        if (behaviorContext)
+        {
+            behaviorContext->Class<MultiplayerComponent>("MultiplayerComponent")
+                ->Attribute(AZ::Script::Attributes::Module, "multiplayer")
+                ->Attribute(AZ::Script::Attributes::Category, "Multiplayer")
+
+                ->Method("Is Authority", [](AZ::EntityId id) -> bool {
+                    AZ::Entity* entity = AZ::Interface<AZ::ComponentApplicationRequests>::Get()->FindEntity(id);
+                    if (!entity)
+                    {
+                        AZ_Warning( "MultiplayerComponent", false, "MultiplayerComponent IsAuthority failed. The entity with id %s doesn't exist, please provide a valid entity id.", id.ToString().c_str())
+                        return false;
+                    }
+
+                    MultiplayerComponent* multiplayerComponent = entity->FindComponent<MultiplayerComponent>();
+                    if (!multiplayerComponent)
+                    {
+                        AZ_Warning( "MultiplayerComponent", false, "MultiplayerComponent IsAuthority failed. Entity '%s' (id: %s) is missing a MultiplayerComponent, make sure this entity contains a component which derives from MultiplayerComponent.", entity->GetName().c_str(), id.ToString().c_str())
+                        return false;
+                    }
+                    return multiplayerComponent->IsAuthority();
+                })
+                ->Method("Is Autonomous", [](AZ::EntityId id) -> bool {
+                    AZ::Entity* entity = AZ::Interface<AZ::ComponentApplicationRequests>::Get()->FindEntity(id);
+                    if (!entity)
+                    {
+                        AZ_Warning( "MultiplayerComponent", false, "MultiplayerComponent IsAutonomous failed. The entity with id %s doesn't exist, please provide a valid entity id.", id.ToString().c_str())
+                        return false;
+                    }
+
+                    MultiplayerComponent* multiplayerComponent = entity->FindComponent<MultiplayerComponent>();
+                    if (!multiplayerComponent)
+                    {
+                        AZ_Warning(
+                            "MultiplayerComponent", false,
+                            "MultiplayerComponent IsAutonomous failed. Entity '%s' (id: %s) is missing a MultiplayerComponent, make sure this entity contains a component which derives from MultiplayerComponent.",
+                            entity->GetName().c_str(), id.ToString().c_str()) return false;
+                    }
+                    return multiplayerComponent->IsAutonomous();
+                })
+                ->Method("Is Client", [](AZ::EntityId id) -> bool {
+                    AZ::Entity* entity = AZ::Interface<AZ::ComponentApplicationRequests>::Get()->FindEntity(id);
+                    if (!entity)
+                    {
+                        AZ_Warning( "MultiplayerComponent", false, "MultiplayerComponent IsClient failed. The entity with id %s doesn't exist, please provide a valid entity id.", id.ToString().c_str())
+                        return false;
+                    }
+
+                    MultiplayerComponent* multiplayerComponent = entity->FindComponent<MultiplayerComponent>();
+                    if (!multiplayerComponent)
+                    {
+                        AZ_Warning("MultiplayerComponent", false, "MultiplayerComponent IsClient failed. Entity '%s' (id: %s) is missing a MultiplayerComponent, make sure this entity contains a component which derives from MultiplayerComponent.", entity->GetName().c_str(), id.ToString().c_str())
+                        return false;
+                    }
+                    return multiplayerComponent->IsClient();
+                })
+                ->Method("Is Server", [](AZ::EntityId id) -> bool {
+                    AZ::Entity* entity = AZ::Interface<AZ::ComponentApplicationRequests>::Get()->FindEntity(id);
+                    if (!entity)
+                    {
+                        AZ_Warning( "MultiplayerComponent", false, "MultiplayerComponent IsServer failed. The entity with id %s doesn't exist, please provide a valid entity id.", id.ToString().c_str())
+                        return false;
+                    }
+
+                    MultiplayerComponent* multiplayerComponent = entity->FindComponent<MultiplayerComponent>();
+                    if (!multiplayerComponent)
+                    {
+                        AZ_Warning("MultiplayerComponent", false, "MultiplayerComponent IsServer failed. Entity '%s' (id: %s) is missing a MultiplayerComponent, make sure this entity contains a component which derives from MultiplayerComponent.", entity->GetName().c_str(), id.ToString().c_str())
+                        return false;
+                    }
+                    return multiplayerComponent->IsServer();
+                })
+            ;
+        }
     }
 
     void MultiplayerComponent::GetRequiredServices(AZ::ComponentDescriptor::DependencyArrayType& required)