package org.springblade.common.cache;

import org.apache.logging.log4j.util.Strings;
import org.springblade.core.cache.utils.CacheUtil;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.core.tool.utils.StringPool;
import org.springblade.modules.system.entity.*;
import org.springblade.modules.system.service.*;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

import static org.springblade.common.cache.CacheNames.ARTICLE_KEY;
import static org.springblade.core.cache.constant.CacheConstant.SYS_CACHE;

/**
 * 系统缓存
 *
 * @author Chill
 */
public class SysCache {
	private static final String MENU_ID = "menu:id:";
	private static final String DEPT_ID = "dept:id:";
	private static final String DEPT_NAME = "dept:name:";
	private static final String DEPT_NAME_FUZZY = "dept:nameFuzzy:";
	private static final String DEPT_NAME_ID = "deptName:id:";
	private static final String DEPT_NAMES_ID = "deptNames:id:";
	private static final String DEPT_CHILD_ID = "deptChild:id:";
	private static final String DEPT_CHILDIDS_ID = "deptChildIds:id:";
	private static final String REGION_CHILD_CODE = "regionChild:code:";
	private static final String REGION_CHILDCODES_CODE = "regionChildCodes:code:";
	private static final String POST_ID = "post:id:";
	private static final String POST_NAME = "post:name:";
	private static final String POST_NAME_FUZZY = "post:nameFuzzy:";
	private static final String POST_NAME_ID = "postName:id:";
	private static final String POST_NAMES_ID = "postNames:id:";
	private static final String ROLE_ID = "role:id:";
	private static final String ROLE_NAME = "role:name:";
	private static final String ROLE_NAME_ID = "roleName:id:";
	private static final String ROLE_NAMES_ID = "roleNames:id:";
	private static final String ROLE_ALIAS_ID = "roleAlias:id:";
	private static final String ROLE_ALIASES_ID = "roleAliases:id:";
	public static final String TENANT_ID = "tenant:id:";
	public static final String TENANT_TENANT_ID = "tenant:tenantId:";
	public static final String TENANT_PACKAGE_ID = "tenant:packageId:";

	private static final IMenuService menuService;
	private static final IDeptService deptService;
	private static final IPostService postService;
	private static final IRoleService roleService;
	private static final IRegionService regionService;
	private static final ITenantService tenantService;
	private static final ITenantPackageService tenantPackageService;

	static {
		menuService = SpringUtil.getBean(IMenuService.class);
		deptService = SpringUtil.getBean(IDeptService.class);
		postService = SpringUtil.getBean(IPostService.class);
		roleService = SpringUtil.getBean(IRoleService.class);
		regionService = SpringUtil.getBean(IRegionService.class);
		tenantService = SpringUtil.getBean(ITenantService.class);
		tenantPackageService = SpringUtil.getBean(ITenantPackageService.class);
	}

	/**
	 * 获取菜单
	 *
	 * @param id 主键
	 * @return 菜单
	 */
	public static Menu getMenu(Long id) {
		return CacheUtil.get(SYS_CACHE, MENU_ID, id, () -> menuService.getById(id));
	}

	/**
	 * 获取部门
	 *
	 * @param id 主键
	 * @return 部门
	 */
	public static Dept getDept(Long id) {
		return CacheUtil.get(SYS_CACHE, DEPT_ID, id, () -> deptService.getById(id));
	}

	/**
	 * 获取部门id
	 *
	 * @param tenantId  租户id
	 * @param deptNames 部门名
	 * @return 部门id
	 */
	public static String getDeptIds(String tenantId, String deptNames) {
		return CacheUtil.get(SYS_CACHE, DEPT_NAME, tenantId + StringPool.DASH + deptNames, () -> deptService.getDeptIds(tenantId, deptNames));
	}

	/**
	 * 获取部门id
	 *
	 * @param tenantId  租户id
	 * @param deptNames 部门名模糊查询
	 * @return 部门id
	 */
	public static String getDeptIdsByFuzzy(String tenantId, String deptNames) {
		return CacheUtil.get(SYS_CACHE, DEPT_NAME_FUZZY, tenantId + StringPool.DASH + deptNames, () -> deptService.getDeptIdsByFuzzy(tenantId, deptNames));
	}

	/**
	 * 获取部门名
	 *
	 * @param id 主键
	 * @return 部门名
	 */
	public static String getDeptName(Long id) {
		return CacheUtil.get(SYS_CACHE, DEPT_NAME_ID, id, () -> deptService.getById(id).getDeptName());
	}


	/**
	 * 获取部门名集合
	 *
	 * @param deptIds 主键集合
	 * @return 部门名
	 */
	public static List<String> getDeptNames(String deptIds) {
		return CacheUtil.get(SYS_CACHE, DEPT_NAMES_ID, deptIds, () -> deptService.getDeptNames(deptIds));
	}

	/**
	 * 获取子部门集合
	 *
	 * @param deptId 主键
	 * @return 子部门
	 */
	public static List<Dept> getDeptChild(Long deptId) {
		return CacheUtil.get(SYS_CACHE, DEPT_CHILD_ID, deptId, () -> deptService.getDeptChild(deptId));
	}

	/**
	 * 获取子部门ID集合
	 *
	 * @param deptIds 主键
	 * @return 子部门ID
	 */
	public static List<Long> getDeptChildIds(String deptIds) {
		if (Strings.isBlank(deptIds)) {
			return null;
		}
		List<Long> list = new ArrayList<>();
		List<String> deptIdList = new ArrayList<>(Arrays.asList(deptIds.split(",")));
		for (String deptId : deptIdList) {
			List<Long> deptIdLists = CacheUtil.get(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, List.class);
			if (deptIdLists == null) {
				deptIdLists = new ArrayList<>();
				List<Dept> deptChild = getDeptChild(Long.parseLong(deptId));
				if (deptChild != null) {
					List<Long> collect = deptChild.stream().map(Dept::getId).collect(Collectors.toList());
					deptIdLists.addAll(collect);
				}
				deptIdLists.add(Long.parseLong(deptId));
				CacheUtil.put(SYS_CACHE, DEPT_CHILDIDS_ID, deptId, deptIdLists);
			}
			list.addAll(deptIdLists);
		}
		// 去重
		List<Long> collect = list.stream().distinct().collect(Collectors.toList());
		// 返回
		return collect;
	}

	/**
	 * 获取岗位
	 *
	 * @param id 主键
	 * @return
	 */
	public static Post getPost(Long id) {
		return CacheUtil.get(SYS_CACHE, POST_ID, id, () -> postService.getById(id));
	}

	/**
	 * 获取岗位id
	 *
	 * @param tenantId  租户id
	 * @param postNames 岗位名
	 * @return
	 */
	public static String getPostIds(String tenantId, String postNames) {
		return CacheUtil.get(SYS_CACHE, POST_NAME, tenantId + StringPool.DASH + postNames, () -> postService.getPostIds(tenantId, postNames));
	}

	/**
	 * 获取岗位id
	 *
	 * @param tenantId  租户id
	 * @param postNames 岗位名模糊查询
	 * @return
	 */
	public static String getPostIdsByFuzzy(String tenantId, String postNames) {
		return CacheUtil.get(SYS_CACHE, POST_NAME_FUZZY, tenantId + StringPool.DASH + postNames, () -> postService.getPostIdsByFuzzy(tenantId, postNames));
	}

	/**
	 * 获取岗位名
	 *
	 * @param id 主键
	 * @return 岗位名
	 */
	public static String getPostName(Long id) {
		return CacheUtil.get(SYS_CACHE, POST_NAME_ID, id, () -> postService.getById(id).getPostName());
	}

	/**
	 * 获取岗位名集合
	 *
	 * @param postIds 主键集合
	 * @return 岗位名
	 */
	public static List<String> getPostNames(String postIds) {
		return CacheUtil.get(SYS_CACHE, POST_NAMES_ID, postIds, () -> postService.getPostNames(postIds));
	}

	/**
	 * 获取角色
	 *
	 * @param id 主键
	 * @return Role
	 */
	public static Role getRole(Long id) {
		return CacheUtil.get(SYS_CACHE, ROLE_ID, id, () -> roleService.getById(id));
	}

	/**
	 * 获取角色id
	 *
	 * @param tenantId  租户id
	 * @param roleNames 角色名
	 * @return
	 */
	public static String getRoleIds(String tenantId, String roleNames) {
		return CacheUtil.get(SYS_CACHE, ROLE_NAME, tenantId + StringPool.DASH + roleNames, () -> roleService.getRoleIds(tenantId, roleNames));
	}

	/**
	 * 获取角色名
	 *
	 * @param id 主键
	 * @return 角色名
	 */
	public static String getRoleName(Long id) {
		return CacheUtil.get(SYS_CACHE, ROLE_NAME_ID, id, () -> roleService.getById(id).getRoleName());
	}

	/**
	 * 获取角色名集合
	 *
	 * @param roleIds 主键集合
	 * @return 角色名
	 */
	public static List<String> getRoleNames(String roleIds) {
		return CacheUtil.get(SYS_CACHE, ROLE_NAMES_ID, roleIds, () -> roleService.getRoleNames(roleIds));
	}

	/**
	 * 获取角色别名
	 *
	 * @param id 主键
	 * @return 角色别名
	 */
	public static String getRoleAlias(Long id) {
		return CacheUtil.get(SYS_CACHE, ROLE_ALIAS_ID, id, () -> roleService.getById(id).getRoleAlias());
	}

	/**
	 * 获取角色别名集合
	 *
	 * @param roleIds 主键集合
	 * @return 角色别名
	 */
	public static List<String> getRoleAliases(String roleIds) {
		return CacheUtil.get(SYS_CACHE, ROLE_ALIASES_ID, roleIds, () -> roleService.getRoleAliases(roleIds));
	}

	/**
	 * 获取租户
	 *
	 * @param id 主键
	 * @return Tenant
	 */
	public static Tenant getTenant(Long id) {
		return CacheUtil.get(SYS_CACHE, TENANT_ID, id, () -> tenantService.getById(id), Boolean.FALSE);
	}

	/**
	 * 获取租户
	 *
	 * @param tenantId 租户id
	 * @return Tenant
	 */
	public static Tenant getTenant(String tenantId) {
		return CacheUtil.get(SYS_CACHE, TENANT_TENANT_ID, tenantId, () -> tenantService.getByTenantId(tenantId), Boolean.FALSE);
	}

	/**
	 * 获取租户产品包
	 *
	 * @param tenantId 租户id
	 * @return Tenant
	 */
	public static TenantPackage getTenantPackage(String tenantId) {
		Tenant tenant = getTenant(tenantId);
		return CacheUtil.get(SYS_CACHE, TENANT_PACKAGE_ID, tenantId, () -> tenantPackageService.getById(tenant.getPackageId()), Boolean.FALSE);
	}

	/**
	 * 获取下级所有区域code
	 *
	 * @return regionCode
	 */
	public static List<String> getRegionChildCodesByDeptId(String deptId,String communityCode) {
		//多个部门按逗号分割
		List<String> deptIdList = Arrays.asList(deptId.split(","));

		//所有行政区划code
		List<String> allRegionList = new ArrayList<>();
		deptIdList.forEach(id->{
			List<String> list = new ArrayList<>();
			// 查询对应的区域编号code
			Dept dept = deptService.getById(id);
			if (null!=dept && !Strings.isBlank(dept.getRegionCode()) && !AuthUtil.isAdministrator()){
				list = getRegionChildCodes(dept.getRegionCode(),communityCode);
				//行政区划不为空添加进集合
				if (list.size()>0){
					allRegionList.addAll(list);
				}
			}
		});

		// 去重
		List<String> collect = allRegionList.stream().distinct().collect(Collectors.toList());
		return collect;
	}

	/**
	 * 获取民警（公安）下级所有区域code
	 *
	 * @return regionCode
	 */
	public static List<String> getPoliceRegionChildCodesByDeptId(String deptId,String communityCode) {
		List<String> regionCodeList = new ArrayList<>();
		if (!Strings.isBlank(deptId)) {
			//多个部门按逗号分割
			List<String> deptIdList = Arrays.asList(deptId.split(","));

			//所有行政区划code
			List<String> allRegionList = new ArrayList<>();
			deptIdList.forEach(id -> {
				List<String> list = new ArrayList<>();
				// 查询对应的区域编号code
				Dept dept = deptService.getById(id);
				// 只取公安的
				if (dept.getDeptNature() == 1) {
					if (null != dept && !Strings.isBlank(dept.getRegionCode()) && !AuthUtil.isAdministrator()) {
						list = getPoliceRegionChildCodes(dept.getRegionCode(),communityCode);
						//行政区划不为空添加进集合
						if (list.size() > 0) {
							allRegionList.addAll(list);
						}
					}
				}
			});
			// 去重
			regionCodeList = allRegionList.stream().distinct().collect(Collectors.toList());
		}
		return regionCodeList;
	}


	/**
	 * 查询综治网格/公安相关的网格，社区编号集合
	 * @param deptId
	 * @param communityCode
	 * @param roleName
	 * @return
	 */
	public static List<String> getGridRegionChildCodesByDeptId(String deptId, String communityCode,String roleName) {
		List<String> regionCodeList = new ArrayList<>();
		if (!Strings.isBlank(deptId)) {
			//多个部门按逗号分割
			List<String> deptIdList = Arrays.asList(deptId.split(","));
			List<String> finalRegionCodeList = regionCodeList;
			String key = AuthUtil.getUserId().toString();
			if (!Strings.isBlank(roleName)){
				key = key + ":" + roleName;
			}
			String finalKey = key;
			deptIdList.forEach(id -> {
				List<String> list = new ArrayList<>();
				// 查询对应的区域编号code
				Dept dept = deptService.getById(id);
				if (null != dept) {
					if ((roleName.equals("mj") && dept.getDeptNature() == 1) ||
						((roleName.equals("wgy") || roleName.equals("wzcj")) && dept.getDeptNature() == 2)) {
						if (!AuthUtil.isAdministrator()) {
							if (dept.getRegionCode() == null) {
								return;
							}
							list = CacheUtil.get(SYS_CACHE, REGION_CHILDCODES_CODE, finalKey, List.class);
							if (list == null || list.size() == 0) {
								list = new ArrayList<>();
								List<Region> deptChild = getGridRegionChild(dept.getRegionCode(), communityCode, roleName);
								if (deptChild != null) {
									List<String> collect = deptChild.stream().map(Region::getCode).collect(Collectors.toList());
									list.addAll(collect);
								}
							}
							//行政区划不为空添加进集合
							if (list.size() > 0) {
								finalRegionCodeList.addAll(list);
							}
						}
					}
				}
			});
			// 去重
			regionCodeList = finalRegionCodeList.stream().distinct().collect(Collectors.toList());
			CacheUtil.put(SYS_CACHE, REGION_CHILDCODES_CODE, key, regionCodeList);
		}
		return regionCodeList;
	}


	/**
	 * 获取下级所有区域code
	 *
	 * @return regionCode
	 */
	public static List<String> getRegionChildCodes(String regionCode,String communityCode) {
		if (regionCode == null) {
			return null;
		}
		List<String> regionCodeList = CacheUtil.get(SYS_CACHE, REGION_CHILDCODES_CODE, regionCode, List.class);
		if (regionCodeList == null || regionCodeList.size()==0) {
			regionCodeList = new ArrayList<>();
			List<Region> deptChild = getRegionChild(regionCode,communityCode);
			if (deptChild != null) {
				List<String> collect = deptChild.stream().map(Region::getCode).collect(Collectors.toList());
				regionCodeList.addAll(collect);
			}
			regionCodeList.add(regionCode);
			CacheUtil.put(SYS_CACHE, REGION_CHILDCODES_CODE, regionCode, regionCodeList);
		}
		return regionCodeList;
	}

	/**
	 * 获取公安下级所有区域code
	 *
	 * @return regionCode
	 */
	public static List<String> getPoliceRegionChildCodes(String regionCode,String communityCode) {
		if (regionCode == null) {
			return null;
		}
		List<String> regionCodeList = CacheUtil.get(SYS_CACHE, REGION_CHILDCODES_CODE, regionCode, List.class);
		if (regionCodeList == null || regionCodeList.size()==0) {
			regionCodeList = new ArrayList<>();
			List<Region> deptChild = getPoliceRegionChild(regionCode,communityCode);
			if (deptChild != null) {
				List<String> collect = deptChild.stream().map(Region::getCode).collect(Collectors.toList());
				regionCodeList.addAll(collect);
			}
			regionCodeList.add(regionCode);
			CacheUtil.put(SYS_CACHE, REGION_CHILDCODES_CODE, regionCode, regionCodeList);
		}
		return regionCodeList;
	}

	/**
	 * 获取综治下级区域
	 * @param regionCode
	 * @return
	 */
	private static List<Region> getGridRegionChild(String regionCode, String communityCode,String roleName) {
		return CacheUtil.get(SYS_CACHE, REGION_CHILD_CODE, regionCode, () -> regionService.getGridRegionChild(regionCode,communityCode,roleName));
	}

	/**
	 * 获取民警下级区域
	 * @param regionCode
	 * @return
	 */
	private static List<Region> getPoliceRegionChild(String regionCode,String communityCode) {
		return CacheUtil.get(SYS_CACHE, REGION_CHILD_CODE, regionCode, () -> regionService.getPoliceRegionChild(regionCode,communityCode));
	}

	/**
	 * 获取下级区域
	 * @param regionCode
	 * @return
	 */
	private static List<Region> getRegionChild(String regionCode,String communityCode) {
		return CacheUtil.get(SYS_CACHE, REGION_CHILD_CODE, regionCode, () -> regionService.getRegionChild(regionCode,communityCode));
	}

	/**
	 * 查询当前文章范围对应的社区编号字符串集合
	 * @param articleRange
	 * @return
	 */
	public static String getAllCommunityNameListString(String articleRange,String id) {
		return CacheUtil.get(ARTICLE_KEY, "id", id ,() -> regionService.getAllCommunityNameListString(articleRange));
	}
}