KPI / fw-morax

package cn.fw.morax.service.biz;

import cn.fw.common.exception.BusinessException;
import cn.fw.ehr.sdk.api.enums.RoleManagerEnum;
import cn.fw.morax.common.constant.Constant;
import cn.fw.morax.common.utils.ExcelDataUtil;
import cn.fw.morax.common.utils.PublicUtil;
import cn.fw.morax.domain.db.SettingDraft;
import cn.fw.morax.domain.db.eval.EvalGroupRewardRankLog;
import cn.fw.morax.domain.db.eval.EvalIndicatorValue;
import cn.fw.morax.domain.db.kpi.IndicatorUserValue;
import cn.fw.morax.domain.db.kpi.KpiGroup;
import cn.fw.morax.domain.dto.AbstractLaddersDto;
import cn.fw.morax.domain.enums.DataTypeEnum;
import cn.fw.morax.domain.enums.IndicatorValueTypeEnum;
import cn.fw.morax.domain.enums.SettingDraftStatusEnum;
import cn.fw.morax.domain.enums.SettingDraftTypeEnum;
import cn.fw.morax.domain.vo.kpi.IndicatorUserValueVO;
import cn.fw.morax.domain.vo.kpi.KpiIndicatorRankStaffVO;
import cn.fw.morax.rpc.ehr.EhrRpcService;
import cn.fw.morax.rpc.ehr.dto.StaffBaseInfoDTO;
import cn.fw.morax.service.data.SettingDraftService;
import cn.hutool.core.io.IoUtil;
import cn.hutool.poi.excel.ExcelUtil;
import cn.hutool.poi.excel.ExcelWriter;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.math.BigDecimal;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.time.*;
import java.time.temporal.TemporalAdjusters;
import java.util.*;
import java.util.stream.Collectors;

@Slf4j
@RequiredArgsConstructor
@Service
public class CommonService {

    private final EhrRpcService ehrRpcService;
    private final SettingDraftService settingDraftService;

    /**
     * 检查阶梯
     *
     * @param dtos
     */
    public static void checkLadders(List<? extends AbstractLaddersDto> dtos) {
        if (PublicUtil.isEmpty(dtos)) {
            return;
        }
        String errorPromptName = dtos.get(0).getBusinessName();
        checkLadders(dtos, errorPromptName);
    }

    /**
     * 检查阶梯
     *
     * @param dtos
     */
    public static void checkLadders(List<? extends AbstractLaddersDto> dtos, String errorPromptName) {
        if (PublicUtil.isEmpty(dtos)) {
            return;
        }
        BigDecimal lastUpper = null;
        //阶梯值校验  标准分不能超过指标基础分，阶梯值必须连续
        for (AbstractLaddersDto laddersDto : dtos) {
            BigDecimal lower = laddersDto.getLower();
            BigDecimal upper = laddersDto.getUpper();
            //上一条的下限要等于下一条的上限
            if (PublicUtil.isNotEmpty(lastUpper) && (lastUpper.compareTo(lower) != 0) ) {
                throw new BusinessException("【" + errorPromptName + "】的阶梯上限与下限不是连续，请编辑后重试");
            }
            //每条上限大于下限
            if (lower.compareTo(upper) >= 0) {
                throw new BusinessException("【" + errorPromptName + "】的阶梯下限大于或等于阶梯上限，请修改错误阶梯");
            }
            lastUpper = laddersDto.getUpper();
        }
    }


    /**
     * 导出文件
     *
     * @param response
     * @param rows
     * @param fileName
     */
    public static <T> void downloadExcel(HttpServletResponse response, List<T> rows, String fileName, Class<?> source) {
        ExcelWriter writer = ExcelUtil.getWriter();
        ExcelDataUtil.setExcelHeadAndWidth(writer, source, Boolean.TRUE);
        writer.setOnlyAlias(true);
        writer.write(rows, true);
        ExcelDataUtil.setAutoSizeColumn(writer.getSheet());
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.setHeader("Content-Disposition","attachment;filename=" + fileName);
        ServletOutputStream out= null;
        try {
            out = response.getOutputStream();
        } catch (IOException e) {
            e.printStackTrace();
        }
        writer.flush(out, true);
        writer.close();
        IoUtil.close(out);
    }

    /**
     * 导出文件
     *
     * @param response
     * @param fileName
     */
    public static <T> void downloadTemplateExcel(HttpServletResponse response, String fileName, Class<?> source) {
        ExcelWriter writer = ExcelUtil.getWriter();
        Map<String, String> headMap = ExcelDataUtil.setExcelHeadAndWidth(writer, source, Boolean.FALSE);
        if (headMap.isEmpty()) {
            return;
        }
        writer.writeHeadRow(headMap.values());
        writer.setRowHeight(0, 25);
//        ExcelDataUtil.setAutoSizeColumn(writer.getSheet(), headMap.size());
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.setHeader("Content-Disposition","attachment;filename=" + fileName);
        ServletOutputStream out= null;
        try {
            out = response.getOutputStream();
        } catch (IOException e) {
            e.printStackTrace();
        }
        writer.flush(out, true);
        writer.close();
        IoUtil.close(out);
    }

    /**
     * 获取编码后的文件名
     *
     * @param request
     * @param fileName
     * @return
     */
    public static String getEncodeName(HttpServletRequest request, String fileName) {
        String userAgent = request.getHeader("User-Agent");
        if (userAgent.contains("MSIE") || userAgent.contains("Trident") || userAgent.contains("Postman")) {
            try {
                fileName = URLEncoder.encode(fileName, "UTF-8");
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        } else {// 非IE浏览器的处理：
            fileName = new String(fileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1);
        }
        return fileName;
    }

    /**
     * 实时查询在职员工
     *
     * @param userId
     * @return
     */
    public List<StaffBaseInfoDTO> queryRealTimeManageStaffs(Long userId) {
        List<StaffBaseInfoDTO> managerStaffs = ehrRpcService.queryManageStaffs(userId);
        if (PublicUtil.isEmpty(managerStaffs)) {
            managerStaffs = new ArrayList<>();
        }
        List<StaffBaseInfoDTO> selectVOS = managerStaffs.stream()
                .filter(staff -> (! staff.getId().equals(userId)))
                .collect(Collectors.toList());
        return selectVOS;
    }

    /**
     * 查询离职、在职员工
     *
     * @param userId
     * @param yearMonth
     * @return
     */
    public List<StaffBaseInfoDTO> queryManageStaffs(Long userId, YearMonth yearMonth) {
        List<StaffBaseInfoDTO> managerStaffs = ehrRpcService.queryManageStaffs(userId);
        if (PublicUtil.isEmpty(managerStaffs)) {
            managerStaffs = new ArrayList<>();
        }
        LocalDate startDate = LocalDate.of(yearMonth.getYear(), yearMonth.getMonth(), 1);
        LocalDate endDate = startDate.with(TemporalAdjusters.lastDayOfMonth());
        LocalDateTime startDateTime = LocalDateTime.of(startDate, LocalTime.MIN);
        LocalDateTime endDateTime = LocalDateTime.of(endDate, LocalTime.MAX);
        Date firstDayOfMonth = Date.from(startDateTime.atZone(ZoneId.systemDefault()).toInstant());
        Date endDayOfMonth = Date.from(endDateTime.atZone(ZoneId.systemDefault()).toInstant());
        List<StaffBaseInfoDTO> leaveManagerStaffs = ehrRpcService.queryLeaveManageStaffs(userId, firstDayOfMonth, endDayOfMonth);
        if (PublicUtil.isNotEmpty(leaveManagerStaffs)) {
            managerStaffs.addAll(leaveManagerStaffs);
        }
        List<StaffBaseInfoDTO> selectVOS = managerStaffs.stream()
                .filter(staff -> (! staff.getId().equals(userId)))
                .collect(Collectors.toList());
        return selectVOS;
    }

    /**
     * 查询离职、在职员工id
     *
     * @param userId
     * @param yearMonth
     * @return
     */
    public List<Long> queryManageStaffIds(Long userId, YearMonth yearMonth) {
        List<StaffBaseInfoDTO> staffBaseInfoDTOS = this.queryManageStaffs(userId, yearMonth);
        if (PublicUtil.isNotEmpty(staffBaseInfoDTOS)) {
            return staffBaseInfoDTOS.stream().map(StaffBaseInfoDTO::getId).collect(Collectors.toList());
        }
        return new ArrayList<>();
    }

    /**
     * 查询离职员工id
     *
     * @param shopId
     * @param yearMonth
     * @return
     */
    public List<Long> queryShopLeaveStaffIds(Long shopId, YearMonth yearMonth) {
        List<Long> shopIds = new ArrayList<Long>(){{add(shopId);}};

        LocalDate startDate = LocalDate.of(yearMonth.getYear(), yearMonth.getMonth(), 1);
        LocalDate endDate = startDate.with(TemporalAdjusters.lastDayOfMonth());
        LocalDateTime startDateTime = LocalDateTime.of(startDate, LocalTime.MIN);
        LocalDateTime endDateTime = LocalDateTime.of(endDate, LocalTime.MAX);
        Date firstDayOfMonth = Date.from(startDateTime.atZone(ZoneId.systemDefault()).toInstant());
        Date endDayOfMonth = Date.from(endDateTime.atZone(ZoneId.systemDefault()).toInstant());

        List<StaffBaseInfoDTO> staffBaseInfoDTOS = ehrRpcService.queryShopLeaveStaff(shopIds, firstDayOfMonth, endDayOfMonth);
        if (PublicUtil.isNotEmpty(staffBaseInfoDTOS)) {
            return staffBaseInfoDTOS.stream().map(StaffBaseInfoDTO::getId).collect(Collectors.toList());
        }
        return new ArrayList<>();
    }

    /**
     * 查询具体指标值
     * @param indicatorCode
     * @param indicatorUserValue
     * @return
     */
    public BigDecimal queryIndicatorValue(String indicatorCode, IndicatorUserValue indicatorUserValue) {
        if (PublicUtil.isEmpty(indicatorUserValue)) {
            return BigDecimal.ZERO;
        }
        BigDecimal value = BigDecimal.ZERO;
        //条件计算类型 是车系id：销售台数
        if (IndicatorValueTypeEnum.CONDITION.equals(indicatorUserValue.getValueType())) {
            return value;
        }
        try {
            JSONObject jsonObject = JSONObject.parseObject(indicatorUserValue.getIndicatorValue());
            value = Optional.ofNullable(jsonObject.getBigDecimal(indicatorCode)).orElse(BigDecimal.ZERO);
        } catch (Exception e) {
            log.error("[{}]指标值转化失败", indicatorCode, e);
        }
        return value;
    }

    /**
     * 查询具体指标值
     * @param indicatorCode
     * @param indicatorValue
     * @return
     */
    public BigDecimal queryEvalIndicatorValue(String indicatorCode, EvalIndicatorValue indicatorValue) {
        if (PublicUtil.isEmpty(indicatorValue)) {
            return BigDecimal.ZERO;
        }
        BigDecimal value = BigDecimal.ZERO;
        try {
            value = Optional.ofNullable(indicatorValue.getIndicatorValue()).orElse(BigDecimal.ZERO);
        } catch (Exception e) {
            log.error("[{}]指标值转化失败", indicatorCode, e);
        }
        return value;
    }

    /**
     * 查询具体指标值
     * @param indicatorCode
     * @param indicatorUserValue
     * @return
     */
    public BigDecimal queryIndicatorValueVO(String indicatorCode, IndicatorUserValueVO indicatorUserValue) {
        BigDecimal value = BigDecimal.ZERO;
        //条件计算类型 是车系id：销售台数
        if (IndicatorValueTypeEnum.CONDITION.equals(indicatorUserValue.getValueType())) {
            return value;
        }
        try {
            JSONObject jsonObject = JSONObject.parseObject(indicatorUserValue.getIndicatorValue());
            value = Optional.ofNullable(jsonObject.getBigDecimal(indicatorCode)).orElse(BigDecimal.ZERO);
            if (IndicatorValueTypeEnum.RATIO.equals(indicatorUserValue.getValueType())) {
                value = value.multiply(Constant.ONE_HUNDRED);
            }
        } catch (Exception e) {
            log.error("[{}]指标值转化失败", indicatorCode, e);
        }
        return value;
    }

    /**
     * 查询具体指标值
     * @param indicatorCode
     * @param indicatorUserValue
     * @return
     */
    public BigDecimal queryEvalIndicatorValueVO(String indicatorCode, EvalIndicatorValue indicatorUserValue) {
        BigDecimal value = BigDecimal.ZERO;
        //条件计算类型 是车系id：销售台数
        try {
            value = Optional.ofNullable(indicatorUserValue.getIndicatorValue()).orElse(BigDecimal.ZERO);
            if (DataTypeEnum.RATIO.equals(indicatorUserValue.getDataType())) {
                value = value.multiply(Constant.ONE_HUNDRED);
            }
        } catch (Exception e) {
            log.error("[{}]指标值转化失败", indicatorCode, e);
        }
        return value;
    }

    /**
     * 是否拥有管理角色
     *
     * @return
     */
    public Boolean hasManagerRole(List<String> roleCodes) {
        if (PublicUtil.isEmpty(roleCodes)) {
            return Boolean.FALSE;
        }
        HashSet<String> roleCodeSet = new HashSet<>(roleCodes);
        return Arrays.stream(RoleManagerEnum.values())
                .anyMatch(roleManagerEnum -> roleCodeSet.contains(roleManagerEnum.getCode()));
    }

    /**
     * 获取重名的草稿
     *
     * @param name
     * @param type
     * @param excludeDraftId
     * @return
     */
    public List<SettingDraft> getEditDraftByName(String name, SettingDraftTypeEnum type, Long excludeDraftId) {
        List<SettingDraft> settingDrafts = settingDraftService.list(Wrappers.<SettingDraft>lambdaQuery()
                .eq(SettingDraft::getName, name)
                .eq(SettingDraft::getType, type)
                .ne(SettingDraft::getStatus, SettingDraftStatusEnum.RELEASE_APPROVAL_AGREE)
                .ne(PublicUtil.isNotEmpty(excludeDraftId), SettingDraft::getId, excludeDraftId)
                .eq(SettingDraft::getYn, Boolean.TRUE)
        );
        return settingDrafts;
    }

    /**
     * 设置排名序号
     *
     * @return
     */
    public void calcRank(List<EvalGroupRewardRankLog> rankLogs) {
        int rank = 1;
        BigDecimal lastIndicatorValue = null;
        for (EvalGroupRewardRankLog rankLog : rankLogs) {
            //初始化
            if (PublicUtil.isEmpty(lastIndicatorValue)) {
                rankLog.setRank(rank);
                lastIndicatorValue = rankLog.getReachValue();
                continue;
            }
            if (lastIndicatorValue.compareTo(rankLog.getReachValue()) != 0) {
                rank++;
            }
            rankLog.setRank(rank);
            lastIndicatorValue = rankLog.getReachValue();
        }
    }

}