From b204fd9b0e2e0264e532d7d20887f32dd24d83fe Mon Sep 17 00:00:00 2001
From: qwerltaz <msmarcinpl@gmail.com>
Date: Wed, 2 Apr 2025 20:37:34 +0200
Subject: [PATCH] add: Dragon rift color changes based on charge (#36216)

* use dragon rift sprite colours

* Entity<T>
---
 Content.Server/Dragon/DragonRiftSystem.cs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/Content.Server/Dragon/DragonRiftSystem.cs b/Content.Server/Dragon/DragonRiftSystem.cs
index 998834835e..9cab018fd7 100644
--- a/Content.Server/Dragon/DragonRiftSystem.cs
+++ b/Content.Server/Dragon/DragonRiftSystem.cs
@@ -13,6 +13,7 @@ using Robust.Shared.Serialization.Manager;
 using System.Numerics;
 using Robust.Shared.Audio;
 using Robust.Shared.Audio.Systems;
+using Robust.Shared.GameStates;
 using Robust.Shared.Utility;
 
 namespace Content.Server.Dragon;
@@ -33,11 +34,20 @@ public sealed class DragonRiftSystem : EntitySystem
     {
         base.Initialize();
 
+        SubscribeLocalEvent<DragonRiftComponent, ComponentGetState>(OnGetState);
         SubscribeLocalEvent<DragonRiftComponent, ExaminedEvent>(OnExamined);
         SubscribeLocalEvent<DragonRiftComponent, AnchorStateChangedEvent>(OnAnchorChange);
         SubscribeLocalEvent<DragonRiftComponent, ComponentShutdown>(OnShutdown);
     }
 
+    private void OnGetState(Entity<DragonRiftComponent> ent, ref ComponentGetState args)
+    {
+        args.State = new DragonRiftComponentState
+        {
+            State = ent.Comp.State,
+        };
+    }
+
     public override void Update(float frameTime)
     {
         base.Update(frameTime);
-- 
2.51.2