63 lines
1.7 KiB
C#
63 lines
1.7 KiB
C#
using Content.Server.Objectives.Components;
|
|
using Content.Shared.Objectives.Components;
|
|
using Content.Shared.Mind;
|
|
using Content.Shared.Mind.Components;
|
|
|
|
namespace Content.Server.Objectives.Systems;
|
|
|
|
/// <summary>
|
|
/// Handles <see cref="CodeConditionComponent"/> progress and provides API for systems to use.
|
|
/// </summary>
|
|
public sealed class CodeConditionSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SharedMindSystem _mind = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<CodeConditionComponent, ObjectiveGetProgressEvent>(OnGetProgress);
|
|
}
|
|
|
|
private void OnGetProgress(Entity<CodeConditionComponent> ent, ref ObjectiveGetProgressEvent args)
|
|
{
|
|
args.Progress = ent.Comp.Completed ? 1f : 0f;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns whether an objective is completed.
|
|
/// </summary>
|
|
public bool IsCompleted(Entity<CodeConditionComponent?> ent)
|
|
{
|
|
if (!Resolve(ent, ref ent.Comp))
|
|
return false;
|
|
|
|
return ent.Comp.Completed;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets an objective's completed field.
|
|
/// </summary>
|
|
public void SetCompleted(Entity<CodeConditionComponent?> ent, bool completed = true)
|
|
{
|
|
if (!Resolve(ent, ref ent.Comp))
|
|
return;
|
|
|
|
ent.Comp.Completed = completed;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets a mob's objective to complete.
|
|
/// </summary>
|
|
public void SetCompleted(Entity<MindContainerComponent?> mob, string prototype, bool completed = true)
|
|
{
|
|
if (_mind.GetMind(mob, mob.Comp) is not {} mindId)
|
|
return;
|
|
|
|
if (!_mind.TryFindObjective(mindId, prototype, out var obj))
|
|
return;
|
|
|
|
SetCompleted(obj.Value, completed);
|
|
}
|
|
}
|