iot/projects/IoTNode/Areas/IoTCenter/Controllers/UserController.cs

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

namespace IoT.Shared.Areas.Admin.Controllers
{
    [Authorize]
    [ApiController]
    [Route("IoTCenter/[controller]/[action]")]
    [Area("IoTCenter")]
    public class UserController : CrudController<User, EditUserModel>
    {
        private readonly RoleController _ajax;

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

        public override IQueryable<User> Include(IQueryable<User> query)
        {
            return query.Include(o => o.UserRoles).ThenInclude(o => o.Role);
        }

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

        public override void ToDisplayModel(User entity, EditUserModel model)
        {
            model.Roles = entity.UserRoles.Select(o => o.RoleId).ToList();
            entity.UserRoles.ForEach(o => ViewData.Add(o.RoleId, o.Role.Name));
        }

        public override void ToEditModel(User entity, EditUserModel model)
        {
            if (entity != null)
            {
                model.Roles = entity.UserRoles.Select(o => o.RoleId).ToList();
            }
            this.ViewData.MultiSelectList(o => model.Roles, () => this._ajax.GetRoleMultiSelectList(model.Roles));
        }

        public override void ToEntity(EditUserModel model, User entity)
        {
            foreach (var id in entity.UserRoles.Select(o => o.RoleId).ToList())
            {
                if (!model.Roles.Any(o => o == id))
                {
                    entity.UserRoles.RemoveAll(o => o.RoleId == id);
                }
            }
            foreach (var id in model.Roles)
            {
                if (!entity.UserRoles.Any(o => o.RoleId == id))
                {
                    entity.UserRoles.Add(new UserRole { RoleId = id });
                }
            }
        }
    }
}