Remotely/Server/Auth/TwoFactorRequiredHandler.cs

using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Identity;
using Remotely.Server.Services;
using Remotely.Shared.Entities;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;

namespace Remotely.Server.Auth;

public class TwoFactorRequiredHandler : AuthorizationHandler<TwoFactorRequiredRequirement>
{
    private readonly UserManager<RemotelyUser> _userManager;
    private readonly IApplicationConfig _appConfig;

    public TwoFactorRequiredHandler(UserManager<RemotelyUser> userManager, IApplicationConfig appConfig)
    {
        _userManager = userManager;
        _appConfig = appConfig;
    }

    protected override async Task HandleRequirementAsync(AuthorizationHandlerContext context, TwoFactorRequiredRequirement requirement)
    {
        if (context.User.Identity?.IsAuthenticated == true && _appConfig.Require2FA)
        {
            var user = await _userManager.GetUserAsync(context.User);
            if (user?.TwoFactorEnabled != true)
            {
                context.Fail();
                return;
            }
        }
        context.Succeed(requirement);
    }
}