Yi.Admin/Yi.Framework/Yi.Framework.Service/UserService.cs

using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
using System.Threading.Tasks;
using Yi.Framework.Interface;
using Yi.Framework.Model;
using Yi.Framework.Model.Models;

namespace Yi.Framework.Service
{
    public partial class UserService : BaseService<user>, IUserService
    {
        private IRoleService _roleService;
        public UserService(DbContext Db, IRoleService roleService) : base(Db)
        {
            _roleService = roleService;
        }

        public async Task<bool> EmailIsExsit(string emailAddress)
        {
            var userList = await GetAllEntitiesTrueAsync();
            var is_email = userList.Where(u => u.email == emailAddress).FirstOrDefault();
            if (is_email == null)
            {
                return true;
            }
            return false;
        }

        /// <summary>
        ///  大优化，直接传用户id即可
        /// </summary>
        /// <param name="_user"></param>
        /// <returns></returns>
        public async Task<List<menu>> GetMenusByUser(user _user)
        {
            var role_data = await GetRolesByUser(_user);
            var menuList = new List<menu>();
            role_data.ForEach(u =>
            {
                var menu_data = _roleService.GetMenusByRole(u);
                menuList = menuList.Concat(menu_data.GetAwaiter().GetResult()).ToList();
            });
            menuList.ForEach(u => u.roles = null);
            return menuList;
        }

        /// <summary>
        /// 大优化只用id即可，这个方法和GetMenusByUser几乎没有任何区别
        /// </summary>
        /// <param name="_user"></param>
        /// <returns></returns>
        public async Task<List<mould>> GetMouldByUser(user _user)
        {
            var menu = await GetMenusByUser(_user);
            var mouldList = menu.Select(u => u.mould).ToList();
            return mouldList;
        }


        public async Task<List<role>> GetRolesByUser(user _user)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles)
                .Where(u => u.id == _user.id && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync();
            var roleList = user_data.roles.Where(u => u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).ToList();
            roleList.ForEach(u => u.users = null);
            return roleList;
        }

        public async Task<user> Login(user _user)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles).Where(u => u.username == _user.username && u.password == _user.password &&
               u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync();
            return user_data;
        }

        public async Task<bool> Register(user _user)
        {
            var user_data = await GetEntity(u => u.username == _user.username);
            if (user_data != null)
            {
                return false;
            }
            return await AddAsync(_user);
        }

        /// <summary>
        /// 优化，如果是传的id，或者是传一个对象，不是要返回全部信息，那么可知直接认为该id一定存在
        /// </summary>
        /// <param name="roleIds"></param>
        /// <param name="userIds"></param>
        /// <returns></returns>
        public async Task<bool> SetRolesByUser(List<int> roleIds, List<int> userIds)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles).Where(u => userIds.Contains(u.id) ).ToListAsync();
            if (user_data == null)
            {
                return false;
            }
            var roleList = await _Db.Set<role>().Where(u => roleIds.Contains(u.id)).ToListAsync();
            foreach (var item in user_data)
            {
                item.roles = roleList;
            }
            return await UpdateListAsync(user_data);
        }

        /// <summary>
        /// 有可简化的GetUserById
        /// </summary>
        /// <param name="_user"></param>
        /// <returns></returns>
        public async Task<List<menu>> GetMenuByUser(user _user)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles).ThenInclude(u => u.menus).Where(u => u.id == _user.id ).FirstOrDefaultAsync();
            List<menu> menu_data = new();
            foreach (var role in user_data.roles)
            {
                var menu = role.menus.ToList();
                menu.ForEach(u => u.roles = null);
                menu_data = menu_data.Concat(menu).ToList();
            }
            return menu_data;
        }


        public async Task<user> GetUserInfoById(int user_id)
        {
            var user_data = await GetUserById(user_id);
            user_data.roles.ToList().ForEach(u => u.users = null);
            return user_data;
        }

        /// <summary>
        /// 这个方法可以被超级多个其他方法利用，这个是重要的
        /// </summary>
        /// <param name="user_id"></param>
        /// <returns></returns>
        public async Task<user> GetUserById(int user_id)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles)
                .Where(u => u.id == user_id && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).FirstOrDefaultAsync();
            return user_data;
        }

        /// <summary>
        /// 有可简化的GetUserById
        /// </summary>
        /// <param name="user_id"></param>
        /// <returns></returns>
        public async Task<List<menu>> GetMenuById(int user_id)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles).Where(u => u.id == user_id).FirstOrDefaultAsync();
            List<menu> menu_data = new();

            foreach (var role in user_data.roles)
            {
                var menu = await _roleService.GetMenusByRole(role);
                menu.ForEach(u => u.roles = null);
                menu_data = menu_data.Union(menu).OrderByDescending(u => u.sort).ToList();
            }
            //menu_data为角色所有的菜单，不是一个递归的啊

            var allMenuIds = menu_data.Select(u => u.id).ToList();
            var topMenu = menu_data.Where(u => u.is_top == (short)Common.Enum.ShowFlagEnum.Show);

            //现在要开始关联菜单了

            List<menu> endMenu = new List<menu>();
            foreach (var item in topMenu)
            {
                var p = await _Db.Set<menu>().Where(u => u.id == item.id).Include(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ThenInclude(u => u.children).ToListAsync();
                endMenu = endMenu.Union(p).ToList();
            }


            return TopMenuBuild2(TopMenuBuild(endMenu, allMenuIds));
        }
        /// <summary>
        /// 这种就命名的话，改成MenuIconAndShowBuild
        /// </summary>
        /// <param name="menu_data"></param>
        /// <param name="allMenuIds"></param>
        /// <returns></returns>
        private List<menu> TopMenuBuild(List<menu> menu_data, List<int> allMenuIds)
        {

            for (int i = menu_data.Count() - 1; i >= 0; i--)
            {

                if (menu_data[i].icon == null)
                {
                    menu_data[i].icon = "Yi";
                }

                if (!allMenuIds.Contains(menu_data[i].id) || menu_data[i].is_delete == (short)Common.Enum.DelFlagEnum.Deleted || menu_data[i].is_show == (short)Common.Enum.ShowFlagEnum.NoShow)
                {
                    menu_data.Remove(menu_data[i]);
                }
                else if (menu_data[i].children != null)
                {
                    menu_data[i].children = TopMenuBuild(menu_data[i].children.ToList(), allMenuIds);
                }
            }
            return menu_data;
        }

        /// <summary>
        /// 这种就命名的话，改成MenuChildrenBuild,这个方法应该可以提到core层，这是一个公用的方法
        /// </summary>
        /// <param name="menu_data"></param>
        /// <returns></returns>
        private List<menu> TopMenuBuild2(List<menu> menu_data)
        {

            for (int i = menu_data.Count() - 1; i >= 0; i--)
            {
                if (menu_data[i].children.Count() == 0)
                {
                    menu_data[i].children = null;
                }
                else if (menu_data[i].children != null)
                {
                    menu_data[i].children = TopMenuBuild2(menu_data[i].children.ToList());
                }
            }
            return menu_data;
        }



        public async Task<List<menu>> GetMenuByUserId(string router, int userId, List<int> menuIds)
        {
            var user_data = await _Db.Set<user>().Include(u => u.roles).ThenInclude(u => u.menus).ThenInclude(u => u.children)
                 .ThenInclude(u => u.mould).Where(u => u.id == userId).FirstOrDefaultAsync();
            var roleList = user_data.roles.ToList();
            roleList.ForEach(u => u.users = null);
            List<menu> menu_data = new();
            foreach (var item in roleList)
            {
                item.menus.ToList().ForEach(u => u.roles = null);
                var menuData = item.menus.Where(u => u.router.ToUpper() == router.ToUpper()).FirstOrDefault();
                menu_data = menuData.children?.Where(u => menuIds.Contains(u.id) && u.is_delete == (short)Common.Enum.DelFlagEnum.Normal).ToList();
                if (menu_data != null) { break; }
            }
            return menu_data;
        }
    }
}