using System.Collections.Generic; using System.Linq; using System.Text.RegularExpressions; using System.Threading.Tasks; using Content.Server.Tools.Components; using Content.Server.UserInterface; using Content.Shared.Configurable; using Content.Shared.Interaction; using Content.Shared.Tool; using Robust.Server.GameObjects; using Robust.Shared.GameObjects; using Robust.Shared.Serialization; using Robust.Shared.Serialization.Manager.Attributes; using Robust.Shared.ViewVariables; namespace Content.Server.Configurable { [RegisterComponent] [ComponentReference(typeof(SharedConfigurationComponent))] public class ConfigurationComponent : SharedConfigurationComponent, IInteractUsing, ISerializationHooks { [ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(ConfigurationUiKey.Key); [DataField("keys")] private List _keys = new(); [ViewVariables] private readonly Dictionary _config = new(); [DataField("validation")] private readonly Regex _validation = new ("^[a-zA-Z0-9 ]*$", RegexOptions.Compiled); void ISerializationHooks.BeforeSerialization() { _keys = _config.Keys.ToList(); } void ISerializationHooks.AfterDeserialization() { foreach (var key in _keys) { _config.Add(key, string.Empty); } } protected override void OnAdd() { base.OnAdd(); if (UserInterface != null) { UserInterface.OnReceiveMessage += UserInterfaceOnReceiveMessage; } } protected override void OnRemove() { base.OnRemove(); if (UserInterface != null) { UserInterface.OnReceiveMessage -= UserInterfaceOnReceiveMessage; } } public string? GetConfig(string name) { return _config.GetValueOrDefault(name); } protected override void Startup() { base.Startup(); UpdateUserInterface(); } async Task IInteractUsing.InteractUsing(InteractUsingEventArgs eventArgs) { if (UserInterface == null || !eventArgs.User.TryGetComponent(out ActorComponent? actor)) return false; if (!eventArgs.Using.TryGetComponent(out var tool)) return false; if (!await tool.UseTool(eventArgs.User, Owner, 0.2f, ToolQuality.Multitool)) return false; OpenUserInterface(actor); return true; } private void UserInterfaceOnReceiveMessage(ServerBoundUserInterfaceMessage serverMsg) { var message = serverMsg.Message; var config = new Dictionary(_config); if (message is ConfigurationUpdatedMessage msg) { foreach (var key in config.Keys) { var value = msg.Config.GetValueOrDefault(key); if (value == null || _validation != null && !_validation.IsMatch(value) && value != string.Empty) continue; _config[key] = value; } SendMessage(new ConfigUpdatedComponentMessage(config)); } } private void UpdateUserInterface() { UserInterface?.SetState(new ConfigurationBoundUserInterfaceState(_config)); } public void OpenUserInterface(ActorComponent actor) { UpdateUserInterface(); UserInterface?.Open(actor.PlayerSession); UserInterface?.SendMessage(new ValidationUpdateMessage(_validation.ToString()), actor.PlayerSession); } private static void FillConfiguration(List list, Dictionary configuration, T value){ for (var index = 0; index < list.Count; index++) { configuration.Add(list[index], value); } } } }