SetStationAiName for testing

Content.Server/Administration/Commands/SetStationAiName.cs

@@ -0,0 +1,57 @@
+using Content.Server.Station.Systems;
+using Content.Shared.Administration;
+using Content.Shared.Roles;
+using Robust.Shared.Console;
+using Robust.Shared.Prototypes;
+
+
+namespace Content.Server.Administration.Commands;
+
+
+[AdminCommand(AdminFlags.Admin)]
+public sealed class GetStationAiNameCommand : IConsoleCommand
+{
+    [Dependency] private readonly IEntityManager _entManager = default!;
+    [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
+    [Dependency] private readonly StationSpawningSystem _spawning = default!;
+
+    private ProtoId<JobPrototype> _stationAIJob = "StationAi";
+
+    public string Command => "setstationainame";
+    public string Description => Loc.GetString("set-station-ai-name-command-description");
+    public string Help => Loc.GetString("set-station-ai-name-command-help-text", ("command", Command));
+
+    public void Execute(IConsoleShell shell, string argStr, string[] args)
+    {
+        if (args.Length != 1)
+        {
+            shell.WriteError(Loc.GetString("shell-wrong-arguments-number"));
+            return;
+        }
+
+        if (!int.TryParse(args[0], out var entInt))
+        {
+            shell.WriteLine(Loc.GetString("shell-entity-uid-must-be-number"));
+            return;
+        }
+
+        var netEntity = new NetEntity(entInt);
+
+        if (!_entManager.TryGetEntity(netEntity, out var target))
+        {
+            shell.WriteLine(Loc.GetString("shell-invalid-entity-id"));
+            return;
+        }
+
+        var hasStationAi = _prototypeManager.TryIndex(_stationAIJob, out var job);
+
+        if (!hasStationAi)
+        {
+            shell.WriteLine(Loc.GetString("set-station-ai-name-command-no-station-ai"));
+            return;
+        }
+
+        _spawning.EquipJobName(target.Value, job!);
+        shell.WriteLine(Loc.GetString("shell-command-success"));
+    }
+}

Content.Server/Clothing/Systems/LoadoutSystem.cs

@@ -2,6 +2,7 @@
 using Content.Server.GameTicking;
 using Content.Server.Paint;
 using Content.Server.Players.PlayTimeTracking;
+using Content.Server.Station.Systems;
 using Content.Shared.CCVar;
 using Content.Shared.Clothing.Loadouts.Prototypes;
 using Content.Shared.Clothing.Loadouts.Systems;
@@ -33,6 +34,7 @@ public sealed class LoadoutSystem : EntitySystem
     [Dependency] private readonly ISerializationManager _serialization = default!;
     [Dependency] private readonly IRobustRandom _random = default!;
     [Dependency] private readonly IComponentFactory _componentFactory = default!;
+    [Dependency] private readonly StationSpawningSystem _stationSpawning = default!;
 
 
     public override void Initialize()
@@ -43,10 +45,13 @@ public override void Initialize()
 
     private void OnPlayerSpawnComplete(PlayerSpawnCompleteEvent ev)
     {
-        if (ev.JobId == null ||
-            !_configurationManager.GetCVar(CCVars.GameLoadoutsEnabled))
+        if (ev.JobId == null
+            || !_protoMan.TryIndex<JobPrototype>(ev.JobId, out var job)
+            || !_configurationManager.GetCVar(CCVars.GameLoadoutsEnabled))
             return;
 
+        _stationSpawning.EquipJobName(ev.Mob, job);
+
         ApplyCharacterLoadout(
             ev.Mob,
             ev.JobId,
@@ -108,7 +113,6 @@ public void ApplyCharacterLoadout(
                 function.OnPlayerSpawn(uid, loadout.Item1, _componentFactory, EntityManager, _serialization);
         }
 
-
         // Pick the heirloom
         if (heirlooms.Any())
         {

Content.Shared/Roles/JobPrototype.cs

@@ -1,5 +1,6 @@
 using Content.Shared.Access;
 using Content.Shared.Customization.Systems;
+using Content.Shared.Dataset;
 using Content.Shared.Players.PlayTimeTracking;
 using Content.Shared.Roles;
 using Content.Shared.StatusIcon;
@@ -98,6 +99,13 @@ public sealed partial class JobPrototype : IPrototype
         [DataField("startingGear", customTypeSerializer: typeof(PrototypeIdSerializer<StartingGearPrototype>))]
         public string? StartingGear { get; private set; }
 
+        /// <summary>
+        ///     If this has a value, it will randomly set the entity name of the
+        ///     entity upon spawn based on the dataset.
+        /// </summary>
+        [DataField]
+        public ProtoId<LocalizedDatasetPrototype>? NameDataset;
+
         /// <summary>
         /// Use this to spawn in as a non-humanoid (borg, test subject, etc.)
         /// Starting gear will be ignored.

Content.Shared/Station/SharedStationSpawningSystem.cs

@@ -1,4 +1,5 @@
 using System.Linq;
+using Content.Shared.Dataset;
 using Content.Shared.Hands.Components;
 using Content.Shared.Hands.EntitySystems;
 using Content.Shared.Inventory;
@@ -18,9 +19,10 @@ public abstract class SharedStationSpawningSystem : EntitySystem
     [Dependency] protected readonly IPrototypeManager PrototypeManager = default!;
     [Dependency] private readonly IRobustRandom _random = default!;
     [Dependency] protected readonly InventorySystem InventorySystem = default!;
-    [Dependency] private   readonly SharedHandsSystem _handsSystem = default!;
-    [Dependency] private   readonly SharedStorageSystem _storage = default!;
-    [Dependency] private   readonly SharedTransformSystem _xformSystem = default!;
+    [Dependency] private readonly SharedHandsSystem _handsSystem = default!;
+    [Dependency] private readonly SharedStorageSystem _storage = default!;
+    [Dependency] private readonly SharedTransformSystem _xformSystem = default!;
+    [Dependency] private readonly MetaDataSystem _metadata = default!;
 
     private EntityQuery<HandsComponent> _handsQuery;
     private EntityQuery<InventoryComponent> _inventoryQuery;
@@ -39,11 +41,13 @@ public override void Initialize()
     /// <summary>
     /// Applies the role's name as applicable to the entity.
     /// </summary>
-    public void EquipRoleName(EntityUid entity, RoleLoadout loadout, RoleLoadoutPrototype roleProto)
+    public void EquipJobName(EntityUid entity, JobPrototype job)
     {
         string? name = null;
 
-        if (string.IsNullOrEmpty(name) && PrototypeManager.TryIndex(roleProto.NameDataset, out var nameData))
+        if (string.IsNullOrEmpty(name)
+            && job.NameDataset.HasValue
+            && PrototypeManager.TryIndex(job.NameDataset.Value, out var nameData))
         {
             name = Loc.GetString(_random.Pick(nameData.Values));
         }

Resources/Locale/en-US/administration/commands/set-station-ai-name-command.ftl

@@ -0,0 +1,3 @@
+set-station-ai-name-command-description = Sets a specific uid to be given a Station AI name at random.
+set-station-ai-name-command-help-text = Usage: {$command} <entityUid>
+set-station-ai-name-command-no-station-ai = No station AI job prototype found.

Resources/Prototypes/Roles/Jobs/Science/borg.yml

@@ -12,6 +12,7 @@
   icon: JobIconStationAi
   supervisors: job-supervisors-rd
   jobEntity: StationAiBrain
+  nameDataset: NamesAI
 
 - type: job
   id: Borg
@@ -24,4 +25,4 @@
   canBeAntag: false
   icon: JobIconBorg
   supervisors: job-supervisors-rd
-  jobEntity: PlayerBorgGeneric
+  jobEntity: PlayerBorgGeneric