RobustToolbox/Robust.Shared/ViewVariables/ViewVariablesUtility.cs

using System;
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using System.Reflection;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.Utility;

namespace Robust.Shared.ViewVariables
{
    public static class ViewVariablesUtility
    {
        /// <summary>
        ///     Check whether this type has any fields or properties with the <see cref="ViewVariablesAttribute"/>,
        ///     i.e. whether there are any members that are visible in a VV window.
        /// </summary>
        /// <remarks>
        ///     This is quite an expensive operation, don't use it lightly.
        /// </remarks>
        /// <param name="type">The type to check.</param>
        /// <returns>True if there are members with the attribute, false otherwise.</returns>
        public static bool TypeHasVisibleMembers(Type type)
        {
            return type.GetAllFields().Cast<MemberInfo>().Concat(type.GetAllProperties())
                .Any(f => TryGetViewVariablesAccess(f, out _));
        }

        /// <summary>
        /// Gets the <see cref="VVAccess"/> defined for the member, if defined.
        /// </summary>
        /// <param name="info">The member to check access for.</param>
        /// <param name="access">The found access. Will be null if no access is defined</param>
        /// <returns>True if access is defined, false if not.</returns>
        public static bool TryGetViewVariablesAccess(MemberInfo info, [NotNullWhen(true)] out VVAccess? access)
        {
            if (info.TryGetCustomAttribute<ViewVariablesAttribute>(out var vv))
            {
                access = vv.Access;
                return true;
            }

            if (info.HasCustomAttribute<DataFieldAttribute>() || info.HasCustomAttribute<IncludeDataFieldAttribute>())
            {
                access = VVAccess.ReadWrite;
                return true;
            }

            access = null;
            return false;
        }
    }
}