package com.project.common.utils;

import com.project.common.annotation.Sensitive;

import java.lang.reflect.Field;
import java.util.regex.Pattern;

/**
 * 功能描述：
 *
 * @author ZQN
 * @version 1.0   2025-03-25 17:04
 */
public class SensitiveUtil {

    public static Object desensitize(Object obj) {
        if (obj == null) {
            return null;
        }
        Class<?> clazz = obj.getClass();
        Field[] fields = clazz.getDeclaredFields();
        for (Field field : fields) {
            if (field.isAnnotationPresent(Sensitive.class)) {
                Sensitive sensitive = field.getAnnotation(Sensitive.class);
                field.setAccessible(true);
                try {
                    Object value = field.get(obj);
                    if (value != null && value instanceof String) {
                        String strValue = (String) value;
                        String desensitizedValue = desensitizeByType(strValue, sensitive.type());
                        field.set(obj, desensitizedValue);
                    }
                } catch (IllegalAccessException e) {
                    e.printStackTrace();
                }
            }
        }
        return obj;
    }

    private static String desensitizeByType(String value, Sensitive.SensitiveType type) {
        switch (type) {
            case PHONE:
                return desensitizePhone(value);
            case EMAIL:
                return desensitizeEmail(value);
            case ID_CARD:
                return desensitizeIdCard(value);
            default:
                return desensitizeDefault(value);
        }
    }

    private static String desensitizeDefault(String value) {
        if (value.length() <= 1) {
            return value;
        }
        StringBuilder sb = new StringBuilder();
        sb.append(value.charAt(0));
        if (value.length() > 2) {
            for (int i = 0; i < value.length() - 2; i++) {
                sb.append("*");
            }
            sb.append(value.charAt(value.length() - 1));
        } else {
            sb.append("*");
        }
        return sb.toString();
    }

    private static String desensitizePhone(String phone) {
        if (Pattern.matches("^\\d{11}$", phone)) {
            return phone.replaceAll("(\\d{3})\\d{4}(\\d{4})", "$1****$2");
        }
        return phone;
    }

    private static String desensitizeEmail(String email) {
        if (email.contains("@")) {
            String[] parts = email.split("@");
            String name = parts[0];
            if (name.length() <= 1) {
                return email;
            }
            StringBuilder sb = new StringBuilder();
            sb.append(name.charAt(0));
            for (int i = 0; i < name.length() - 2; i++) {
                sb.append("*");
            }
            sb.append(name.charAt(name.length() - 1));
            return sb.toString() + "@" + parts[1];
        }
        return email;
    }

    private static String desensitizeIdCard(String idCard) {
        if (Pattern.matches("^\\d{18}$", idCard)) {
            return idCard.replaceAll("(\\d{4})\\d{10}(\\d{4})", "$1**********$2");
        }
        return idCard;
    }
}