EmploymentSite/src/main/java/com/hz/employmentsite/services/impl/baseSettings/SiteUserImpl.java

package com.hz.employmentsite.services.impl.baseSettings;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.hz.employmentsite.filter.exception.BaseException;
import com.hz.employmentsite.mapper.*;
import com.hz.employmentsite.mapper.cquery.UserInfoCQuery;
import com.hz.employmentsite.model.*;
import com.hz.employmentsite.services.service.baseSettings.SiteUserService;
import com.hz.employmentsite.services.service.system.DictionaryService;
import com.hz.employmentsite.util.StringUtils;
import com.hz.employmentsite.vo.baseSettings.SiteUserVo;
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.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.UUID;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

@Service("SiteUserService")
public class SiteUserImpl implements SiteUserService {

    @Autowired
    private UserInfoCQuery userInfoCQuery;

    @Autowired
    private StringUtils stringUtils;

    @Autowired
    private PcSiteUserMapper pcSiteUserMapper;

    @Autowired
    private SysUserMapper sysUserMapper;

    @Autowired
    private SysRoleMapper sysRoleMapper;

    @Autowired
    private SysUserSysRoleMapper sysUserSysRoleMapper;

    @Autowired
    private PcSiteMapper siteMapper;

    @Autowired
    private DictionaryService dictionaryService;

    private final String MOBILE_REGEX = "0?(13|14|15|18|17)[0-9]{9}";

    private final String ID_CARD_REGEX = "\\d{17}[\\d|x]|\\d{15}";


    @Override
    public PageInfo<SiteUserVo> getList(int pageIndex, int pageSize, List<String> siteUserIDList, String siteUserName, String siteID, String roleName, String regionCode, String userNo) {
        PageHelper.startPage(pageIndex, pageSize);
        List<SiteUserVo> dataList = userInfoCQuery.getList(stringUtils.ListToInSql(siteUserIDList), siteUserName, siteID, roleName, regionCode, userNo);
        PageInfo<SiteUserVo> result = new PageInfo(dataList);
        return result;
    }

    @Override
    public PageInfo<SiteUserVo> getListBySiteID(int pageIndex, int pageSize, String siteID) {
        PageHelper.startPage(pageIndex, pageSize);
        List<SiteUserVo> dataList = userInfoCQuery.getListBySiteID(siteID);
        PageInfo<SiteUserVo> result = new PageInfo(dataList);
        return result;
    }

    @Override
    public SiteUserVo get(String siteUserID) {
        List<String> ids = new ArrayList<>();
        if (stringUtils.IsNullOrEmpty(siteUserID)) {
            return null;
        }else{
            ids.add(siteUserID);
        }
        SiteUserVo model = userInfoCQuery.getList(stringUtils.ListToInSql(ids), null, null, null, null, null).stream().findFirst().orElse(null);
        return model;
    }

    @Override
    public List<SelectProps> getDataList() {
        var dataList = getList(1, 99999, null, null, null, null, null, null).getList();
        List<SelectProps> dataResult = new ArrayList<>();
        for (var item: dataList) {
            SelectProps  itemResult = new SelectProps();
            itemResult.setText(item.siteUserName);
            itemResult.setValue(item.userID);
            dataResult.add(itemResult);
        }
        return dataResult;
    }

    @Override
    public Integer save(SiteUserVo data, String userId) {
        int result = 0;

        // 手机号查重
        List<String> byMobileSiteId = pcSiteUserMapper.selectByMobile(data.mobile);
        List<String> byUserNoSiteId = pcSiteUserMapper.selectByUserNo(data.userNo);
        List<String> byIdCardSiteId = pcSiteUserMapper.selectByIdCard(data.idCard);

        SiteUserVo record = get(data.siteUserID);
        PcSiteUser dbData = record == null ? null : pcSiteUserMapper.selectByPrimaryKey(record.siteUserID);
        Boolean isExist = dbData != null;
        SysUser curSysUserData = data.userID != null ? sysUserMapper.selectByPrimaryKey(data.userID) : null;
        Boolean isLinkUser = curSysUserData != null;
        SysRoleExample roleExp = null;
        /*注意：这里的data.roleID为用户类型ID*/
        if (!isExist) {
            // 添加时有重复数据立即报错
            if (!byMobileSiteId.isEmpty()) {
                throw new BaseException("500", "该手机号已绑定其他驿站人员");
            }
            if (!byUserNoSiteId.isEmpty()) {
                throw new BaseException("500", "该工号已绑定其他驿站人员");
            }
            if (!byIdCardSiteId.isEmpty()) {
                throw new BaseException("500", "该身份证已绑定其他驿站人员");
            }

            //添加
            //处理用户表
            curSysUserData = new SysUser();
            curSysUserData.setUserID(UUID.randomUUID().toString());
            curSysUserData.setName(data.getSiteUserName());
            curSysUserData.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes()).toUpperCase());
            curSysUserData.setLoginID(StrToPinYin(data.getSiteUserName()));
            curSysUserData.setUserTypeID(data.getRoleID());
            curSysUserData.setCreateTime(new Date());
            curSysUserData.setCreateBy(userId);
            curSysUserData.setRecordStatus(1);
            result += sysUserMapper.insert(curSysUserData);

            //处理站点人员表
            dbData = new PcSiteUser();
            dbData.setUserID(curSysUserData.getUserID());
            dbData.setSiteUserID(data.getSiteUserID());
            dbData.setSiteID(data.getSiteID());
            dbData.setGender(data.getGender());
            dbData.setSiteUserName(data.getSiteUserName());
            dbData.setGender(data.getGender());
            dbData.setRoleID(data.getRoleID());
            dbData.setMobile(data.getMobile());
            dbData.setUserNo(data.getUserNo());
            dbData.setIDCard(data.getIdCard());
            dbData.setCreateUserID(userId);
            dbData.setCreateTime(new Date());
            result += pcSiteUserMapper.insert(dbData);
        } else {
            // 修改时，ID列表中没有参数中的ID才报错
            if (!byMobileSiteId.isEmpty() && !byMobileSiteId.contains(data.siteUserID)) {
                throw new BaseException("500", "该手机号已绑定其他驿站人员");
            }
            if (!byUserNoSiteId.isEmpty() && !byUserNoSiteId.contains(data.siteUserID)) {
                throw new BaseException("500", "该工号已绑定其他驿站人员");
            }
            if (!byIdCardSiteId.isEmpty() && !byIdCardSiteId.contains(data.siteUserID)) {
                throw new BaseException("500", "该身份证已绑定其他驿站人员");
            }

            //修改
            //处理用户表
            if (isLinkUser) {
                curSysUserData.setName(data.siteUserName);
                curSysUserData.setUserTypeID(data.roleID);
                curSysUserData.setLoginID(StrToPinYin(data.siteUserName));
                curSysUserData.setUpdateTime(new Date());
                curSysUserData.setUpdateBy(userId);
                result += sysUserMapper.updateByPrimaryKey(curSysUserData);
            }

            //处理站点人员表
            dbData.setUserID(data.userID);
            dbData.setSiteUserID(data.siteUserID);
            dbData.setSiteID(data.getSiteID());
            dbData.setGender(data.gender);
            dbData.setSiteUserName(data.siteUserName);
            dbData.setGender(data.gender);
            dbData.setRoleID(data.roleID);
            dbData.setMobile(data.mobile);
            dbData.setUserNo(data.userNo);
            dbData.setIDCard(data.idCard);
            dbData.setUpdateBy(userId);
            dbData.setUpdateTime(new Date());
            result += pcSiteUserMapper.updateByPrimaryKey(dbData);

        }

        return result;
    }

    @Override
    public Integer delete(List<String> idList){
        int result = 0;
        PcSiteUserExample siteUserExp = new PcSiteUserExample();
        siteUserExp.or().andSiteUserIDIn(idList);
        var curSiteUserList = pcSiteUserMapper.selectByExample(siteUserExp).stream().toList();
        result += pcSiteUserMapper.deleteByExample(siteUserExp);
        if (curSiteUserList != null && curSiteUserList.size() > 0){
            for(PcSiteUser curSiteUser : curSiteUserList) {
                SysUserSysRoleExample userRoleExp = new SysUserSysRoleExample();
                userRoleExp.or().andUserIDEqualTo(curSiteUser.getUserID());
                result += sysUserSysRoleMapper.deleteByExample(userRoleExp);
                SysUserExample userExp = new SysUserExample();
                userExp.or().andUserIDEqualTo(curSiteUser.getUserID());
                result += sysUserMapper.deleteByExample(userExp);
            }
        }
        return result;
    }

    /**
     * 数据导入
     *
     * @param dataList    excel数据
     * @param loginUserId 当前登录用户ID
     * @return 导入结果
     */
    @Override
    public List<SiteUserVo> importSiteUser(List<SiteUserVo> dataList, String loginUserId) {
        if (dataList.size() <= 0)
            throw new BaseException("", "请添加导入数据！");
        // 所属驿站
        List<PcSite> siteList = siteMapper.selectByExample(null);
        // 已存在的驿站工作人员
        List<PcSiteUser> pcSiteUsers = pcSiteUserMapper.selectByExample(null);
        // 用户类型字典数据
        List<SysDictionaryItem> userTypeList = dictionaryService.getDictionaryItemList("UserType");
        // 性别字典数据
        List<SysDictionaryItem> genderList = dictionaryService.getDictionaryItemList("Gender");

        // 最终结果
        List<SiteUserVo> resultList = new ArrayList<>();

        dataList.forEach(item -> {
            String errorInfo = "";

            // 数据内容为空判断
            if (stringUtils.IsNullOrEmpty(item.siteUserName)) {
                errorInfo += "请填写姓名！";
            }
            if (stringUtils.IsNullOrEmpty(item.userNo)) {
                errorInfo += "请填写工号！";
            } else {
                String siteUserID = pcSiteUsers.stream()
                        .filter(it -> it.getUserNo() != null && it.getUserNo().equals(item.getUserNo().trim()))
                        .findFirst()
                        .orElse(new PcSiteUser())
                        .getSiteUserID();
                if (!stringUtils.IsNullOrEmpty(siteUserID)) {
                    errorInfo += "工号已绑定其他驿站人员！";
                }
            }
            if (stringUtils.IsNullOrEmpty(item.genderName)) {
                errorInfo += "请填写性别！";
            } else {
                item.gender = genderList.stream().filter(it -> it.getName().equals(item.getGenderName().trim()))
                        .findFirst().orElse(new SysDictionaryItem()).getValue();
                if (item.gender == null) {
                    errorInfo += "性别不存在！";
                }
            }
            if (stringUtils.IsNullOrEmpty(item.mobile)) {
                errorInfo += "请填写联系电话！";
            } else {
                if (Pattern.matches(MOBILE_REGEX, item.mobile)) {
                    String siteUserID = pcSiteUsers.stream()
                            .filter(it -> it.getMobile() != null && it.getMobile().equals(item.getMobile().trim()))
                            .findFirst()
                            .orElse(new PcSiteUser())
                            .getSiteUserID();
                    if (!stringUtils.IsNullOrEmpty(siteUserID)) {
                        errorInfo += "联系电话已绑定其他驿站人员！";
                    }
                } else {
                    errorInfo += "联系电话格式错误！";
                }
            }
            if (stringUtils.IsNullOrEmpty(item.idCard)) {
                errorInfo += "请填写身份证号码！";
            } else {
                if (Pattern.matches(ID_CARD_REGEX, item.idCard)) {
                    String siteUserID = pcSiteUsers.stream()
                            .filter(it -> it.getIDCard() != null && it.getIDCard().equals(item.getIdCard().trim()))
                            .findFirst()
                            .orElse(new PcSiteUser())
                            .getSiteUserID();
                    if (!stringUtils.IsNullOrEmpty(siteUserID)) {
                        errorInfo += "身份证已绑定其他驿站人员！";
                    }
                } else {
                    errorInfo += "身份证格式错误！";
                }
            }
            if (stringUtils.IsNullOrEmpty(item.roleName)) {
                errorInfo += "请填写用户类型！";
            } else {
                item.roleID = userTypeList.stream()
                        .filter(it -> it.getName().equals(item.getRoleName().trim()))
                        .findFirst()
                        .orElse(new SysDictionaryItem())
                        .getValue();
                if (item.roleID == null) {
                    errorInfo += "用户类型不存在！";
                }
            }
            if (stringUtils.IsNullOrEmpty(item.siteName)) {
                errorInfo += "请填写所属驿站！";
            } else {
                item.siteID = siteList.stream()
                        .filter(it -> it.getSiteName().equals(item.getSiteName().trim()))
                        .findFirst()
                        .orElse(new PcSite())
                        .getSiteID();
                if (stringUtils.IsNullOrEmpty(item.siteID)) {
                    errorInfo += "驿站不存在！";
                }
            }

            // 检查结果填充
            if (stringUtils.IsNullOrEmpty(errorInfo)) {
                item.setSiteUserID(UUID.randomUUID().toString());
                resultList.add(item);
            } else {
                item.setErrorMessage(errorInfo);
            }
        });

        // 返回结果
        if (dataList.stream().filter(it -> !stringUtils.IsNullOrEmpty(it.errorMessage)).collect(Collectors.toList()).size() > 0)
            return dataList;

        // 保存到数据
        resultList.forEach(item -> {
            save(item, loginUserId);
        });
        return null;
    }

    public String StrToPinYin(String chinese) {
        String pinyinStr = "";
        char[] newChar = chinese.toCharArray();
        HanyuPinyinOutputFormat defaultFormat = new HanyuPinyinOutputFormat();
        defaultFormat.setCaseType(HanyuPinyinCaseType.LOWERCASE);
        defaultFormat.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
        for (int i = 0; i < newChar.length; i++) {
            if (newChar[i] > 128) {
                try {
                    pinyinStr += PinyinHelper.toHanyuPinyinStringArray(newChar[i], defaultFormat)[0];
                } catch (BadHanyuPinyinOutputFormatCombination e) {
                    e.printStackTrace();
                }
            }else{
                pinyinStr += newChar[i];
            }
        }
        return pinyinStr;
    }
}