feat: 角色模板改为数据库管理支持前端自定义

2025-12-03 20:38:26 +08:00
parent 19137f3cf7
commit 6a84141799
28 changed files with 901 additions and 652 deletions
--- a/src/Infrastructure/TakeoutSaaS.Infrastructure/Identity/Persistence/EfRoleTemplateRepository.cs
+++ b/src/Infrastructure/TakeoutSaaS.Infrastructure/Identity/Persistence/EfRoleTemplateRepository.cs
@@ -0,0 +1,117 @@
+using System.Collections.Generic;
+using System.Linq;
+using Microsoft.EntityFrameworkCore;
+using TakeoutSaaS.Domain.Identity.Entities;
+using TakeoutSaaS.Domain.Identity.Repositories;
+
+namespace TakeoutSaaS.Infrastructure.Identity.Persistence;
+
+/// <summary>
+/// 角色模板仓储实现。
+/// </summary>
+public sealed class EfRoleTemplateRepository(IdentityDbContext dbContext) : IRoleTemplateRepository
+{
+    public Task<IReadOnlyList<RoleTemplate>> GetAllAsync(bool? isActive, CancellationToken cancellationToken = default)
+    {
+        var query = dbContext.RoleTemplates.AsNoTracking();
+        if (isActive.HasValue)
+        {
+            query = query.Where(x => x.IsActive == isActive.Value);
+        }
+
+        return query
+            .OrderBy(x => x.TemplateCode)
+            .ToListAsync(cancellationToken)
+            .ContinueWith(t => (IReadOnlyList<RoleTemplate>)t.Result, cancellationToken);
+    }
+
+    public Task<RoleTemplate?> FindByCodeAsync(string templateCode, CancellationToken cancellationToken = default)
+    {
+        var normalized = templateCode.Trim();
+        return dbContext.RoleTemplates.AsNoTracking().FirstOrDefaultAsync(x => x.TemplateCode == normalized, cancellationToken);
+    }
+
+    public Task<IReadOnlyList<RoleTemplatePermission>> GetPermissionsAsync(long roleTemplateId, CancellationToken cancellationToken = default)
+    {
+        return dbContext.RoleTemplatePermissions.AsNoTracking()
+            .Where(x => x.RoleTemplateId == roleTemplateId)
+            .ToListAsync(cancellationToken)
+            .ContinueWith(t => (IReadOnlyList<RoleTemplatePermission>)t.Result, cancellationToken);
+    }
+
+    public async Task<IDictionary<long, IReadOnlyList<RoleTemplatePermission>>> GetPermissionsAsync(IEnumerable<long> roleTemplateIds, CancellationToken cancellationToken = default)
+    {
+        var ids = roleTemplateIds.Distinct().ToArray();
+        if (ids.Length == 0)
+        {
+            return new Dictionary<long, IReadOnlyList<RoleTemplatePermission>>();
+        }
+
+        var permissions = await dbContext.RoleTemplatePermissions.AsNoTracking()
+            .Where(x => ids.Contains(x.RoleTemplateId))
+            .ToListAsync(cancellationToken);
+
+        return permissions
+            .GroupBy(x => x.RoleTemplateId)
+            .ToDictionary(g => g.Key, g => (IReadOnlyList<RoleTemplatePermission>)g.ToList());
+    }
+
+    public async Task AddAsync(RoleTemplate template, IEnumerable<string> permissionCodes, CancellationToken cancellationToken = default)
+    {
+        template.TemplateCode = template.TemplateCode.Trim();
+        template.Name = template.Name.Trim();
+        await dbContext.RoleTemplates.AddAsync(template, cancellationToken);
+        await ReplacePermissionsInternalAsync(template, permissionCodes, cancellationToken);
+    }
+
+    public async Task UpdateAsync(RoleTemplate template, IEnumerable<string> permissionCodes, CancellationToken cancellationToken = default)
+    {
+        template.TemplateCode = template.TemplateCode.Trim();
+        template.Name = template.Name.Trim();
+        dbContext.RoleTemplates.Update(template);
+        await ReplacePermissionsInternalAsync(template, permissionCodes, cancellationToken);
+    }
+
+    public async Task DeleteAsync(long roleTemplateId, CancellationToken cancellationToken = default)
+    {
+        var entity = await dbContext.RoleTemplates.FirstOrDefaultAsync(x => x.Id == roleTemplateId, cancellationToken);
+        if (entity != null)
+        {
+            var permissions = dbContext.RoleTemplatePermissions.Where(x => x.RoleTemplateId == roleTemplateId);
+            dbContext.RoleTemplatePermissions.RemoveRange(permissions);
+            dbContext.RoleTemplates.Remove(entity);
+        }
+    }
+
+    public Task SaveChangesAsync(CancellationToken cancellationToken = default)
+        => dbContext.SaveChangesAsync(cancellationToken);
+
+    private async Task ReplacePermissionsInternalAsync(RoleTemplate template, IEnumerable<string> permissionCodes, CancellationToken cancellationToken)
+    {
+        // 确保模板已持久化，便于 FK 正确填充
+        if (!dbContext.Entry(template).IsKeySet || template.Id == 0)
+        {
+            await dbContext.SaveChangesAsync(cancellationToken);
+        }
+
+        var normalized = permissionCodes
+            .Where(code => !string.IsNullOrWhiteSpace(code))
+            .Select(code => code.Trim())
+            .Distinct(StringComparer.OrdinalIgnoreCase)
+            .ToArray();
+
+        var existing = await dbContext.RoleTemplatePermissions
+            .Where(x => x.RoleTemplateId == template.Id)
+            .ToListAsync(cancellationToken);
+
+        dbContext.RoleTemplatePermissions.RemoveRange(existing);
+
+        var toAdd = normalized.Select(code => new RoleTemplatePermission
+        {
+            RoleTemplateId = template.Id,
+            PermissionCode = code
+        });
+
+        await dbContext.RoleTemplatePermissions.AddRangeAsync(toAdd, cancellationToken);
+    }
+}