using System;
using System.Threading.Tasks;


namespace AIRuntime.Runtime.Capabilities
{
    /// <summary>
    /// Records when capability pipeline execution finishes.
    /// Runs for successful, failed, and cancelled executions.
    /// </summary>
    public sealed class TimingMiddleware : ICapabilityMiddleware
    {
        public async Task InvokeAsync(
            CapabilityContext context,
            CapabilityDelegate next)
        {
            if (context == null)
                throw new ArgumentNullException(nameof(context));

            try
            {
                await next(context);
            }
            finally
            {
                context.FinishedUtc =
                    DateTime.UtcNow;
            }
        }
    }
}
