package com.ruoyi.user.service.impl;

import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.constant.Constants;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.user.entity.Region;
import com.ruoyi.user.mapper.RegionMapper;
import com.ruoyi.user.service.RegionService;
import com.ruoyi.user.vo.RegionVo;
import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinCaseType;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

/**
 * <p>
 * 省市区三级联动 服务实现类
 * </p>
 *
 * @author hjl
 * @since 2024-06-12
 */
@Service
public class RegionServiceImpl extends ServiceImpl<RegionMapper, Region> implements RegionService {

    @Override
    public Object urbanArea(String keyword) {
        LambdaQueryChainWrapper<Region> wrapper = this.lambdaQuery()
                .ne(Region::getParentId, Constants.ZERO);
        if (StringUtils.isNotBlank(keyword)) {
            wrapper.like(Region::getName, keyword);
        }
        // 获取所有市区信息
        List<Region> reginList = wrapper.orderByAsc(Region::getId).list();
        // 根据市区首字母分组排序
        return newSort(reginList);
    }

    /**
     * 获取汉字串拼音，英文字符不变
     *
     * @param chinese 汉字串
     * @return 汉语拼音
     */
    public String getFullSpell(String chinese) {
        StringBuilder pybf = new StringBuilder();
        char[] arr = chinese.toCharArray();
        HanyuPinyinOutputFormat defaultFormat = new HanyuPinyinOutputFormat();
        defaultFormat.setCaseType(HanyuPinyinCaseType.UPPERCASE);
        defaultFormat.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
        for (char c : arr) {
            if (c > 128) {
                try {
                    pybf.append(PinyinHelper.toHanyuPinyinStringArray(c, defaultFormat)[0]);
                } catch (BadHanyuPinyinOutputFormatCombination e) {
                    e.printStackTrace();
                }
            } else {
                pybf.append(c);
            }
        }
        return pybf.toString();
    }

    private Object newSort(List<Region> reginList) {
        // 按照城市名称排序，并分组
        Map<String, List<RegionVo>> map = new TreeMap<>();
        for (int i = 1; i <= 26; i++) {
            // 循环找出首字母一样的数据
            String letter = String.valueOf((char) (96 + i)).toUpperCase();
            List<RegionVo> regionVos = new ArrayList<>();
            for (Region region : reginList) {
                String firstLetter = getFullSpell(region.getName()).substring(0, 1);
                if (letter.equals(firstLetter)) {
                    RegionVo regionVo = new RegionVo(region.getName(), region.getCode());
                    regionVos.add(regionVo);
                }
                if (!regionVos.isEmpty()) {
                    map.put(letter, regionVos);
                }
            }
        }
        return map;
    }

}