JavaScript妙用localeCompare实现获取汉字拼音首字母

/**
 * 获取汉字的拼音首字母
 * @param chineseChar 中文字符串，若传入多字符则只取第一个字符
 * @param useUpperCase 是否返回大写字母，默认为false（小写）
 * @returns 拼音首字母，若无法识别则返回空字符串
 */
export const getTheFirstLetterForPinyin = (chineseChar: string = '', useUpperCase: boolean = false): string => {
    // 兼容性检查：确保浏览器支持 localeCompare 方法
    if (!String.prototype.localeCompare) {
        return '';
    }

    // 参数验证：确保输入为有效字符串
    if (typeof chineseChar !== 'string' || !chineseChar.length) {
        return '';
    }

    // 准备用于比较的字母表和基准汉字
    // 注：这些基准汉字分别对应A、B、C...等拼音首字母的起始位置
    const letters = 'ABCDEFGHJKLMNOPQRSTWXYZ'.split('');
    const zh = '阿八嚓哒妸发旮哈讥咔垃痳拏噢妑七呥扨它穵夕丫帀'.split('');

    let firstLetter = '';
    const firstChar = chineseChar[0];

    // 处理字母和数字：直接返回原字符
    if (/^\w/.test(firstChar)) {
        firstLetter = firstChar;
    } else {
        // 处理汉字：通过比较确定拼音首字母范围
        letters.some((item, index) => {
            // 检查当前字符是否在当前基准汉字与下一个基准汉字之间
            if (firstChar.localeCompare(zh[index]) >= 0 && 
                (index === letters.length - 1 || firstChar.localeCompare(zh[index + 1]) < 0)) {
                firstLetter = item;
                return true;
            }
            return false;
        });
    }

    // 根据参数决定返回大写还是小写字母
    return useUpperCase ? firstLetter.toUpperCase() : firstLetter.toLowerCase();
}

// 获取单个汉字的拼音首字母
getTheFirstLetterForPinyin('你'); // => 'n'
getTheFirstLetterForPinyin('好', true); // => 'H'
getTheFirstLetterForPinyin('123'); // => '1'
getTheFirstLetterForPinyin(''); // => ''

/**
 * 获取整个字符串的拼音首字母缩写
 * @param str 输入字符串
 * @returns 拼音首字母缩写
 */
function getPinyinInitials(str: string): string {
    return str.split('')
        .map(char => getTheFirstLetterForPinyin(char))
        .join('');
}

getPinyinInitials('你好世界'); // => 'nhsj'
getPinyinInitials('JavaScript中文教程'); // => 'javascriptzwjc'
getPinyinInitials('加油123'); // => 'jy123'

// 将联系人列表按拼音首字母分组
function groupContactsByInitial(contacts: {name: string}[]): Record<string, {name: string}[]> {
    const groups: Record<string, {name: string}[]> = {};
    
    // 按首字母分组
    contacts.forEach(contact => {
        const initial = getTheFirstLetterForPinyin(contact.name, true);
        if (!groups[initial]) {
            groups[initial] = [];
        }
        groups[initial].push(contact);
    });
    
    return groups;
}

const contacts = [
    {name: '张三'},
    {name: '李四'},
    {name: '王五'},
    {name: '赵六'}
];

// 按首字母分组
const groupedContacts = groupContactsByInitial(contacts);

// 还可以进一步通过 localeCompare 实现组内排序，此处暂时不做操作
/*
结果：
{
    'L': [{name: '李四'}],
    'W': [{name: '王五'}],
    'Z': [{name: '张三'}, {name: '赵六'}]
}
*/