// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using Microsoft.AspNetCore.Mvc.ApplicationModels;
using Microsoft.AspNetCore.Mvc.ViewFeatures.Infrastructure;
 
namespace Microsoft.AspNetCore.Mvc.ViewFeatures.Filters;
 
internal sealed class TempDataApplicationModelProvider : IApplicationModelProvider
{
    private readonly TempDataSerializer _tempDataSerializer;
 
    public TempDataApplicationModelProvider(TempDataSerializer tempDataSerializer)
    {
        _tempDataSerializer = tempDataSerializer;
    }
 
    /// <inheritdoc />
    /// <remarks>This order ensures that <see cref="TempDataApplicationModelProvider"/> runs after the <see cref="DefaultApplicationModelProvider"/>.</remarks>
    public int Order => -1000 + 10;
 
    /// <inheritdoc />
    public void OnProvidersExecuted(ApplicationModelProviderContext context)
    {
    }
 
    /// <inheritdoc />
    public void OnProvidersExecuting(ApplicationModelProviderContext context)
    {
        ArgumentNullException.ThrowIfNull(context);
 
        foreach (var controllerModel in context.Result.Controllers)
        {
            var modelType = controllerModel.ControllerType.AsType();
 
            var tempDataProperties = SaveTempDataPropertyFilterBase.GetTempDataProperties(_tempDataSerializer, modelType);
            if (tempDataProperties == null)
            {
                continue;
            }
 
            var filter = new ControllerSaveTempDataPropertyFilterFactory(tempDataProperties);
            controllerModel.Filters.Add(filter);
        }
    }
}