package cn.fw.morax.service.biz.eval;

import cn.fw.ehr.sdk.api.enums.StaffShopTypeEnum;
import cn.fw.morax.common.utils.PublicUtil;
import cn.fw.morax.domain.db.eval.*;
import cn.fw.morax.domain.enums.EvalScopeEnum;
import cn.fw.morax.domain.enums.EvalShopPoolStatusEnum;
import cn.fw.morax.rpc.ehr.EhrRpcService;
import cn.fw.morax.rpc.ehr.dto.StaffBaseInfoDTO;
import cn.fw.morax.rpc.ehr.dto.StaffShopInfoDTO;
import cn.fw.morax.rpc.oop.OopRpcService;
import cn.fw.morax.rpc.oop.dto.ShopDTO;
import cn.fw.morax.service.data.eval.EvalGroupRewardDimService;
import cn.fw.morax.service.data.eval.EvalShopPoolService;
import cn.fw.morax.service.data.eval.EvalUserPoolService;
import cn.fw.morax.service.data.salary.SalaryGroupUserService;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.map.MultiKeyMap;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.YearMonth;
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * @author : kurisu
 * @version : 1.0
 * @className : SalaryPoolCommonService
 * @description : 绩效池公用方法
 * @date : 2022-09-27 11:40
 */
@Slf4j
@Service
@AllArgsConstructor
public class EvalPoolCommonService {

    private SalaryGroupUserService salaryGroupUserService;
    private EvalUserPoolService evalUserPoolService;
    private EvalGroupRewardDimService evalGroupRewardDimService;
    private EvalShopPoolService evalShopPoolService;
    private EhrRpcService ehrRpcService;
    private OopRpcService oopRpcService;

    public Map<Long, EvalUserPool> inspectionUserPoolMap(EvalGroup evalGroup, EvalGroupRankStage evalGroupRankStage, List<EvalGroupUser> users, LocalDate localDate) {
        if (PublicUtil.isEmpty(users)) {
            return new HashMap<>();
        }
        Set<Long> userIds = users.stream().map(EvalGroupUser::getUserId).collect(Collectors.toSet());
        List<EvalUserPool> pools = evalUserPoolService.list(Wrappers.<EvalUserPool>lambdaQuery()
                .eq(EvalUserPool::getEvalGroupRankStageId, evalGroupRankStage.getId())
                .eq(EvalUserPool::getMonthly, YearMonth.from(localDate))
                .in(EvalUserPool::getUserId, userIds)
                .eq(EvalUserPool::getYn, Boolean.TRUE)
        );

        //人员可能在考评组中出现多次（调岗）
        MultiKeyMap<Long, EvalUserPool> userShopPostShopIdMap = new MultiKeyMap<>();
        for (EvalUserPool pool : pools) {
            userShopPostShopIdMap.put(pool.getUserId(), pool.getShopId(), pool.getPostId(), pool);
        }

        for (EvalGroupUser user : users) {
            if (! userShopPostShopIdMap.containsKey(user.getUserId(), user.getShopId(), user.getPostId())) {
                EvalUserPool userPool = inspectionUserPool(user);
                userShopPostShopIdMap.put(user.getUserId(), user.getShopId(), user.getPostId(), userPool);
            }
        }

        Map<Long, EvalUserPool> userPoolMap = userShopPostShopIdMap.values().stream().collect(Collectors.toMap(EvalUserPool::getUserId, Function.identity(), (v1, v2) -> {
            log.error("同一个考评组、考评人员出现两条考评池数据。用户id：{}，考评组：{}", v1.getUserId(), evalGroup);
            return v1;
        }));

        return userPoolMap;
    }


    public Map<Long, EvalShopPool> inspectionShopPools(EvalGroup evalGroup, LocalDate localDate, List<Long> distShopIds) {
        Set<Long> evalGroupShopIds = new HashSet<>(evalGroup.getShopIds());
        List<EvalShopPool> pools = evalShopPoolService.list(Wrappers.<EvalShopPool>lambdaQuery()
                .eq(EvalShopPool::getEvalGroupId, evalGroup.getId())
                .in(EvalShopPool::getShopId, evalGroupShopIds)
                .eq(EvalShopPool::getMonthly, YearMonth.from(localDate))
                .eq(EvalShopPool::getYn, Boolean.TRUE)
        );

        Map<Long, EvalShopPool> shopPoolMap = pools.stream().collect(Collectors.toMap(EvalShopPool::getShopId, Function.identity(), (v1, v2) -> {
            log.error("同一个考评组、考评门店出现两条考评池数据。门店id：{}，考评组：{}", v1.getShopId(), evalGroup);
            return v1;
        }));

        for (Long shopId : evalGroupShopIds) {
            if (! shopPoolMap.containsKey(shopId)) {
                EvalShopPoolStatusEnum status = (distShopIds.contains(shopId))
                        ? EvalShopPoolStatusEnum.CALC_REWARD : EvalShopPoolStatusEnum.NO_REWARD;
                EvalShopPool shopPool = this.createShopPoolData(shopId, evalGroup, localDate, status);
                shopPoolMap.put(shopId, shopPool);
            }
        }
        return shopPoolMap;
    }

    @Transactional(rollbackFor = Exception.class)
    public EvalShopPool createShopPoolData(Long shopId, EvalGroup evalGroup, LocalDate localDate, EvalShopPoolStatusEnum status) {
        ShopDTO shopDTO = oopRpcService.queryShop(shopId);
        EvalShopPool evalShopPool = new EvalShopPool();
        evalShopPool.setEgc(evalGroup.getEgc());
        evalShopPool.setEvalGroupId(evalGroup.getId());
        evalShopPool.setEvalGroupRankId(evalGroup.getEvalGroupRankId());
        evalShopPool.setEvalGroupRankStageId(evalGroup.getEvalGroupRankStageId());
//        evalShopPool.setEvalId(evalGroup.getEvalId());
        evalShopPool.setStatus(status);
        evalShopPool.setShopId(shopId);
        evalShopPool.setShopName(shopDTO.getShortName());
        evalShopPool.setScore(BigDecimal.ZERO);
        evalShopPool.setScoreRatio(BigDecimal.ZERO);
        evalShopPool.setMonthly(YearMonth.from(localDate));
        evalShopPool.setReward(BigDecimal.ZERO);
        evalShopPool.setGroupId(evalGroup.getGroupId());
        evalShopPool.setYn(Boolean.TRUE);
        evalShopPoolService.save(evalShopPool);
        return evalShopPool;
    }

    /**
     * 查询薪酬池
     *
     * @param user
     * @return
     */
    @Transactional(rollbackFor = Exception.class)
    public EvalUserPool inspectionUserPool(EvalGroupUser user) {
        EvalUserPool evalUserPool = evalUserPoolService.getOne(Wrappers.<EvalUserPool>lambdaQuery()
                        .eq(EvalUserPool::getEvalGroupRankStageId, user.getEvalGroupRankStageId())
                        .eq(EvalUserPool::getUserId, user.getUserId())
                        .eq(EvalUserPool::getPostId, user.getPostId())
                        .eq(EvalUserPool::getShopId, user.getShopId())
                        .eq(EvalUserPool::getMonthly, YearMonth.from(user.getDataDate()))
                        .eq(EvalUserPool::getYn, Boolean.TRUE)
                , Boolean.FALSE);
        if (PublicUtil.isNotEmpty(evalUserPool)) {
            return evalUserPool;
        }
//        evalUserPool = evalUserPoolService.getOne(Wrappers.<EvalUserPool>lambdaQuery()
//                        .eq(EvalUserPool::getEvalGroupRankStageId, user.getEvalGroupRankStageId())
//                        .eq(EvalUserPool::getUserId, user.getUserId())
//                        .eq(EvalUserPool::getMonthly, YearMonth.from(user.getDataDate()).minusMonths(1))
//                        .eq(EvalUserPool::getYn, Boolean.TRUE)
//                , Boolean.FALSE);
//        if (PublicUtil.isNotEmpty(evalUserPool)) {
//            evalUserPoolService.update(Wrappers.<EvalUserPool>lambdaUpdate()
//                    .eq(EvalUserPool::getId, evalUserPool.getId())
//                    .set(EvalUserPool::getMonthly, YearMonth.from(user.getDataDate()))
//                    .set(EvalUserPool::getUpdateTime, new Date())
//            );
//            evalUserPool.setMonthly(YearMonth.from(user.getDataDate()));
//            return evalUserPool;
//        }
        return this.createUserPoolData(user);
    }

    /**
     * 获取员工社保公积金门店
     *
     * @param userId
     */
    private StaffShopInfoDTO getFundShopInfo(Long userId) {
        StaffBaseInfoDTO staff = ehrRpcService.queryStaffBaseInfo(userId);
        if (PublicUtil.isEmpty(staff) || PublicUtil.isEmpty(staff.getStaffShopList())) {
            log.error("员工没有社保公积金门店，userId:{}", userId);
        }
        return staff.getStaffShopList().stream()
                .filter(shop -> StaffShopTypeEnum.FUND_SHOP.getValue().equals(shop.getType()))
                .findFirst()
                .orElse(null);

    }

    @Transactional(rollbackFor = Exception.class)
    public EvalUserPool modifyPoolData(EvalUserPool pool, EvalGroupUser user) {
        pool.setEvalGroupId(user.getEvalGroupId());
        pool.setShopId(user.getShopId());
        pool.setShopName(user.getShopName());
        pool.setEgc(user.getEgc());
        evalUserPoolService.updateById(pool);
        return pool;
    }

    @Transactional(rollbackFor = Exception.class)
    public EvalUserPool createUserPoolData(EvalGroupUser user) {
        EvalUserPool evalUserPool = new EvalUserPool();
//        evalUserPool.setEvalId(user.getEvalGroupRankId());
        evalUserPool.setEgc(user.getEgc());
        evalUserPool.setEvalGroupId(user.getEvalGroupId());
        evalUserPool.setEvalGroupRankId(user.getEvalGroupRankId());
        evalUserPool.setEvalGroupRankStageId(user.getEvalGroupRankStageId());
        evalUserPool.setUserId(user.getUserId());
        evalUserPool.setUserName(user.getUserName());
        evalUserPool.setPostId(user.getPostId());
        evalUserPool.setPostName(user.getPostName());
        evalUserPool.setShopId(user.getShopId());
        evalUserPool.setShopName(user.getShopName());
        evalUserPool.setScore(BigDecimal.ZERO);
        evalUserPool.setScoreRatio(BigDecimal.ZERO);
        evalUserPool.setInclusion(Boolean.FALSE.equals(user.getIgnored()));
//        evalUserPool.setScoreRatio(BigDecimal.ZERO);
        evalUserPool.setReward(BigDecimal.ZERO);
        evalUserPool.setMonthly(YearMonth.from(user.getDataDate()));
//        evalUserPool.setRegular(Boolean.FALSE);
        evalUserPool.setGroupId(user.getGroupId());
        evalUserPool.setYn(Boolean.TRUE);
        StaffShopInfoDTO staffShopInfoDTO = this.getFundShopInfo(user.getUserId());
        if (PublicUtil.isNotEmpty(staffShopInfoDTO)) {
            evalUserPool.setFundShopId(staffShopInfoDTO.getShopId());
            evalUserPool.setFundShopName(staffShopInfoDTO.getShopName());
        }
        evalUserPoolService.save(evalUserPool);
        return evalUserPool;
    }

}