src/QueryCat.Backend.Core/Fetch/ClassRowsFrameBuilderExtensions.cs - develop - krasninja/querycat

src/QueryCat.Backend.Core/Fetch/ClassRowsFrameBuilderExtensions.cs

43 строки2 KB

Ivan Kozhin

Allow to select from CLR enumerables

15 апр 2024, 09:34

15 апр 2024, 09:34b1a8d0e

using System.ComponentModel;
using System.Diagnostics.CodeAnalysis;
using System.Reflection;
using QueryCat.Backend.Core.Types;

namespace QueryCat.Backend.Core.Fetch;

/// <summary>
/// Extensions for <see cref="ClassRowsFrameBuilder{TClass}" />.
/// </summary>
public static class ClassRowsFrameBuilderExtensions
{
    /// <summary>
    /// Add all public properties as columns.
    /// </summary>
    /// <returns>The instance of <see cref="ClassRowsFrameBuilder{TClass}" />.</returns>
    public static ClassRowsFrameBuilder<TClass> AddPublicProperties
        <[DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicProperties)] TClass>(
        this ClassRowsFrameBuilder<TClass> builder) where TClass : class
    {
        AddPublicProperties(builder, null, out _);
        return builder;
    }

    internal static void AddPublicProperties
        <[DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicProperties)] TClass>(
        this ClassRowsFrameBuilder<TClass> builder,
        [DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicProperties)] Type? type,
        out List<PropertyInfo> properties) where TClass : class
    {
        var props = (type ?? typeof(TClass)).GetProperties().Where(p => p.CanRead);
        properties = new List<PropertyInfo>();
        foreach (var propertyInfo in props)
        {
            var description = propertyInfo.GetCustomAttributes<DescriptionAttribute>()
                .Select(a => a.Description).FirstOrDefault();
            var dataType = Converter.ConvertFromSystem(propertyInfo.PropertyType);
            builder.AddProperty(propertyInfo.Name, dataType,
                obj => VariantValue.CreateFromObject(propertyInfo.GetValue(obj)), description);
            properties.Add(propertyInfo);
        }
    }
}