package com.doumee.service.business.impl;

import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.DispatchUnitMapper;
import com.doumee.dao.business.DuSolutionMapper;
import com.doumee.dao.business.DuWorktypeMapper;
import com.doumee.dao.business.WorktypeMapper;
import com.doumee.dao.business.model.DispatchUnit;
import com.doumee.dao.business.model.DuSolution;
import com.doumee.dao.business.model.DuWorktype;
import com.doumee.dao.business.model.Worktype;
import com.doumee.service.business.DuWorktypeService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.Objects;

/**
 * 派遣单位方案管工种关联信息表Service实现
 * @author 江蹄蹄
 * @date 2024/01/16 10:03
 */
@Service
public class DuWorktypeServiceImpl implements DuWorktypeService {

    @Autowired
    private DuWorktypeMapper duWorktypeMapper;

    @Autowired
    private WorktypeMapper worktypeMapper;

    @Autowired
    private DispatchUnitMapper dispatchUnitMapper;

    @Autowired
    private DuSolutionMapper duSolutionMapper;

    @Override
    public Integer create(DuWorktype duWorktype) {
        duWorktypeMapper.insert(duWorktype);
        return duWorktype.getId();
    }

    @Override
    public void deleteById(Integer id) {
        DuWorktype dbDuWorkType = duWorktypeMapper.selectById(id);
        if(Objects.isNull(dbDuWorkType)){
            throw new BusinessException(ResponseStatus.DATA_EMPTY);
        }
        if(!dbDuWorkType.getStatus().equals(Constants.TWO)){
            throw new BusinessException(ResponseStatus.DATA_ERRO.getCode(),"数据状态错误，无法进行删除");
        }
        DispatchUnit dispatchUnit = dispatchUnitMapper.selectById(dbDuWorkType.getDispatchUnitId());
        if(Objects.isNull(dispatchUnit)){
            throw new BusinessException(ResponseStatus.DATA_EMPTY);
        }
        duWorktypeMapper.deleteById(id);
        if(duWorktypeMapper.selectCount(new QueryWrapper<DuWorktype>().lambda()
                .eq(DuWorktype::getDispatchUnitId,dispatchUnit.getId())
                .eq(DuWorktype::getDuSolutionId,dbDuWorkType.getDuSolutionId()))<=0){
            duSolutionMapper.deleteById(dbDuWorkType.getDuSolutionId());
        }
        if(duWorktypeMapper.selectCount(new QueryWrapper<DuWorktype>().lambda()
                .eq(DuWorktype::getDispatchUnitId,dispatchUnit.getId())
                .eq(DuWorktype::getStatus,Constants.ZERO)
                .ne(DuWorktype::getId,dbDuWorkType.getId())
        )>Constants.ZERO){
            dispatchUnit.setWorktypeStatus(0);
        }else  if(duWorktypeMapper.selectCount(new QueryWrapper<DuWorktype>().lambda()
                .eq(DuWorktype::getDispatchUnitId,dispatchUnit.getId())
                .eq(DuWorktype::getStatus,Constants.TWO)
                .ne(DuWorktype::getId,dbDuWorkType.getId())
        )>Constants.ZERO){
            dispatchUnit.setWorktypeStatus(2);
        }else{
            dispatchUnit.setWorktypeStatus(1);
        }
        dispatchUnitMapper.updateById(dispatchUnit);
    }

    @Override
    public void delete(DuWorktype duWorktype) {
        UpdateWrapper<DuWorktype> deleteWrapper = new UpdateWrapper<>(duWorktype);
        duWorktypeMapper.delete(deleteWrapper);
    }

    @Override
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) {
            return;
        }
        duWorktypeMapper.deleteBatchIds(ids);
    }

    @Override
    public void updateById(DuWorktype duWorktype) {
        duWorktypeMapper.updateById(duWorktype);
    }

    @Override
    public void updateByIdInBatch(List<DuWorktype> duWorktypes) {
        if (CollectionUtils.isEmpty(duWorktypes)) {
            return;
        }
        for (DuWorktype duWorktype: duWorktypes) {
            this.updateById(duWorktype);
        }
    }

    @Override
    public DuWorktype findById(Integer id) {
        return duWorktypeMapper.selectById(id);
    }

    @Override
    public DuWorktype findOne(DuWorktype duWorktype) {
        QueryWrapper<DuWorktype> wrapper = new QueryWrapper<>(duWorktype);
        return duWorktypeMapper.selectOne(wrapper);
    }

    @Override
    public List<DuWorktype> findList(DuWorktype duWorktype) {
        QueryWrapper<DuWorktype> wrapper = new QueryWrapper<>(duWorktype);
        return duWorktypeMapper.selectList(wrapper);
    }
  
    @Override
    public PageData<DuWorktype> findPage(PageWrap<DuWorktype> pageWrap) {
        IPage<DuWorktype> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
        QueryWrapper<DuWorktype> queryWrapper = new QueryWrapper<>();
        Utils.MP.blankToNull(pageWrap.getModel());
        if (pageWrap.getModel().getId() != null) {
            queryWrapper.lambda().eq(DuWorktype::getId, pageWrap.getModel().getId());
        }
        if (pageWrap.getModel().getCreator() != null) {
            queryWrapper.lambda().eq(DuWorktype::getCreator, pageWrap.getModel().getCreator());
        }
        if (pageWrap.getModel().getCreateDate() != null) {
            queryWrapper.lambda().ge(DuWorktype::getCreateDate, Utils.Date.getStart(pageWrap.getModel().getCreateDate()));
            queryWrapper.lambda().le(DuWorktype::getCreateDate, Utils.Date.getEnd(pageWrap.getModel().getCreateDate()));
        }
        if (pageWrap.getModel().getEditor() != null) {
            queryWrapper.lambda().eq(DuWorktype::getEditor, pageWrap.getModel().getEditor());
        }
        if (pageWrap.getModel().getEditDate() != null) {
            queryWrapper.lambda().ge(DuWorktype::getEditDate, Utils.Date.getStart(pageWrap.getModel().getEditDate()));
            queryWrapper.lambda().le(DuWorktype::getEditDate, Utils.Date.getEnd(pageWrap.getModel().getEditDate()));
        }
        if (pageWrap.getModel().getIsdeleted() != null) {
            queryWrapper.lambda().eq(DuWorktype::getIsdeleted, pageWrap.getModel().getIsdeleted());
        }
        if (pageWrap.getModel().getRemark() != null) {
            queryWrapper.lambda().eq(DuWorktype::getRemark, pageWrap.getModel().getRemark());
        }
        if (pageWrap.getModel().getStatus() != null) {
            queryWrapper.lambda().eq(DuWorktype::getStatus, pageWrap.getModel().getStatus());
        }
        if (pageWrap.getModel().getSortnum() != null) {
            queryWrapper.lambda().eq(DuWorktype::getSortnum, pageWrap.getModel().getSortnum());
        }
        if (pageWrap.getModel().getDuSolutionId() != null) {
            queryWrapper.lambda().eq(DuWorktype::getDuSolutionId, pageWrap.getModel().getDuSolutionId());
        }
        if (pageWrap.getModel().getWorkTypeId() != null) {
            queryWrapper.lambda().eq(DuWorktype::getWorkTypeId, pageWrap.getModel().getWorkTypeId());
        }
        if (pageWrap.getModel().getVideoUrl() != null) {
            queryWrapper.lambda().eq(DuWorktype::getVideoUrl, pageWrap.getModel().getVideoUrl());
        }
        if (pageWrap.getModel().getCheckDate() != null) {
            queryWrapper.lambda().ge(DuWorktype::getCheckDate, Utils.Date.getStart(pageWrap.getModel().getCheckDate()));
            queryWrapper.lambda().le(DuWorktype::getCheckDate, Utils.Date.getEnd(pageWrap.getModel().getCheckDate()));
        }
        if (pageWrap.getModel().getCheckInfo() != null) {
            queryWrapper.lambda().eq(DuWorktype::getCheckInfo, pageWrap.getModel().getCheckInfo());
        }
        if (pageWrap.getModel().getCheckUserId() != null) {
            queryWrapper.lambda().eq(DuWorktype::getCheckUserId, pageWrap.getModel().getCheckUserId());
        }
        if (pageWrap.getModel().getStatus2() != null) {
            queryWrapper.lambda().eq(DuWorktype::getStatus2, pageWrap.getModel().getStatus2());
        }
        for(PageWrap.SortData sortData: pageWrap.getSorts()) {
            if (sortData.getDirection().equalsIgnoreCase(PageWrap.DESC)) {
                queryWrapper.orderByDesc(sortData.getProperty());
            } else {
                queryWrapper.orderByAsc(sortData.getProperty());
            }
        }
        return PageData.from(duWorktypeMapper.selectPage(page, queryWrapper));
    }

    @Override
    public long count(DuWorktype duWorktype) {
        QueryWrapper<DuWorktype> wrapper = new QueryWrapper<>(duWorktype);
        return duWorktypeMapper.selectCount(wrapper);
    }
}