iot/projects/Platform/Areas/UserCenter/Controllers/RoleController.cs

using Infrastructure.Application;
using Infrastructure.Data;
using Infrastructure.Extensions;
using Infrastructure.Web.Mvc;
using IoT.Shared.Application.Domain.Entities;
using IoT.Shared.Application.Models;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Microsoft.EntityFrameworkCore;
using System.Linq;

namespace IoT.Shared.Areas.UserCenter.Controllers
{
    [Authorize]
    [ApiController]
    [Route("UserCenter/[controller]/[action]")]
    [Area("UserCenter")]
    [ControllerScope(ControllerScopeType.Platform)]
    public class RoleController : CrudController<Role, EditRoleModel>
    {
        private readonly PermissionController _ajax;

        public RoleController(IRepository<Role> repo, PermissionController ajax) : base(repo)
        {
            this._ajax = ajax;
        }

        public override IQueryable<Role> Include(IQueryable<Role> query)
        {
            return query.Include(o => o.RolePermissions).ThenInclude(o => o.Permission).ThenInclude(o => o.Category).ThenInclude(o => o.Parent);
        }

        public override IQueryable<Role> Query(PagedListModel<EditRoleModel> model, IQueryable<Role> query)
        {
            return base.Query(model, query)
                .WhereIf(!string.IsNullOrEmpty(model.Query.Name), o => o.Name.Contains(model.Query.Name))
                .OrderBy(o => o.Name);
        }

        public override void ToDisplayModel(Role entity, EditRoleModel model)
        {
            model.Permissions = entity.RolePermissions.Select(o => o.PermissionId).ToList();
            entity.RolePermissions.ForEach(o => ViewData.Add(o.PermissionId, o.Permission.Name));
        }

        public override void ToEditModel(Role entity, EditRoleModel model)
        {
            if (entity != null)
            {
                model.Permissions = entity.RolePermissions.Select(o => o.PermissionId).ToList();
            }
            this.ViewData.MultiSelectList(o => model.Permissions, () => this._ajax.GetPermissionMultiSelectList(model.Permissions));
        }

        public override void ToEntity(EditRoleModel model, Role entity)
        {
            foreach (var id in entity.RolePermissions.Select(o => o.PermissionId).ToList())
            {
                if (!model.Permissions.Any(o => o == id))
                {
                    entity.RolePermissions.RemoveAll(o => o.PermissionId == id);
                }
            }
            foreach (var id in model.Permissions)
            {
                if (!entity.RolePermissions.Any(o => o.PermissionId == id))
                {
                    entity.RolePermissions.Add(new RolePermission { PermissionId = id });
                }
            }
        }
    }
}