80 lines
3.1 KiB
C#
80 lines
3.1 KiB
C#
// Licensed to the .NET Foundation under one or more agreements.
|
|
// The .NET Foundation licenses this file to you under the MIT license.
|
|
#nullable disable
|
|
|
|
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using TightWiki.Models;
|
|
|
|
namespace TightWiki.Areas.Identity.Pages.Account.Manage
|
|
{
|
|
public class GenerateRecoveryCodesModel : PageModelBase
|
|
{
|
|
private readonly UserManager<IdentityUser> _userManager;
|
|
private readonly ILogger<GenerateRecoveryCodesModel> _logger;
|
|
|
|
public GenerateRecoveryCodesModel(SignInManager<IdentityUser> signInManager,
|
|
UserManager<IdentityUser> userManager,
|
|
ILogger<GenerateRecoveryCodesModel> logger)
|
|
: base(signInManager)
|
|
{
|
|
_userManager = userManager;
|
|
_logger = logger;
|
|
}
|
|
|
|
/// <summary>
|
|
/// This API supports the ASP.NET Core Identity default UI infrastructure and is not intended to be used
|
|
/// directly from your code. This API may change or be removed in future releases.
|
|
/// </summary>
|
|
[TempData]
|
|
public string[] RecoveryCodes { get; set; }
|
|
|
|
/// <summary>
|
|
/// This API supports the ASP.NET Core Identity default UI infrastructure and is not intended to be used
|
|
/// directly from your code. This API may change or be removed in future releases.
|
|
/// </summary>
|
|
[TempData]
|
|
public string StatusMessage { get; set; }
|
|
|
|
public async Task<IActionResult> OnGetAsync()
|
|
{
|
|
var user = await _userManager.GetUserAsync(User);
|
|
if (user == null)
|
|
{
|
|
return NotFound($"Unable to load user with ID '{_userManager.GetUserId(User)}'.");
|
|
}
|
|
|
|
var isTwoFactorEnabled = await _userManager.GetTwoFactorEnabledAsync(user);
|
|
if (!isTwoFactorEnabled)
|
|
{
|
|
throw new InvalidOperationException($"Cannot generate recovery codes for user because they do not have 2FA enabled.");
|
|
}
|
|
|
|
return Page();
|
|
}
|
|
|
|
public async Task<IActionResult> OnPostAsync()
|
|
{
|
|
var user = await _userManager.GetUserAsync(User);
|
|
if (user == null)
|
|
{
|
|
return NotFound($"Unable to load user with ID '{_userManager.GetUserId(User)}'.");
|
|
}
|
|
|
|
var isTwoFactorEnabled = await _userManager.GetTwoFactorEnabledAsync(user);
|
|
var userId = await _userManager.GetUserIdAsync(user);
|
|
if (!isTwoFactorEnabled)
|
|
{
|
|
throw new InvalidOperationException($"Cannot generate recovery codes for user as they do not have 2FA enabled.");
|
|
}
|
|
|
|
var recoveryCodes = await _userManager.GenerateNewTwoFactorRecoveryCodesAsync(user, 10);
|
|
RecoveryCodes = recoveryCodes.ToArray();
|
|
|
|
_logger.LogInformation("User with ID '{UserId}' has generated new 2FA recovery codes.", userId);
|
|
StatusMessage = "You have generated new recovery codes.";
|
|
return RedirectToPage($"{GlobalConfiguration.BasePath}/Identity/ShowRecoveryCodes");
|
|
}
|
|
}
|
|
}
|