Access groups + aghost all access (#6671)
This commit is contained in:
19
Content.Shared/Access/AccessGroupPrototype.cs
Normal file
19
Content.Shared/Access/AccessGroupPrototype.cs
Normal file
@@ -0,0 +1,19 @@
|
||||
using Content.Shared.Access.Components;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.Set;
|
||||
|
||||
namespace Content.Shared.Access;
|
||||
|
||||
/// <summary>
|
||||
/// Contains a list of access tags that are part of this group.
|
||||
/// Used by <see cref="AccessComponent"/> to avoid boilerplate.
|
||||
/// </summary>
|
||||
[Prototype("accessGroup")]
|
||||
public sealed class AccessGroupPrototype : IPrototype
|
||||
{
|
||||
[DataField("id", required: true)]
|
||||
public string ID { get; } = default!;
|
||||
|
||||
[DataField("tags", required: true, customTypeSerializer:typeof(PrototypeIdHashSetSerializer<AccessLevelPrototype>))]
|
||||
public HashSet<string> Tags = default!;
|
||||
}
|
||||
@@ -13,10 +13,12 @@ namespace Content.Shared.Access.Components
|
||||
/// </summary>
|
||||
[RegisterComponent]
|
||||
[Friend(typeof(AccessSystem))]
|
||||
public class AccessComponent : Component
|
||||
public sealed class AccessComponent : Component
|
||||
{
|
||||
[DataField("tags", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<AccessLevelPrototype>))]
|
||||
[ViewVariables]
|
||||
public HashSet<string> Tags = new();
|
||||
|
||||
[DataField("groups", customTypeSerializer: typeof(PrototypeIdHashSetSerializer<AccessGroupPrototype>))]
|
||||
public HashSet<string> Groups = new();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,11 +1,33 @@
|
||||
using Content.Shared.Access.Components;
|
||||
using Robust.Shared.GameObjects;
|
||||
using System.Collections.Generic;
|
||||
using Robust.Shared.Prototypes;
|
||||
|
||||
namespace Content.Shared.Access.Systems
|
||||
{
|
||||
public class AccessSystem : EntitySystem
|
||||
public sealed class AccessSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<AccessComponent, ComponentInit>(OnAccessInit);
|
||||
}
|
||||
|
||||
private void OnAccessInit(EntityUid uid, AccessComponent component, ComponentInit args)
|
||||
{
|
||||
// Add all tags in groups to the list of tags.
|
||||
foreach (var group in component.Groups)
|
||||
{
|
||||
if (!_prototypeManager.TryIndex<AccessGroupPrototype>(group, out var proto))
|
||||
continue;
|
||||
|
||||
component.Tags.UnionWith(proto.Tags);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Replaces the set of access tags we have with the provided set.
|
||||
/// </summary>
|
||||
@@ -20,5 +42,21 @@ namespace Content.Shared.Access.Systems
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public bool TryAddGroups(EntityUid uid, IEnumerable<string> newGroups, AccessComponent? access = null)
|
||||
{
|
||||
if (!Resolve(uid, ref access))
|
||||
return false;
|
||||
|
||||
foreach (var group in newGroups)
|
||||
{
|
||||
if (!_prototypeManager.TryIndex<AccessGroupPrototype>(group, out var proto))
|
||||
continue;
|
||||
|
||||
access.Tags.UnionWith(proto.Tags);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user