package com.panzhihua.service_equipment.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.panzhihua.common.model.dtos.equipment.UnionInspectionDto;
import com.panzhihua.common.model.vos.R;
import com.panzhihua.service_equipment.dao.UnionInspectionMapper;
import com.panzhihua.service_equipment.dao.UnionUserMapper;
import com.panzhihua.service_equipment.model.dos.UnionInspection;
import com.panzhihua.service_equipment.model.dos.UnionReport;
import com.panzhihua.service_equipment.model.dos.UnionUser;
import com.panzhihua.service_equipment.service.UnionInspectionService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.time.LocalDateTime;

@Slf4j
@Service
public class UnionInspectionServiceImpl extends ServiceImpl<UnionInspectionMapper, UnionInspection> implements UnionInspectionService {

    @Resource
    private UnionUserMapper unionUserMapper;


    /**
     * 新增户外劳工站巡查
     *
     * @param unionInspectionDto
     * @return 新增结果
     */
    @Override
    public R add(UnionInspectionDto unionInspectionDto) {
        UnionInspection unionInspection = new UnionInspection();
        BeanUtils.copyProperties(unionInspectionDto, unionInspection);
        UnionUser unionUser=null;
        if (unionInspectionDto.getIsApplets().equals(1))
        {
            unionUser = unionUserMapper
                .selectOne(new QueryWrapper<UnionUser>().eq("user_id", unionInspection.getCreateUserId()));
        }else if (unionInspectionDto.getIsApplets().equals(3)) {
            unionUser = unionUserMapper
                    .selectOne(new QueryWrapper<UnionUser>().eq("id", unionInspection.getCreateUnionUserId()));
        }
        if (unionUser!=null){
            unionInspection.setCreateUnionUserId(unionUser.getId());
            unionInspection.setCreatePhone(unionUser.getPhone());
        }
        unionInspection.setStatus(1);
        unionInspection.setCreateAt(LocalDateTime.now());
        if (baseMapper.insert(unionInspection) > 0) {
            return R.ok();
        }
        return R.fail();
    }

    /**
     * 分页户外劳工站巡查
     *
     * @param unionInspectionDto
     * @return 动态结果
     */
    @Override
    public R<IPage<UnionReport>> query(UnionInspectionDto unionInspectionDto) {
        Page page = new Page(unionInspectionDto.getPageNum(), unionInspectionDto.getPageSize());
        QueryWrapper<UnionInspection> unionOutdoorLaborDynamicQueryWrapper = new QueryWrapper<>();
        if (unionInspectionDto.getIsApplets().equals(1)){
            unionOutdoorLaborDynamicQueryWrapper.eq("create_user_id",unionInspectionDto.getCreateUserId()).eq("type",2);
            if (unionInspectionDto.getStatus()!=null)
                unionOutdoorLaborDynamicQueryWrapper.eq("status",unionInspectionDto.getStatus());
        }else if(unionInspectionDto.getIsApplets().equals(3)){
            unionOutdoorLaborDynamicQueryWrapper.eq("create_union_user_id",unionInspectionDto.getCreateUnionUserId()).eq("type",2);
            if (unionInspectionDto.getStatus()!=null)
                unionOutdoorLaborDynamicQueryWrapper.eq("status",unionInspectionDto.getStatus());
        }
        Page pageResult = baseMapper.selectPage(page, unionOutdoorLaborDynamicQueryWrapper);
        pageResult.setTotal(pageResult.getRecords().size());
        return R.ok(pageResult);
    }
}