64 lines
2.3 KiB
C#
64 lines
2.3 KiB
C#
using Content.Server.Administration;
|
|
using Content.Server.AI.Components;
|
|
using Content.Server.AI.Utility;
|
|
using Content.Server.AI.Utility.AiLogic;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.Movement.Components;
|
|
using Robust.Shared.Console;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
|
|
namespace Content.Server.AI.Commands
|
|
{
|
|
[AdminCommand(AdminFlags.Fun)]
|
|
public class AddAiCommand : IConsoleCommand
|
|
{
|
|
public string Command => "addai";
|
|
public string Description => "Add an ai component with a given processor to an entity.";
|
|
public string Help => "Usage: addai <entityId> <behaviorSet1> <behaviorSet2>..."
|
|
+ "\n entityID: Uid of entity to add the AiControllerComponent to. Open its VV menu to find this."
|
|
+ "\n behaviorSet: Name of a behaviorset to add to the component on initialize.";
|
|
|
|
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
|
{
|
|
if(args.Length < 1)
|
|
{
|
|
shell.WriteLine("Wrong number of args.");
|
|
return;
|
|
}
|
|
|
|
var entId = new EntityUid(int.Parse(args[0]));
|
|
|
|
if (!IoCManager.Resolve<IEntityManager>().TryGetEntity(entId, out var ent))
|
|
{
|
|
shell.WriteLine($"Unable to find entity with uid {entId}");
|
|
return;
|
|
}
|
|
|
|
if (IoCManager.Resolve<IEntityManager>().HasComponent<AiControllerComponent>(ent.Uid))
|
|
{
|
|
shell.WriteLine("Entity already has an AI component.");
|
|
return;
|
|
}
|
|
|
|
// TODO: IMover refffaaccctttooorrr
|
|
if (IoCManager.Resolve<IEntityManager>().HasComponent<IMoverComponent>(ent.Uid))
|
|
{
|
|
IoCManager.Resolve<IEntityManager>().RemoveComponent<IMoverComponent>(ent.Uid);
|
|
}
|
|
|
|
var comp = IoCManager.Resolve<IEntityManager>().AddComponent<UtilityAi>(ent);
|
|
var behaviorManager = IoCManager.Resolve<INpcBehaviorManager>();
|
|
|
|
for (var i = 1; i < args.Length; i++)
|
|
{
|
|
var bSet = args[i];
|
|
behaviorManager.AddBehaviorSet(comp, bSet, false);
|
|
}
|
|
|
|
behaviorManager.RebuildActions(comp);
|
|
shell.WriteLine("AI component added.");
|
|
}
|
|
}
|
|
}
|