Kavita/API/Data/ManualMigrations/MigrateChangeRestrictionRoles.cs

using System.Threading.Tasks;
using API.Constants;
using API.Entities;
using Microsoft.AspNetCore.Identity;
using Microsoft.Extensions.Logging;

namespace API.Data.ManualMigrations;

/// <summary>
/// New role introduced in v0.6. Adds the role to all users.
/// </summary>
public static class MigrateChangeRestrictionRoles
{
    /// <summary>
    /// Will not run if any users have the <see cref="PolicyConstants.ChangeRestrictionRole"/> role already
    /// </summary>
    /// <param name="unitOfWork"></param>
    /// <param name="userManager"></param>
    /// <param name="logger"></param>
    public static async Task Migrate(IUnitOfWork unitOfWork, UserManager<AppUser> userManager, ILogger<Program> logger)
    {
        var usersWithRole = await userManager.GetUsersInRoleAsync(PolicyConstants.ChangeRestrictionRole);
        if (usersWithRole.Count != 0) return;

        logger.LogCritical("Running MigrateChangeRestrictionRoles migration");

        var allUsers = await unitOfWork.UserRepository.GetAllUsersAsync();
        foreach (var user in allUsers)
        {
            await userManager.RemoveFromRoleAsync(user, PolicyConstants.ChangeRestrictionRole);
            await userManager.AddToRoleAsync(user, PolicyConstants.ChangeRestrictionRole);
        }

        logger.LogInformation("MigrateChangeRestrictionRoles migration complete");
    }
}