OpenAuth.Net/OpenAuth.App/ModuleManagerApp.cs

using Infrastructure;
using OpenAuth.App.ViewModel;
using OpenAuth.Domain;
using OpenAuth.Domain.Interface;
using System;
using System.Collections.Generic;
using System.Linq;

namespace OpenAuth.App
{
    public class ModuleManagerApp
    {
        private IModuleRepository _repository;
        private IRelevanceRepository _relevanceRepository;

        public ModuleManagerApp(IModuleRepository repository,
            IRelevanceRepository relevanceRepository)
        {
            _repository = repository;
            _relevanceRepository = relevanceRepository;
        }

        /// <summary>
        /// 加载一个节点下面的所有
        /// </summary>
        public dynamic Load(int parentId, int pageindex, int pagesize)
        {
            IEnumerable<Module> Modules;
            int total = 0;
            if (parentId == 0)
            {
                Modules = _repository.LoadModules(pageindex, pagesize);
                total = _repository.GetCount();
            }
            else
            {
                var ids = GetSubOrgIds(parentId);
                Modules = _repository.LoadInOrgs(pageindex, pagesize, ids);
                total = _repository.GetModuleCntInOrgs(ids);
            }

            return new
            {
                total = total,
                list = Modules,
                pageCurrent = pageindex
            };
        }

        /// <summary>
        /// 为树型结构提供数据
        /// </summary>
        public List<Module> LoadForTree()
        {
            return _repository.Find(null).ToList();
        }

        /// <summary>
        /// 以组合的方式显示所有的模块信息
        /// </summary>
        /// <param name="parentId">The parent unique identifier.</param>
        /// <returns>List{ModuleView}.</returns>
        public List<ModuleView> LoadByParent(int parentId)
        {
            var modules = new List<ModuleView>();
            var roots = _repository.Find(u => u.ParentId == parentId);
            foreach (var module in roots)
            {
                ModuleView mv = module;
                mv.Childern = LoadByParent(module.Id);
                modules.Add(mv);
            }
            return modules;
        }

        public Module Find(int id)
        {
            var module = _repository.FindSingle(u => u.Id == id);
            if (module == null) return new Module();

            return module;
        }

        public void Delete(int id)
        {
            var del = _repository.FindSingle(u => u.Id == id);
            if (del == null) return;

            _repository.Delete(u => u.CascadeId.Contains(del.CascadeId));
        }

        public void AddOrUpdate(Module vm)
        {
            Module model = new Module();
            vm.CopyTo(model);  //copy一次，防止成员为null的情况
            ChangeModuleCascade(model);
            if (model.Id == 0)
            {
                _repository.Add(model);
            }
            else
            {
                _repository.Update(model);
            }
        }

        #region 用户/角色分配模块

        /// <summary>
        /// 加载特定用户的模块
        /// TODO:这里会加载用户及用户角色的所有模块，“为用户分配模块”功能会给人一种混乱的感觉，但可以接受
        /// </summary>
        /// <param name="userId">The user unique identifier.</param>
        public List<Module> LoadForUser(int userId)
        {
            //用户角色
            var userRoleIds =
                _relevanceRepository.Find(u => u.FirstId == userId && u.Key == "UserRole").Select(u => u.SecondId).ToList();

            //用户角色与自己分配到的模块ID
            var moduleIds =
                _relevanceRepository.Find(
                    u =>
                        (u.FirstId == userId && u.Key == "UserModule") ||
                        (u.Key == "RoleModule" && userRoleIds.Contains(u.FirstId))).Select(u => u.SecondId).ToList();

            //var moduleIds =
            //    _relevanceRepository.Find(u => u.FirstId == userId && u.Key == "UserModule")
            //        .Select(u => u.SecondId)
            //        .ToList();
            if (!moduleIds.Any()) return new List<Module>();
            return _repository.Find(u => moduleIds.Contains(u.Id)).ToList();
        }

        /// <summary>
        /// 为特定的用户分配模块
        /// </summary>
        /// <param name="userId">The user unique identifier.</param>
        /// <param name="ids">模块ID</param>
        public void AssignModuleForUser(int userId, int[] ids)
        {
            _relevanceRepository.DeleteBy("UserModule", userId);
            _relevanceRepository.AddRelevance("UserModule", ids.ToLookup(u => userId));
        }

        /// <summary>
        /// 加载特定角色的模块
        /// </summary>
        /// <param name="roleId">The role unique identifier.</param>
        public List<Module> LoadForRole(int roleId)
        {
            var moduleIds =
                _relevanceRepository.Find(u => u.FirstId == roleId && u.Key == "RoleModule")
                    .Select(u => u.SecondId)
                    .ToList();
            if (!moduleIds.Any()) return new List<Module>();
            return _repository.Find(u => moduleIds.Contains(u.Id)).ToList();
        }

        /// <summary>
        /// 为特定的角色分配模块
        /// </summary>
        /// <param name="roleId">The user unique identifier.</param>
        /// <param name="ids">模块ID</param>
        public void AssignModuleForRole(int roleId, int[] ids)
        {
            _relevanceRepository.DeleteBy("RoleModule", roleId);
            _relevanceRepository.AddRelevance("RoleModule", ids.ToLookup(u => roleId));
        }

        #endregion 用户/角色分配模块

        #region 私有方法

        //根据同一级中最大的语义ID

        private int[] GetSubOrgIds(int parentId)
        {
            var parent = _repository.FindSingle(u => u.Id == parentId);
            var orgs = _repository.Find(u => u.CascadeId.Contains(parent.CascadeId)).Select(u => u.Id).ToArray();
            return orgs;
        }

        //修改对象的级联ID
        private void ChangeModuleCascade(Module module)
        {
            string cascadeId;
            int currentCascadeId = 1;  //当前结点的级联节点最后一位
            var sameLevels = _repository.Find(o => o.ParentId == module.ParentId && o.Id != module.Id);
            foreach (var obj in sameLevels)
            {
                int objCascadeId = int.Parse(obj.CascadeId.Split('.').Last());
                if (currentCascadeId <= objCascadeId) currentCascadeId = objCascadeId + 1;
            }

            if (module.ParentId != 0)
            {
                var parentOrg = _repository.FindSingle(o => o.Id == module.ParentId);
                if (parentOrg != null)
                {
                    cascadeId = parentOrg.CascadeId + "." + currentCascadeId;
                    module.ParentName = parentOrg.Name;
                }
                else
                {
                    throw new Exception("未能找到该组织的父节点信息");
                }
            }
            else
            {
                cascadeId = "0." + currentCascadeId;
                module.ParentName = "根节点";
            }

            module.CascadeId = cascadeId;
        }

        #endregion 私有方法
    }
}