pco/sp-server/src/main/java/com/pj/project/tb_costomer/TbCostomerService.java

package com.pj.project.tb_costomer;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.IService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.pj.project.tb_init_permission.TbInitPermission;
import com.pj.project.tb_init_permission.TbInitPermissionService;
import com.pj.project4sp.SP;
import com.pj.project4sp.admin.SpAdmin;
import com.pj.project4sp.admin.SpAdminMapper;
import com.pj.project4sp.admin.SpAdminService;
import com.pj.project4sp.role.SpRole;
import com.pj.project4sp.role.SpRoleMapper;
import com.pj.project4sp.role4permission.SpRolePermissionMapper;
import com.pj.project4sp.role4permission.SpRolePermissionService;
import com.pj.utils.so.SoMap;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Service: tb_costomer -- 客户管理
 *
 * @author qzy
 */
@Service
@Transactional
public class TbCostomerService extends ServiceImpl<TbCostomerMapper, TbCostomer> implements IService<TbCostomer> {

    /**
     * 底层 Mapper 对象
     */
    @Autowired
    TbCostomerMapper tbCostomerMapper;
    @Resource
    private SpRoleMapper spRoleMapper;
    @Resource
    private SpRolePermissionMapper spRolePermissionMapper;
    @Resource
    private TbInitPermissionService tbInitPermissionService;
    @Resource
    private SpAdminMapper spAdminMapper;
    @Resource
    private SpAdminService spAdminService;

    /**
     * 增
     */
    int add(TbCostomer t) {
        if (!validateName(t)) {
            throw new RuntimeException("该企业名称已存在");
        }
        t.setCreareTime(new Date()).setJudgeContent("平台创建，直接通过").setJudgeTime(new Date());
        this.save(t);
        SpRole spRole = new SpRole();
        spRole.setCustomerId(t.getId()).setInfo("企业/客户管理员")
                .setName("管理员").setType("customerAdmin");
        spRoleMapper.add(spRole);
        long roleId = SP.publicMapper.getPrimarykey();
        List<String> codeList = new ArrayList<>();
        for (String type : StrUtil.splitTrim(t.getType(), ",")) {
            SoMap perSoMap = SoMap.getRequestSoMap();
            perSoMap.put("type", type);
            List<TbInitPermission> permissions = tbInitPermissionService.getList(perSoMap);
            codeList.addAll(permissions.stream().map(TbInitPermission::getCode).distinct().collect(Collectors.toList()));
        }
        spRolePermissionMapper.saveRolePer(roleId, codeList);
        return 1;
    }

    private boolean validateName(TbCostomer t) {
        String costomerName = t.getName();
        TbCostomer db = this.findByName(costomerName);
        if (db == null) {
            return true;
        }
        return StrUtil.equals(db.getId(), t.getId());
    }

    public TbCostomer findByName(String name) {
        QueryWrapper<TbCostomer> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("name", name);
        return this.getOne(queryWrapper);
    }

    /**
     * 删
     */
    int delete(String id) {
        this.removeById(id);
        SoMap soMap = SoMap.getRequestSoMap();
        soMap.put("customerId", id);
        List<SpRole> roles = spRoleMapper.getList(soMap);
        if (roles.isEmpty()) {
            return 1;
        }
        spRoleMapper.removeByCustomerId(id);
        spRolePermissionMapper.removeByRoleIds(roles.stream().map(SpRole::getId).collect(Collectors.toList()));
        spAdminMapper.removeByCustomerId(id);
        return 1;
    }

    /**
     * 改
     */
    int update(TbCostomer t) {
        if (!validateName(t)) {
            throw new RuntimeException("名称已存在");
        }
        this.updateById(t);
        return 1;
    }


    /**
     * 查集合 - 根据条件（参数为空时代表忽略指定条件）
     */
    public List<TbCostomer> getList(SoMap so) {

        return tbCostomerMapper.getList(so);
    }


    public int updateStatus(String id, int value) {
        int line = SP.publicMapper.updateColumnById(TbCostomer.TABLE_NAME, "status", value, id);
        int status = value == 1 ? 1 : 2;
        spAdminMapper.updateCustomerAdmin(id, status);
        return line;
    }

    public void judge(String id, int status, String judgeContent) {
        TbCostomer costomer = this.getById(id);
        costomer.setJudgeContent(judgeContent)
                .setJudgeStatus(status).setJudgeTime(new Date());
        this.updateById(costomer);

        //审核不通过禁用用户
        int isBan = status == 3 ? 2 : 1;//账号状态(1=正常, 2=禁用)
        List<SpAdmin> spAdminList = spAdminMapper.getByCostomerId(costomer.getId());
        for (SpAdmin spAdmin : spAdminList) {
            if (spAdmin != null) {
                spAdmin.setStatus(isBan);
                spAdminMapper.updateCustomerAdmin(costomer.getId(), isBan);
            }
        }
    }

    public void register(TbCostomer costomer) {
        costomer.setJudgeStatus(1).setStatus(1);
        this.add(costomer);

        //新增一个默认用户账号 账号：联系人/密码：手机号后六位
        SpAdmin spAdmin = new SpAdmin();
        spAdmin.setId(0L);
        spAdmin.setCustomerId(costomer.getId());
        spAdmin.setName(costomer.getDutyPeople());
        spAdmin.setPassword(costomer.getPhone().substring(costomer.getPhone().length() - 6));
        SoMap so = new SoMap();
        so.put("customerId", costomer.getId());
        List<SpRole> roleList = spRoleMapper.getList(so);
        List<Long> roleIdList = roleList.stream().map(role -> role.getId()).collect(Collectors.toList());
        String roleId = roleIdList.stream().map(String::valueOf).collect(Collectors.joining(","));
        spAdmin.setRoleId(roleId);
        spAdminService.addWhenRegister(spAdmin);
        spAdminMapper.updateCustomerAdmin(costomer.getId(), 2);
    }

    public List<String> findByMessageTypeOpenid(String businessType, String messageType) {
        return tbCostomerMapper.findByMessageTypeOpenid(businessType, messageType);

    }

    /**
     * 添加客户权限
     *
     * @param perStr
     */
    public void addMenu(String customerTypeStr, String perStr) {
        List<Integer> customerTypeList = StrUtil.splitTrim(customerTypeStr, ",").stream().map(Integer::new).collect(Collectors.toList());
        List<TbCostomer> costomerList = this.findCustomerList(customerTypeList);
        List<String> customerIdList = costomerList.stream().map(TbCostomer::getId).collect(Collectors.toList());
        List<SpAdmin> spAdminList = spAdminService.findByCustomerIdList(customerIdList);
        List<String> roleList = spAdminList.stream().map(SpAdmin::getRoleId).collect(Collectors.toList());
        //权限列表
        List<String> list = StrUtil.splitTrim(perStr, ",");
        //初始化新增客户时的权限
        list.forEach(code -> {
            List<TbInitPermission> initPermissions = tbInitPermissionService.findByCode(code);
            if (initPermissions.isEmpty()) {
                TbInitPermission tbInitPermission = new TbInitPermission();
                tbInitPermission.setCode(code).setType(TbCostomer.CustomerEnum.BUSINESS_TYPE.getType());
                initPermissions.add(tbInitPermission);
            } else {
                initPermissions.forEach(tbInitPermission -> {
                    tbInitPermission.setCode(tbInitPermission.getCode() + "," + code);
                });
            }
            tbInitPermissionService.saveOrUpdateBatch(initPermissions);
        });
        //增加旧客户的权限
        roleList.forEach(roleId -> {
            spRolePermissionMapper.saveRolePer(Long.valueOf(roleId), list);
        });

    }

    /**
     * 获取客户列表
     *
     * @return
     */
    private List<TbCostomer> findCustomerList(List<Integer> list) {
        QueryWrapper<TbCostomer> ew = new QueryWrapper<>();
        ew.lambda().in(TbCostomer::getType, list);
        return list(ew);
    }

    /**
     * 删除对应客户、合作伙伴的权限
     *
     * @param customerTypeStr
     * @param perStr
     */
    public void delCustomerMenu(String customerTypeStr, String perStr) {
        List<Integer> customerTypeList = StrUtil.splitTrim(customerTypeStr, ",").stream().map(Integer::new).collect(Collectors.toList());
        List<TbCostomer> costomerList = this.findCustomerList(customerTypeList);
        List<String> customerIdList = costomerList.stream().map(TbCostomer::getId).collect(Collectors.toList());
        List<SpAdmin> spAdminList = spAdminService.findByCustomerIdList(customerIdList);
        List<String> roleList = spAdminList.stream().map(SpAdmin::getRoleId).collect(Collectors.toList());
        //权限列表
        List<String> list = StrUtil.splitTrim(perStr, ",");
        //删除角色-权限表数据
        spRolePermissionMapper.deleteByRoleAndCode(roleList, list);
        //删除tb_init_permission
        List<TbInitPermission> initPermissionList = tbInitPermissionService.findByCode(list);
        initPermissionList.forEach(tbInitPermission -> {
            String type = tbInitPermission.getType();
            List<String> typeList = StrUtil.splitTrim(type, ",");
            typeList.removeIf(typeStr -> customerTypeList.contains(Integer.valueOf(typeStr)));
            if (typeList.isEmpty()) {
                tbInitPermissionService.removeById(tbInitPermission.getId());
            } else {
                tbInitPermission.setType(String.join(",", typeList));
                tbInitPermissionService.updateById(tbInitPermission);
            }
        });
    }

    public void addAdminMenu(String perStr) {
        StrUtil.splitTrim(perStr, ",").forEach(perCode -> {
            tbCostomerMapper.addAdminPermission(perCode);
        });
    }
}