package cc.mrbird.febs.server.hr.service.impl;

import cc.mrbird.febs.common.core.constant.ModuleCode;
import cc.mrbird.febs.common.core.exception.FebsException;
import cc.mrbird.febs.common.core.utils.FebsUtil;
import cc.mrbird.febs.common.core.utils.MyUtil;
import cc.mrbird.febs.common.core.utils.SequenceUtil;
import cc.mrbird.febs.server.hr.entity.*;
import cc.mrbird.febs.server.hr.mapper.EmpAccessoryMapper;
import cc.mrbird.febs.server.hr.properties.FebsServerHrProperties;
import cc.mrbird.febs.server.hr.service.IEmpAccessoryService;
import cc.mrbird.febs.server.hr.service.IEmpBaseInfoService;
import cc.mrbird.febs.server.hr.service.ILabelService;
import cc.mrbird.febs.server.hr.vo.EmpAccessoryVO;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.lang.Dict;
import cn.hutool.core.lang.Snowflake;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.ZipUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.aspectj.weaver.ast.Or;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.annotation.Propagation;
import lombok.RequiredArgsConstructor;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import cc.mrbird.febs.common.core.entity.QueryRequest;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.util.*;

/**
 * 附件管理 Service实现
 *
 * @author yz
 * @date 2021-02-15 17:07:19
 */
@Service
@RequiredArgsConstructor
@Transactional(propagation = Propagation.SUPPORTS, readOnly = true)
public class EmpAccessoryServiceImpl extends ServiceImpl<EmpAccessoryMapper, EmpAccessory> implements IEmpAccessoryService {
    private final ILabelService labelService;
    private final EmpAccessoryMapper empAccessoryMapper;
    private final FebsServerHrProperties properties;
    private final String operatorId = Optional.ofNullable(FebsUtil.getCurrentUser())
            .map(u -> u.getUserId().toString())
            .orElse("1");
    private final IEmpBaseInfoService empBaseInfoService;
    /**
     * 参数2为数据中心ID 参数1为终端ID
     */
    private Snowflake snowflake = IdUtil.getSnowflake(ModuleCode.HR_FIlE, 1);

    @Override
    public IPage<Map<String, Object>> findEmpAccessorys(QueryRequest request, EmpAccessoryVO vo) {
         QueryWrapper<EmpAccessoryVO> queryWrapper = new QueryWrapper<>();
        if (StrUtil.isNotBlank(vo.getBasic())) {
            queryWrapper.like("a.empNumb", vo.getBasic()).or().like("a.deptName", vo.getBasic()).or().like("a.certificateNumb", vo.getBasic()).or().like("a.certificateNumb", vo.getBasic()).or().like("a.jobName", vo.getBasic());
        }
        if (StrUtil.isNotBlank(vo.getEmpName())) {
            queryWrapper.like("a.empName", vo.getEmpName());
        }
        if (StrUtil.isNotBlank(vo.getEmpNumb())) {
            queryWrapper.like("a.empNumb", vo.getEmpNumb());
        }
        if (StrUtil.isNotBlank(vo.getCertificateNumb())) {
            queryWrapper.like("a.certificateNumb", vo.getCertificateNumb());
        }
        if (StrUtil.isNotBlank(vo.getDeptName())) {
            queryWrapper.like("a.deptName", vo.getDeptName());
        }
        if (StrUtil.isNotBlank(vo.getEntryDate())) {
            queryWrapper.between("a.entryDate", vo.getEntryDate().split(",")[0], vo.getEntryDate().split(",")[1]);
        }
        if (StrUtil.isNotBlank(vo.getAge())) {
            String[] arr = vo.getAge().split(",");
            for (String s : arr) {
                if (StrUtil.equals(s, "1")) {
                    queryWrapper.between("a.age", 18, 29);
                }  if (StrUtil.equals(s, "2")) {
                    queryWrapper.between("a.age", 30, 39);
                }  if (StrUtil.equals(s, "3")) {
                    queryWrapper.between("a.age", 40, 49);
                }  if (StrUtil.equals(s, "4")) {
                    queryWrapper.between("a.age", 50, 59);
                }  if (StrUtil.equals(s, "5")) {
                    queryWrapper.between("a.age", 60, 69);
                }  if (StrUtil.equals(s, "6")) {
                    queryWrapper.ge("a.age", 69);
                }
            }
        }
        if (StrUtil.isNotBlank(vo.getArchivesStatus())) {
                queryWrapper.in("a.archivesStatus", vo.getArchivesStatus());
        }
        if (StrUtil.isNotBlank(vo.getEducation())) {
                queryWrapper.in("a.education", vo.getEducation());
        }
        if (StrUtil.isNotBlank(vo.getSex())) {
                queryWrapper.in("a.sex", vo.getSex());
        }
        if (StrUtil.isNotBlank(vo.getInsuranceType())) {
                queryWrapper.in("a.insuranceType", vo.getInsuranceType());
        }
        if (StrUtil.isNotBlank(vo.getPolitics())) {
                queryWrapper.in("a.politics", vo.getPolitics());
        }
        if (StrUtil.isNotBlank(vo.getEmpStatus())) {
            queryWrapper.in("a.empStatus", vo.getEmpStatus());
        }
        queryWrapper.eq("a.delFlag", 0);
        Page<EmpAccessory> page = new Page<>(request.getPageNum(), request.getPageSize());
        String querySql = empAccessoryMapper.getQuerySql();
        IPage<Map<String, Object>> mapIPage = empAccessoryMapper.selectPageVo(page, queryWrapper, querySql);
        return mapIPage;
    }

    @Override
    public List<EmpAccessory> findEmpAccessorys(EmpAccessory empAccessory) {
        LambdaQueryWrapper<EmpAccessory> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(EmpAccessory::getDelFlag, 0);
        if (null != empAccessory.getLabelid()) {
            queryWrapper.eq(EmpAccessory::getLabelid, empAccessory.getLabelid());
        }
        if (StrUtil.isNotBlank(empAccessory.getFilesname())) {
            queryWrapper.like(EmpAccessory::getFilesname, empAccessory.getFilesname());
        }
        if (null != empAccessory.getEmpid()) {
            queryWrapper.eq(EmpAccessory::getEmpid, empAccessory.getEmpid());
        }
        return this.baseMapper.selectList(queryWrapper);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void createEmpAccessory(Long labelid, MultipartFile file, Long empId) throws FebsException, IOException {
        EmpBaseInfo empBaseInfo = empBaseInfoService.getById(empId);
        String nameAndCertificateNumb = empBaseInfo.getEmpName() + "_" + empBaseInfo.getCertificateNumb() + "/";
        Dict dict = MyUtil.filesUpload(file, properties.getUploadSinglePath() + nameAndCertificateNumb, snowflake.nextIdStr());
        EmpAccessory accessory = new EmpAccessory();
        accessory.setCreatetime(new Date());
        accessory.setCreator(operatorId);
        accessory.setDelFlag(0);
        accessory.setEmpid(empId);
        accessory.setFilesaddress(dict.getStr("newName"));
        accessory.setFilesformat(StrUtil.isBlank(dict.getStr("suffix")) ? dict.getStr("suffix") : dict.getStr("suffix").substring(1));
        accessory.setAccessoryid(SequenceUtil.generateId(0L, ModuleCode.HR_FIlE));
        accessory.setFilesname(dict.getStr("fileName"));
        accessory.setLabelid(labelid);
        accessory.setModifier(operatorId);
        accessory.setModifytime(new Date());
        accessory.setVersion(0);
        this.save(accessory);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateEmpAccessory(EmpAccessory empAccessory) {
        this.saveOrUpdate(empAccessory);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteEmpAccessory(String accessoryids) {
        String[] arr = accessoryids.split(",");
        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
        wapper.in(EmpAccessory::getAccessoryid,arr);
        wapper.set(EmpAccessory::getDelFlag,1);
        wapper.set(EmpAccessory::getModifier,operatorId);
        wapper.set(EmpAccessory::getModifytime,new Date());
        this.update(wapper);

        LambdaQueryWrapper<EmpAccessory> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.in(EmpAccessory::getAccessoryid,arr);
        List<EmpAccessory> list = this.list(queryWrapper);
        list.parallelStream().forEach(i-> {
            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(i.getEmpid());
            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
            FileUtil.del(properties.getUploadSinglePath()+nameAndCertificateNumb+"/" + i.getFilesaddress());
        });
    }


    @Override
    public void download(String accessoryids, HttpServletRequest request, HttpServletResponse response) {
        String[] arr = accessoryids.split(",");
        List<File> files = new ArrayList<>();
        List<String> fileName = new ArrayList<>();
        Arrays.stream(arr).forEach(i -> {
            EmpAccessory accessory = this.getById(i);
            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(accessory.getEmpid());
            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
            files.add(new File(properties.getUploadSinglePath() + nameAndCertificateNumb + accessory.getFilesaddress()));
            fileName.add(accessory.getFilesname());
        });
        MyUtil.download(request, response, files, fileName);

    }

    @Override
    public void mvFiles(String accessoryids,Long labelid,Long empId) {
        String[] arr = accessoryids.split(",");

        LambdaUpdateWrapper<EmpAccessory> wapper = new LambdaUpdateWrapper<>();
        wapper.in(EmpAccessory::getAccessoryid, arr);
        wapper.set(EmpAccessory::getModifier,operatorId);
        wapper.set(EmpAccessory::getModifytime,new Date());
        wapper.set(EmpAccessory::getLabelid,labelid);
        this.update(wapper);

    }

    @Override
    public void singledownload(String empIds, Long labelid, HttpServletRequest request, HttpServletResponse response) throws IOException {
        String[] arr = empIds.split(",");
        //主文件夹
        String mianFileStr = properties.getUploadSinglePath()+snowflake.nextIdStr()+"/";
        FileUtil.mkdir(mianFileStr);
        Arrays.stream(arr).forEach(i -> {
            LambdaQueryWrapper<EmpAccessory> lambdaQueryWrapper = new LambdaQueryWrapper<>();
            lambdaQueryWrapper.eq(EmpAccessory::getEmpid,i);
            lambdaQueryWrapper.eq(EmpAccessory::getDelFlag,0);
            if (!labelid.equals(-1L)){
                lambdaQueryWrapper.eq(EmpAccessory::getLabelid,labelid);
            }
            List<EmpAccessory> accessory = this.list(lambdaQueryWrapper);
            EmpBaseInfo empBaseInfo = empBaseInfoService.getById(i);
            String nameAndCertificateNumb = empBaseInfo.getEmpName()+"_"+empBaseInfo.getCertificateNumb()+"/";
            //创建个人文件夹
            String singlefileStr = mianFileStr+nameAndCertificateNumb;
            FileUtil.mkdir(singlefileStr);
            accessory.stream().forEach(a->{
                Label label = labelService.getById(a.getLabelid());
                FileUtil.copy(properties.getUploadSinglePath() + nameAndCertificateNumb + a.getFilesaddress(),singlefileStr+label.getLabelname()+"/"+a.getFilesname(),true);
            });
        });
        ZipUtil.zip(mianFileStr);
        try {
            MyUtil.downloadFile(new File(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString()),response);
        } catch (IOException e) {
            log.error("下载文件异常",e);
        }
        FileUtil.del(new StringBuilder().append(mianFileStr, 0, mianFileStr.length() - 1).append(".zip").toString());
        FileUtil.del(mianFileStr);
    }


    @Override
    public Integer getLabelCount(Long labelId) {
            LambdaQueryWrapper<EmpAccessory> wrapper = new LambdaQueryWrapper<>();
            wrapper.eq(EmpAccessory::getDelFlag,0);
            wrapper.eq(EmpAccessory::getLabelid,labelId);
        return this.count(wrapper);
    }
}