Add completions to addobjective and localize it (#30456)

* Add completions to addobjective and localise it

* Cleanup

* Fix

* Make a manager to handle subscribtion completion options

This is so we can unsubscribe to prototype reloads properly

* Convert the manager into a system

* Move the system into the systems folder

I forgor

* Merge CompletionsSystem into ObjectivesSystem
This commit is contained in:
nikthechampiongr
2024-09-08 07:28:43 +00:00
committed by GitHub
parent b2007bce19
commit 4b8bedaeef
5 changed files with 109 additions and 39 deletions

View File

@@ -1,56 +1,73 @@
using Content.Server.Administration;
using System.Linq;
using Content.Server.Administration;
using Content.Shared.Administration;
using Content.Shared.Mind;
using Content.Shared.Objectives.Components;
using Content.Shared.Prototypes;
using Robust.Server.Player;
using Robust.Shared.Console;
using Robust.Shared.Prototypes;
namespace Content.Server.Objectives.Commands
namespace Content.Server.Objectives.Commands;
[AdminCommand(AdminFlags.Admin)]
public sealed class AddObjectiveCommand : LocalizedEntityCommands
{
[AdminCommand(AdminFlags.Admin)]
public sealed class AddObjectiveCommand : IConsoleCommand
[Dependency] private readonly IPlayerManager _players = default!;
[Dependency] private readonly IPrototypeManager _prototypes = default!;
[Dependency] private readonly SharedMindSystem _mind = default!;
[Dependency] private readonly ObjectivesSystem _objectives = default!;
public override string Command => "addobjective";
public override void Execute(IConsoleShell shell, string argStr, string[] args)
{
[Dependency] private readonly IEntityManager _entityManager = default!;
public string Command => "addobjective";
public string Description => "Adds an objective to the player's mind.";
public string Help => "addobjective <username> <objectiveID>";
public void Execute(IConsoleShell shell, string argStr, string[] args)
if (args.Length != 2)
{
if (args.Length != 2)
{
shell.WriteLine("Expected exactly 2 arguments.");
return;
}
shell.WriteError(Loc.GetString(Loc.GetString("cmd-addobjective-invalid-args")));
return;
}
var mgr = IoCManager.Resolve<IPlayerManager>();
if (!mgr.TryGetSessionByUsername(args[0], out var data))
{
shell.WriteLine("Can't find the playerdata.");
return;
}
if (!_players.TryGetSessionByUsername(args[0], out var data))
{
shell.WriteError(Loc.GetString("cmd-addobjective-player-not-found"));
return;
}
var minds = _entityManager.System<SharedMindSystem>();
if (!minds.TryGetMind(data, out var mindId, out var mind))
{
shell.WriteLine("Can't find the mind.");
return;
}
if (!_mind.TryGetMind(data, out var mindId, out var mind))
{
shell.WriteError(Loc.GetString("cmd-addobjective-mind-not-found"));
return;
}
if (!IoCManager.Resolve<IPrototypeManager>()
.TryIndex<EntityPrototype>(args[1], out var proto) ||
!proto.TryGetComponent<ObjectiveComponent>(out _))
{
shell.WriteLine($"Can't find matching objective prototype {args[1]}");
return;
}
if (!_prototypes.TryIndex<EntityPrototype>(args[1], out var proto) ||
!proto.HasComponent<ObjectiveComponent>())
{
shell.WriteError(Loc.GetString("cmd-addobjective-objective-not-found", ("obj", args[1])));
return;
}
if (!minds.TryAddObjective(mindId, mind, args[1]))
{
// can fail for other reasons so dont pretend to be right
shell.WriteLine("Failed to add the objective. Maybe requirements dont allow that objective to be added.");
}
if (!_mind.TryAddObjective(mindId, mind, args[1]))
{
// can fail for other reasons so dont pretend to be right
shell.WriteError(Loc.GetString("cmd-addobjective-adding-failed"));
}
}
public override CompletionResult GetCompletion(IConsoleShell shell, string[] args)
{
if (args.Length == 1)
{
var options = _players.Sessions.OrderBy(c => c.Name).Select(c => c.Name).ToArray();
return CompletionResult.FromHintOptions(options, Loc.GetString("cmd-addobjective-player-completion"));
}
if (args.Length != 2)
return CompletionResult.Empty;
return CompletionResult.FromHintOptions(
_objectives.Objectives(),
Loc.GetString(Loc.GetString("cmd-add-objective-obj-completion")));
}
}