Spider queen antag v0.2

Content.Server/Roles/RoleSystem.cs

@@ -1,3 +1,4 @@
+using Content.Server.SS220.Roles;
 using Content.Shared.Roles;
 using Content.Shared.SS220.DarkReaper;
 using Content.Shared.SS220.MindSlave;
@@ -24,6 +25,8 @@ public override void Initialize()
         SubscribeAntagEvents<MindSlaveRoleComponent>();
         //SS220 DarkReaper
         SubscribeAntagEvents<DarkReaperRoleComponent>();
+        //SS220 Spider queen
+        SubscribeAntagEvents<SpiderQueenRoleComponent>();
     }
 
     public string? MindGetBriefing(EntityUid? mindId)

Content.Server/SS220/Markers/AntagSpawnMarkerComponent.cs

@@ -0,0 +1,7 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+namespace Content.Server.SS220.Markers;
+
+[RegisterComponent]
+public sealed partial class AntagSpawnMarkerComponent : Component
+{
+}

Content.Server/SS220/Objectives/Components/CreateCocoonsConditionComponent.cs

@@ -0,0 +1,7 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+namespace Content.Server.SS220.Objectives.Components;
+
+[RegisterComponent]
+public sealed partial class CreateCocoonsConditionComponent : Component
+{
+}

Content.Server/SS220/Objectives/Systems/CreateCocoonsConditionSystem.cs

@@ -0,0 +1,50 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+using Content.Server.Objectives.Systems;
+using Content.Server.SS220.Objectives.Components;
+using Content.Server.SS220.Roles;
+using Content.Shared.Mind;
+using Content.Shared.Objectives.Components;
+using Content.Shared.SS220.SpiderQueen.Components;
+
+namespace Content.Server.SS220.Objectives.Systems;
+
+public sealed partial class CreateCocoonsConditionSystem : EntitySystem
+{
+    [Dependency] private readonly NumberObjectiveSystem _number = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<CreateCocoonsConditionComponent, ObjectiveGetProgressEvent>(OnGetProgress);
+    }
+
+    private void OnGetProgress(Entity<CreateCocoonsConditionComponent> ent, ref ObjectiveGetProgressEvent args)
+    {
+        args.Progress = GetProgress(args.MindId, args.Mind, _number.GetTarget(ent.Owner));
+    }
+
+    private float GetProgress(EntityUid mindId, MindComponent mind, int target)
+    {
+        if (!TryComp<SpiderQueenRoleComponent>(mindId, out var spiderQueenRole))
+            return 0f;
+
+        if (spiderQueenRole.IsCreateCocoonsCompletedOnce)
+            return 1f;
+
+        var mobUid = mind.CurrentEntity;
+        if (mobUid is null ||
+            !TryComp<SpiderQueenComponent>(mobUid, out var spiderQueen))
+            return 0f;
+
+        if (spiderQueen.CocoonsList.Count >= target)
+        {
+            spiderQueenRole.IsCreateCocoonsCompletedOnce = true;
+            return 1f;
+        }
+        else
+        {
+            return 0f;
+        }
+    }
+}

Content.Server/SS220/Roles/SpiderQueenRoleComponent.cs

@@ -0,0 +1,11 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+using Content.Shared.Roles;
+
+namespace Content.Server.SS220.Roles;
+
+[RegisterComponent, ExclusiveAntagonist]
+public sealed partial class SpiderQueenRoleComponent : AntagonistRoleComponent
+{
+    [ViewVariables]
+    public bool IsCreateCocoonsCompletedOnce = false;
+}

Content.Server/SS220/SpiderQueen/SpiderQueenRuleComponent.cs

@@ -0,0 +1,15 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+using Content.Shared.Whitelist;
+
+namespace Content.Server.SS220.SpiderQueen;
+
+[RegisterComponent]
+public sealed partial class SpiderQueenRuleComponent : Component
+{
+    /// <summary>
+    /// Spawn on a random entity that passed whitelist.
+    /// If null - spawn on a random tile.
+    /// </summary>
+    [DataField]
+    public EntityWhitelist? MarkersWhitelist;
+}

Content.Server/SS220/SpiderQueen/SpiderQueenRuleSystem.cs

@@ -0,0 +1,111 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+using Content.Server.Antag;
+using Content.Server.GameTicking.Rules;
+using Content.Server.Mind;
+using Content.Server.Respawn;
+using Content.Server.Roles;
+using Content.Server.SS220.Markers;
+using Content.Server.SS220.Roles;
+using Content.Server.Station.Components;
+using Content.Server.Station.Systems;
+using Content.Shared.Mind;
+using Content.Shared.SS220.SpiderQueen.Components;
+using Content.Shared.Whitelist;
+using Robust.Shared.Map;
+
+namespace Content.Server.SS220.SpiderQueen;
+
+public sealed class SpiderQueenRuleSystem : GameRuleSystem<SpiderQueenRuleComponent>
+{
+    [Dependency] private readonly EntityWhitelistSystem _entityWhitelist = default!;
+    [Dependency] private readonly SharedTransformSystem _transform = default!;
+    [Dependency] private readonly StationSystem _station = default!;
+    [Dependency] private readonly SpecialRespawnSystem _specialRespawn = default!;
+    [Dependency] private readonly MindSystem _mind = default!;
+    [Dependency] private readonly AntagSelectionSystem _antag = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<SpiderQueenRuleComponent, AntagSelectLocationEvent>(OnAntagSelectLocation);
+        SubscribeLocalEvent<SpiderQueenRuleComponent, AfterAntagEntitySelectedEvent>(AfterEntitySelected);
+
+        SubscribeLocalEvent<SpiderQueenRoleComponent, GetBriefingEvent>(OnGetBriefing);
+    }
+
+    private void OnAntagSelectLocation(Entity<SpiderQueenRuleComponent> ent, ref AntagSelectLocationEvent args)
+    {
+        if (args.Handled)
+            return;
+
+        List<MapCoordinates> validCoordinates = new();
+        var query = EntityQueryEnumerator<AntagSpawnMarkerComponent>();
+        while (query.MoveNext(out var uid, out _))
+        {
+            if (_entityWhitelist.IsWhitelistFail(ent.Comp.MarkersWhitelist, uid) ||
+                !TryComp<TransformComponent>(uid, out var transform))
+                continue;
+
+            validCoordinates.Add(_transform.ToMapCoordinates(transform.Coordinates));
+        }
+
+        if (validCoordinates.Count > 0)
+            args.Coordinates = validCoordinates;
+        else
+        {
+            EntityUid? grid = null;
+            EntityUid? map = null;
+            foreach (var station in _station.GetStationsSet())
+            {
+                if (!TryComp<StationDataComponent>(station, out var data))
+                    continue;
+
+                grid = _station.GetLargestGrid(data);
+                if (!grid.HasValue)
+                    continue;
+
+                map = Transform(grid.Value).MapUid;
+                if (!map.HasValue)
+                    continue;
+
+                break;
+            }
+
+            if (grid.HasValue && map.HasValue &&
+                _specialRespawn.TryFindRandomTile(grid.Value, map.Value, 30, out var randomCoords))
+                args.Coordinates.Add(_transform.ToMapCoordinates(randomCoords));
+        }
+    }
+
+    private void AfterEntitySelected(Entity<SpiderQueenRuleComponent> ent, ref AfterAntagEntitySelectedEvent args)
+    {
+        var spider = args.EntityUid;
+        if (!_mind.TryGetMind(spider, out var mindId, out var mind))
+            return;
+
+        var briefing = Loc.GetString("spider-queen-role-greeting");
+        _antag.SendBriefing(spider, briefing, null, null);
+    }
+
+    private void OnGetBriefing(Entity<SpiderQueenRoleComponent> ent, ref GetBriefingEvent args)
+    {
+        if (!TryComp<MindComponent>(ent.Owner, out var mind))
+            return;
+
+        var briefing = Loc.GetString("spider-queen-role-greeting");
+
+        if (mind.OwnedEntity is { } spider &&
+            TryComp<SpiderQueenComponent>(spider, out var spiderQueen))
+        {
+            var current = spiderQueen.CurrentMana;
+            var max = spiderQueen.MaxMana;
+
+            briefing = string.Format("{0}\n{1}", briefing,
+                Loc.GetString("spider-queen-role-mana-information", ("current", current), ("max", max)));
+        }
+
+        args.Append(briefing);
+    }
+}
+