import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

/**
 * 數(shù)據脫敏工具類
 *
 * @date 2024/10/30 13:44
 */

@Slf4j
public class DataDesensitizationUtils {

    public static final String STAR_1 = "*";
    public static final String STAR_2 = "**";
    public static final String STAR_3 = "***";
    public static final String STAR_4 = "****";

    /**
     * 手機號脫敏
     *
     * @param mobile 手機號
     * @return 返回值
     */
    public static String mobileDesensitize(String mobile) {
        // 如果手機號為空則返回空字符串
        if (StringUtils.isBlank(mobile)) {
            return "";
        }

        // 取手機號最后4位
        String right = StringUtils.right(mobile, 4);
        // 手機號除了最后4位都用*代替
        String leftPad = StringUtils.leftPad(right, StringUtils.length(mobile), STAR_1);
        // 移除前面默認的三位*
        String removeStart = StringUtils.removeStart(leftPad, STAR_3);
        // 取手機號前3位
        String left = StringUtils.left(mobile, 3);
        // 字符串鏈接并返回
        return left.concat(removeStart);
    }

    /**
     * 姓名脫敏
     *
     * @param name 姓名
     * @return 返回值
     */
    public static String nameDesensitize(String name) {
        // 名字為空則返回空字符串
        if (StringUtils.isBlank(name)) {
            return "";
        }

        // 名字長度小于等于3個字符，則將前面一個字符用*代替
        int length = 3;
        if (name.length() <= length) {
            int len = name.length() - 1;
            String right = StringUtils.right(name, len);
            return STAR_1.concat(right);
        }

        // 名字長度大于3個字符，則將前兩個字符用*代替
        int len = name.length() - 2;
        String right = StringUtils.right(name, len);
        return STAR_2.concat(right);
    }

    /**
     * 地址脫敏
     *
     * @param detailAddress 地址
     * @return 返回值
     */
    public static String detailAddressDesensitize(String detailAddress) {
        int length = 10;
        String desensitizationText = STAR_4;
        // 地址為空或長度小于10個字符則默認返回*
        if (StringUtils.isBlank(detailAddress) || detailAddress.length() <= length) {
            return desensitizationText;
        } else {
            // 把地址10個字符后的所有字符替換為*
            StringBuilder address = new StringBuilder(detailAddress);
            int start = address.length() - 10;
            int end = address.length();
            return address.replace(start, end, desensitizationText).toString();
        }
    }

	/**
     * 對指定字符串的部分內容進行脫敏處理，替換為指定數(shù)量的星號。
     *
     * @param start       需要脫敏的起始位置（包含）
     * @param end         需要脫敏的結束位置（不包含）
     * @param content     原始字符串
     * @param maxStarSize 最大星號數(shù)量，超出部分將被截斷
     * @return 處理后的字符串
     */
    public static String contentDesensitize(int start, int end, String content, int maxStarSize) {
        // 字符串為空則返回空字符串
        if (StringUtils.isBlank(content)) {
            return "";
        }

        // 起始位置不能小于0
        start = Math.max(0, start);
        // 結束位置不能大于字符串最大長度
        end = Math.min(content.length(), end);

        // 如果結束位置小于起始位置或者結束位置大于原始串長度，則返回原始串。
        if (end < start) {
            return content;
        }

        // 取兩者最小值，即中間最多出現(xiàn)maxStarSize個*號
        int min = Math.min(end - start, maxStarSize);
        // 取兩者最大值作為重復次數(shù)，即保證*號重復次數(shù)不為負數(shù)
        int count = Math.max(0, min);
        // 計算count次數(shù)的*號字符串，repeat是JDK11+提供的方法
        // JDK8可以使用以下方法 
        // String repeat = StringUtils.repeat(STAR_1, count);
        String repeat = STAR_1.repeat(count);
        // 將字符串start與end中間的明文替換為*號
        return new StringBuilder(content)
                .replace(start, end, repeat)
                .toString();
    }

    /**
     * 對指定字符串的部分內容進行脫敏處理，替換為指定數(shù)量的星號。
     *
     * @param start   需要脫敏的起始位置（包含）
     * @param end     需要脫敏的結束位置（不包含）
     * @param content 原始字符串
     * @return 處理后的字符串
     */
    public static String contentDesensitize(int start, int end, String content) {
        // 默認最大*號數(shù)量即中間替換字符數(shù)量
        return contentDesensitize(start, end, content, end - start);
    }

}

@Slf4j
@ExtendWith(MockitoExtension.class)
public class DataDesensitizationUtilsTest {

    @Test
    public void testMobileDesensitize() {
        String mobile = "13011112222";
        String result = DataDesensitizationUtils.mobileDesensitize(mobile);
        log.info("手機號脫敏結果：{}", result);
    }

    @Test
    public void testNameDesensitize() {
        String name = "諸葛孔明";
        String result = DataDesensitizationUtils.nameDesensitize(name);
        log.info("姓名脫敏結果：{}", result);
    }

    @Test
    public void testAddressDesensitize() {
        String address = "北京首鋼國際會展中心9號館、10號館";
        String result = DataDesensitizationUtils.detailAddressDesensitize(address);
        log.info("地址脫敏結果：{}", result);
    }

	@Test
    public void testStringUtils() {
        String param = "北京首鋼國際會展中心9號館、10號館";
        String result1 = StringUtils.left(param, 3);
        log.info("處理結果1：{}", result1);

        String result2 = StringUtils.right(param, 4);
        log.info("處理結果2：{}", result2);

        String result3 = StringUtils.leftPad(result2, param.length(), "*");
        log.info("處理結果3：{}", result3);
    }
    
	@Test
    public void testContentDesensitize() {
        String param = "諸葛孔明";
        String result1 = DataDesensitizationUtils.contentDesensitize(1, 7, param);
        log.info("處理結果1：{}", result1);
        String result2 = DataDesensitizationUtils.contentDesensitize(1, param.length() - 7, param, 6);
        log.info("處理結果2：{}", result2);
        String result3 = DataDesensitizationUtils.contentDesensitize(0, param.length() + 7, param, 6);
        log.info("處理結果3：{}", result3);
    }

}