iotgateway/WalkingTec.Mvvm/WalkingTec.Mvvm.Core/Support/DataPrivilegeInfo.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using WalkingTec.Mvvm.Core.Extensions;

namespace WalkingTec.Mvvm.Core
{
    /// <summary>
    /// 数据权限信息接口
    /// </summary>
    public interface IDataPrivilege
    {
        //数据权限关联的表名
        string ModelName { get; set; }
        //数据权限名称
        string PrivillegeName { get; set; }

        Type ModelType { get; set; }
        //获取数据权限的下拉菜单
        List<ComboSelectListItem> GetItemList (WTMContext wtmcontext, string filter = null, List<string> ids = null);
        List<string> GetTreeParentIds(WTMContext wtmcontext, List<DataPrivilege> dps);
        List<string> GetTreeSubIds(WTMContext wtmcontext, List<string> pids);
    }

    /// <summary>
    /// 数据权限信息
    /// </summary>
    /// <typeparam name="T">关联表类</typeparam>
    public class DataPrivilegeInfo<T> : IDataPrivilege where T : TopBasePoco
    {
        //数据权限关联的表名
        public string ModelName { get; set; }
        //数据权限名称
        public string PrivillegeName { get; set; }
        //显示字段
        private Expression<Func<T, string>> _displayField;
        //where过滤条件
        private Expression<Func<T, bool>> _where;
        public Type ModelType { get; set; }

        public DataPrivilegeInfo(string name, Expression<Func<T, string>> displayField, Expression<Func<T, bool>> where = null)
        {
            ModelType = typeof(T);
            ModelName = ModelType.Name;
            PrivillegeName = name;
            _displayField = displayField;
            _where = where;
        }

        /// <summary>
        /// 获取数据权限的下拉菜单
        /// </summary>
        /// <param name="wtmcontext">wtm context</param>
        /// <param name="filter">filter</param>
        /// <param name="ids">ids</param>
        /// <returns>数据权限关联表的下拉菜单</returns>
        public List<ComboSelectListItem> GetItemList(WTMContext wtmcontext, string filter = null,List<string> ids= null)
        {
            var user = wtmcontext?.LoginUserInfo;
            Expression<Func<T, bool>> where = null;

            if (ids != null)
            {
                where = ids.GetContainIdExpression<T>();
            }
            else
            {
                if (string.IsNullOrEmpty(filter) == false)
                {
                    ChangePara cp = new ChangePara();
                    ParameterExpression pe = Expression.Parameter(typeof(T));
                    //var toString = Expression.Call(cp.Change(_displayField.Body, pe), "Cast", new Type[] { typeof(string)});
                    var tolower = Expression.Call(cp.Change(_displayField.Body, pe), "ToLower", new Type[] { });
                    var exp = Expression.Call(tolower, "Contains", null, Expression.Constant(filter.ToLower()));
                    where = Expression.Lambda<Func<T, bool>>(exp, pe);
                    if (_where != null)
                    {
                        var temp = cp.Change(_where.Body, pe);
                        var together = Expression.And(where.Body, temp);
                        where = Expression.Lambda<Func<T, bool>>(together, pe);
                    }
                }
                else
                {
                    where = _where;
                }
                if (where == null)
                {
                    where = x => 1 == 1;
                }
            }
            List<ComboSelectListItem> rv = new List<ComboSelectListItem>();
            if (user.Roles?.Where(x => x.RoleCode == "001").FirstOrDefault() == null && user.DataPrivileges?.Where(x => x.RelateId == null).FirstOrDefault() == null)
            {
                rv = wtmcontext.DC.Set<T>().CheckIDs(user.DataPrivileges.Select(y => y.RelateId).ToList()).Where(where).GetSelectListItems(wtmcontext, _displayField, null, ignorDataPrivilege: true);
            }
            else
            {
                rv = wtmcontext.DC.Set<T>().Where(where).GetSelectListItems(wtmcontext, _displayField, null, ignorDataPrivilege: true);
            }
            return rv;
        }

        public List<string> GetTreeParentIds(WTMContext wtmcontext, List<DataPrivilege> dps)
        {
            var ids = dps.Where(x => x.TableName == this.ModelName).Select(x => x.RelateId).ToList();
            var idscheck = ids.GetContainIdExpression<T>();
            var modified = wtmcontext.DC.Set<T>().Where(idscheck).CheckNotNull("ParentId").DynamicSelect("ParentId");
            var skipids = modified.ToList();
            return skipids;
        }

        public List<string> GetTreeSubIds(WTMContext wtmcontext, List<string> pids)
        {
            ParameterExpression pe = Expression.Parameter(typeof(T));
            Expression parentid = Expression.Property(pe, typeof(T).GetSingleProperty("ParentId"));
            return wtmcontext.DC.Set<T>().Where(pids.GetContainIdExpression<T>(parentid)).DynamicSelect("ID").ToList();
        }
    }


}