d4e9049576
* Added reset all (non-free) skills * I don't know why I didn't do this the first time * Formatting fixes * Removed old code and made removal function iterate in reverse order
109 lines
3.2 KiB
C#
109 lines
3.2 KiB
C#
using System.Linq;
|
|
using Content.Shared._CP14.Skill.Components;
|
|
using Content.Shared._CP14.Skill.Prototypes;
|
|
using Content.Shared.Administration;
|
|
using Content.Shared.Administration.Managers;
|
|
using Content.Shared.FixedPoint;
|
|
using Content.Shared.Verbs;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Shared._CP14.Skill;
|
|
|
|
public abstract partial class CP14SharedSkillSystem
|
|
{
|
|
[Dependency] private readonly ISharedAdminManager _admin = default!;
|
|
[Dependency] private readonly IPrototypeManager _proto = default!;
|
|
|
|
private IEnumerable<CP14SkillPrototype>? _allSkills;
|
|
private IEnumerable<CP14SkillTreePrototype>? _allTrees;
|
|
private void InitializeAdmin()
|
|
{
|
|
SubscribeLocalEvent<CP14SkillStorageComponent, GetVerbsEvent<Verb>>(OnGetAdminVerbs);
|
|
|
|
SubscribeLocalEvent<PrototypesReloadedEventArgs>(OnPrototypeReloaded);
|
|
|
|
UpdateCachedSkill();
|
|
}
|
|
|
|
private void OnPrototypeReloaded(PrototypesReloadedEventArgs ev)
|
|
{
|
|
if (!ev.WasModified<CP14SkillPrototype>())
|
|
return;
|
|
|
|
UpdateCachedSkill();
|
|
}
|
|
|
|
private void UpdateCachedSkill()
|
|
{
|
|
_allSkills = _proto.EnumeratePrototypes<CP14SkillPrototype>();
|
|
_allTrees = _proto.EnumeratePrototypes<CP14SkillTreePrototype>();
|
|
}
|
|
|
|
|
|
private void OnGetAdminVerbs(Entity<CP14SkillStorageComponent> ent, ref GetVerbsEvent<Verb> args)
|
|
{
|
|
if (!_admin.HasAdminFlag(args.User, AdminFlags.Admin))
|
|
return;
|
|
|
|
if (_allSkills is null || _allTrees is null)
|
|
return;
|
|
|
|
var target = args.Target;
|
|
|
|
//Add Skill
|
|
foreach (var skill in _allSkills)
|
|
{
|
|
if (ent.Comp.LearnedSkills.Contains(skill))
|
|
continue;
|
|
|
|
var name = Loc.GetString(GetSkillName(skill));
|
|
args.Verbs.Add(new Verb
|
|
{
|
|
Text = name,
|
|
Message = name + ": " + Loc.GetString(GetSkillDescription(skill)),
|
|
Category = VerbCategory.CP14AdminSkillAdd,
|
|
Icon = skill.Icon,
|
|
Act = () =>
|
|
{
|
|
TryAddSkill(target, skill);
|
|
},
|
|
});
|
|
}
|
|
|
|
//Remove Skill
|
|
foreach (var skill in ent.Comp.LearnedSkills)
|
|
{
|
|
if (!_proto.TryIndex(skill, out var indexedSkill))
|
|
continue;
|
|
|
|
var name = Loc.GetString(GetSkillName(skill));
|
|
args.Verbs.Add(new Verb
|
|
{
|
|
Text = name,
|
|
Message = name + ": " + Loc.GetString(GetSkillDescription(skill)),
|
|
Category = VerbCategory.CP14AdminSkillRemove,
|
|
Icon = indexedSkill.Icon,
|
|
Act = () =>
|
|
{
|
|
TryRemoveSkill(target, skill);
|
|
},
|
|
});
|
|
}
|
|
|
|
//Reset/Remove All Skills
|
|
args.Verbs.Add(new Verb
|
|
{
|
|
Text = "Reset skills",
|
|
Message = "Remove all learned skills",
|
|
Category = VerbCategory.CP14AdminSkillRemove,
|
|
Icon = new SpriteSpecifier.Rsi(new("/Textures/_CP14/Interface/Misc/reroll.rsi"), "reroll"),
|
|
Act = () =>
|
|
{
|
|
TryResetSkills(target);
|
|
},
|
|
});
|
|
|
|
}
|
|
}
|