package org.springblade.common.desensitization;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.util.DesensitizedUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.JSONPath;
import liquibase.util.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springblade.core.secure.utils.AuthUtil;
import org.springframework.stereotype.Component;

import static cn.hutool.core.util.DesensitizedUtil.DesensitizedType.FIXED_PHONE;

/**
 * 数据脱敏切面
 * @author zhongrj
 * @date 2023-09-11
 */
@Slf4j
@Aspect
@Component
public class DesensitizationAspect {

	/**
	 * 脱敏环绕通知
	 * @param pjp
	 * @param desensitizationWord
	 * @return
	 * @throws Throwable
	 */
	@Around("@annotation(desensitizationWord)")
	public Object around(ProceedingJoinPoint pjp, DesensitizationWord desensitizationWord) throws Throwable{
		Object object = pjp.proceed();
		// 判断是否需要脱敏处理
		if (isDesensitization()) {
			try {
				Class<?> aClass = object.getClass();
				// 改变返回值
				Desensitization[] value = desensitizationWord.value();
				JSONObject ob = JSON.parseObject(JSON.toJSONString(object));

				// 循环处理每一个 json 路径下的值
				for (Desensitization s : value) {
					replace(ob, s);
				}
				object = JSON.parseObject(JSON.toJSONString(ob), aClass);
			} catch (Exception e) {
				log.info("数据脱敏失败：" + e.getMessage());
			}
		}
		return object;
	}

	/**
	 * 判断是否需要脱敏
	 * @return
	 */
	private boolean isDesensitization() {
		// 获取用户角色，根据角色判断是否进行脱敏处理
		String userRole = AuthUtil.getUserRole();
		// 如果用户角色为null，则需要脱敏
		if (null!=userRole && !userRole.equals("")){
			// 判断角色是否为民警角色，如果是则不需要脱敏，否则需要脱敏处理
			if (userRole.equals("民警")){
				return false;
			}
		}
		return true;
	}

	/**
	 * 脱敏
	 */
	private Object sensitiveByString(Object value) {
		if (StringUtils.isNotEmpty(value.toString())) {
			String st = Convert.toStr(value);
			st = st.substring(0, st.length() - 3 > 0 ? 3 : st.length()) + "****" + st.substring(Math.max(st.length() - 4, 0));
			return st;
		}
		return value;
	}

	/**
	 * 敏感数据替换
	 *
	 * @param jsonObject
	 * @param s
	 */
	private void replace(JSONObject jsonObject, Desensitization s) {
		// 只有传入的 JSON 路径在 这个 JSONObject 中才会进行脱敏处理
		if (JSONPath.contains(jsonObject, s.jsonPath())) {

			// 查询是否有数组 列表 类型的数据需要脱敏
			int index = s.jsonPath().lastIndexOf("[*]");
			if (index > -1) {
				String prefix = StrUtil.subPre(s.jsonPath(), index);
				String suffix = StrUtil.subSuf(s.jsonPath(), index + 3);
				// 提取json 路径下的 数组\链表 元素
				Object eval = JSONPath.eval(jsonObject, prefix);

				// 将数组\链表 元素 转为 JSONArray 方便做 统一格式处理
				JSONArray jsonArray = (JSONArray) eval;
				int size = jsonArray.size();
				for (int i = 0; i < size; i++) {
					// 由于脱敏数组内部的参数传入格式为 ：jsonPath = "$.data.records[*].username"
					// 所以需要重新组装 jsonPath 将 * 号 替换成具体的值
					String indexJsonPath = StrUtil.strBuilder().append(prefix).append("[").append(i).append("]").append(suffix).toString();
					// 使用 cn.hutool.core.convert Convert.toStr 转换为字符串 如果给定的值为null，或者转换失败，返回默认值null，这样可以减少报错，避免程序异常
					String desensitized = Convert.toStr(JSONPath.eval(jsonObject, indexJsonPath));
					if (StrUtil.isBlank(desensitized)) {
						continue;
					}
					// 如果是默认指定 则使用默认方式脱敏
					if (s.desensitizedType() == FIXED_PHONE) {
						desensitized = sensitiveByString(desensitized).toString();
					}else {
						// 否则使用 cn.hutool.core.util 进行数据脱敏
						desensitized = DesensitizedUtil.desensitized(desensitized, s.desensitizedType());
					}

					// 使用JSON 路径操作，将已经脱敏的新数据，放入之前未脱敏的数据地址处，替换未脱敏数据
					JSONPath.set(jsonObject, indexJsonPath, desensitized);
				}
			} else {
				// 使用 cn.hutool.core.convert Convert.toStr 转换为字符串 如果给定的值为null，或者转换失败，返回默认值null，这样可以减少报错，避免程序异常
				Object eval = JSONPath.eval(jsonObject, Convert.toStr(s.jsonPath()));
				String desensitized = "";
				if (s.desensitizedType() == FIXED_PHONE) {
					desensitized = sensitiveByString(Convert.toStr(eval)).toString();
				}else {
					desensitized = DesensitizedUtil.desensitized(Convert.toStr(eval), s.desensitizedType());
				}
				JSONPath.set(jsonObject, s.jsonPath(), desensitized);
			}
		}
	}
}