package com.pz.system.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.pz.common.core.page.TableDataInfo;
import com.pz.common.core.domain.PageQuery;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.pz.common.enums.TotalOrderStatus;
import com.pz.common.exception.ServiceException;
import com.pz.merchant.domain.vo.SonOrderVo;
import com.pz.merchant.service.ISonOrderService;
import com.pz.system.domain.DbwzOrder;
import com.pz.system.domain.TotalOrder;
import com.pz.system.domain.bo.TotalOrderBo;
import com.pz.system.mapper.TotalOrderMapper;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import com.pz.system.domain.bo.ZqghOrderBo;
import com.pz.system.domain.vo.ZqghOrderVo;
import com.pz.system.domain.ZqghOrder;
import com.pz.system.mapper.ZqghOrderMapper;
import com.pz.system.service.IZqghOrderService;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;

/**
 * 诊前挂号订单Service业务层处理
 *
 * @author ruoyi
 * @date 2023-09-12
 */
@RequiredArgsConstructor
@Service
public class ZqghOrderServiceImpl implements IZqghOrderService, ISonOrderService {

    private final ZqghOrderMapper baseMapper;

    private final TotalOrderMapper totalOrderMapper;

    /**
     * 查询诊前挂号订单
     */
    @Override
    public ZqghOrderVo queryById(Integer id) {
        return baseMapper.selectVoById(id);
    }

    /**
     * 查询诊前挂号订单列表
     */
    @Override
    public TableDataInfo<ZqghOrderVo> queryPageList(ZqghOrderBo bo, PageQuery pageQuery) {
        Page<ZqghOrderVo> result = baseMapper.findZqghOrderVoPage(pageQuery.build(), bo);
        return TableDataInfo.build(result);
    }

    /**
     * 查询诊前挂号订单列表
     */
    @Override
    public List<ZqghOrderVo> queryList(ZqghOrderBo bo) {
        LambdaQueryWrapper<ZqghOrder> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    @Override
    public List<ZqghOrderVo> queryAppList(ZqghOrderBo bo) {
        LambdaQueryWrapper<ZqghOrder> lqw = buildQueryWrapper(bo);
        List<ZqghOrderVo> zqghOrderVos = baseMapper.selectVoList(lqw);

        //初始化预约集合
        Map<String, Integer> countMap = new HashMap<>();
        countMap.put(bo.getTimeOfAppointmentArr().get(0),0);
        countMap.put(bo.getTimeOfAppointmentArr().get(1),0);
        countMap.put(bo.getTimeOfAppointmentArr().get(2),0);
        countMap.put(bo.getTimeOfAppointmentArr().get(3),0);
        countMap.put(bo.getTimeOfAppointmentArr().get(4),0);
        countMap.put(bo.getTimeOfAppointmentArr().get(5),0);

        //查询订单医生预约订单
        for (ZqghOrderVo obj : zqghOrderVos) {
            String propertyValue = obj.getTimeOfAppointment();
            countMap.put(propertyValue, countMap.getOrDefault(propertyValue, 0) + 1);
        }
        List<ZqghOrderVo> zqghOrderVos1 = new ArrayList<>();
        // 遍历 Map
        for (Map.Entry<String, Integer> entry : countMap.entrySet()) {

            ZqghOrderVo zqghOrderVo = new ZqghOrderVo();
            zqghOrderVo.setTimeOfAppointment(entry.getKey());
            zqghOrderVo.setNum(entry.getValue());
            zqghOrderVos1.add(zqghOrderVo);

        }
        return zqghOrderVos1;
    }

    private LambdaQueryWrapper<ZqghOrder> buildQueryWrapper(ZqghOrderBo bo) {
        Map<String, Object> params = bo.getParams();
        LambdaQueryWrapper<ZqghOrder> lqw = Wrappers.lambdaQuery();
        lqw.eq(bo.getOrderId() != null, ZqghOrder::getOrderId, bo.getOrderId());
        lqw.eq(bo.getHid() != null, ZqghOrder::getHid, bo.getHid());
        lqw.eq(bo.getDid() != null, ZqghOrder::getDid, bo.getDid());
        lqw.eq(bo.getDoctorId() != null, ZqghOrder::getDid, bo.getDoctorId());
        lqw.eq(bo.getDoctorId() != null, ZqghOrder::getDoctorId, bo.getDoctorId());
        lqw.eq(StringUtils.isNotBlank(bo.getTimeOfAppointment()), ZqghOrder::getTimeOfAppointment, bo.getTimeOfAppointment());
        lqw.eq(bo.getStatus() != null, ZqghOrder::getStatus, bo.getStatus());
        lqw.eq(bo.getOverTime() != null, ZqghOrder::getOverTime, bo.getOverTime());
        lqw.eq(StringUtils.isNotBlank(bo.getVoucher()), ZqghOrder::getVoucher, bo.getVoucher());
        lqw.eq(bo.getIsCal() != null, ZqghOrder::getIsCal, bo.getIsCal());
        lqw.in(bo.getTimeOfAppointmentArr() != null, ZqghOrder::getTimeOfAppointment, bo.getTimeOfAppointmentArr());
        return lqw;
    }

    /**
     * 新增诊前挂号订单
     */
    @Override
    public Boolean insertByBo(ZqghOrderBo bo) {
        ZqghOrder add = BeanUtil.toBean(bo, ZqghOrder.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setId(add.getId());
        }
        return flag;
    }

    /**
     * 修改诊前挂号订单
     */
    @Override
    public Boolean updateByBo(ZqghOrderBo bo) {
        ZqghOrder update = BeanUtil.toBean(bo, ZqghOrder.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(ZqghOrder entity) {
        // TODO 做一些数据校验,如唯一约束
    }

    /**
     * 批量删除诊前挂号订单
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<Integer> ids, Boolean isValid) {
        if (isValid) {
            // TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteBatchIds(ids) > 0;
    }

    @Override
    public SonOrderVo getSonOrderSimpleDataByTotalId(Integer totalId) {
        return baseMapper.selectSonOrderInfoByTotalId(totalId);
    }


    /**
     * 切换子订单状态
     *
     * @param totalId 主订单ID
     * @param target  子订单目标状态
     * @return 操作结果
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean switchSonOrderStatus(Long totalId, Integer target) {
        ZqghOrder sonOrder = baseMapper.selectOne(Wrappers.<ZqghOrder>lambdaQuery().eq(ZqghOrder::getOrderId, totalId));
        Objects.requireNonNull(sonOrder, "子订单不存在,请检查");
        sonOrder.setStatus(target);
        totalOrderMapper.update(null,new LambdaUpdateWrapper<TotalOrder>().set(TotalOrder::getSuborderStatus,target).eq(TotalOrder::getId,totalId));
        return baseMapper.updateById(sonOrder) > 0;
    }

    @Override
    public String getFinishTime(Integer totalId) {
        return baseMapper.selectFinishTime(totalId);
    }
}