import cn.hutool.extra.pinyin.PinyinUtil;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * 中文姓名转拼音邮箱账号工具
 *
 * 支持：
 * - 单姓：张三 → san.zhang
 * - 复姓：诸葛孔明 → kongming.zhuge
 * - 单名：张伟 → wei.zhang
 */
public class PinyinEmailUtil {

    /**
     * 中国常见复姓集合（可根据需要扩充）
     */
    private static final Set<String> COMPOUND_SURNAMES = new HashSet<>(Arrays.asList(
            "欧阳", "太史", "端木", "上官", "司马", "东方", "独孤", "南宫",
            "万俟", "闻人", "夏侯", "诸葛", "尉迟", "公羊", "赫连", "澹台",
            "皇甫", "宗政", "濮阳", "公冶", "太叔", "申屠", "公孙", "慕容",
            "仲孙", "钟离", "长孙", "宇文", "司徒", "鲜于", "司空", "闾丘",
            "子车", "亓官", "司寇", "巫马", "公西", "颛孙", "壤驷", "公良",
            "漆雕", "乐正", "宰父", "谷梁", "拓跋", "夹谷", "轩辕", "令狐",
            "段干", "百里", "呼延", "东郭", "南门", "羊舌", "微生", "公户",
            "公玉", "公仪", "梁丘", "公仲", "公上", "公门", "公山", "公坚",
            "左丘", "公伯", "西门", "公祖", "第五", "公乘", "贯丘", "公皙",
            "南荣", "东里", "东宫", "仲长", "子书", "子桑", "即墨", "达奚",
            "褚师"
    ));

    /**
     * 将中文姓名转换为拼音邮箱账号
     *
     * @param chineseName 中文姓名（如 "诸葛孔明"）
     * @param domain      邮箱域名（如 "company.com"）
     * @return 邮箱账号（如 "kongming.zhuge@company.com"）
     */
    public static String toEmail(String chineseName, String domain) {
        if (chineseName == null || chineseName.length() < 2) {
            throw new IllegalArgumentException("姓名至少需要2个字符");
        }

        String surname;
        String givenName;

        // 判断是否为复姓（取前两个字匹配）
        String firstTwo = chineseName.substring(0, 2);
        if (COMPOUND_SURNAMES.contains(firstTwo)) {
            surname = firstTwo;
            givenName = chineseName.substring(2);
        } else {
            surname = chineseName.substring(0, 1);
            givenName = chineseName.substring(1);
        }

        // 转拼音（去掉声调，转小写）
        String surnamePinyin = PinyinUtil.getPinyin(surname, "");
        String givenNamePinyin = PinyinUtil.getPinyin(givenName, "");

        return givenNamePinyin + "." + surnamePinyin + "@" + domain;
    }

    /**
     * 仅生成邮箱前缀（不含域名）
     */
    public static String toEmailPrefix(String chineseName) {
        String prefix = toEmail(chineseName, "").replace("@", "");
        // 去除末尾可能多余的点号
        if (prefix.endsWith(".")) {
            prefix = prefix.substring(0, prefix.length() - 1);
        }
        return prefix;
    }
}