// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Net.Http;
using AngleSharp.Dom.Html;
 
namespace Microsoft.AspNetCore.Identity.FunctionalTests.Account;
 
public class Login : DefaultUIPage
{
    private readonly IHtmlFormElement _loginForm;
    private readonly IHtmlAnchorElement _forgotPasswordLink;
    private readonly IHtmlAnchorElement _reconfirmLink;
    private readonly IHtmlFormElement _externalLoginForm;
    private readonly IHtmlElement _contosoButton;
    private readonly IHtmlElement _loginButton;
 
    public Login(
        HttpClient client,
        IHtmlDocument login,
        DefaultUIContext context)
        : base(client, login, context)
    {
        _loginForm = HtmlAssert.HasForm("#account", login);
        _loginButton = HtmlAssert.HasElement("#login-submit", login);
        _forgotPasswordLink = HtmlAssert.HasLink("#forgot-password", login);
        _reconfirmLink = HtmlAssert.HasLink("#resend-confirmation", login);
        if (Context.ContosoLoginEnabled)
        {
            _externalLoginForm = HtmlAssert.HasForm("#external-account", login);
            _contosoButton = HtmlAssert.HasElement("button[value=Contoso]", login);
        }
    }
 
    public async Task<Contoso.Login> ClickLoginWithContosoLinkAsync()
    {
        var externalFormResponse = await Client.SendAsync(_externalLoginForm, _contosoButton);
        var goToContosoLogin = ResponseAssert.IsRedirect(externalFormResponse);
        var contosoLoginResponse = await Client.GetAsync(goToContosoLogin);
 
        var contosoLogin = await ResponseAssert.IsHtmlDocumentAsync(contosoLoginResponse);
 
        return new Contoso.Login(Client, contosoLogin, Context);
    }
 
    public async Task<ForgotPassword> ClickForgotPasswordLinkAsync()
    {
        var response = await Client.GetAsync(_forgotPasswordLink.Href);
        var forgotPassword = await ResponseAssert.IsHtmlDocumentAsync(response);
 
        return new ForgotPassword(Client, forgotPassword, Context);
    }
 
    public async Task<ResendEmailConfirmation> ClickReconfirmEmailLinkAsync()
    {
        var response = await Client.GetAsync(_reconfirmLink.Href);
        var forgotPassword = await ResponseAssert.IsHtmlDocumentAsync(response);
 
        return new ResendEmailConfirmation(Client, forgotPassword, Context);
    }
 
    public async Task<Index> LoginValidUserAsync(string userName, string password)
    {
        var loggedIn = await SendLoginForm(userName, password);
 
        var loggedInLocation = ResponseAssert.IsRedirect(loggedIn);
        Assert.Equal(Index.Path, loggedInLocation.ToString());
        var indexResponse = await Client.GetAsync(loggedInLocation);
        var index = await ResponseAssert.IsHtmlDocumentAsync(indexResponse);
        return new Index(
            Client,
            index,
            Context.WithAuthenticatedUser().WithPasswordLogin());
    }
 
    public async Task LoginWrongPasswordAsync(string userName, string password)
    {
        var failedLogin = await SendLoginForm(userName, password);
 
        ResponseAssert.IsOK(failedLogin);
        var content = await failedLogin.Content.ReadAsStringAsync();
        Assert.Contains("Invalid login attempt.", content);
    }
 
    public async Task<DefaultUIPage> LockoutUserAsync(string userName, string password)
    {
        var loginAttempt = await SendLoginForm(userName, password);
 
        var lockedOut = ResponseAssert.IsRedirect(loginAttempt);
        Assert.Equal("/Identity/Account/Lockout", lockedOut.ToString());
 
        var lockedOutResponse = await Client.GetAsync(lockedOut);
        var lockout = await ResponseAssert.IsHtmlDocumentAsync(lockedOutResponse);
        return new DefaultUIPage(Client, lockout, Context);
    }
 
    private async Task<HttpResponseMessage> SendLoginForm(string userName, string password)
    {
        return await Client.SendAsync(_loginForm, _loginButton, new Dictionary<string, string>()
        {
            ["Input_Email"] = userName,
            ["Input_Password"] = password
        });
    }
 
    public async Task<LoginWith2fa> PasswordLoginValidUserWith2FaAsync(string userName, string password)
    {
        var loggedIn = await SendLoginForm(userName, password);
 
        var loggedInLocation = ResponseAssert.IsRedirect(loggedIn);
        Assert.StartsWith(LoginWith2fa.Path, loggedInLocation.ToString());
        var loginWithTwoFactorResponse = await Client.GetAsync(loggedInLocation);
        var loginWithTwoFactor = await ResponseAssert.IsHtmlDocumentAsync(loginWithTwoFactorResponse);
 
        return new LoginWith2fa(Client, loginWithTwoFactor, Context);
    }
}