74 lines
2.4 KiB
C#
74 lines
2.4 KiB
C#
using Content.Shared.Mind;
|
|
using Content.Shared.Roles;
|
|
using Content.Shared.Roles.Jobs;
|
|
using Content.Shared.Store;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.Set;
|
|
|
|
namespace Content.Server.Store.Conditions;
|
|
|
|
/// <summary>
|
|
/// Allows a store entry to be filtered out based on the user's job.
|
|
/// Supports both blacklists and whitelists
|
|
/// </summary>
|
|
public sealed partial class BuyerDepartmentCondition : ListingCondition
|
|
{
|
|
/// <summary>
|
|
/// A whitelist of department prototypes that can purchase this listing. Only one needs to be found.
|
|
/// </summary>
|
|
[DataField("whitelist", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<DepartmentPrototype>))]
|
|
public HashSet<string>? Whitelist;
|
|
|
|
/// <summary>
|
|
/// A blacklist of department prototypes that can purchase this listing. Only one needs to be found.
|
|
/// </summary>
|
|
[DataField("blacklist", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<DepartmentPrototype>))]
|
|
public HashSet<string>? Blacklist;
|
|
|
|
public override bool Condition(ListingConditionArgs args)
|
|
{
|
|
var prototypeManager = IoCManager.Resolve<IPrototypeManager>();
|
|
|
|
var ent = args.EntityManager;
|
|
var minds = ent.System<SharedMindSystem>();
|
|
|
|
// this is for things like surplus crate
|
|
if (!minds.TryGetMind(args.Buyer, out var mindId, out _))
|
|
return true;
|
|
|
|
var jobs = ent.System<SharedJobSystem>();
|
|
jobs.MindTryGetJob(mindId, out var job, out _);
|
|
|
|
if (Blacklist != null && job?.Prototype != null)
|
|
{
|
|
foreach (var department in prototypeManager.EnumeratePrototypes<DepartmentPrototype>())
|
|
{
|
|
if (department.Roles.Contains(job.Prototype.Value) && Blacklist.Contains(department.ID))
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (Whitelist != null)
|
|
{
|
|
var found = false;
|
|
|
|
if (job?.Prototype != null)
|
|
{
|
|
foreach (var department in prototypeManager.EnumeratePrototypes<DepartmentPrototype>())
|
|
{
|
|
if (department.Roles.Contains(job.Prototype.Value) && Whitelist.Contains(department.ID))
|
|
{
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!found)
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|