Add the possibility to register multiple interceptors

Having the interceptors registered with the
ITypeRegistrar also enables the usage of ITypeResolver
in interceptors.

Author: nils-a

src/Spectre.Console.Cli/ConfiguratorExtensions.cs

@@ -229,7 +229,7 @@ public static IConfigurator SetInterceptor(this IConfigurator configurator, ICom
             throw new ArgumentNullException(nameof(configurator));
         }
 
-        configurator.Settings.Interceptor = interceptor;
+        configurator.Settings.Registrar.RegisterInstance<ICommandInterceptor>(interceptor);
         return configurator;
     }
 

src/Spectre.Console.Cli/ICommandAppSettings.cs

@@ -50,6 +50,7 @@ public interface ICommandAppSettings
     /// Gets or sets the <see cref="ICommandInterceptor"/> used
     /// to intercept settings before it's being sent to the command.
     /// </summary>
+    [Obsolete("Register the interceptor with the ITypeRegistrar.")]
     ICommandInterceptor? Interceptor { get; set; }
 
     /// <summary>

src/Spectre.Console.Cli/ICommandInterceptor.cs

@@ -12,5 +12,25 @@ public interface ICommandInterceptor
     /// </summary>
     /// <param name="context">The intercepted <see cref="CommandContext"/>.</param>
     /// <param name="settings">The intercepted <see cref="CommandSettings"/>.</param>
-    void Intercept(CommandContext context, CommandSettings settings);
+    void Intercept(CommandContext context, CommandSettings settings)
+#if NETSTANDARD2_0
+    ;
+#else
+    {
+    }
+#endif
+
+    /// <summary>
+    /// Intercepts a command result before it's passed as the result.
+    /// </summary>
+    /// <param name="context">The intercepted <see cref="CommandContext"/>.</param>
+    /// <param name="settings">The intercepted <see cref="CommandSettings"/>.</param>
+    /// <param name="result">The result from the command execution.</param>
+    void InterceptResult(CommandContext context, CommandSettings settings, ref int result)
+#if NETSTANDARD2_0
+    ;
+#else
+    {
+    }
+#endif
 }

src/Spectre.Console.Cli/Internal/CommandExecutor.cs

@@ -121,7 +121,7 @@ private static string ResolveApplicationVersion(IConfiguration configuration)
             VersionHelper.GetVersion(Assembly.GetEntryAssembly());
     }
 
-    private static Task<int> Execute(
+    private static async Task<int> Execute(
         CommandTree leaf,
         CommandTree tree,
         CommandContext context,
@@ -130,7 +130,19 @@ private static Task<int> Execute(
     {
         // Bind the command tree against the settings.
         var settings = CommandBinder.Bind(tree, leaf.Command.SettingsType, resolver);
-        configuration.Settings.Interceptor?.Intercept(context, settings);
+        var interceptors =
+            ((IEnumerable<ICommandInterceptor>?)resolver.Resolve(typeof(IEnumerable<ICommandInterceptor>))
+            ?? Array.Empty<ICommandInterceptor>()).ToList();
+#pragma warning disable CS0618 // Type or member is obsolete
+        if (configuration.Settings.Interceptor != null)
+        {
+            interceptors.Add(configuration.Settings.Interceptor);
+        }
+#pragma warning restore CS0618 // Type or member is obsolete
+        foreach (var interceptor in interceptors)
+        {
+            interceptor.Intercept(context, settings);
+        }
 
         // Create and validate the command.
         var command = leaf.CreateCommand(resolver);
@@ -141,6 +153,12 @@ private static Task<int> Execute(
         }
 
         // Execute the command.
-        return command.Execute(context, settings);
+        var result = await command.Execute(context, settings);
+        foreach (var interceptor in interceptors)
+        {
+            interceptor.InterceptResult(context, settings, ref result);
+        }
+
+        return result;
     }
 }

src/Spectre.Console.Cli/Internal/Configuration/CommandAppSettings.cs

@@ -8,6 +8,7 @@ internal sealed class CommandAppSettings : ICommandAppSettings
     public int MaximumIndirectExamples { get; set; }
     public bool ShowOptionDefaultValues { get; set; }
     public IAnsiConsole? Console { get; set; }
+    [Obsolete("Register the interceptor with the ITypeRegistrar.")]
     public ICommandInterceptor? Interceptor { get; set; }
     public ITypeRegistrarFrontend Registrar { get; set; }
     public CaseSensitivity CaseSensitivity { get; set; }

src/Spectre.Console.Testing/Cli/CallbackCommandInterceptor.cs

@@ -21,4 +21,11 @@ public void Intercept(CommandContext context, CommandSettings settings)
     {
         _callback(context, settings);
     }
+
+#if NETSTANDARD2_0
+    /// <inheritdoc/>
+    public void InterceptResult(CommandContext context, CommandSettings settings, ref int result)
+    {
+    }
+#endif
 }

test/Spectre.Console.Cli.Tests/Unit/CommandAppTests.Interceptor.cs

@@ -0,0 +1,90 @@
+namespace Spectre.Console.Tests.Unit.Cli;
+
+public sealed partial class CommandAppTests
+{
+    public sealed class Interceptor
+    {
+        public sealed class NoCommand : Command<NoCommand.Settings>
+        {
+            public sealed class Settings : CommandSettings
+            {
+            }
+
+            public override int Execute(CommandContext context, Settings settings)
+            {
+                return 0;
+            }
+        }
+
+        public sealed class MyInterceptor : ICommandInterceptor
+        {
+            private readonly Action<CommandContext, CommandSettings> _action;
+
+            public MyInterceptor(Action<CommandContext, CommandSettings> action)
+            {
+                _action = action;
+            }
+
+            public void Intercept(CommandContext context, CommandSettings settings)
+            {
+                _action(context, settings);
+            }
+        }
+
+        public sealed class MyResultInterceptor : ICommandInterceptor
+        {
+            private readonly Func<CommandContext, CommandSettings, int, int> _function;
+
+            public MyResultInterceptor(Func<CommandContext, CommandSettings, int, int> function)
+            {
+                _function = function;
+            }
+
+            public void InterceptResult(CommandContext context, CommandSettings settings, ref int result)
+            {
+                result = _function(context, settings, result);
+            }
+        }
+
+        [Fact]
+        public void Should_Run_The_Interceptor()
+        {
+            // Given
+            var count = 0;
+            var app = new CommandApp<NoCommand>();
+            var interceptor = new MyInterceptor((_, _) =>
+            {
+                count += 1;
+            });
+            app.Configure(config => config.SetInterceptor(interceptor));
+
+            // When
+            app.Run(Array.Empty<string>());
+
+            // Then
+            count.ShouldBe(1); // to be sure
+        }
+
+        [Fact]
+        public void Should_Run_The_ResultInterceptor()
+        {
+            // Given
+            var count = 0;
+            const int Expected = 123;
+            var app = new CommandApp<NoCommand>();
+            var interceptor = new MyResultInterceptor((_, _, _) =>
+            {
+                count += 1;
+                return Expected;
+            });
+            app.Configure(config => config.SetInterceptor(interceptor));
+
+            // When
+            var actual = app.Run(Array.Empty<string>());
+
+            // Then
+            count.ShouldBe(1);
+            actual.ShouldBe(Expected);
+        }
+    }
+}