Re-organize all projects (#4166)
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
#nullable enable
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using Content.Shared.Administration;
|
||||
using Robust.Shared.Console;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
|
||||
namespace Content.Server.Administration.Commands
|
||||
{
|
||||
[AdminCommand(AdminFlags.Mapping)]
|
||||
public class FindEntitiesWithComponents : IConsoleCommand
|
||||
{
|
||||
public string Command => "findentitieswithcomponents";
|
||||
public string Description => "Finds entities with all of the specified components.";
|
||||
public string Help => $"{Command} <componentName1> <componentName2>...";
|
||||
|
||||
public void Execute(IConsoleShell shell, string argStr, string[] args)
|
||||
{
|
||||
if (args.Length == 0)
|
||||
{
|
||||
shell.WriteLine($"Invalid amount of arguments: {args.Length}.\n{Help}");
|
||||
return;
|
||||
}
|
||||
|
||||
var components = new List<Type>();
|
||||
var componentFactory = IoCManager.Resolve<IComponentFactory>();
|
||||
var invalidArgs = new List<string>();
|
||||
|
||||
foreach (var arg in args)
|
||||
{
|
||||
if (!componentFactory.TryGetRegistration(arg, out var registration))
|
||||
{
|
||||
invalidArgs.Add(arg);
|
||||
continue;
|
||||
}
|
||||
|
||||
components.Add(registration.Type);
|
||||
}
|
||||
|
||||
if (invalidArgs.Count > 0)
|
||||
{
|
||||
shell.WriteLine($"No component found for component names: {string.Join(", ", invalidArgs)}");
|
||||
return;
|
||||
}
|
||||
|
||||
var entityManager = IoCManager.Resolve<IEntityManager>();
|
||||
var query = new MultipleTypeEntityQuery(components);
|
||||
var entityIds = new HashSet<string>();
|
||||
|
||||
foreach (var entity in entityManager.GetEntities(query))
|
||||
{
|
||||
if (entity.Prototype == null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
entityIds.Add(entity.Prototype.ID);
|
||||
}
|
||||
|
||||
if (entityIds.Count == 0)
|
||||
{
|
||||
shell.WriteLine($"No entities found with components {string.Join(", ", args)}.");
|
||||
return;
|
||||
}
|
||||
|
||||
shell.WriteLine($"{entityIds.Count} entities found:\n{string.Join("\n", entityIds)}");
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user