import { ref, unref } from 'vue';
|
|
export interface MaskOptions {
|
filler: string;
|
ignoreChar: string;
|
maskType: number;
|
prefixLimit: number;
|
prefixSpecifyChar: string;
|
prefixType: number;
|
suffixLimit: number;
|
suffixSpecifyChar: string;
|
suffixType: number;
|
unrealMaskLength: number;
|
useUnrealMask: boolean;
|
}
|
|
/**
|
* 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: MaskOptions = {
|
filler: '*', // 填充符号
|
ignoreChar: '', // 显示字符
|
maskType: 1, // 掩码规则
|
prefixLimit: 0, // 开头字数
|
prefixSpecifyChar: '', // 开头字符
|
prefixType: 1, // 开头显示
|
suffixLimit: 0, // 结尾字数
|
suffixSpecifyChar: '', // 结尾字符
|
suffixType: 1, // 结尾显示
|
unrealMaskLength: 1, // 虚拟掩码长度
|
useUnrealMask: false, // 虚拟掩码
|
};
|
|
export function useTextMask(options?: Nullable<Partial<MaskOptions>>) {
|
const config = ref({ ...defaultMaskOptions, ...options });
|
|
// 全掩盖
|
function maskAll(str: string): string {
|
return unref(config).filler.repeat(str.length);
|
}
|
// 姓名 显示前1个字,后1个字
|
function maskName(str: string): string {
|
if (str.length <= 1) return str;
|
const prefix = str[0];
|
if (str.length === 2) return prefix + unref(config).filler;
|
const suffix = str.slice(-1);
|
const maskedChars = unref(config).filler.repeat(str.length - 2);
|
return prefix + maskedChars + suffix;
|
}
|
// 手机号 - 显示前3位,后4位
|
function maskPhoneNumber(str: string): string {
|
if (str.length <= 7) return str;
|
const prefix = str.slice(0, 3);
|
const suffix = str.slice(-4);
|
const maskedChars = unref(config).filler.repeat(str.length - 7);
|
return prefix + maskedChars + suffix;
|
}
|
// 邮箱 - 显示前3个字,@和之后的字
|
function maskEmailAddress(str: string): string {
|
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 = unref(config).filler.repeat(maskedCharsLength);
|
return prefix + maskedChars + suffix;
|
}
|
// 身份证 - 显示前6位,后3位,虚拟为4位
|
function maskIdNumber(str: string): string {
|
if (str.length <= 9) return str;
|
const prefix = str.slice(0, 6);
|
const suffix = str.slice(-3);
|
const maskedChars = unref(config).filler.repeat(4);
|
return prefix + maskedChars + suffix;
|
}
|
// IP地址-显示第1段IP
|
function maskIPAddress(str: string): string {
|
const segments = str.split('.');
|
if (segments.length === 0) return str;
|
const maskedChars = `.${unref(config).filler.repeat(3)}`.repeat(3);
|
return segments[0] + maskedChars;
|
}
|
// 车牌号-显示前1个字,后2位
|
function maskLicensePlate(str: string): string {
|
if (str.length <= 3) return str;
|
const prefix = str[0];
|
const suffix = str.slice(-2);
|
const maskedChars = unref(config).filler.repeat(str.length - 3);
|
return prefix + maskedChars + suffix;
|
}
|
// 银行卡号-显示前6位,后4位
|
function maskBankCard(str: string): string {
|
if (str.length <= 10) return str;
|
const prefix = str.slice(0, 6);
|
const suffix = str.slice(-4);
|
const maskedChars = unref(config).filler.repeat(str.length - 10);
|
return prefix + maskedChars + suffix;
|
}
|
// 自定义掩码规则
|
function maskCustom(str: string): string {
|
let prefixLength = 0;
|
let suffixLength = 0;
|
if (unref(config).prefixType === 2)
|
prefixLength = unref(config).prefixLimit || 0;
|
if (
|
(unref(config).prefixType === 3 || unref(config).prefixType === 4) &&
|
unref(config).prefixSpecifyChar
|
) {
|
const specifyCharIndex = str.indexOf(unref(config).prefixSpecifyChar);
|
if (specifyCharIndex !== -1)
|
prefixLength =
|
unref(config).prefixType === 3
|
? specifyCharIndex
|
: specifyCharIndex + unref(config).prefixSpecifyChar.length;
|
}
|
if (unref(config).suffixType === 2)
|
suffixLength = unref(config).suffixLimit || 0;
|
if (
|
(unref(config).suffixType === 3 || unref(config).suffixType === 4) &&
|
unref(config).suffixSpecifyChar
|
) {
|
const specifyCharIndex = str.indexOf(unref(config).suffixSpecifyChar);
|
if (specifyCharIndex !== -1)
|
suffixLength =
|
unref(config).suffixType === 3
|
? str.length -
|
specifyCharIndex -
|
unref(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 (unref(config).ignoreChar) {
|
const ignoreCharList = unref(config).ignoreChar.split(',');
|
const middleStr = str.slice(prefixLength, str.length - suffixLength);
|
const reg = new RegExp(
|
`(${ignoreCharList.map((o) => o.replaceAll('*', String.raw`\*`)).join('|')})`,
|
'g',
|
);
|
let list = middleStr.split(reg);
|
list = list.map((o) => {
|
if (o && !ignoreCharList.includes(o)) {
|
const maskedLength = unref(config).useUnrealMask
|
? unref(config).unrealMaskLength || 1
|
: o.length;
|
o = unref(config).filler.repeat(maskedLength);
|
}
|
return o;
|
});
|
middleChar = list.join('');
|
} else {
|
const maskedLength = unref(config).useUnrealMask
|
? unref(config).unrealMaskLength || 1
|
: str.length - prefixLength - suffixLength;
|
middleChar = unref(config).filler.repeat(maskedLength);
|
}
|
|
return prefix + middleChar + suffix;
|
}
|
|
// 获取掩码后文本
|
function getMaskedText(str: string): string {
|
if (!str) return '';
|
if (unref(config).maskType === 1) return maskAll(str);
|
if (unref(config).maskType === 2) return maskName(str);
|
if (unref(config).maskType === 3) return maskPhoneNumber(str);
|
if (unref(config).maskType === 4) return maskEmailAddress(str);
|
if (unref(config).maskType === 5) return maskIdNumber(str);
|
if (unref(config).maskType === 6) return maskIPAddress(str);
|
if (unref(config).maskType === 7) return maskLicensePlate(str);
|
if (unref(config).maskType === 8) return maskBankCard(str);
|
if (unref(config).maskType === 0) return maskCustom(str);
|
return str;
|
}
|
|
return {
|
getMaskedText,
|
maskAll,
|
maskBankCard,
|
maskCustom,
|
maskEmailAddress,
|
maskIdNumber,
|
maskIPAddress,
|
maskLicensePlate,
|
maskName,
|
maskPhoneNumber,
|
};
|
}
|