WrapIfFirstCallMatchesPolicy

WrapIfFirstCallMatchesPolicy.cs

public class WrapIfFirstCallMatchesPolicy<TWrapperBehavior> : IConfigurationAction where TWrapperBehavior : IActionBehavior
{
    readonly Func<ActionCall, bool> _filter;

    public WrapIfFirstCallMatchesPolicy(Func<ActionCall, bool> filter )
    {
        _filter = filter;
    }

    public bool Matches(BehaviorChain chain)
    {
        var firstCall = chain.FirstCall();

        return firstCall != null && _filter(firstCall);
    }
        
    public void Configure(BehaviorGraph graph)
    {
        graph.Behaviors.Where(Matches).Each(c => c.Prepend(Wrapper.For<TWrapperBehavior>()));
    }
}