// 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.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
 
namespace Microsoft.AspNetCore.Server.IISIntegration;
 
internal sealed class IISSetupFilter : IStartupFilter
{
    private readonly string _pairingToken;
    private readonly PathString _pathBase;
    private readonly bool _isWebsocketsSupported;
 
    internal IISSetupFilter(string pairingToken, PathString pathBase, bool isWebsocketsSupported)
    {
        _pairingToken = pairingToken;
        _pathBase = pathBase;
        _isWebsocketsSupported = isWebsocketsSupported;
    }
 
    public Action<IApplicationBuilder> Configure(Action<IApplicationBuilder> next)
    {
        return app =>
        {
            app.UsePathBase(_pathBase);
            app.UseForwardedHeaders();
            app.UseMiddleware<IISMiddleware>(_pairingToken, _isWebsocketsSupported);
            next(app);
        };
    }
}