package com.lxzn.nc.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.lxzn.auth.JwtUtil;
import com.lxzn.framework.domain.filesystem.response.FileUploadResult;
import com.lxzn.framework.domain.nc.*;
import com.lxzn.framework.domain.nc.request.DocInfoQueryRequest;
import com.lxzn.framework.domain.nc.request.DocInfoUploadRequest;
import com.lxzn.framework.domain.nc.response.DocumentCode;
import com.lxzn.framework.domain.ucenter.response.UcenterCode;
import com.lxzn.framework.exception.ExceptionCast;
import com.lxzn.framework.model.response.CommonCode;
import com.lxzn.framework.model.response.QueryListResponseResult;
import com.lxzn.framework.model.response.QueryPageResponseResult;
import com.lxzn.framework.model.response.ResponseResult;
import com.lxzn.framework.utils.BeanMapper;
import com.lxzn.framework.utils.ValidateUtil;
import com.lxzn.framework.utils.file.FileUtil;
import com.lxzn.nc.dao.DocInfoMapper;
import com.lxzn.nc.service.*;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.List;

@Service
public class DocInfoServiceImpl extends ServiceImpl<DocInfoMapper,DocInfo> implements IDocInfoService {
    @Autowired
    private IDocClassificationService docClassificationService;
    @Autowired
    private IDocFileService docFileService;
    @Autowired
    private IFileOperateService fileOperateService;
    @Autowired
    private IDocRelativeService docRelativeService;

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean addDocInfo(MultipartFile file, DocInfoUploadRequest uploadRequest) {
        if(file == null || file.isEmpty() || uploadRequest == null)
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        if(!ValidateUtil.validateString(uploadRequest.getAttributionId()) || !ValidateUtil.validateInteger(uploadRequest.getAttributionType()))
            ExceptionCast.cast(DocumentCode.DOC_ATTRIBUTE_ERROR);
        if(!ValidateUtil.validateString(uploadRequest.getDocClassCode()))
            ExceptionCast.cast(DocumentCode.DOC_CLASS_ERROR);
        DocClassification docClass = docClassificationService.getByCode(uploadRequest.getDocClassCode());
        if(docClass == null)
            ExceptionCast.cast(DocumentCode.DOC_CLASS_ERROR);
        String fileName = FileUtil.getFilenameNonSuffix(file.getOriginalFilename());
        String suffixFile = FileUtil.getFileSuffix(file.getOriginalFilename());
        DocInfo en = findByAttrAndDocName(fileName, uploadRequest.getAttributionType(), uploadRequest.getAttributionId(),suffixFile);
        if(en != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_EXIST);
        FileUploadResult fileUploadResult = FileUtil.uploadFile(file);
        if(fileUploadResult == null)
            ExceptionCast.cast(DocumentCode.DOC_UPLOAD_ERROR);
        DocInfo docInfo = new DocInfo();
        String docId = IdWorker.getIdStr();
        docInfo.setDocId(docId);
        docInfo.setDocName(fileUploadResult.getFileName());
        docInfo.setDocSuffix(fileUploadResult.getFileSuffix());
        docInfo.setDocStatus(1);
        DocRelative docRelative = new DocRelative();
        docRelative.setAttributionId(uploadRequest.getAttributionId());
        docRelative.setDocId(docInfo.getDocId());
        docRelative.setAttributionType(uploadRequest.getAttributionType());
        docRelative.setClassificationId(docClass.getClassificationId());
        boolean b = docRelativeService.save(docRelative);
        if(!b)
            ExceptionCast.cast(DocumentCode.DOC_UPLOAD_ERROR);
        DocFile docFile = new DocFile();
        docFile.setDocId(docInfo.getDocId());
        docFile.setFileName(fileUploadResult.getFileName());
        docFile.setFileEncodeName(fileUploadResult.getEncodeFileName());
        docFile.setFilePath(fileUploadResult.getFilePath());
        docFile.setFileSize(fileUploadResult.getFileSize());
        docFile.setFileSuffix(fileUploadResult.getFileSuffix());
        b =  docFileService.addDocFile(docFile);
        if(!b)
            ExceptionCast.cast(DocumentCode.DOC_UPLOAD_ERROR);
        docInfo.setPublishVersion(docFile.getDocVersion());
        docInfo.setPublishFileId(docFile.getFileId());
        return super.save(docInfo);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean editDocInfo(String id, DocInfo docInfo) {
        if(!ValidateUtil.validateString(id) || docInfo == null)
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        DocInfo asDoc = findByAttrAndDocNameNotThis(docInfo.getDocName(), docInfo.getAttributionType(), docInfo.getAttributionId(), id,docInfo.getDocSuffix());
        if(asDoc != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_EXIST);
        docInfo.setDocId(id);
        docInfo.setDocStatus(null);
        docInfo.setDocSuffix(null);
        docInfo.setClassificationId(null);
        docInfo.setPublishFileId(null);
        docInfo.setPublishVersion(null);
        return super.updateById(docInfo);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(en.getDocStatus() != 1)
            ExceptionCast.cast(DocumentCode.DOC_DEL_ERROR);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_OPERATING);
        boolean b = docFileService.deleteByDocId(id);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = docRelativeService.deleteByDocId(id);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        return super.removeById(id);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean pushDocFile(String id, MultipartFile file) {
        if(file == null || file.isEmpty() || !ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        String userId = JwtUtil.getUserId();
        if(!ValidateUtil.validateString(userId))
            ExceptionCast.cast(UcenterCode.UCENTER_ACCOUNT_NOT_EXIST);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate == null || !userId.equals(operate.getCreateUser()))
            ExceptionCast.cast(DocumentCode.DOC_OPERATE_PUSH_FAIL);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        List<DeviceInfo> deviceList = docRelativeService.findDeviceByDocId(en.getDocId());
        if(deviceList != null && !deviceList.isEmpty())
            ExceptionCast.cast(DocumentCode.DOC_DEVICE_EXIST);
        String suffix = FileUtil.getFileSuffix(file.getOriginalFilename());
        if(ValidateUtil.validateString(en.getDocSuffix())) {
            if(!en.getDocSuffix().equals(suffix)) {
                ExceptionCast.cast(DocumentCode.DOC_SUFFIX_ERROR);
            }
        }else {
            if(ValidateUtil.validateString(suffix)) {
                ExceptionCast.cast(DocumentCode.DOC_SUFFIX_ERROR);
            }
        }
        FileUploadResult fileUploadResult = FileUtil.uploadFile(file);
        DocFile docFile = new DocFile();
        docFile.setDocId(id);
        docFile.setFileName(fileUploadResult.getFileName());
        docFile.setFileEncodeName(fileUploadResult.getEncodeFileName());
        docFile.setFilePath(fileUploadResult.getFilePath());
        docFile.setFileSize(fileUploadResult.getFileSize());
        docFile.setFileSuffix(fileUploadResult.getFileSuffix());
        boolean b = fileOperateService.deleteByDocId(id);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b=  docFileService.addDocFile(docFile);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        en.setPublishVersion(docFile.getDocVersion());
        en.setPublishFileId(docFile.getFileId());
        return super.updateById(en);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public DocFile checkDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(en.getDocStatus() != 1)
            ExceptionCast.cast(DocumentCode.DOC_OPERATE_FAIL);
        if(!ValidateUtil.validateString(en.getPublishFileId()))
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NONE);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_OPERATING);
        operate = new FileOperate();
        operate.setDocId(id);
        boolean b = fileOperateService.save(operate);
        if(!b)
            ExceptionCast.cast(DocumentCode.DOC_OPERATE_FAIL);
        DocFile fileInfo = docFileService.getById(en.getPublishFileId());
        if(fileInfo == null)
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NOT_EXIST);
        return fileInfo;
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean publishDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(en.getDocStatus() != 1)
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_ERROR);
        if(!ValidateUtil.validateString(en.getPublishFileId()))
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NONE);
        DocFile pubFile = docFileService.getById(en.getPublishFileId());
        if(pubFile == null)
            ExceptionCast.cast(DocumentCode.DOC_FILE_ERROR);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_OPERATING);
        DocFile old = docFileService.getDocFileNearest(en.getDocId());
        if(old == null)
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NONE);
        if(old.getDocVersion().contains(".")) {
            DocFile docFile = new DocFile();
            BeanMapper.copy(pubFile, docFile);
            docFile.setFileId(null);
            docFile.setDocVersion(old.getDocVersion().substring(0, 1));
            boolean b =  docFileService.save(docFile);
            if(!b)
                ExceptionCast.cast(CommonCode.FAIL);
            return super.lambdaUpdate().eq(DocInfo::getDocId, id).set(DocInfo::getDocStatus, 2).set(DocInfo::getPublishFileId, docFile.getFileId())
                    .set(DocInfo::getPublishVersion, docFile.getDocVersion()).update();
        }else {
            return super.lambdaUpdate().eq(DocInfo::getDocId, id).set(DocInfo::getDocStatus, 2).set(DocInfo::getPublishFileId, old.getFileId())
                    .set(DocInfo::getPublishVersion, old.getDocVersion()).update();
        }
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean republishDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(en.getDocStatus() != 2)
            ExceptionCast.cast(DocumentCode.DOC_REPUBLISH_ERROR);
        return super.lambdaUpdate().eq(DocInfo::getDocId, id).set(DocInfo::getDocStatus, 1).update();
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean pigeonholeDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(en.getDocStatus() == 3)
            ExceptionCast.cast(DocumentCode.DOC_PIGEONHOLE_ERROR);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate != null)
            ExceptionCast.cast(DocumentCode.DOC_IS_OPERATING);
        return super.lambdaUpdate().eq(DocInfo::getDocId, id).set(DocInfo::getDocStatus, 3).update();
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean cancelPullDocInfo(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        String userId = JwtUtil.getUserId();
        if(!ValidateUtil.validateString(userId))
            ExceptionCast.cast(UcenterCode.UCENTER_ACCOUNT_NOT_EXIST);
        FileOperate operate = fileOperateService.getByDocId(id);
        if(operate == null || !userId.equals(operate.getCreateUser()))
            ExceptionCast.cast(DocumentCode.DOC_CANCEL_PULL_ERROR);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        return fileOperateService.deleteByDocId(id);
    }

    @Override
    public QueryPageResponseResult<DocInfo> findPageList(int page, int size, DocInfoQueryRequest docQuery) {
        if(page < 1 || size < 1) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }
        if(docQuery == null || !ValidateUtil.validateString(docQuery.getAttributionId())
                || !ValidateUtil.validateInteger(docQuery.getAttributionType()) ||
                !ValidateUtil.validateString(docQuery.getDocClassCode())) {
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        }
        DocClassification docClassification = docClassificationService.getByCode(docQuery.getDocClassCode());
        if(docClassification == null) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }
        IPage<DocInfo> pageData = new Page<>(page, size);

//        LambdaQueryWrapper<DocInfo> lambdaQuery = Wrappers.lambdaQuery();
//        lambdaQuery.eq(DocInfo::getAttributionType, docQuery.getAttributionType()).eq(DocInfo::getAttributionId, docQuery.getAttributionId())
//                .eq(DocInfo::getClassificationId, docClassification.getClassificationId());
        QueryWrapper<DocInfo> queryWrapper = Wrappers.query();
        queryWrapper.eq("r.attribution_type", docQuery.getAttributionType()).eq("r.attribution_id", docQuery.getAttributionId())
                .eq("r.classification_id", docClassification.getClassificationId());
        if(ValidateUtil.validateString(docQuery.getAscStr())) {
            String[] ascArr = docQuery.getAscStr().split(",");
            ((Page<DocInfo>) pageData).setAsc(ascArr);
        }
        if(ValidateUtil.validateString(docQuery.getDescStr())) {
            String[] descStr = docQuery.getDescStr().split(",");
            ((Page<DocInfo>) pageData).setDesc(descStr);
        }
        if (StringUtils.isBlank(docQuery.getAscStr()) || StringUtils.isBlank(docQuery.getDescStr())) {
            queryWrapper.orderByDesc("u.create_time");
        }
        IPage<DocInfo> docInfoIPage = super.getBaseMapper().findDocExtList(pageData, queryWrapper);
        return new QueryPageResponseResult<>(CommonCode.SUCCESS, docInfoIPage);
    }

    @Override
    public DocInfo getByDocAttrAndDocId(String docId, Integer attrType, String attributeId) {
        if(!ValidateUtil.validateString(docId) || !ValidateUtil.validateString(attributeId) || !ValidateUtil.validateInteger(attrType))
            return null;
        List<DocInfo> list = super.getBaseMapper().getByDocAttrAndDocId(docId, attrType, attributeId);
        if(list == null || list.size() != 1)
            return null;
        return list.get(0);
    }

    @Override
    public DocFile previewDocFile(String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        DocFile fileInfo = docFileService.getById(en.getPublishFileId());
        if(fileInfo == null)
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NOT_EXIST);
        return fileInfo;
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByProductId(String productId) {
        if(!ValidateUtil.validateString(productId))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        Integer attrType = 1;
        boolean b = docFileService.deleteByDocAttr(attrType, productId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = deleteByDocAttr(attrType, productId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        return docRelativeService.deleteByDocAttr(attrType, productId);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByComponentId(String componentId) {
        if(!ValidateUtil.validateString(componentId))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        Integer attrType = 2;
        boolean b = docFileService.deleteByDocAttr(attrType, componentId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = deleteByDocAttr(attrType, componentId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        return docRelativeService.deleteByDocAttr(attrType, componentId);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByPartsId(String partsId) {
        if(!ValidateUtil.validateString(partsId))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        Integer attrType = 3;
        boolean b = docFileService.deleteByDocAttr(attrType, partsId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = deleteByDocAttr(attrType, partsId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        return docRelativeService.deleteByDocAttr(attrType, partsId);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByDeviceId(String deviceId) {
        if(!ValidateUtil.validateString(deviceId))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        Integer attrType = 4;
        /*boolean b = docFileService.deleteByDocAttr(attrType, deviceId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = deleteByDocAttr(attrType, deviceId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);*/
        return docRelativeService.deleteByDocAttr(attrType, deviceId);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByProcessId(String processId) {
        if(!ValidateUtil.validateString(processId))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        Integer attrType = 4;
        boolean b = docFileService.deleteByDocAttr(attrType, processId);
        if(!b)
            ExceptionCast.cast(CommonCode.FAIL);
        b = deleteByDocAttr(attrType, processId);
        if(!b) {
            ExceptionCast.cast(CommonCode.FAIL);
        }
        return docRelativeService.deleteByDocAttr(attrType, processId);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByDocAttr(Integer attrType, String attrId) {
        int i = super.getBaseMapper().deleteByDocAttr(attrType, attrId);
        if(i >= 0) {
            return true;
        }
        return false;
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public ResponseResult pullDocFile(HttpServletRequest request, HttpServletResponse response, String id) {
        DocFile docFile = checkDocInfo(id);
        if(!ValidateUtil.validateString(docFile.getFilePath()) || !ValidateUtil.validateString(docFile.getFileName()) ||
                /*!ValidateUtil.validateString(docFile.getFileSuffix()) ||*/ !ValidateUtil.validateString(docFile.getFileEncodeName()))
            return new ResponseResult(CommonCode.FAIL);
        String fileName;
        if(ValidateUtil.validateString(docFile.getFileSuffix())) {
            fileName = docFile.getFileName() + "." + docFile.getFileSuffix();
        }else {
            fileName = docFile.getFileName();
        }
        String filePath = docFile.getFilePath();
        String fileEncodeName = docFile.getFileEncodeName();
        FileUtil.downLoadFile(response, fileEncodeName, filePath, fileName);
        return null;
    }

    @Override
    public DocInfo findByAttrAndDocName(String docName, Integer attrType, String attrId,String suffixFile) {
        List<DocInfo> docInfoList = super.getBaseMapper().findByAttrAndDocName(docName, attrType, attrId,suffixFile);
        if(docInfoList == null || docInfoList.isEmpty())
            return null;
        return docInfoList.get(0);
    }

    @Override
    public DocInfo findByAttrAndDocNameNotThis(String docName, Integer attrType, String attrId, String docId,String docSuffix) {
        List<DocInfo> docInfoList = super.getBaseMapper().findByAttrAndDocName(docName, attrType, attrId,docSuffix);
        if(docInfoList == null || docInfoList.isEmpty())
            return null;
        for(DocInfo docInfo : docInfoList) {
            if(!docId.equals(docInfo.getDocId())) {
                return docInfo;
            }
        }
        return null;
    }

    @Override
    public List<DocInfo> getByDocId(String docId) {
        return super.getBaseMapper().getByDocId(docId);
    }

    @Override
    public QueryPageResponseResult<DocInfo> findPageListByDevice(int page, int size, DocInfoQueryRequest docQuery) {
        if(page < 1 || size < 1) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }
        if(docQuery == null || !ValidateUtil.validateString(docQuery.getAttributionId())
                || !ValidateUtil.validateInteger(docQuery.getAttributionType()) ||
                !ValidateUtil.validateString(docQuery.getDocClassCode())) {
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        }
        DocClassification docClassification = docClassificationService.getByCode(docQuery.getDocClassCode());
        if(docClassification == null) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }
        IPage<DocInfo> pageData = new Page<>(page, size);
        QueryWrapper<DocInfo> queryWrapper = Wrappers.query();
        queryWrapper.eq("r.attribution_type", docQuery.getAttributionType()).eq("r.attribution_id", docQuery.getAttributionId())
                .eq("r.classification_id", docClassification.getClassificationId());
        if(ValidateUtil.validateString(docQuery.getAscStr())) {
            String[] ascArr = docQuery.getAscStr().split(",");
            ((Page<DocInfo>) pageData).setAsc(ascArr);
        }
        if(ValidateUtil.validateString(docQuery.getDescStr())) {
            String[] descStr = docQuery.getDescStr().split(",");
            ((Page<DocInfo>) pageData).setDesc(descStr);
        }
        IPage<DocInfo> docInfoIPage = super.getBaseMapper().findDeviceDocExtList(pageData, queryWrapper);
        return new QueryPageResponseResult<>(CommonCode.SUCCESS, docInfoIPage);
    }

    @Override
    public QueryListResponseResult<DocInfo> findList(DocInfoQueryRequest docQuery) {
        if(docQuery == null || !ValidateUtil.validateString(docQuery.getAttributionId())
                || !ValidateUtil.validateInteger(docQuery.getAttributionType()) ||
                !ValidateUtil.validateString(docQuery.getDocClassCode())) {
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        }
        DocClassification docClassification = docClassificationService.getByCode(docQuery.getDocClassCode());
        if(docClassification == null) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }

        QueryWrapper<DocInfo> queryWrapper = Wrappers.query();
        queryWrapper.eq("r.attribution_type", docQuery.getAttributionType()).eq("r.attribution_id", docQuery.getAttributionId())
                .eq("r.classification_id", docClassification.getClassificationId());
        List<DocInfo> docInfoIPage = super.getBaseMapper().findDocExtList(queryWrapper);
        return new QueryListResponseResult<>(CommonCode.SUCCESS, docInfoIPage);
    }

    @Override
    public List<DocInfo> getByProcessIds(List<ProcessStream> streams) {
        List<String> ids = new ArrayList<>();
        for(ProcessStream s : streams) {
            ids.add(s.getProcessId());
        }
        DocClassification docClassification = docClassificationService.getByCode("NC");
        if(docClassification == null) {
            ExceptionCast.cast(CommonCode.INVALID_PAGE);
        }
        QueryWrapper<DocInfo> queryWrapper = Wrappers.query();
        queryWrapper.in("r.attribution_id", ids)
                .eq("r.classification_id", docClassification.getClassificationId())
        .ne("u.doc_status", 3).isNotNull("u.publish_file_id");
        return super.getBaseMapper().findDocExtList(queryWrapper);
    }

    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean batchRemoveDocInfo(String[] docIds) {
        if(docIds == null || docIds.length < 1)
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        for(String docId : docIds) {
            this.deleteDocInfo(docId);
        }
        return true;
    }

    @Override
    public ResponseResult downloadDocFile(HttpServletRequest request, HttpServletResponse response, String id) {
        if(!ValidateUtil.validateString(id))
            ExceptionCast.cast(CommonCode.INVALID_PARAM);
        DocInfo en = super.getById(id);
        if(en == null)
            ExceptionCast.cast(DocumentCode.DOC_NOT_EXIST);
        if(!ValidateUtil.validateString(en.getPublishFileId()))
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NONE);
        DocFile docFile = docFileService.getById(en.getPublishFileId());
        if(docFile == null)
            ExceptionCast.cast(DocumentCode.DOC_PUBLISH_FILE_NOT_EXIST);
        if(!ValidateUtil.validateString(docFile.getFilePath()) || !ValidateUtil.validateString(docFile.getFileName()) ||
                /*!ValidateUtil.validateString(docFile.getFileSuffix()) ||*/ !ValidateUtil.validateString(docFile.getFileEncodeName()))
            return new ResponseResult(CommonCode.FAIL);
        String fileName;
        if(ValidateUtil.validateString(docFile.getFileSuffix())) {
            fileName = docFile.getFileName() + "." + docFile.getFileSuffix();
        }else {
            fileName = docFile.getFileName();
        }
        String filePath = docFile.getFilePath();
        String fileEncodeName = docFile.getFileEncodeName();
        FileUtil.downLoadFile(response, fileEncodeName, filePath, fileName);
        return null;
    }
}