HTCloud/HT.Cloud.Service/Infrastructure/BaseService.cs

using SqlSugar;
using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;
using HT.Cloud.Code;
using HT.Cloud.DataBase;
using HT.Cloud.Domain.SystemManage;
using HT.Cloud.Domain.SystemOrganize;

namespace HT.Cloud.Service
{
	public class BaseService<T> : RepositoryBase<T> where T : class, new()
	{
		// 用户信息
		public OperatorModel currentuser;

		// 用于当前表操作
		protected RepositoryBase<T> repository;

		// 用于其他表操作
		protected ISqlSugarClient _context;

		public BaseService(ISqlSugarClient context) : base(context)
		{
			currentuser = OperatorProvider.Provider.GetCurrent();
			_context = context;
			repository = this;
			if (currentuser == null)
			{
				currentuser = new OperatorModel();
			}
			else
			{
				var entityType = typeof(T);
				if (entityType.IsDefined(typeof(TenantAttribute), false))
				{
					var tenantAttribute = entityType.GetCustomAttribute<TenantAttribute>(false)!;
					repository.ChangeEntityDb(tenantAttribute.configId);
				}
				else if (_context.AsTenant().IsAnyConnection(currentuser.DbNumber))
				{
					repository.ChangeEntityDb(currentuser.DbNumber);
				}
				else
				{
					var dblist = DBInitialize.GetConnectionConfigs(false);
                    _context.AsTenant().AddConnection(dblist.FirstOrDefault(a => a.ConfigId.ToString() == currentuser.DbNumber));
                    repository.ChangeEntityDb(currentuser.DbNumber);
					repository.Db.DefaultConfig();
				}
			}
		}

		/// <summary>
		///  获取当前登录用户的数据访问权限(单表)
		/// </summary>
		/// <param name=""parameterName>linq表达式参数的名称，如u=>u.name中的"u"</param>
		/// <param name=""moduleName>菜单名称</param>
		/// <param name=""query>查询</param>
		/// <returns></returns>
		protected ISugarQueryable<T> GetDataPrivilege(string parametername, string moduleName = "", ISugarQueryable<T> query = null)
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			if (query == null)
			{
				query = repository.IQueryable();
			}
			if (!CheckDataPrivilege(moduleName))
			{
				return GetFieldsFilterDataNew(parametername, query, moduleName);
			}
			var rule = repository.Db.Queryable<DataPrivilegeRuleEntity>().WithCache().First(u => u.F_ModuleCode == moduleName && u.F_EnabledMark == true && u.F_DeleteMark == false);
			if (rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINUSER) ||
											 rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINROLE) ||
											 rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINORG))
			{
				//即把{loginUser} =='xxxxxxx'换为 loginUser.User.Id =='xxxxxxx'，从而把当前登录的用户名与当时设计规则时选定的用户id对比
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINUSER, currentuser.UserId);
				var roles = currentuser.RoleId;
				//var roles = loginUser.Roles.Select(u => u.Id).ToList();//多角色
				//roles.Sort();
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINROLE,
					roles);
				var orgs = currentuser.OrganizeId;
				//var orgs = loginUser.Orgs.Select(u => u.Id).ToList();//多部门
				//orgs.Sort();
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINORG,
					orgs);
			}
			query = query.GenerateFilter(parametername,
				JsonHelper.ToObject<List<FilterList>>(rule.F_PrivilegeRules));
			return GetFieldsFilterDataNew(parametername, query, moduleName);
		}

		/// <summary>
		///  获取当前登录用户的数据访问权限(复杂查询)
		/// </summary>
		/// <param name=""parameterName>linq表达式参数的名称，如u=>u.name中的"u"</param>
		/// <param name=""moduleName>菜单名称</param>
		/// <param name=""query>查询</param>
		/// <returns></returns>
		protected ISugarQueryable<TEntity> GetDataPrivilege<TEntity>(string parametername, string moduleName = "", ISugarQueryable<TEntity> query = null)
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			if (!CheckDataPrivilege(moduleName))
			{
				return GetFieldsFilterDataNew(parametername, query, moduleName);
			}
			var rule = repository.Db.Queryable<DataPrivilegeRuleEntity>().WithCache().First(u => u.F_ModuleCode == moduleName && u.F_EnabledMark == true && u.F_DeleteMark == false);
			if (rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINUSER) ||
											 rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINROLE) ||
											 rule.F_PrivilegeRules.Contains(Define.DATAPRIVILEGE_LOGINORG))
			{
				//即把{loginUser} =='xxxxxxx'换为 loginUser.User.Id =='xxxxxxx'，从而把当前登录的用户名与当时设计规则时选定的用户id对比
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINUSER, currentuser.UserId);
				var roles = currentuser.RoleId;
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINROLE,
					roles);
				var orgs = currentuser.OrganizeId;
				rule.F_PrivilegeRules = rule.F_PrivilegeRules.Replace(Define.DATAPRIVILEGE_LOGINORG,
					orgs);
			}
			query = query.GenerateFilter(parametername,
				JsonHelper.ToObject<List<FilterList>>(rule.F_PrivilegeRules));
			return GetFieldsFilterDataNew(parametername, query, moduleName);
		}

		/// <summary>
		///  获取当前登录用户是否需要数据控制
		/// </summary>
		/// <param name=""moduleName>菜单名称</param>
		/// <returns></returns>
		protected bool CheckDataPrivilege(string moduleName = "")
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			if (currentuser.IsAdmin) return false;  //超级管理员特权
			var rule = repository.Db.Queryable<DataPrivilegeRuleEntity>().WithCache().First(u => u.F_ModuleCode == moduleName && u.F_EnabledMark == true && u.F_DeleteMark == false);
			////系统菜单也不需要数据权限 跟字段重合取消这样处理
			//var module = UnitWork.FindEntity<ModuleEntity>(u => u.F_EnCode == moduleName).GetAwaiter().GetResult();
			if (rule == null)
			{
				return false; //没有设置数据规则，那么视为该资源允许被任何主体查看
			}
			//if (rule == null|| module.F_IsPublic==true)
			//{
			//    return false; //没有设置数据规则，那么视为该资源允许被任何主体查看
			//}
			return true;
		}

		/// <summary>
		///  soul数据反向模板化
		/// </summary>
		/// <param name=""dic>集合</param>
		/// <param name=""pagination>分页</param>
		/// <returns></returns>
		protected SoulPage<TEntity> ChangeSoulData<TEntity>(Dictionary<string, Dictionary<string, string>> dic, SoulPage<TEntity> pagination)
		{
			List<FilterSo> filterSos = pagination.getFilterSos();
			filterSos = FormatData(dic, filterSos);
			pagination.filterSos = filterSos.ToJson();
			return pagination;
		}

		protected List<FilterSo> FormatData(Dictionary<string, Dictionary<string, string>> dic, List<FilterSo> filterSos)
		{
			foreach (var item in filterSos)
			{
				if (item.mode == "condition" && dic.ContainsKey(item.field) && dic[item.field].Values.Contains(item.value))
				{
					item.value = dic[item.field].FirstOrDefault(a => a.Value == item.value).Key;
				}
				if (item.children != null && item.children.Count > 0)
				{
					item.children = FormatData(dic, item.children);
				}
			}
			return filterSos;
		}

		/// <summary>
		///  字段权限处理
		/// </summary>
		///<param name=""list>数据列表</param>
		/// <param name=""moduleName>菜单名称</param>
		/// <returns></returns>
		protected List<TEntity> GetFieldsFilterData<TEntity>(List<TEntity> list, string moduleName = "")
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			//管理员跳过
			if (currentuser.IsAdmin)
			{
				return list;
			}
			//系统菜单跳过
			var module = repository.Db.Queryable<ModuleEntity>().First(u => u.F_EnCode == moduleName);
			//判断是否需要字段权限
			if (module == null || module.F_IsFields == false)
			{
				return list;
			}
			//空list直接返回
			if (list.Count == 0)
			{
				return list;
			}
			var rolelist = currentuser.RoleId.Split(',');
			var rule = repository.Db.Queryable<RoleAuthorizeEntity>().Where(u => rolelist.Contains(u.F_ObjectId) && u.F_ItemType == 3).Select(a => a.F_ItemId).Distinct().ToList();
			var fieldsList = repository.Db.Queryable<ModuleFieldsEntity>().Where(u => (rule.Contains(u.F_Id) || u.F_IsPublic == true) && u.F_EnabledMark == true && u.F_ModuleId == module.F_Id).Select(u => u.F_EnCode).ToList();
			//反射获取主键
			PropertyInfo pkProp = typeof(TEntity).GetProperties().Where(p => p.GetCustomAttributes(typeof(SugarColumn), false).Length > 0).First();
			var idName = "F_Id";
			if (pkProp != null)
			{
				idName = pkProp.Name;
			}
			fieldsList.Add(idName);
			fieldsList = fieldsList.Distinct().ToList();
			return DataTableHelper.ListFilter(list, fieldsList);
		}

		/// <summary>
		///  字段权限处理
		/// </summary>
		///<param name=""entity>数据</param>
		/// <param name=""moduleName>菜单名称</param>
		/// <returns></returns>
		protected TEntity GetFieldsFilterData<TEntity>(TEntity entity, string moduleName = "")
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			//管理员跳过
			if (currentuser.IsAdmin)
			{
				return entity;
			}
			//系统菜单跳过
			var module = repository.Db.Queryable<ModuleEntity>().First(u => u.F_EnCode == moduleName);
			//判断是否需要字段权限
			if (module == null || module.F_IsFields == false)
			{
				return entity;
			}
			//空对象直接返回
			if (entity == null)
			{
				return entity;
			}
			var rolelist = currentuser.RoleId.Split(',');
			var rule = repository.Db.Queryable<RoleAuthorizeEntity>().Where(u => rolelist.Contains(u.F_ObjectId) && u.F_ItemType == 3).Select(a => a.F_ItemId).Distinct().ToList();
			var fieldsList = repository.Db.Queryable<ModuleFieldsEntity>().Where(u => (rule.Contains(u.F_Id) || u.F_IsPublic == true) && u.F_EnabledMark == true && u.F_ModuleId == module.F_Id).Select(u => u.F_EnCode).ToList();
			//反射获取主键
			PropertyInfo pkProp = typeof(TEntity).GetProperties().Where(p => p.GetCustomAttributes(typeof(SugarColumn), false).Length > 0).First();
			var idName = "F_Id";
			if (pkProp != null)
			{
				idName = pkProp.Name;
			}
			fieldsList.Add(idName);
			fieldsList = fieldsList.Distinct().ToList();
			List<TEntity> list = new List<TEntity>();
			list.Add(entity);
			return DataTableHelper.ListFilter(list, fieldsList)[0];
		}

		/// <summary>
		///  字段权限处理
		/// </summary>
		///<param name=""query>数据列表</param>
		/// <param name=""moduleName>菜单名称</param>
		/// <returns></returns>
		protected ISugarQueryable<TEntity> GetFieldsFilterDataNew<TEntity>(string parametername, ISugarQueryable<TEntity> query, string moduleName = "")
		{
			moduleName = string.IsNullOrEmpty(moduleName) ? ReflectionHelper.GetModuleName() : moduleName;
			//管理员跳过
			if (currentuser.IsAdmin)
			{
				return query;
			}
			//系统菜单跳过
			var module = repository.Db.Queryable<ModuleEntity>().First(u => u.F_EnCode == moduleName);
			//判断是否需要字段权限
			if (module == null || module.F_IsFields == false)
			{
				return query;
			}
			var rolelist = currentuser.RoleId.Split(',');
			var rule = repository.Db.Queryable<RoleAuthorizeEntity>().Where(u => rolelist.Contains(u.F_ObjectId) && u.F_ItemType == 3).Select(a => a.F_ItemId).Distinct().ToList();
			var fieldsList = repository.Db.Queryable<ModuleFieldsEntity>().Where(u => (rule.Contains(u.F_Id) || u.F_IsPublic == true) && u.F_EnabledMark == true && u.F_ModuleId == module.F_Id).Select(u => u.F_EnCode).ToList();
			//反射获取主键
			PropertyInfo pkProp = typeof(TEntity).GetProperties().Where(p => p.GetCustomAttributes(typeof(SugarColumn), false).Length > 0).First();
			var idName = "F_Id";
			if (pkProp != null)
			{
				idName = pkProp.Name;
			}
			fieldsList.Add(idName);
			fieldsList = fieldsList.Distinct().ToList();
			//可以构建lambda
			var parameter = Expression.Parameter(typeof(TEntity), parametername);
			var bindings = fieldsList
			.Select(name => name.Trim())
			.Select(name => Expression.Bind(
			typeof(TEntity).GetProperty(name),
			Expression.Property(parameter, name)
			));
			var newT = Expression.MemberInit(Expression.New(typeof(TEntity)), bindings);
			var lambda = Expression.Lambda<Func<TEntity, TEntity>>(newT, parameter);
			query = query.Select(lambda);
			//chloe扩展方法
			//List<string> ignoreList = new List<string>();
			//         foreach (var item in typeof(TEntity).GetProperties())
			//         {
			//             if (!fieldsList.Contains(item.Name))
			//             {
			//                 ignoreList.Add(item.Name);
			//             }
			//         }
			//         query = query.Ignore(ignoreList.ToArray());
			return query;
		}
	}
}