package com.cloudroam.service.impl;

import cn.hutool.core.util.IdUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.cloudroam.common.enumeration.FileReTypeEnum;
import com.cloudroam.dto.file.FileImageInfo;
import com.cloudroam.model.FileDO;
import com.cloudroam.model.LinFileReDO;
import com.cloudroam.mapper.LinFileReMapper;
import com.cloudroam.service.FileService;
import com.cloudroam.service.LinFileReService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 文件关联关系表 服务实现类
 * </p>
 *
 * @author generator@TaoJie
 * @since 2024-08-06
 */
@Service
public class LinFileReServiceImpl extends ServiceImpl<LinFileReMapper, LinFileReDO> implements LinFileReService {

    @Autowired
    private FileService fileService;

    @Override
    public List<String> getByReIdType(String reid, FileReTypeEnum fileReTypeEnum) {
        QueryWrapper<LinFileReDO> fileReDOQueryWrapper=new QueryWrapper<>();
        fileReDOQueryWrapper.lambda()
                .eq(StringUtils.isNotBlank(reid),LinFileReDO::getReId,reid)
                .eq(!Objects.isNull(fileReTypeEnum),LinFileReDO::getType,fileReTypeEnum.getCode());
        final List<LinFileReDO> fileReDOList = baseMapper.selectList(fileReDOQueryWrapper);

        final List<String> fileIds = fileReDOList.stream().map(fileReDO -> fileReDO.getFileId()).collect(Collectors.toList());
        return fileIds;
    }

    @Override
    public List<FileDO> getFileByReIdType(String reid, FileReTypeEnum fileReTypeEnum) {
        if(StringUtils.isNotBlank(reid)){
            final List<String> fileIds = this.getByReIdType(reid, fileReTypeEnum);
            if(!CollectionUtils.isEmpty(fileIds)){
                final List<FileDO> fileDOS = fileService.getBaseMapper().selectBatchIds(fileIds);
                return fileDOS;
            }
        }
        return null;
    }

    @Transactional
    @Override
    public boolean saveFileReByReIdType(String reId,List<FileImageInfo> ganteImgList, FileReTypeEnum fileReTypeEnum) {

        // 甘特图
        // 先删除甘特图信息
        QueryWrapper<LinFileReDO> linFileReDOQueryWrapper=new QueryWrapper<>();
        linFileReDOQueryWrapper.lambda()
                .eq(LinFileReDO::getReId,reId)
                .eq(LinFileReDO::getType,fileReTypeEnum.getCode())
        ;
        remove(linFileReDOQueryWrapper);

        if(CollectionUtils.isEmpty(ganteImgList)) return false;

        // 保存甘特图信息
        final List<LinFileReDO> fileReDOList = ganteImgList.stream().map(fileImageInfo -> {
            LinFileReDO linFileReDO = new LinFileReDO();
            linFileReDO.setId(IdUtil.simpleUUID());
            linFileReDO.setFileId(String.valueOf(fileImageInfo.getImgId()));
            linFileReDO.setReId(reId);
            linFileReDO.setType(fileReTypeEnum.getCode());
            return linFileReDO;
        }).collect(Collectors.toList());
        saveBatch(fileReDOList);
        return true;
    }

    @Override
    public List<FileImageInfo> transFileToFileImageInfo(List<FileDO> fileList) {
        if(!CollectionUtils.isEmpty(fileList)){
            List<FileImageInfo> fileImageInfos=fileList.stream().map(fileDO -> {
                FileImageInfo fileImageInfo=new FileImageInfo();
                fileImageInfo.setId(String.valueOf(fileDO.getId()));
                fileImageInfo.setImgId(Integer.valueOf(fileDO.getId()));
                fileImageInfo.setDisplay(fileDO.getPath());
                return fileImageInfo;
            }).collect(Collectors.toList());
            return fileImageInfos;
        }else{
            return null;
        }
    }

    @Transactional
    @Override
    public boolean saveFileAnyReByReIdType(String reId, List<Integer> fileList, FileReTypeEnum fileReTypeEnum) {

        QueryWrapper<LinFileReDO> removeWrapper=new QueryWrapper<>();
        removeWrapper.lambda()
                .eq(LinFileReDO::getReId,reId)
                .eq(LinFileReDO::getType,fileReTypeEnum.getCode())
        ;
        remove(removeWrapper);

        if(CollectionUtils.isEmpty(fileList)) return false;

        if(!CollectionUtils.isEmpty(fileList)){
            List<LinFileReDO> fileReDOList = fileList.stream().map(file -> {
                LinFileReDO fileReDO = new LinFileReDO();
                fileReDO.setId(IdUtil.simpleUUID());
                fileReDO.setFileId(String.valueOf(file));
                fileReDO.setReId(reId);
                fileReDO.setType(fileReTypeEnum.getCode());
                return fileReDO;
            }).collect(Collectors.toList());
            if(!CollectionUtils.isEmpty(fileReDOList)){
                return saveBatch(fileReDOList);
            }

        }
        return true;
    }


}