mirror of https://github.com/dahall/Vanara.git
124 lines
6.0 KiB
C#
124 lines
6.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Vanara.Extensions;
|
|
// ReSharper disable MemberCanBePrivate.Global
|
|
// ReSharper disable UnusedMember.Global
|
|
|
|
namespace Vanara.InteropServices
|
|
{
|
|
/// <summary>Actions that can be taken with a corresponding type.</summary>
|
|
[Flags]
|
|
public enum CorrepsondingAction
|
|
{
|
|
/// <summary>No actions may be taken.</summary>
|
|
None = 0,
|
|
/// <summary>The type can be retrieved.</summary>
|
|
Get = 1,
|
|
/// <summary>The type can be set.</summary>
|
|
Set = 2,
|
|
/// <summary>The type can be retrieved and set.</summary>
|
|
GetSet = 3,
|
|
/// <summary>Throw a <see cref="Exception"/> if this enumeration value is used.</summary>
|
|
Exception = 4
|
|
}
|
|
|
|
/// <summary>
|
|
/// Attribute for enum values that provides information about corresponding types and related actions. Useful for Get/Set methods that use an enumeration
|
|
/// value to determine the type to get or set.
|
|
/// </summary>
|
|
/// <seealso cref="System.Attribute"/>
|
|
[AttributeUsage(AttributeTargets.Field | AttributeTargets.Class, AllowMultiple = true)]
|
|
public class CorrespondingTypeAttribute : Attribute
|
|
{
|
|
/// <summary>Initializes a new instance of the <see cref="CorrespondingTypeAttribute"/> class.</summary>
|
|
/// <param name="typeRef">The type that corresponds to this enumeration value.</param>
|
|
/// <param name="action">The actions allowed for the type.</param>
|
|
public CorrespondingTypeAttribute(Type typeRef, CorrepsondingAction action = CorrepsondingAction.GetSet)
|
|
{
|
|
TypeRef = typeRef;
|
|
Action = action;
|
|
}
|
|
|
|
/// <summary>Initializes a new instance of the <see cref="CorrespondingTypeAttribute"/> class.</summary>
|
|
/// <param name="action">The actions allowed for the type.</param>
|
|
public CorrespondingTypeAttribute(CorrepsondingAction action)
|
|
{
|
|
Action = action;
|
|
}
|
|
|
|
/// <summary>Gets the action allowed for the type.</summary>
|
|
/// <value>The action allowed for the type.</value>
|
|
public CorrepsondingAction Action { get; }
|
|
|
|
/// <summary>Gets the type that corresponds to this enumeration value.</summary>
|
|
/// <value>The type that corresponds to this enumeration value.</value>
|
|
public Type TypeRef { get; }
|
|
|
|
/// <summary>Determines whether this instance can get the type for the specified enum value or class.</summary>
|
|
/// <param name="value">The enumeration value or class instance.</param>
|
|
/// <param name="typeRef">The type supplied by the user to validate.</param>
|
|
/// <returns><c>true</c> if this instance can get the specified type; otherwise, <c>false</c>.</returns>
|
|
public static bool CanGet(object value, Type typeRef)
|
|
{
|
|
return GetAttrForObj(value).Any(a => a.Action.IsFlagSet(CorrepsondingAction.Get) && a.TypeRef == typeRef);
|
|
}
|
|
|
|
/// <summary>Determines whether this type can get the specified reference type.</summary>
|
|
/// <param name="type">The class type.</param>
|
|
/// <param name="typeRef">The type supplied by the user to validate.</param>
|
|
/// <returns><c>true</c> if this type can get the specified reference type; otherwise, <c>false</c>.</returns>
|
|
public static bool CanGet(Type type, Type typeRef)
|
|
{
|
|
return GetAttrForType(type).Any(a => a.Action.IsFlagSet(CorrepsondingAction.Get) && a.TypeRef == typeRef);
|
|
}
|
|
|
|
/// <summary>Determines whether this instance can set the type for the specified enum value or class.</summary>
|
|
/// <param name="value">The enumeration value or class instance.</param>
|
|
/// <param name="typeRef">The type supplied by the user to validate.</param>
|
|
/// <returns><c>true</c> if this instance can set the specified type; otherwise, <c>false</c>.</returns>
|
|
public static bool CanSet(object value, Type typeRef)
|
|
{
|
|
return GetAttrForObj(value).Any(a => a.Action.IsFlagSet(CorrepsondingAction.Set) && a.TypeRef == typeRef);
|
|
}
|
|
|
|
/// <summary>Determines whether this type can set the specified reference type.</summary>
|
|
/// <param name="type">The class type.</param>
|
|
/// <param name="typeRef">The type supplied by the user to validate.</param>
|
|
/// <returns><c>true</c> if this type can set the specified reference type; otherwise, <c>false</c>.</returns>
|
|
public static bool CanSet(Type type, Type typeRef)
|
|
{
|
|
return GetAttrForType(type).Any(a => a.Action.IsFlagSet(CorrepsondingAction.Set) && a.TypeRef == typeRef);
|
|
}
|
|
|
|
/// <summary>Gets the corresponding types for the supplied enumeration value.</summary>
|
|
/// <param name="enumValue">The enumeration value or class.</param>
|
|
/// <returns>The types defined by the attribute.</returns>
|
|
public static IEnumerable<Type> GetCorrespondingTypes(object enumValue) => GetAttrForObj(enumValue).Select(a => a.TypeRef);
|
|
|
|
/// <summary>Gets the corresponding types for the supplied enumeration value.</summary>
|
|
/// <param name="type">The enumeration value or class.</param>
|
|
/// <returns>The types defined by the attribute.</returns>
|
|
public static IEnumerable<Type> GetCorrespondingTypes(Type type) => GetAttrForType(type).Select(a => a.TypeRef);
|
|
|
|
private static IEnumerable<CorrespondingTypeAttribute> GetAttrForObj(object value)
|
|
{
|
|
if (value == null) throw new ArgumentNullException(nameof(value));
|
|
var valueType = value.GetType();
|
|
if (!valueType.IsEnum && !valueType.IsClass) throw new ArgumentException("Value must be an enumeration or class value.", nameof(value));
|
|
var attr = (valueType.IsEnum ? valueType.GetField(value.ToString()).GetCustomAttributes<CorrespondingTypeAttribute>() : valueType.GetCustomAttributes<CorrespondingTypeAttribute>());
|
|
if (!attr.Any()) throw new InvalidOperationException("Value must have the CorrespondingTypeAttribute defined.");
|
|
if (attr.Any(a => a.Action == CorrepsondingAction.Exception)) throw new Exception();
|
|
return attr;
|
|
}
|
|
|
|
private static IEnumerable<CorrespondingTypeAttribute> GetAttrForType(Type type)
|
|
{
|
|
if (type == null) throw new ArgumentNullException(nameof(type));
|
|
var attr = type.GetCustomAttributes<CorrespondingTypeAttribute>();
|
|
if (!attr.Any()) throw new InvalidOperationException("Type must have the CorrespondingTypeAttribute defined.");
|
|
if (attr.Any(a => a.Action == CorrepsondingAction.Exception)) throw new Exception();
|
|
return attr;
|
|
}
|
|
}
|
|
} |