package com.gk.hotwork.Service.ServiceImpl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.hotwork.Domain.TaskEnclosure;
import com.gk.hotwork.Domain.UserInfo;
import com.gk.hotwork.Domain.Vo.TaskCheck;
import com.gk.hotwork.Mapper.TaskEnclosureMapper;
import com.gk.hotwork.Service.TaskEnclosureService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@Service("taskEnclosureService")
public class TaskEnclosureServiceImpl extends ServiceImpl<TaskEnclosureMapper,TaskEnclosure> implements TaskEnclosureService {
    @Override
    public List<TaskEnclosure> getListByTaskCode(String code) {
        LambdaQueryWrapper<TaskEnclosure> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEnclosure::getTaskcode,code);
        return baseMapper.selectList(wrapper);
    }

    @Override
    public void removeByTaskCode(String code) {
        LambdaQueryWrapper<TaskEnclosure> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEnclosure::getTaskcode,code);
        baseMapper.delete(wrapper);
    }

    @Override
    public void removeByTaskCodeAndType(String code, String type) {
        LambdaQueryWrapper<TaskEnclosure> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEnclosure::getTaskcode,code);
        wrapper.eq(TaskEnclosure::getType,type);
        baseMapper.delete(wrapper);
    }

    @Override
    public List<TaskEnclosure> getFileByCodeAndType(String code,String type) {
        LambdaQueryWrapper<TaskEnclosure> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEnclosure::getTaskcode,code);
        wrapper.eq(TaskEnclosure::getType,type);
        return baseMapper.selectList(wrapper);
    }

    @Override
    public void removeSubFile(String taskCode) {
        LambdaQueryWrapper<TaskEnclosure> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(TaskEnclosure::getTaskcode,taskCode);
        List<String>list = new ArrayList<>();
        list.add("动土方案文件");
        list.add("断路地段示意图");
        list.add("盲板位置图");
        wrapper.in(TaskEnclosure::getType,list);
        baseMapper.delete(wrapper);
    }

    @Override
    public void removeAndSave(TaskCheck taskCheck, UserInfo userInfo) {
        if (taskCheck.getResources() != null && taskCheck.getResources().size() != 0){
            removeByTaskCodeAndType(taskCheck.getCode(),"检查整改问题");
            List<TaskEnclosure> resources = taskCheck.getResources();
            resources.forEach(item->{
                item.setTaskcode(taskCheck.getCode());
                item.setCreatedat(new Date());
                item.setCreatedby(userInfo.getRealname());
            });
            this.saveBatch(taskCheck.getResources());
        }
    }
}