mixtape/zero.Core/Collections/CollectionInterceptorHandler.cs

using FluentValidation;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading.Tasks;
using zero.Core.Entities;

namespace zero.Core.Collections
{
  public class CollectionInterceptorHandler : ICollectionInterceptorHandler
  {
    /// <inheritdoc />
    public IEnumerable<ICollectionInterceptor> Interceptors { get; private set; }


    public CollectionInterceptorHandler(IEnumerable<ICollectionInterceptor> items)
    {
      Interceptors = items;
    }


    /// <inheritdoc />
    public async Task<EntityResult<T>> Handle<T>(Expression<Func<ICollectionInterceptor, Task<EntityResult<T>>>> intercept)
    {
      Func<ICollectionInterceptor, Task<EntityResult<T>>> func = default;

      foreach (ICollectionInterceptor interceptor in ForType(typeof(T)))
      {
        if (func == default)
        {
          func = intercept.Compile();
        }

        EntityResult<T> result = await func(interceptor);

        if (result != default)
        {
          return result;
        }
      }

      return default;
    }


    /// <inheritdoc />
    public async Task Handle<T>(Expression<Func<ICollectionInterceptor, Task>> intercept)
    {
      Func<ICollectionInterceptor, Task> func = default;

      foreach (ICollectionInterceptor interceptor in ForType(typeof(T)))
      {
        if (func == default)
        {
          func = intercept.Compile();
        }

        await func(interceptor);
      }
    }


    /// <summary>
    /// Get all interceptors for a certain type
    /// </summary>
    IEnumerable<ICollectionInterceptor> ForType(Type targetType)
    {
      return Interceptors.Where(item => item.Types.Count == 0 || item.Types.Any(type => targetType.IsAssignableFrom(type)));
    }
  }


  public interface ICollectionInterceptorHandler
  {
    /// <summary>
    /// All registered interceptors
    /// </summary>
    IEnumerable<ICollectionInterceptor> Interceptors { get; }

    /// <summary>
    /// Calls all matching interceptors with the specified expression
    /// </summary>
    Task<EntityResult<T>> Handle<T>(Expression<Func<ICollectionInterceptor, Task<EntityResult<T>>>> intercept);

    /// <summary>
    /// Calls all matching interceptors with the specified expression
    /// </summary>
    Task Handle<T>(Expression<Func<ICollectionInterceptor, Task>> intercept);
  }
}