308 lines
10 KiB
JavaScript
308 lines
10 KiB
JavaScript
|
class AddressDiscern {
|
|||
|
|
// 初始化数据源
|
|||
|
|
constructor(dataSource = {}) {
|
|||
|
|
this.dataSource = dataSource;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 智能解析收货地址
|
|||
|
|
discern(text) {
|
|||
|
|
if (!text) return { code: -1, msg: "地址文本不能为空" };
|
|||
|
|
// 清理文本，去除多余空格
|
|||
|
|
text = text.trim();
|
|||
|
|
// 先解析地址，因为地址最容易识别
|
|||
|
|
const positionRes = this.regionDiscern(text);
|
|||
|
|
if (positionRes.code !== 0) return positionRes;
|
|||
|
|
// 再解析姓名和手机号
|
|||
|
|
const { name, mobile } = this.nameMobileDiscern(positionRes.not_address_text);
|
|||
|
|
const status = name && mobile && positionRes.data ? 1 : 0;
|
|||
|
|
let msg = "ok";
|
|||
|
|
if (status === 0) {
|
|||
|
|
msg = "未识别：";
|
|||
|
|
if (!name) msg += "姓名、";
|
|||
|
|
if (!mobile) msg += "手机号、";
|
|||
|
|
msg = msg.slice(0, -1);
|
|||
|
|
}
|
|||
|
|
const res = {
|
|||
|
|
code: 0,
|
|||
|
|
msg,
|
|||
|
|
status, // 1 表示全部识别成功，0 表示部分成功
|
|||
|
|
data: {
|
|||
|
|
name, // 姓名
|
|||
|
|
mobile, // 手机号
|
|||
|
|
position: positionRes.data, // 省市区街道信息
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
return res;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 智能识别省市区
|
|||
|
|
regionDiscern(addressText) {
|
|||
|
|
const { provinces, citys, areas } = this.dataSource;
|
|||
|
|
const province = {};
|
|||
|
|
const city = {};
|
|||
|
|
const area = {};
|
|||
|
|
let address = "";
|
|||
|
|
|
|||
|
|
if (!addressText) return { code: -1, msg: "地址文本不能为空" };
|
|||
|
|
addressText = addressText.trim();
|
|||
|
|
|
|||
|
|
// 获取省份索引
|
|||
|
|
const findProvinceIndex = () => {
|
|||
|
|
const index = provinces.findIndex(({ name }) => addressText.includes(name.substring(0, 2)));
|
|||
|
|
return index;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 获取地级市索引
|
|||
|
|
const findCityIndex = (citys) => {
|
|||
|
|
const index = citys.findIndex(({ name }) => addressText.includes(name.slice(0, -1)));
|
|||
|
|
return index;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 获取县级市索引
|
|||
|
|
const findAreaIndex = (areas) => {
|
|||
|
|
const index = areas.findIndex(({ name }) => {
|
|||
|
|
const reg = name.length > 2 ? `${name}|${name.slice(0, -1)}` : name;
|
|||
|
|
const areaRegExp = new RegExp(reg);
|
|||
|
|
if (areaRegExp.test(addressText)) {
|
|||
|
|
address = addressText.replace(areaRegExp, "{{~}}").split("{{~}}")[1] || "";
|
|||
|
|
address = address.split(new RegExp("[^\\u4e00-\\u9fa5a-zA-Z0-9+-（）()]+", "g"))[0];
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
});
|
|||
|
|
return index;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 通过市倒推省份
|
|||
|
|
const findProvinceByCity = () => {
|
|||
|
|
for (let i = 0; i < citys.length; i++) {
|
|||
|
|
const index = findCityIndex(citys[i]);
|
|||
|
|
if (index !== -1) {
|
|||
|
|
return { provinceIndex: i, cityIndex: index };
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return { provinceIndex: -1, cityIndex: -1 };
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 通过区倒推市和省份
|
|||
|
|
const findProvinceByArea = () => {
|
|||
|
|
for (let i = 0; i < areas.length; i++) {
|
|||
|
|
for (let j = 0; j < areas[i].length; j++) {
|
|||
|
|
const index = findAreaIndex(areas[i][j]);
|
|||
|
|
if (index !== -1) {
|
|||
|
|
return { provinceIndex: i, cityIndex: j, areaIndex: index };
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return { provinceIndex: -1, cityIndex: -1, areaIndex: -1 };
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// 省
|
|||
|
|
let provinceIndex = findProvinceIndex();
|
|||
|
|
let cityIndex = -1;
|
|||
|
|
let areaIndex = -1;
|
|||
|
|
|
|||
|
|
if (provinceIndex === -1) {
|
|||
|
|
// 如果省份没有找到，通过市倒推省份
|
|||
|
|
const cityResult = findProvinceByCity();
|
|||
|
|
provinceIndex = cityResult.provinceIndex;
|
|||
|
|
cityIndex = cityResult.cityIndex;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (provinceIndex === -1) {
|
|||
|
|
// 如果市没有找到，通过区倒推市和省份
|
|||
|
|
const areaResult = findProvinceByArea();
|
|||
|
|
provinceIndex = areaResult.provinceIndex;
|
|||
|
|
cityIndex = areaResult.cityIndex;
|
|||
|
|
areaIndex = areaResult.areaIndex;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (provinceIndex === -1) return { code: -1, msg: "省份没有找到，请输入正确的地址" };
|
|||
|
|
|
|||
|
|
Object.assign(province, provinces[provinceIndex]);
|
|||
|
|
|
|||
|
|
const cityList = citys[provinceIndex];
|
|||
|
|
// 市
|
|||
|
|
if (cityIndex === -1) {
|
|||
|
|
cityIndex = findCityIndex(cityList);
|
|||
|
|
if (cityIndex === -1) return { code: -1, msg: "地级市没有找到，请输入正确的地址" };
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Object.assign(city, cityList[cityIndex]);
|
|||
|
|
|
|||
|
|
// 区
|
|||
|
|
const areaList = areas[provinceIndex][cityIndex];
|
|||
|
|
if (areaIndex === -1) {
|
|||
|
|
areaIndex = findAreaIndex(areaList);
|
|||
|
|
if (areaIndex === -1) return { code: -1, msg: "县级市没有找到，请输入正确的地址" };
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Object.assign(area, areaList[areaIndex]);
|
|||
|
|
|
|||
|
|
const formatted_address = `${province.name}${city.name}${area.name}${address}`;
|
|||
|
|
const provinceName = province.name.substring(0, 2); // 省份名称前两个字
|
|||
|
|
const not_address_text = (addressText.substring(0, addressText.indexOf(provinceName)) + addressText.substring(addressText.indexOf(address) + address.length)).trim();
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
msg: "ok",
|
|||
|
|
data: {
|
|||
|
|
province,
|
|||
|
|
city,
|
|||
|
|
area,
|
|||
|
|
address,
|
|||
|
|
formatted_address
|
|||
|
|
},
|
|||
|
|
not_address_text
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 智能解析姓名和手机号
|
|||
|
|
nameMobileDiscern(text) {
|
|||
|
|
if (!text) return { name: "", mobile: "" };
|
|||
|
|
|
|||
|
|
let name = "";
|
|||
|
|
let mobile = "";
|
|||
|
|
|
|||
|
|
// 手机号正则匹配（支持多种格式）
|
|||
|
|
const mobilePatterns = [
|
|||
|
|
new RegExp('1\\d{2}([\\s-]?\\d{4}){2}', 'g')
|
|||
|
|
];
|
|||
|
|
let mobileMatchText;
|
|||
|
|
// 查找手机号
|
|||
|
|
for (const pattern of mobilePatterns) {
|
|||
|
|
const matches = text.match(pattern);
|
|||
|
|
if (matches && matches.length > 0) {
|
|||
|
|
// 取第一个匹配的手机号，去除格式符号
|
|||
|
|
mobile = matches[0].replace(new RegExp("[\\s\\-\\.]", "g"), "");
|
|||
|
|
mobileMatchText = matches[0];
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 从文本中移除手机号，便于后续识别姓名
|
|||
|
|
let cleanText = text;
|
|||
|
|
if (mobile) {
|
|||
|
|
cleanText = cleanText.replace(mobileMatchText, "");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 处理常见的标签前缀（扩展更多模式）
|
|||
|
|
const labelPatterns = [
|
|||
|
|
new RegExp("^姓名[：:]\\s*"),
|
|||
|
|
new RegExp("^收货人[：:]\\s*"),
|
|||
|
|
new RegExp("^收件人[：:]\\s*"),
|
|||
|
|
new RegExp("^联系人[：:]\\s*"),
|
|||
|
|
new RegExp("^name[：:]\\s*", "i"),
|
|||
|
|
new RegExp("^收[：:]\\s*"),
|
|||
|
|
new RegExp("^人[：:]\\s*"),
|
|||
|
|
new RegExp("^电话[：:]\\s*"),
|
|||
|
|
new RegExp("^手机[：:]\\s*"),
|
|||
|
|
new RegExp("^tel[：:]\\s*", "i"),
|
|||
|
|
new RegExp("^phone[：:]\\s*", "i"),
|
|||
|
|
new RegExp("^mobile[：:]\\s*", "i"),
|
|||
|
|
new RegExp("^地址[：:]\\s*"),
|
|||
|
|
new RegExp("^address[：:]\\s*", "i"),
|
|||
|
|
new RegExp("^收货地址[：:]\\s*"),
|
|||
|
|
new RegExp("^收件地址[：:]\\s*"),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
// 移除标签前缀
|
|||
|
|
for (const pattern of labelPatterns) {
|
|||
|
|
cleanText = cleanText.replace(pattern, "");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 清理多余的空格和标点
|
|||
|
|
cleanText = cleanText.replace(new RegExp("\\s+", "g"), " ").trim();
|
|||
|
|
|
|||
|
|
// 姓名识别规则（按优先级排序，更具体的模式在前）
|
|||
|
|
const namePatterns = [
|
|||
|
|
// 英文+称谓（如：vk先生、John先生、Mary女士等）- 最具体，优先匹配
|
|||
|
|
new RegExp("[A-Za-z]+[\\u4e00-\\u9fa5]{1,3}", "g"),
|
|||
|
|
// 中文姓名（2-6个汉字，支持复姓和少数民族姓名）
|
|||
|
|
new RegExp("[\\u4e00-\\u9fa5]{2,6}", "g"),
|
|||
|
|
// 英文姓名（支持多种格式：全名、简称、首字母等）
|
|||
|
|
new RegExp("[A-Za-z]+(\\s+[A-Za-z]+)*", "g"),
|
|||
|
|
];
|
|||
|
|
|
|||
|
|
// 查找姓名
|
|||
|
|
for (const pattern of namePatterns) {
|
|||
|
|
const matches = cleanText.match(pattern);
|
|||
|
|
if (matches && matches.length > 0) {
|
|||
|
|
// 过滤掉一些明显不是姓名的词
|
|||
|
|
const validNames = matches.filter(match => {
|
|||
|
|
// 过滤掉常见的地址词汇
|
|||
|
|
const addressWords = ['省', '市', '区', '县', '镇', '村', '路', '街', '号', '楼', '室', '单元', '栋', '层', '小区', '大厦', '广场', '花园'];
|
|||
|
|
const isAddressWord = addressWords.some(word => match.includes(word));
|
|||
|
|
|
|||
|
|
// 过滤掉常见的标签词汇
|
|||
|
|
const labelWords = ['姓名', '收货人', '收件人', '联系人', '收', '人', '电话', '手机', '地址', '收货', '收件', '联系'];
|
|||
|
|
const isLabelWord = labelWords.some(word => match.includes(word));
|
|||
|
|
|
|||
|
|
// 过滤掉纯数字
|
|||
|
|
const isNumber = new RegExp("^\\d+$").test(match);
|
|||
|
|
|
|||
|
|
// 过滤掉太短的词（但允许英文简称，如vk、ab等）
|
|||
|
|
const isTooShort = match.length < 2;
|
|||
|
|
|
|||
|
|
// 过滤掉太长的词（中文超过8个字，英文超过25个字符可能是地址或其他信息）
|
|||
|
|
// 增加长度限制以支持"英文+称谓"的格式
|
|||
|
|
const isChinese = new RegExp("[\\u4e00-\\u9fa5]").test(match);
|
|||
|
|
const isEnglishWithTitle = new RegExp("^[A-Za-z]+[\\u4e00-\\u9fa5]{1,3}$").test(match);
|
|||
|
|
const isTooLong = isEnglishWithTitle ? false : (isChinese ? match.length > 8 : match.length > 25);
|
|||
|
|
|
|||
|
|
// 过滤掉包含特殊字符的词（但允许英文中的空格和中文）
|
|||
|
|
const hasSpecialChars = isChinese ?
|
|||
|
|
new RegExp("[^\\u4e00-\\u9fa5a-zA-Z\\s]").test(match) :
|
|||
|
|
new RegExp("[^a-zA-Z\\s]").test(match);
|
|||
|
|
|
|||
|
|
return !isAddressWord && !isLabelWord && !isNumber && !isTooShort && !isTooLong && !hasSpecialChars;
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
if (validNames.length > 0) {
|
|||
|
|
// 优先选择较短的姓名（通常姓名不会太长）
|
|||
|
|
name = validNames.sort((a, b) => a.length - b.length)[0];
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果没有通过正则找到姓名，尝试其他方法
|
|||
|
|
if (!name) {
|
|||
|
|
// 尝试从剩余文本中提取可能的姓名
|
|||
|
|
const remainingText = cleanText.trim();
|
|||
|
|
if (remainingText) {
|
|||
|
|
// 按空格或标点符号分割
|
|||
|
|
const parts = remainingText.split(new RegExp("[\\s，,。.！!？?；;：:]"));
|
|||
|
|
for (const part of parts) {
|
|||
|
|
const trimmedPart = part.trim();
|
|||
|
|
// 检查是否是中文姓名（2-6个汉字）、英文姓名（2-20个字母）或英文+称谓
|
|||
|
|
const isChineseName = new RegExp("^[\\u4e00-\\u9fa5]{2,6}$").test(trimmedPart);
|
|||
|
|
const isEnglishName = new RegExp("^[a-zA-Z]{2,20}$").test(trimmedPart);
|
|||
|
|
const isEnglishWithTitle = new RegExp("^[a-zA-Z]+[\\u4e00-\\u9fa5]{1,3}$").test(trimmedPart);
|
|||
|
|
|
|||
|
|
if (isChineseName || isEnglishName || isEnglishWithTitle) {
|
|||
|
|
// 再次过滤地址词汇和标签词汇
|
|||
|
|
const addressWords = ['省', '市', '区', '县', '镇', '村', '路', '街', '号', '楼', '室', '单元', '栋', '层', '小区', '大厦', '广场', '花园'];
|
|||
|
|
const labelWords = ['姓名', '收货人', '收件人', '联系人', '收', '人', '电话', '手机', '地址', '收货', '收件', '联系'];
|
|||
|
|
const isAddressWord = addressWords.some(word => trimmedPart.includes(word));
|
|||
|
|
const isLabelWord = labelWords.some(word => trimmedPart.includes(word));
|
|||
|
|
if (!isAddressWord && !isLabelWord) {
|
|||
|
|
name = trimmedPart;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 最终清理：去除姓名中的多余空格
|
|||
|
|
if (name) {
|
|||
|
|
name = name.replace(new RegExp("\\s+", "g"), "").trim();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return { name, mobile };
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
export default AddressDiscern;
|