/**
 * 獲取漢字的拼音首字母
 * @param chineseChar 中文字符串，若傳入多字符則只取第一個字符
 * @param useUpperCase 是否返回大寫字母，默認(rèn)為false（小寫）
 * @returns 拼音首字母，若無法識別則返回空字符串
 */
export const getTheFirstLetterForPinyin = (chineseChar: string = '', useUpperCase: boolean = false): string => {
    // 兼容性檢查：確保瀏覽器支持 localeCompare 方法
    if (!String.prototype.localeCompare) {
        return '';
    }

    // 參數(shù)驗證：確保輸入為有效字符串
    if (typeof chineseChar !== 'string' || !chineseChar.length) {
        return '';
    }

    // 準(zhǔn)備用于比較的字母表和基準(zhǔn)漢字
    // 注：這些基準(zhǔn)漢字分別對應(yīng)A、B、C...等拼音首字母的起始位置
    const letters = 'ABCDEFGHJKLMNOPQRSTWXYZ'.split('');
    const zh = '阿八嚓噠妸發(fā)旮哈譏咔垃痳拏噢妑七呥扨它穵夕丫帀'.split('');

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

    // 處理字母和數(shù)字：直接返回原字符
    if (/^\w/.test(firstChar)) {
        firstLetter = firstChar;
    } else {
        // 處理漢字：通過比較確定拼音首字母范圍
        letters.some((item, index) => {
            // 檢查當(dāng)前字符是否在當(dāng)前基準(zhǔn)漢字與下一個基準(zhǔn)漢字之間
            if (firstChar.localeCompare(zh[index]) >= 0 && 
                (index === letters.length - 1 || firstChar.localeCompare(zh[index + 1]) < 0)) {
                firstLetter = item;
                return true;
            }
            return false;
        });
    }

    // 根據(jù)參數(shù)決定返回大寫還是小寫字母
    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'

// 將聯(lián)系人列表按拼音首字母分組
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);

// 還可以進(jìn)一步通過 localeCompare 實現(xiàn)組內(nèi)排序，此處暫時不做操作
/*
結(jié)果：
{
    'L': [{name: '李四'}],
    'W': [{name: '王五'}],
    'Z': [{name: '張三'}, {name: '趙六'}]
}
*/