| | | 1 | | using Elsa.Common; |
| | | 2 | | using Elsa.Identity.Contracts; |
| | | 3 | | using Elsa.Identity.Options; |
| | | 4 | | using JetBrains.Annotations; |
| | | 5 | | using Microsoft.Extensions.Logging; |
| | | 6 | | using Microsoft.Extensions.Options; |
| | | 7 | | |
| | | 8 | | namespace Elsa.Identity.HostedServices; |
| | | 9 | | |
| | | 10 | | /// <summary> |
| | | 11 | | /// Hosted service that initializes the admin user and role from <see cref="DefaultAdminUserOptions"/> configuration if |
| | | 12 | | /// </summary> |
| | | 13 | | [UsedImplicitly] |
| | 2 | 14 | | public class AdminUserInitializer( |
| | 2 | 15 | | IUserStore userStore, |
| | 2 | 16 | | IRoleStore roleStore, |
| | 2 | 17 | | IUserManager userManager, |
| | 2 | 18 | | IRoleManager roleManager, |
| | 2 | 19 | | IOptions<DefaultAdminUserOptions> options, |
| | 2 | 20 | | ILogger<AdminUserInitializer> logger) |
| | | 21 | | : BackgroundTask |
| | | 22 | | { |
| | | 23 | | public override async Task ExecuteAsync(CancellationToken cancellationToken) |
| | | 24 | | { |
| | 3 | 25 | | var adminUserName = options.Value.AdminUserName; |
| | 3 | 26 | | var adminPassword = options.Value.AdminPassword; |
| | 3 | 27 | | var adminRoleName = options.Value.AdminRoleName; |
| | 3 | 28 | | var adminRolePermissions = options.Value.AdminRolePermissions; |
| | 3 | 29 | | if (string.IsNullOrWhiteSpace(adminRoleName)) |
| | | 30 | | { |
| | 0 | 31 | | logger.LogWarning("AdminRoleName is not configured. Skipping admin role and user creation."); |
| | 0 | 32 | | return; |
| | | 33 | | } |
| | | 34 | | |
| | 3 | 35 | | var existingRole = await roleStore.FindAsync(new() { Id = adminRoleName }, cancellationToken); |
| | | 36 | | |
| | 3 | 37 | | if (existingRole == null) |
| | | 38 | | { |
| | 0 | 39 | | var roleResult = await roleManager.CreateRoleAsync( |
| | 0 | 40 | | adminRoleName, |
| | 0 | 41 | | adminRolePermissions.ToList(), |
| | 0 | 42 | | adminRoleName, |
| | 0 | 43 | | cancellationToken); |
| | | 44 | | |
| | 0 | 45 | | logger.LogInformation("Admin role '{RoleName}' created successfully with {PermissionCount} permissions.", |
| | 0 | 46 | | roleResult.Role.Name, |
| | 0 | 47 | | roleResult.Role.Permissions.Count); |
| | | 48 | | } |
| | | 49 | | else |
| | | 50 | | { |
| | 3 | 51 | | var missingPermissions = adminRolePermissions.Except(existingRole.Permissions, StringComparer.Ordinal).ToArr |
| | | 52 | | |
| | 3 | 53 | | if (missingPermissions.Length == 0) |
| | | 54 | | { |
| | 2 | 55 | | logger.LogInformation("Admin role '{RoleName}' already exists with all configured permissions.", adminRo |
| | | 56 | | } |
| | | 57 | | else |
| | | 58 | | { |
| | 1 | 59 | | existingRole.Permissions = existingRole.Permissions.Concat(missingPermissions).ToList(); |
| | 1 | 60 | | await roleStore.SaveAsync(existingRole, cancellationToken); |
| | 1 | 61 | | logger.LogInformation( |
| | 1 | 62 | | "Admin role '{RoleName}' updated successfully with {PermissionCount} missing configured permissions. |
| | 1 | 63 | | adminRoleName, |
| | 1 | 64 | | missingPermissions.Length); |
| | | 65 | | } |
| | 3 | 66 | | } |
| | | 67 | | |
| | 3 | 68 | | var roleToAssign = adminRoleName; |
| | | 69 | | |
| | | 70 | | // Create user if configured |
| | 3 | 71 | | if (string.IsNullOrWhiteSpace(adminUserName) || string.IsNullOrWhiteSpace(adminPassword)) |
| | | 72 | | { |
| | 3 | 73 | | logger.LogWarning("AdminUserName and/or AdminPassword not configured in DefaultAdminUserOptions. Skipping ad |
| | 3 | 74 | | return; |
| | | 75 | | } |
| | | 76 | | |
| | | 77 | | // Check if user already exists |
| | 0 | 78 | | var existingUser = await userStore.FindAsync(new() { Name = adminUserName }, cancellationToken); |
| | | 79 | | |
| | 0 | 80 | | if (existingUser != null) |
| | | 81 | | { |
| | 0 | 82 | | logger.LogInformation("Admin user '{User}' already exists. Skipping creation.", adminUserName); |
| | 0 | 83 | | return; |
| | | 84 | | } |
| | | 85 | | |
| | | 86 | | // Create the admin user |
| | 0 | 87 | | var result = await userManager.CreateUserAsync( |
| | 0 | 88 | | adminUserName, |
| | 0 | 89 | | adminPassword, |
| | 0 | 90 | | new List<string> { roleToAssign }, |
| | 0 | 91 | | cancellationToken); |
| | | 92 | | |
| | 0 | 93 | | logger.LogInformation("Admin user '{Name}' created successfully with role '{Role}'.", result.User.Name, roleToAs |
| | 3 | 94 | | } |
| | | 95 | | } |