boyue_jnpf/boyue-web-vue3/jnpf-app-vue3/components/Jnpf/Input/useTextMask.js

/**
 * maskType
 * 1 - 全掩盖
 * 2 - 姓名-显示前1个字，后1个字
 * 3 - 手机号-显示前3位，后4位
 * 4 - 邮箱-显示前3个字，@和之后的字
 * 5 - 身份证-显示前6位，后3位，虚拟为4位
 * 6 - IP地址-显示第1段IP
 * 7 - 车牌号-显示前1个字，后2位
 * 8 - 银行卡号-显示前6位，后4位
 * 0 - 自定义规则
 */
export const defaultMaskOptions = {
	filler: '*', // 填充符号
	maskType: 1, // 掩码规则
	prefixType: 1, // 开头显示
	prefixLimit: 0, // 开头字数
	prefixSpecifyChar: '', // 开头字符
	suffixType: 1, // 结尾显示
	suffixLimit: 0, // 结尾字数
	suffixSpecifyChar: '', // 结尾字符
	ignoreChar: '', // 显示字符
	useUnrealMask: false, // 虚拟掩码
	unrealMaskLength: 1, // 虚拟掩码长度
};

export function useTextMask(options = {}) {
	const config = {
		...defaultMaskOptions,
		...options
	};

	// 全掩盖
	function maskAll(str) {
		return config.filler.repeat(str.length);
	}
	//姓名 显示前1个字，后1个字
	function maskName(str) {
		if (str.length <= 1) return str;
		const prefix = str[0];
		if (str.length === 2) return prefix + config.filler;
		const suffix = str.slice(-1);
		const maskedChars = config.filler.repeat(str.length - 2);
		return prefix + maskedChars + suffix;
	}
	// 手机号 - 显示前3位，后4位
	function maskPhoneNumber(str) {
		if (str.length <= 7) return str;
		const prefix = str.slice(0, 3);
		const suffix = str.slice(-4);
		const maskedChars = config.filler.repeat(str.length - 7);
		return prefix + maskedChars + suffix;
	}
	// 邮箱 - 显示前3个字，@和之后的字
	function maskEmailAddress(str) {
		const atIndex = str.indexOf('@');
		if (str.length <= 3 || (atIndex > -1 && atIndex < 3)) return str;
		let suffixLength = 0;
		let maskedCharsLength = str.length - 3;
		if (atIndex > 0) {
			suffixLength = atIndex;
			maskedCharsLength = atIndex - 3;
		}
		const prefix = str.slice(0, 3);
		const suffix = suffixLength ? str.slice(suffixLength) : '';
		const maskedChars = config.filler.repeat(maskedCharsLength);
		return prefix + maskedChars + suffix;
	}
	// 身份证 - 显示前6位，后3位，虚拟为4位
	function maskIdNumber(str) {
		if (str.length <= 9) return str;
		const prefix = str.slice(0, 6);
		const suffix = str.slice(-3);
		const maskedChars = config.filler.repeat(4);
		return prefix + maskedChars + suffix;
	}
	// IP地址-显示第1段IP
	function maskIPAddress(str) {
		const segments = str.split('.');
		if (segments.length < 1) return str;
		const maskedChars = ('.' + config.filler.repeat(3)).repeat(3);
		return segments[0] + maskedChars;
	}
	// 车牌号-显示前1个字，后2位
	function maskLicensePlate(str) {
		if (str.length <= 3) return str;
		const prefix = str[0];
		const suffix = str.slice(-2);
		const maskedChars = config.filler.repeat(str.length - 3);
		return prefix + maskedChars + suffix;
	}
	// 银行卡号-显示前6位，后4位
	function maskBankCard(str) {
		if (str.length <= 10) return str;
		const prefix = str.slice(0, 6);
		const suffix = str.slice(-4);
		const maskedChars = config.filler.repeat(str.length - 10);
		return prefix + maskedChars + suffix;
	}
	// 自定义掩码规则
	function maskCustom(str) {
		let prefixLength = 0,
			suffixLength = 0;
		if (config.prefixType === 2) prefixLength = config.prefixLimit || 0;
		if ((config.prefixType === 3 || config.prefixType === 4) && config.prefixSpecifyChar) {
			let specifyCharIndex = str.indexOf(config.prefixSpecifyChar);
			if (specifyCharIndex > -1) prefixLength = config.prefixType === 3 ? specifyCharIndex :
				specifyCharIndex + config.prefixSpecifyChar.length;
		}
		if (config.suffixType === 2) suffixLength = config.suffixLimit || 0;
		if ((config.suffixType === 3 || config.suffixType === 4) && config.suffixSpecifyChar) {
			let specifyCharIndex = str.indexOf(config.suffixSpecifyChar);
			if (specifyCharIndex > -1) suffixLength = config.suffixType === 3 ?
				str.length - specifyCharIndex - config.suffixSpecifyChar.length :
				str.length - specifyCharIndex;
		}
		if (prefixLength + suffixLength >= str.length) return str;
		const prefix = prefixLength ? str.slice(0, prefixLength) : '';
		const suffix = suffixLength ? str.slice(-suffixLength) : '';
		let middleChar = '';
		if (!config.ignoreChar) {
			const maskedLength = config.useUnrealMask ? config.unrealMaskLength || 1 : str.length - prefixLength -
				suffixLength;
			middleChar = config.filler.repeat(maskedLength);
		} else {
			const ignoreCharList = config.ignoreChar.split(',');
			const middleStr = str.slice(prefixLength, str.length - suffixLength);
			const reg = new RegExp('(' + ignoreCharList.map(o => o.replace(/\*/g, '\\*')).join('|') + ')', 'g');
			let list = middleStr.split(reg);
			list = list.map(o => {
				if (o && !ignoreCharList.includes(o)) {
					const maskedLength = config.useUnrealMask ? config.unrealMaskLength || 1 : o.length;
					o = config.filler.repeat(maskedLength);
				}
				return o;
			});
			middleChar = list.join('');
		}

		return prefix + middleChar + suffix;
	}

	// 获取掩码后文本
	function getMaskedText(str) {
		if (!str) return '';
		if (config.maskType === 1) return maskAll(str);
		if (config.maskType === 2) return maskName(str);
		if (config.maskType === 3) return maskPhoneNumber(str);
		if (config.maskType === 4) return maskEmailAddress(str);
		if (config.maskType === 5) return maskIdNumber(str);
		if (config.maskType === 6) return maskIPAddress(str);
		if (config.maskType === 7) return maskLicensePlate(str);
		if (config.maskType === 8) return maskBankCard(str);
		if (config.maskType === 0) return maskCustom(str);
		return str;
	}

	return {
		maskAll,
		maskName,
		maskPhoneNumber,
		maskEmailAddress,
		maskIdNumber,
		maskIPAddress,
		maskLicensePlate,
		maskBankCard,
		maskCustom,
		getMaskedText
	};
}