45 lines
1.6 KiB
C#
45 lines
1.6 KiB
C#
using Content.Server.Objectives.Interfaces;
|
|
using Content.Shared.Random.Helpers;
|
|
using Content.Shared.Random;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.Objectives
|
|
{
|
|
public sealed class ObjectivesManager : IObjectivesManager
|
|
{
|
|
[Dependency] private IPrototypeManager _prototypeManager = default!;
|
|
[Dependency] private IRobustRandom _random = default!;
|
|
|
|
public ObjectivePrototype? GetRandomObjective(Mind.Mind mind, string objectiveGroupProto)
|
|
{
|
|
if (!_prototypeManager.TryIndex<WeightedRandomPrototype>(objectiveGroupProto, out var groups))
|
|
{
|
|
Logger.Error("Tried to get a random objective, but can't index WeightedRandomPrototype " + objectiveGroupProto);
|
|
return null;
|
|
}
|
|
|
|
// yeah the old 'preventing infinite loops' thing wasn't super elegant either and it mislead people on what exactly it did
|
|
var tries = 0;
|
|
while (tries < 20)
|
|
{
|
|
var groupName = groups.Pick(_random);
|
|
|
|
if (!_prototypeManager.TryIndex<WeightedRandomPrototype>(groupName, out var group))
|
|
{
|
|
Logger.Error("Couldn't index objective group prototype" + groupName);
|
|
return null;
|
|
}
|
|
|
|
if (_prototypeManager.TryIndex<ObjectivePrototype>(group.Pick(_random), out var objective)
|
|
&& objective.CanBeAssigned(mind))
|
|
return objective;
|
|
else
|
|
tries++;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|
|
}
|