// 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;
 
namespace Microsoft.AspNetCore.Mvc.ViewFeatures.Filters;
 
internal sealed class ViewDataAttributeApplicationModelProvider : IApplicationModelProvider
{
    /// <inheritdoc />
    /// <remarks>This order ensures that <see cref="ViewDataAttributeApplicationModelProvider"/> 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 controllerType = controllerModel.ControllerType.AsType();
 
            var viewDataProperties = ViewDataAttributePropertyProvider.GetViewDataProperties(controllerType);
            if (viewDataProperties == null)
            {
                continue;
            }
 
            var filter = new ControllerViewDataAttributeFilterFactory(viewDataProperties);
            controllerModel.Filters.Add(filter);
        }
    }
}