refactor: server CP14KnowledgeSystem

Content.Server/_CP14/Knowledge/CP14KnowledgeAdminUtilitiesSystem.cs

@@ -0,0 +1,71 @@
+using Content.Shared._CP14.Knowledge.Components;
+using Content.Shared._CP14.Knowledge.Prototypes;
+using Content.Shared.Administration;
+using Content.Shared.Administration.Managers;
+using Content.Shared.Database;
+using Content.Shared.Verbs;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server._CP14.Knowledge;
+
+// TODO: Add UI
+public sealed class CP14KnowledgeAdminUtilitiesSystem : EntitySystem
+{
+    [Dependency] private readonly ISharedAdminManager _admin = default!;
+    [Dependency] private readonly IPrototypeManager _prototype = default!;
+    [Dependency] private readonly CP14KnowledgeSystem _knowledge = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<CP14KnowledgeStorageComponent, GetVerbsEvent<Verb>>(AddKnowledgeAdminVerb);
+    }
+
+    private void AddKnowledgeAdminVerb(Entity<CP14KnowledgeStorageComponent> entity, ref GetVerbsEvent<Verb> args)
+    {
+        if (!_admin.HasAdminFlag(args.User, AdminFlags.Admin))
+            return;
+
+        // Remove knowledge
+        foreach (var knowledge in entity.Comp.Knowledge)
+        {
+            if (!_prototype.TryIndex(knowledge, out var indexedKnowledge))
+                continue;
+
+            args.Verbs.Add(new Verb
+            {
+                Text = $"{Loc.GetString(indexedKnowledge.Name)}",
+                Message = Loc.GetString(indexedKnowledge.Desc),
+                Category = VerbCategory.CP14KnowledgeRemove,
+                Act = () =>
+                {
+                    _knowledge.TryRemove(entity.Owner, knowledge);
+                },
+                Impact = LogImpact.High,
+            });
+        }
+
+        // Add knowledge
+        foreach (var knowledge in _prototype.EnumeratePrototypes<CP14KnowledgePrototype>())
+        {
+            // An entity with CP14AllKnowingComponent can't be taught anything,
+            // it already knows everything
+            if (_knowledge.HasKnowledge(entity.Owner, knowledge.ID))
+                continue;
+
+            args.Verbs.Add(new Verb
+            {
+                Text = $"{Loc.GetString(knowledge.Name)}",
+                Message = Loc.GetString(knowledge.Desc),
+                Category = VerbCategory.CP14KnowledgeAdd,
+                Act = () =>
+                {
+                    _knowledge.TryAdd(entity.Owner, knowledge, true);
+                },
+                Impact = LogImpact.High,
+            });
+        }
+    }
+
+}