update userManager logic for organization unit

modules/identity/src/Volo.Abp.Identity.Application.Contracts/Volo/Abp/Identity/IdentityUserCreateOrUpdateDtoBase.cs

@@ -1,4 +1,5 @@
-using System.ComponentModel.DataAnnotations;
+using System;
+using System.ComponentModel.DataAnnotations;
 using JetBrains.Annotations;
 
 namespace Volo.Abp.Identity
@@ -29,5 +30,7 @@ namespace Volo.Abp.Identity
 
         [CanBeNull]
         public string[] RoleNames { get; set; }
+
+        public Guid[] OrganizationUnits { get; set; }
     }
 }

modules/identity/src/Volo.Abp.Identity.Domain.Shared/Volo/Abp/Identity/Localization/en.json

@@ -64,6 +64,7 @@
     "Identity.StaticRoleRenamingErrorMessage": "Static roles can not be renamed.",
     "Identity.StaticRoleDeletionErrorMessage": "Static roles can not be deleted.",
     "Identity.OrganizationUnit.DuplicateDisplayNameWarning": "There is already an organization unit with name {0}. Two units with same name can not be created in same level.",
+    "Identity.OrganizationUnit.MaxUserMembershipCount": "Maximum allowed organization unit membership count for a user",
     "Volo.Abp.Identity:010001": "You can not delete your own account!",
     "Permission:IdentityManagement": "Identity management",
     "Permission:RoleManagement": "Role management",

modules/identity/src/Volo.Abp.Identity.Domain.Shared/Volo/Abp/Identity/Settings/IdentitySettingNames.cs

@@ -40,5 +40,12 @@
             public const string IsUserNameUpdateEnabled = UserPrefix + ".IsUserNameUpdateEnabled";
             public const string IsEmailUpdateEnabled = UserPrefix + ".IsEmailUpdateEnabled";
         }
+
+        public static class OrganizationUnit
+        {
+            private const string OrganizationUnitPrefix = Prefix + ".OrganizationUnit";
+
+            public const string MaxUserMembershipCount = OrganizationUnitPrefix + ".MaxUserMembershipCount";
+        }
     }
 }

modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/AbpIdentitySettingDefinitionProvider.cs

@@ -57,7 +57,7 @@ namespace Volo.Abp.Identity
 
                 new SettingDefinition(
                     IdentitySettingNames.Lockout.LockoutDuration,
-                    (5*60).ToString(), L("DisplayName:Abp.Identity.Lockout.LockoutDuration"), 
+                    (5 * 60).ToString(), L("DisplayName:Abp.Identity.Lockout.LockoutDuration"),
                     L("Description:Abp.Identity.Lockout.LockoutDuration"),
                     true),
 
@@ -88,6 +88,12 @@ namespace Volo.Abp.Identity
                     IdentitySettingNames.User.IsEmailUpdateEnabled,
                     true.ToString(), L("DisplayName:Abp.Identity.User.IsEmailUpdateEnabled"),
                     L("Description:Abp.Identity.User.IsEmailUpdateEnabled"),
+                    true),
+
+                 new SettingDefinition(
+                    IdentitySettingNames.OrganizationUnit.MaxUserMembershipCount,
+                    int.MaxValue.ToString(), L("Identity.OrganizationUnit.MaxUserMembershipCount"),
+                    L("Identity.OrganizationUnit.MaxUserMembershipCount"),
                     true)
             );
         }

modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IdentityUser.cs

@@ -129,7 +129,7 @@ namespace Volo.Abp.Identity
         /// <summary>
         /// Navigation property for this organization units.
         /// </summary>
-        public virtual ICollection<OrganizationUnitUser> OrganizationUnits { get; protected set; }
+        public virtual ICollection<IdentityUserOrganizationUnit> OrganizationUnits { get; protected set; }
 
         protected IdentityUser()
         {
@@ -154,7 +154,7 @@ namespace Volo.Abp.Identity
             Claims = new Collection<IdentityUserClaim>();
             Logins = new Collection<IdentityUserLogin>();
             Tokens = new Collection<IdentityUserToken>();
-            OrganizationUnits = new Collection<OrganizationUnitUser>();
+            OrganizationUnits = new Collection<IdentityUserOrganizationUnit>();
 
             ExtraProperties = new Dictionary<string, object>();
         }
@@ -293,7 +293,7 @@ namespace Volo.Abp.Identity
                 return;
             }
 
-            OrganizationUnits.Add(new OrganizationUnitUser(TenantId, Id, organizationUnitId));
+            OrganizationUnits.Add(new IdentityUserOrganizationUnit(TenantId, Id, organizationUnitId));
         }
 
         public virtual void RemoveOrganizationUnit(Guid organizationUnitId)

modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IdentityUserManager.cs

@@ -9,7 +9,12 @@ using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
 using Volo.Abp.Domain.Entities;
 using Volo.Abp.Domain.Services;
+using Volo.Abp.Identity.Organizations;
+using Volo.Abp.Domain.Repositories;
 using Volo.Abp.Threading;
+using Volo.Abp.Uow;
+using Volo.Abp.Settings;
+using Volo.Abp.Identity.Settings;
 
 namespace Volo.Abp.Identity
 {
@@ -18,6 +23,9 @@ namespace Volo.Abp.Identity
         protected override CancellationToken CancellationToken => _cancellationTokenProvider.Token;
 
         private readonly ICancellationTokenProvider _cancellationTokenProvider;
+        protected IOrganizationUnitRepository _organizationUnitRepository { get; private set; }
+        protected IIdentityUserRepository _identityUserRepository { get; private set; }
+        private readonly ISettingProvider _settingProvider;
 
         public IdentityUserManager(
             IdentityUserStore store,
@@ -28,7 +36,10 @@ namespace Volo.Abp.Identity
             ILookupNormalizer keyNormalizer, IdentityErrorDescriber errors,
             IServiceProvider services,
             ILogger<IdentityUserManager> logger,
-            ICancellationTokenProvider cancellationTokenProvider)
+            ICancellationTokenProvider cancellationTokenProvider,
+            IOrganizationUnitRepository organizationUnitRepository,
+            IIdentityUserRepository identityUserRepository,
+            ISettingProvider settingProvider)
             : base(
                   store,
                   optionsAccessor,
@@ -41,6 +52,9 @@ namespace Volo.Abp.Identity
                   logger)
         {
             _cancellationTokenProvider = cancellationTokenProvider;
+            _organizationUnitRepository = organizationUnitRepository;
+            _identityUserRepository = identityUserRepository;
+            _settingProvider = settingProvider;
         }
 
         public virtual async Task<IdentityUser> GetByIdAsync(Guid id)
@@ -75,5 +89,115 @@ namespace Volo.Abp.Identity
 
             return IdentityResult.Success;
         }
+
+        public virtual async Task<bool> IsInOrganizationUnitAsync(Guid userId, Guid ouId)
+        {
+            return await IsInOrganizationUnitAsync(
+                await GetByIdAsync(userId),
+                await _organizationUnitRepository.GetAsync(ouId)
+                );
+        }
+
+        public virtual Task<bool> IsInOrganizationUnitAsync(IdentityUser user, OrganizationUnit ou)
+        {
+            return Task.FromResult(user.IsInOrganizationUnit(ou.Id));
+        }
+
+        public virtual async Task AddToOrganizationUnitAsync(Guid userId, Guid ouId)
+        {
+            await AddToOrganizationUnitAsync(
+                await GetByIdAsync(userId),
+                await _organizationUnitRepository.GetAsync(ouId)
+                );
+        }
+
+        public virtual async Task AddToOrganizationUnitAsync(IdentityUser user, OrganizationUnit ou)
+        {
+            await _identityUserRepository.EnsureCollectionLoadedAsync(user, u => u.OrganizationUnits, _cancellationTokenProvider.Token).ConfigureAwait(false);
+            
+            var currentOus = user.OrganizationUnits;
+
+            if (currentOus.Any(cou => cou.Id == ou.Id))
+            {
+                return;
+            }
+
+            await CheckMaxUserOrganizationUnitMembershipCountAsync(user.TenantId, currentOus.Count + 1);
+
+            user.AddOrganizationUnit(ou.Id);
+        }
+
+        public virtual async Task RemoveFromOrganizationUnitAsync(Guid userId, Guid ouId)
+        {
+            await RemoveFromOrganizationUnitAsync(
+                await GetByIdAsync(userId),
+                await _organizationUnitRepository.GetAsync(ouId)
+                );
+        }
+
+        public virtual async Task RemoveFromOrganizationUnitAsync(IdentityUser user, OrganizationUnit ou)
+        {
+            await _identityUserRepository.EnsureCollectionLoadedAsync(user, u => u.OrganizationUnits, _cancellationTokenProvider.Token).ConfigureAwait(false);
+
+            user.RemoveOrganizationUnit(ou.Id);
+        }
+
+        public virtual async Task SetOrganizationUnitsAsync(Guid userId, params Guid[] organizationUnitIds)
+        {
+            await SetOrganizationUnitsAsync(
+                await GetByIdAsync(userId),
+                organizationUnitIds
+                );
+        }
+
+        public virtual async Task SetOrganizationUnitsAsync(IdentityUser user, params Guid[] organizationUnitIds)
+        {
+            Check.NotNull(user, nameof(user));
+            Check.NotNull(organizationUnitIds, nameof(organizationUnitIds));
+
+            await CheckMaxUserOrganizationUnitMembershipCountAsync(user.TenantId, organizationUnitIds.Length);
+
+            var currentOus = user.OrganizationUnits;
+
+            //Remove from removed OUs
+            foreach (var currentOu in currentOus)
+            {
+                if (!organizationUnitIds.Contains(currentOu.Id))
+                {
+                    await RemoveFromOrganizationUnitAsync(user.Id, currentOu.Id);
+                }
+            }
+
+            //Add to added OUs
+            foreach (var organizationUnitId in organizationUnitIds)
+            {
+                if (currentOus.All(ou => ou.Id != organizationUnitId))
+                {
+                    await AddToOrganizationUnitAsync(
+                        user,
+                        await _organizationUnitRepository.GetAsync(organizationUnitId)
+                        );
+                }
+            }
+        }
+
+        private async Task CheckMaxUserOrganizationUnitMembershipCountAsync(Guid? tenantId, int requestedCount)
+        {
+            var maxCount = await _settingProvider.GetAsync<int>(IdentitySettingNames.OrganizationUnit.MaxUserMembershipCount).ConfigureAwait(false);
+            if (requestedCount > maxCount)
+            {
+                throw new AbpException(string.Format("Can not set more than {0} organization unit for a user!", maxCount));
+            }
+        }
+
+        [UnitOfWork]
+        public virtual async Task<List<OrganizationUnit>> GetOrganizationUnitsAsync(IdentityUser user)
+        {
+            await _identityUserRepository.EnsureCollectionLoadedAsync(user, u => u.OrganizationUnits, _cancellationTokenProvider.Token).ConfigureAwait(false);
+
+            var ouOfUser = user.OrganizationUnits;
+
+            return await _organizationUnitRepository.GetListAsync(ouOfUser.Select(t => t.OrganizationUnitId));
+        }
     }
 }

modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/Organizations/OrganizationUnitUser.cs → modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/IdentityUserOrganizationUnit.cs

@@ -2,12 +2,12 @@
 using Volo.Abp.Domain.Entities.Auditing;
 using Volo.Abp.MultiTenancy;
 
-namespace Volo.Abp.Identity.Organizations
+namespace Volo.Abp.Identity
 {
     /// <summary>
     /// Represents membership of a User to an OU.
     /// </summary>
-    public class OrganizationUnitUser : CreationAuditedEntity<Guid>, IMultiTenant, ISoftDelete
+    public class IdentityUserOrganizationUnit : CreationAuditedEntity<Guid>, IMultiTenant, ISoftDelete
     {
 
         /// <summary>
@@ -33,7 +33,7 @@ namespace Volo.Abp.Identity.Organizations
         /// <summary>
         /// Initializes a new instance of the <see cref="UserOrganizationUnit"/> class.
         /// </summary>
-        public OrganizationUnitUser()
+        public IdentityUserOrganizationUnit()
         {
 
         }
@@ -44,7 +44,7 @@ namespace Volo.Abp.Identity.Organizations
         /// <param name="tenantId">TenantId</param>
         /// <param name="userId">Id of the User.</param>
         /// <param name="organizationUnitId">Id of the <see cref="OrganizationUnit"/>.</param>
-        public OrganizationUnitUser(Guid? tenantId, Guid userId, Guid organizationUnitId)
+        public IdentityUserOrganizationUnit(Guid? tenantId, Guid userId, Guid organizationUnitId)
         {
             TenantId = tenantId;
             UserId = userId;

modules/identity/src/Volo.Abp.Identity.Domain/Volo/Abp/Identity/Organizations/IOrganizationUnitRepository.cs

@@ -10,5 +10,7 @@ namespace Volo.Abp.Identity.Organizations
         Task<List<OrganizationUnit>> GetChildrenAsync(Guid? parentId);
 
         Task<List<OrganizationUnit>> GetAllChildrenWithParentCodeAsync(string code, Guid? parentId);
+
+        Task<List<OrganizationUnit>> GetListAsync(IEnumerable<Guid> ids);
     }
 }

modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/EfCoreOrganizationUnitRepository.cs

@@ -28,5 +28,10 @@ namespace Volo.Abp.Identity.EntityFrameworkCore
             return await DbSet.Where(ou => ou.Code.StartsWith(code) && ou.Id != parentId.Value)
                 .ToListAsync();
         }
+
+        public async Task<List<OrganizationUnit>> GetListAsync(IEnumerable<Guid> ids)
+        {
+            return await DbSet.Where(t => ids.Contains(t.Id)).ToListAsync();
+        }
     }
 }

modules/identity/src/Volo.Abp.Identity.EntityFrameworkCore/Volo/Abp/Identity/EntityFrameworkCore/IdentityDbContextModelBuilderExtensions.cs

@@ -165,9 +165,9 @@ namespace Volo.Abp.Identity.EntityFrameworkCore
                 b.HasIndex(ou => new { ou.RoleId, ou.OrganizationUnitId });
             });
 
-            builder.Entity<OrganizationUnitUser>(b =>
+            builder.Entity<IdentityUserOrganizationUnit>(b =>
             {
-                b.ToTable(options.TablePrefix + "OrganizationUnitUsers", options.Schema);
+                b.ToTable(options.TablePrefix + "UserOrganizationUnits", options.Schema);
 
                 b.HasKey(ou => new { ou.OrganizationUnitId, ou.UserId });