Access groups + aghost all access (#6671)

This commit is contained in:
mirrorcult
2022-02-11 19:01:14 -07:00
committed by GitHub
parent 47f6f88fdc
commit aa2727c84d
19 changed files with 160 additions and 82 deletions

View 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!;
}

View File

@@ -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();
}
}

View File

@@ -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;
}
}
}