// 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);
        }
    }
}