RobustToolbox/Robust.Client/ViewVariables/ViewVariablesInstance.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Robust.Client.Interfaces.ResourceManagement;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Client.UserInterface.CustomControls;
using Robust.Client.ViewVariables.Traits;
using Robust.Shared.Interfaces.Serialization;
using Robust.Shared.Maths;
using Robust.Shared.Utility;
using Robust.Shared.ViewVariables;

namespace Robust.Client.ViewVariables
{
    /// <summary>
    ///     Controls the behavior of a VV window.
    /// </summary>
    internal abstract class ViewVariablesInstance
    {
        public readonly IViewVariablesManagerInternal ViewVariablesManager;
        protected readonly IRobustSerializer _robustSerializer;

        protected ViewVariablesInstance(IViewVariablesManagerInternal vvm, IRobustSerializer robustSerializer)
        {
            ViewVariablesManager = vvm;
            _robustSerializer = robustSerializer;
        }

        /// <summary>
        ///     Initializes this instance to work on a local object.
        /// </summary>
        /// <param name="window">The window to initialize by adding GUI components.</param>
        /// <param name="obj">The object that is being VV'd</param>
        public abstract void Initialize(SS14Window window, object obj);

        /// <summary>
        ///     Initializes this instance to work on a remote object.
        ///     This is called when the view variables manager has already made a session to the remote object.
        /// </summary>
        /// <param name="window">The window to initialize by adding GUI components.</param>
        /// <param name="blob">The data blob sent by the server for this remote object.</param>
        /// <param name="session">The session connecting to the remote object.</param>
        public virtual void Initialize(SS14Window window, ViewVariablesBlobMetadata blob, ViewVariablesRemoteSession session)
        {
            throw new NotSupportedException();
        }

        /// <summary>
        ///     Invoked to "clean up" the instance, such as closing remote sessions.
        /// </summary>
        public virtual void Close()
        {
        }

        protected internal static IEnumerable<IGrouping<Type, Control>> LocalPropertyList(object obj, IViewVariablesManagerInternal vvm,
            IRobustSerializer robustSerializer)
        {
            var styleOther = false;
            var type = obj.GetType();

            var members = new List<(MemberInfo, VVAccess, object? value, Action<object> onValueChanged, Type)>();

            foreach (var fieldInfo in type.GetAllFields())
            {
                var attr = fieldInfo.GetCustomAttribute<ViewVariablesAttribute>();
                if (attr == null)
                {
                    continue;
                }

                members.Add((fieldInfo, attr.Access, fieldInfo.GetValue(obj), v => fieldInfo.SetValue(obj, v),
                    fieldInfo.FieldType));
            }

            foreach (var propertyInfo in type.GetAllProperties())
            {
                var attr = propertyInfo.GetCustomAttribute<ViewVariablesAttribute>();
                if (attr == null)
                {
                    continue;
                }

                if (!propertyInfo.IsBasePropertyDefinition())
                {
                    continue;
                }

                members.Add((propertyInfo, attr.Access, propertyInfo.GetValue(obj),
                    v => propertyInfo.GetSetMethod(true)!.Invoke(obj, new[] {v}), propertyInfo.PropertyType));
            }

            var groupedSorted = members
                .OrderBy(p => p.Item1.Name)
                .GroupBy(p => p.Item1.DeclaringType!, tuple =>
                {
                    var (memberInfo, access, value, onValueChanged, memberType) = tuple;
                    var data = new ViewVariablesBlobMembers.MemberData
                    {
                        Editable = access == VVAccess.ReadWrite,
                        Name = memberInfo.Name,
                        Type = memberType.AssemblyQualifiedName,
                        TypePretty = TypeAbbreviation.Abbreviate(memberType),
                        Value = value
                    };

                    var propertyEdit = new ViewVariablesPropertyControl(vvm, robustSerializer);
                    propertyEdit.SetStyle(styleOther = !styleOther);
                    var editor = propertyEdit.SetProperty(data);
                    editor.OnValueChanged += onValueChanged;
                    return propertyEdit;
                })
                .OrderByDescending(p => p.Key, TypeHelpers.TypeInheritanceComparer);

            return groupedSorted;
        }

        protected static Control MakeTopBar(string top, string bottom)
        {
            if (top == bottom)
            {
                return new Label {Text = top, ClipText = true};
            }

            //var smallFont =
            //    new VectorFont(IoCManager.Resolve<IResourceCache>().GetResource<FontResource>("/Fonts/CALIBRI.TTF"),
            //        10);

            // Custom ToString() implementation.
            var headBox = new VBoxContainer {SeparationOverride = 0};
            headBox.AddChild(new Label {Text = top, ClipText = true});
            headBox.AddChild(new Label
            {
                Text = bottom,
            //    FontOverride = smallFont,
                FontColorOverride = Color.DarkGray,
                ClipText = true
            });
            return headBox;
        }
    }
}