package org.jeecg.modules.doc.service.impl;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.common.util.MD5Util;
import org.jeecg.modules.doc.component.PathFileUtil;
import org.jeecg.modules.doc.component.UfopConfig;
import org.jeecg.modules.doc.constant.Constant;
import org.jeecg.modules.doc.dto.UploadFileDTO;
import org.jeecg.modules.doc.entity.DocFile;
import org.jeecg.modules.doc.entity.DocFilePath;
import org.jeecg.modules.doc.entity.LbDocEntity;
import org.jeecg.modules.doc.mapper.LbDocMapper;
import org.jeecg.modules.doc.service.IDocFilePathService;
import org.jeecg.modules.doc.service.IDocFileService;
import org.jeecg.modules.doc.service.ILbDocService;
import org.jeecg.modules.doc.util.MD5;
import org.jeecg.modules.doc.util.UFOPUtils;
import org.jeecg.modules.doc.vo.CompareFileVo;
import org.jeecg.modules.doc.vo.DocTree;
import org.jeecg.modules.doc.vo.QiwenFile;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * 测试Service
 */
@Service
public class LbDocServiceImpl extends ServiceImpl<LbDocMapper, LbDocEntity> implements ILbDocService {


    @Override
    public String hello() {
        return "hello jeecg!";
    }

    @Autowired
    private UfopConfig ufopConfig;

    @Autowired
    private IDocFilePathService filePathService;

    @Autowired
    private IDocFileService docFileService;

    @Autowired
    private RestTemplate restTemplate;


    @Override
    public DocTree genDocTree(DocTree docTree) {

        int level = Integer.valueOf(docTree.getKey().substring(0, docTree.getKey().indexOf("-")));
        level++;
        File file = new File(docTree.getPath());
        if (!file.exists()) {
            System.err.println("文件不存在！！");
        }

        docTree.setTitle(file.getName());
        File[] files = file.listFiles();

        if (files.length != 0) {
            List<DocTree> docs = new ArrayList<>();
            DocTree dt = null;
            for (int i = 0; i < files.length; i++) {
                File f = files[i];
                dt = new DocTree();
                dt.setKey(level + "-" + i);
                dt.setTitle(f.getName());
                dt.setPath(f.getPath());
                if (f.isDirectory()) {

                    // genDocTree(f.getAbsolutePath(), level,dt);
                } else {

                    dt.setLeaf(true);

                }
                docs.add(dt);
            }
            docTree.setChildren(docs);
        }

        return docTree;
    }



    public void initDocSize(Map<String, Long> map, String releasePath, DocFilePath parent) {
       // System.out.println("文件：" + fsize +" 文件夹：" + dsize);
        File file = new File(releasePath);
        if (!file.exists()) {
            System.err.println("文件不存在！！");
        }
        File[] files = file.listFiles();
        DocFilePath filePath = null;
        if (files != null && files.length != 0) {
            for (int i = 0; i < files.length; i++) {
                File f = files[i];
                if (!"#recycle".equals(f.getName()) && !"$RECYCLE.BIN".equals(f.getName()) && !"System Volume Information".equals(f.getName())) {
                    if (f.isDirectory()) {
                        map.put("dsize", map.get("dsize")+1);
                        initDocSize(map,f.getPath(), filePath);
                    } else {
                        map.put("fsize", map.get("fsize")+1);
                    }
                }

            }
        }
    }


    public void initDir(Map<String, String> paths, List<DocFilePath> filePathList, String releasePath, DocFilePath parent) {
        File file = new File(releasePath);
        if (!file.exists()) {
            System.err.println("文件不存在！！");
        }
        File[] files = file.listFiles();
        DocFilePath filePath = null;

//        System.out.println("当前文件夹：" + file.getPath());
        if (files != null && files.length != 0) {
            for (int i = 0; i < files.length; i++) {
                File f = files[i];
             //   System.out.println(f.getPath());
              //  System.out.println("r\\/"+f.getPath().replace("\\","/"));
               // System.out.println("ddd" + f.getPath().replace("\\","/").replace(ufopConfig.getReleasePath(),"/07 开发三部"));
                String filepath = f.getPath().replace("\\","/").replace(ufopConfig.getReleasePath(),"");
                //String filepath = QiwenFile.separator+QiwenFile.formatPath(f.getPath().replace("\\","/")).replace(ufopConfig.getReleasePath(),"/07 开发三部");
                filepath = QiwenFile.separator+(filepath.substring(0,filepath.lastIndexOf("/")));
                if (!"#recycle".equals(f.getName()) && !"$RECYCLE.BIN".equals(f.getName()) && !"System Volume Information".equals(f.getName())) {
                    if (f.isDirectory()) {
                        //System.out.println("文件：" + f.getName());
                        filePath = PathFileUtil.getDir(filepath,f.getName(), parent);
                        filePath.setIsDir(Constant.IS_DIR);
                        String s = paths.get(filePath.getFilePath() + filePath.getFileName() + (filePath.getExtendName()==null?"":filePath.getExtendName()) + filePath.getIsDir());

                        if (s == null) {
                            filePathList.add(filePath);
                        } else {
                            filePath.setPathId(s);
                        }

                        initDir(paths, filePathList,f.getPath(), filePath);
//                        if (s != null) {
//                            continue;
//                        }
                    } else {
                        continue;

                    }


                }
            }

        }

    }

    @Override
    public void initDoc(Map<String, String> paths, String releasePath, DocFilePath parent) {

        File file = new File(releasePath);
        if (!file.exists()) {
            System.err.println("文件不存在！！");
        }
        File[] files = file.listFiles();
        DocFilePath filePath = null;
        DocFile docFile = null;
        List<DocFilePath> filePathList = new ArrayList<>();
        if (files != null && files.length != 0) {
            for (int i = 0; i < files.length; i++) {
                File f = files[i];
                String filepath = QiwenFile.separator+QiwenFile.formatPath(f.getPath().replace("\\","/")).replaceFirst(ufopConfig.getReleasePath(),"");
                filepath = QiwenFile.separator+(filepath.substring(0,filepath.lastIndexOf("/")));

                if (!"#recycle".equals(f.getName()) && !"$RECYCLE.BIN".equals(f.getName())
                        && !"System Volume Information".equals(f.getName())
                        // && !"档案资料区".equals(f.getName())
                         && !"02 研发一部".equals(f.getName())
                        && !"03 研发二部".equals(f.getName())
                        && !"04 研发三部".equals(f.getName())
                                               && !"05 开发一部".equals(f.getName())
                          && !"06 开发二部".equals(f.getName())
                         && !"07 开发三部".equals(f.getName())
                        && !"08 工业设计部".equals(f.getName())
                         && !"09 研发管理部".equals(f.getName())
                         && !"10 兰浦公司".equals(f.getName())
                        && !"01 公共区域".equals(f.getName())
                ) {

                    if (f.isDirectory()) {
                        filePath = PathFileUtil.getDir(filepath,f.getName(), parent);
                        filePath.setIsDir(Constant.IS_DIR);
                        initDoc(paths, f.getPath(), filePath);
                    } else {

                        int lastIndex = f.getName().lastIndexOf(".");

                        filePath = PathFileUtil.getDir(filepath,f.getName().substring(0,lastIndex>0?lastIndex:0), parent);
                        //System.out.println("同步文件:" + filePath.getFileName());
                        filePath.setIsDir(Constant.IS_NOT_DIR);
                        filePath.setExtendName(f.getName().substring(f.getName().lastIndexOf(".") + 1));


                        String key = filePath.getFilePath().replace("/192.168.0.9/研发中心", "") + filePath.getFileName() + (filePath.getExtendName() == null ? "" : filePath.getExtendName()) + filePath.getIsDir();
                        String s = paths.get(key);


                        if (s != null) {
                            //System.out.println("该文件已存在");
                            docFile = new DocFile();
                            continue;
//                            docFile.setFileId(s);
                        } else {
                            long length = f.length();
                            if (length > 1000000000l) {
                                log.error("小文件：[" + f.getPath() +"]>>>["+f.length()+ "]");
                                continue;
                            }
                            // String dfId;
                            String md5 = MD5.getMD5(f);

                            // 查询是否已上传过
                            List<DocFile> fileList = docFileService.list(new LambdaQueryWrapper<DocFile>().eq(DocFile::getIdentifier,md5));
                            if (fileList!=null && fileList.size()>0) {
                                docFile = new DocFile();
                                // dfId = fileList.get(0).getFileId();
                                docFile.setFileId(fileList.get(0).getFileId());
                            } else {
                            //    String url = UFOPUtils.ENCRYPT_PATH + "/fileTransfer/encryptUpload";
                                try {

                                    // 1、封装请求头
//                                    HttpHeaders headers = new HttpHeaders();
//                                    MediaType type = MediaType.parseMediaType("multipart/form-data");
//                                    headers.setContentType(type);
//                                    headers.setContentLength(length);
//                                    headers.setContentDispositionFormData("media", f.getName());
//                                    // 2、封装请求体
//                                    MultiValueMap<String, Object> param = new LinkedMultiValueMap<>();
//                                    FileSystemResource resource = new FileSystemResource(f);
//                                    param.add("file", resource);
//                                    param.add("identifier", md5);
//                                    // 3、封装整个请求报文
//                                    HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<>(param, headers);
//                                    // 4、发送请求
//                                    ResponseEntity<DocFile> data = restTemplate.postForEntity(url, formEntity, DocFile.class);
//                                    // 5、请求结果处理
//                                    docFile = data.getBody();
                                    docFile = docFileService.saveFile(f, md5);

                                } catch (Exception e) {
                                    log.error("异常文件：PATH=" + f.getPath() + "[md5]=" + md5);
                                    log.error(e.getMessage());
                                    //e.printStackTrace();
                                    continue;
                                }
                            }
                        }
                        filePath.setFileId(docFile.getFileId());
                    }
                    String s = paths.get(filePath.getFilePath().replace("/192.168.0.9/研发中心","") + filePath.getFileName() + (filePath.getExtendName()==null?"":filePath.getExtendName()) + filePath.getIsDir());
                    if (s != null) {
                        continue;
                    }
                    //filePath.setPathId(s);
//                    try {
//                        filePathService.saveOrUpdate(filePath);
//                    }catch (Exception e) {
//                        log.error("获取相同文件：：："+filePath.getFilePath() + filePath.getFileName() + filePath.getExtendName() + filePath.getIsDir());
//                        log.error("保存文件失败：" + filePath.getFilePath() + ">>" + filePath.getFileName() + ">>>" + filePath.getExtendName());
//                        log.error(e.getMessage());
//                    }
                    filePathList.add(filePath);
                }

            }
            System.out.println("当前文件夹：" + file.getPath());
            System.out.println("保存" + filePathList.size() +"个文件:" + System.currentTimeMillis());
            try {
                filePathService.saveOrUpdateBatch(filePathList);
            } catch (Exception e) {
                log.error("保存失败");
                log.error(e.getMessage());
            }

            System.out.println("保存完成:" +System.currentTimeMillis());
        }
    }





    private static void copyFileUsingStream(File source, File dest) throws IOException {
        InputStream is = null;
        OutputStream os = null;
        try {
            is = new FileInputStream(source);
            os = new FileOutputStream(dest);
            byte[] buffer = new byte[1024];
            int length;
            while ((length = is.read(buffer)) > 0) {
                os.write(buffer, 0, length);
            }
        } finally {
            is.close();
            os.close();
        }

    }


    @Override
    public CompareFileVo compareFile(UploadFileDTO uploadFileDto)  {
        CompareFileVo compareFileVo = new CompareFileVo();
        compareFileVo.setFileName(uploadFileDto.getRelativePath());
        compareFileVo.setIdentifier(uploadFileDto.getIdentifier());
        String relativePath = uploadFileDto.getRelativePath().substring(uploadFileDto.getRelativePath().indexOf("/"));
        String filePath = (uploadFileDto.getFilePath() + relativePath).replace("/" + uploadFileDto.getFilename(), "");
        int lastIndex = uploadFileDto.getFilename().lastIndexOf(".");
        String fileName = uploadFileDto.getFilename().substring(0, lastIndex > 0 ? lastIndex : 0);
        String extendName = uploadFileDto.getFilename().substring(lastIndex + 1);
        List<DocFilePath> list = filePathService.list(new LambdaQueryWrapper<DocFilePath>()

                .eq(DocFilePath::getFilePath, filePath)
                .eq(DocFilePath::getFileName, fileName)
                        .eq(DocFilePath::getExtendName, extendName)
                        .eq(DocFilePath::getDeleteFlag, 0)
                .eq(DocFilePath::getIsDir, 0)
        );
        if (list != null && list.size() == 1) {
            DocFile byId = docFileService.getById(list.get(0).getFileId());
            File file = new File(UFOPUtils.getStaticPath() + byId.getFileUrl());
            String md5 = MD5.getMD5(file);
            compareFileVo.setComIdentifier(md5);
            compareFileVo.setComFileSize(file.length());
        }
        return compareFileVo;

    }

    @Override
    public void checkFile(String filePath) {
            File file = new File(filePath);
            if (!file.exists()) {
                System.err.println("文件不存在！！");
            }

            File[] files = file.listFiles();
            if (files != null && files.length != 0) {
                for (int i = 0; i < files.length; i++) {
                    File f = files[i];




                        if (f.isDirectory()) {
                            System.out.println(f.getPath());

                                checkFile(f.getPath());

                        } else {
                            try {
//                                long l = f.lastModified();
//                                String dat = "2023-10-23";
//                                Date parse = new SimpleDateFormat("yyyy-MM-dd").parse(dat);
//                                Date date = new Date(l);
//                                if (date.after(parse)) {
//                                    continue;
//                                }
//                                String md5 = MD5.getMD5(f);
//                                String fileName = "";
//                                int lastIndex = f.getName().lastIndexOf(".");
//                                if (lastIndex <0) {
////                                fileName = f.getName().substring(0,f.getName().length()-1);
//                                    fileName = f.getName().replace("\uF029","");
//                                } else {
//                                    fileName = f.getName().substring(0, lastIndex);
//                                }
//                                if (!fileName.equals(md5)) {
//
//                                    System.out.println("原MD5:" + fileName);
//                                    System.out.println("现MD5:" + md5);
//                                    docFileService.update(new LambdaUpdateWrapper<DocFile>().set(DocFile::getFileStatus, 0).eq(DocFile::getIdentifier, fileName));
//                                    f.delete();
//                                }

                            }catch (Exception e) {
                                System.out.println("文件名：【" + f.getName()+"】");
                                e.printStackTrace();
                            }


                        }
                        System.out.println("完成！");
                    }
                }
    }
}