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