4dkankan-fusion/src/main/java/com/fdkankan/fusion/down/CaseDownService.java

package com.fdkankan.fusion.down;

import cn.hutool.core.io.FileUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.deepoove.poi.XWPFTemplate;
import com.fdkankan.fusion.common.FilePath;
import com.fdkankan.fusion.common.ResultData;
import com.fdkankan.fusion.common.util.SceneTypeUtil;
import com.fdkankan.fusion.common.util.ShellUtil;
import com.fdkankan.fusion.common.util.UploadToOssUtil;
import com.fdkankan.fusion.entity.*;
import com.fdkankan.fusion.httpClient.LaserService;
import com.fdkankan.fusion.httpClient.response.FdkkResponse;
import com.fdkankan.fusion.request.CaseParam;
import com.fdkankan.fusion.response.DownVo;
import com.fdkankan.fusion.response.DownloadProcessVo;
import com.fdkankan.fusion.response.FusionNumVo;
import com.fdkankan.fusion.response.SceneVo;
import com.fdkankan.fusion.service.*;
import com.fdkankan.fusion.service.impl.DownService;
import com.fdkankan.redis.util.RedisUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.FileOutputStream;
import java.nio.charset.StandardCharsets;
import java.util.HashSet;
import java.util.List;

@Service
@Slf4j
public class CaseDownService {

    @Value("${server.servlet.context-path}")
    String basePath;

    public static String jsonDataName = "data.json";
    public static String caseInfo =     "/case/getInfo?caseId=";
    public static String caseSettingsInfo =     "/caseSettings/info?caseId=";
    public static String hostIcon =     "/edit/hotIcon/list?caseId=";
    public static String caseView =     "/caseView/allList?caseId=";
    public static String caseFiles =      "/caseFiles/allList?caseId=";
    public static String caseFilesType =      "/caseFilesType/allList?caseId=";
    public static String caseScene =      "/case/sceneList?caseId=";
    public static String caseFusion =      "/caseFusion/list?caseId=";
    public static String caseVideoFolder =      "/caseVideoFolder/allList?caseId=";
    public static String caseVideo =      "/caseVideo/allList?folderId=";
    public static String caseTag =      "/caseTag/allList?caseId=";
    public static String caseTagPoint =      "/caseTagPoint/allList?tagId=";
    public static String caseInquest =      "/caseInquest/info?caseId=";
    public static String caseExtractDetail =      "/caseExtractDetail/info?caseId=";
    public static String fusionGuide =      "/fusionGuide/allList?caseId=";
    public static String fusionGuidePath =     "/fusionGuidePath/allList?guideId=";
    public static String fusionMeter =      "/fusionMeter/allList?fusionId=";

    public static String model =      "/model/getInfo?modelId=";
    public static String caseImg =      "/caseImg/getFfmpegImage?caseId=";

    public static String inquestInfoUrl =      "/caseInquestInfo/info?caseId=";
    public static String casePathInfoUrl =      "/casePath/info?caseId=";

    public static String laserData = "/laser/dataset/%s/getDataSet";
    public static String laserDataQuery = "/laser/filter/%s/query?datasetId=%s";


    @Autowired
    ICaseService caseService;
    @Autowired
    ICaseSettingsService caseSettingsService;
    @Autowired
    IFusionNumService fusionNumService;
    @Autowired
    ICaseViewService caseViewService;
    @Autowired
    ICaseVideoFolderService caseVideoFolderService;
    @Autowired
    ICaseVideoService caseVideoService;
    @Autowired
    ICaseFilesService caseFilesService;
    @Autowired
    ICaseFilesTypeService caseFilesTypeService;
    @Autowired
    IHotIconService hotIconService;
    @Autowired
    ICaseTagService caseTagService;
    @Autowired
    ICaseTagPointService caseTagPointService;
    @Autowired
    IFusionGuideService fusionGuideService;
    @Autowired
    IFusionGuidePathService fusionGuidePathService;
    @Autowired
    ICaseInquestService caseInquestService;
    @Autowired
    ICaseExtractDetailService caseExtractDetailService;
    @Autowired
    IFusionMeterService fusionMeterService;
    @Autowired
    IModelService modelService;
    @Autowired
    LaserService laserService;
    @Autowired
    ICaseOfflineService caseOfflineService;
    @Autowired
    ICaseImgService caseImgService;
    @Autowired
    ICaseInquestInfoService caseInquestInfoService;
    @Autowired
    ICasePathService casePathService;

    @Autowired
    RedisUtil redisUtil;

    @Value("${spring.profiles.active}")
    private String environment;
    public static String downProcessKey = "fusion:down:offline:process:caseId:%s";

    public DownVo checkDown(Integer caseId) {
        DownVo downVo = new DownVo();
//        CaseOffline byCaseId = caseOfflineService.getByCaseId(caseId);
//        if(byCaseId != null){
//            downVo.setDownloadStatus(3);
//            downVo.setDownloadUrl(byCaseId.getOfflineUrl());
//        }
        return downVo;
    }
    @Async
    public void downOffline(Integer caseId){
        try {
            String redisKey = String.format(downProcessKey, caseId);
            if( redisUtil.hasKey(redisKey)){
                String res = redisUtil.get(redisKey);
                DownloadProcessVo downloadProcessVo = JSONObject.parseObject(res, DownloadProcessVo.class);
                if(downloadProcessVo.getStatus() != 1003 && (downloadProcessVo.getPercent()== null || downloadProcessVo.getPercent() != 100)){
                    return;
                }
            }

            setRedisProcess(caseId,0);
            //复制前端资源
            cpIndexHtml(caseId);
            setRedisProcess(caseId,10);
            //创建data.json并下载资源
            createDataJson(caseId);
            setRedisProcess(caseId,50);
            //打包zip
            String path = "/mnt/fusion/offline";
            String name = "offline_"+ caseId;
            String zipName = path+File.separator+name+".zip";
            ShellUtil.zipOffline(zipName,name);
            setRedisProcess(caseId,70);
            //上传oss
            String ossUrl = zipName.replace("/mnt/", "");
            ShellUtil.yunUpload(zipName,ossUrl);
            Thread.sleep(2000L);
            FileUtil.del(zipName);
            ossUrl = queryPath + ossUrl;
            caseOfflineService.saveByCase(caseId,ossUrl);
            setRedisProcess(caseId,100,ossUrl);
        }catch (Exception e){
            log.info("down-offline-error:{}",caseId,e);
            setRedisProcess(caseId,0,null,1003);
        }

    }

    public void setRedisProcess(Integer caseId,Integer num){
        setRedisProcess(caseId,num,null,1000);
    }

    public void setRedisProcess(Integer caseId,Integer num,String url){
       setRedisProcess(caseId,num,url,1000);
    }

    public void setRedisProcess(Integer caseId,Integer num,String url,Integer status){
        String redisKey = String.format(downProcessKey, caseId);
        log.info("down-offline-process:{},{},{}",caseId,num,url);
        DownloadProcessVo processVo = new DownloadProcessVo();
        processVo.setStatus(status);
        processVo.setPercent(num);
        processVo.setUrl( url);
        redisUtil.set(redisKey,JSONObject.toJSONString(processVo));
    }

    public DownloadProcessVo process(Integer caseId) {
        DownloadProcessVo downVo = new DownloadProcessVo();
        String redisKey = String.format(downProcessKey, caseId);
        if(redisUtil.hasKey(redisKey)){
            return JSONObject.parseObject(redisUtil.get(redisKey),DownloadProcessVo.class);
        }
        return downVo;
    }

    public void createDataJson(Integer caseId){
        log.info("down-offline-createDataJson:{}",caseId);
        JSONObject jsonObject = new JSONObject();
        CaseParam param = new CaseParam();
        param.setCaseId(caseId);

        //设置案件信息
        jsonObject.put(basePath+inquestInfoUrl+caseId, ResultData.ok(caseInquestInfoService.getByCaseId(caseId)));
        jsonObject.put(basePath+caseInfo+caseId, ResultData.ok(caseService.getInfo(caseId)));

        List<CaseSettings> caseSettings = caseSettingsService.getByCaseId(caseId);
        jsonObject.put(basePath+caseSettingsInfo+caseId, ResultData.ok(caseSettings));
        for (CaseSettings caseSetting : caseSettings) {
            downResource(caseId,caseSetting.getBack());
            downResource(caseId,caseSetting.getCover());
        }

        List<FusionNumVo> listByCaseId = fusionNumService.getListByCaseId(caseId,null);
        jsonObject.put(basePath+caseFusion+caseId, ResultData.ok(listByCaseId));
        for (FusionNumVo fusion : listByCaseId) {
            jsonObject.put(basePath+fusionMeter+fusion.getFusionId(), ResultData.ok(fusionMeterService.getListByFusionId(fusion.getFusionId(),null)));
        }
        for (FusionNumVo fusionNumVo : listByCaseId) {
            //下载媒体库模型
            if(fusionNumVo.getSceneData() != null){
                if(fusionNumVo.getSceneData().getType() ==3 && StringUtils.isNotBlank(fusionNumVo.getSceneData().getModelGlbUrl()) ){
                    downModel(caseId,fusionNumVo.getSceneData().getModelGlbUrl());
                }
            }
        }

        List<SceneVo> sceneVos = caseService.sceneList(param);
        for (SceneVo sceneData : sceneVos) {
            //下载模型
            if(StringUtils.isNotBlank(sceneData.getModelGlbUrl())){
                downModel(caseId,sceneData.getModelGlbUrl());
            }
            if(sceneData.getType() != 3){
                //下载场景离线包
                downSwkk(caseId,sceneData.getNum(),sceneData.getType());
            }
            if(SceneTypeUtil.isLaser(sceneData.getType())){
                FdkkResponse sceneInfo = laserService.getSceneInfo(sceneData.getNum());
                HashSet<String> dataSetIds = new HashSet<>();
                if(sceneInfo != null){
                    JSONArray jsonArray = JSONArray.parseArray(JSONArray.toJSONString(sceneInfo.getData()));
                    JSONArray newJsonArray = new JSONArray();
                    for (Object object : jsonArray) {
                        JSONObject sceneInfoObj = (JSONObject) object;
                        String newPath = String.format(FilePath.OFFLINE_LASER_OSS_PATH, sceneData.getNum(), sceneData.getNum());
                        String oldPath = sceneInfoObj.getString("webBin");
                        sceneInfoObj.put("oldWebBin",oldPath);
                        sceneInfoObj.put("webBin",newPath + oldPath);
                        newJsonArray.add(sceneInfoObj);
                        dataSetIds.add( sceneInfoObj.getString("id"));
                    }
                    sceneInfo.setData(newJsonArray);
                    jsonObject.put(String.format(laserData,sceneData.getNum()),sceneInfo);
                    if(!dataSetIds.isEmpty()){
                        for (String dataSetId : dataSetIds) {
                            FdkkResponse sceneInfo2 = laserService.getSceneInfoQuery(sceneData.getNum(),dataSetId);
                            jsonObject.put(String.format(laserDataQuery,sceneData.getNum(),dataSetId),sceneInfo2);
                        }
                    }

                }

            }
            if(sceneData.getModelId() != null){
                jsonObject.put(basePath+model+sceneData.getModelId(), ResultData.ok(modelService.getInfo(sceneData.getModelId())));
            }
        }
        jsonObject.put(basePath+caseScene+caseId, ResultData.ok(sceneVos));

        List<CaseView> caseViews = caseViewService.allList(caseId, null, null, null, null);
        jsonObject.put(basePath+caseView+caseId, ResultData.ok(caseViews));
        for (CaseView view : caseViews) {
            downResource(caseId,view.getViewImg());
            downResource(caseId,view.getViewImgSmall());
        }

        List<CaseVideoFolder> videoFolders = caseVideoFolderService.getAllList(caseId);
        jsonObject.put(basePath+caseVideoFolder+caseId, ResultData.ok(videoFolders));
        for (CaseVideoFolder videoFolder : videoFolders) {
            downResource(caseId,videoFolder.getVideoFolderCover());
            downResource(caseId,videoFolder.getVideoMergeUrl());

            List<CaseVideo> allList = caseVideoService.getAllList(videoFolder.getVideoFolderId());
            for (CaseVideo video : allList) {
                downResource(caseId,video.getVideoCover());
                downResource(caseId,video.getVideoPath());
            }
            jsonObject.put(basePath+caseVideo+videoFolder.getVideoFolderId(), ResultData.ok(allList));
        }

        List<CaseFiles> caseFilesList = caseFilesService.allList(caseId, null);
        for (CaseFiles files : caseFilesList) {
            downResource(caseId,files.getFilesUrl());
        }
        jsonObject.put(basePath+caseFiles+caseId, ResultData.ok(caseFilesList));


        jsonObject.put(basePath+caseFilesType+caseId, ResultData.ok(caseFilesTypeService.list()));

        List<HotIcon> hotIconList = hotIconService.getListByCaseId(caseId);
        for (HotIcon hotIcon : hotIconList) {
            downResource(caseId,hotIcon.getIconUrl());
        }
        jsonObject.put(basePath+hostIcon+caseId, ResultData.ok(hotIconList));

        List<CaseTag> caseTagList = caseTagService.allList(caseId, null);
        jsonObject.put(basePath+caseTag+caseId, ResultData.ok(caseTagList));
        for (CaseTag tag : caseTagList) {
            jsonObject.put(basePath+caseTagPoint+tag.getTagId(), ResultData.ok(caseTagPointService.allList(tag.getTagId())));
            downResources(caseId,tag.getTagImgUrl());
            downResource(caseId,tag.getHotIconUrl());
        }

        jsonObject.put(basePath+casePathInfoUrl+caseId, ResultData.ok(casePathService.getByCaseId(caseId)));
        List<FusionGuide> fusionGuides = fusionGuideService.getAllList(caseId);
        jsonObject.put(basePath+fusionGuide+caseId, ResultData.ok(fusionGuides));

        for (FusionGuide guide : fusionGuides) {
            downResource(caseId,guide.getCover());
            List<FusionGuidePath> listByGuideId = fusionGuidePathService.getListByGuideId(guide.getFusionGuideId());
            for (FusionGuidePath guidePath : listByGuideId) {
                downResource(caseId,guidePath.getCover());
            }
            jsonObject.put(basePath+fusionGuidePath+guide.getFusionGuideId(), ResultData.ok(listByGuideId));
        }
        CaseInquest caseInquest1 = caseInquestService.getByCaseId(caseId);
        if(caseInquest1 != null){
            XWPFTemplate inquestTemp = caseInquestService.getWordByTemplate(caseInquest1);
            downWordByTemplate(caseId,inquestTemp,"caseInquest.doc");
        }
        jsonObject.put(basePath+caseInquest+caseId, ResultData.ok(caseInquest1));

        CaseExtractDetail caseExtractDetail1 = caseExtractDetailService.getByCaseId(caseId);
        if(caseExtractDetail1 != null){
            XWPFTemplate detailTemp = caseExtractDetailService.getWordByTemplate(caseExtractDetail1);
            downWordByTemplate(caseId,detailTemp,"caseExtractDetail.doc");
        }
        jsonObject.put(basePath+caseExtractDetail+caseId, ResultData.ok(caseExtractDetail1));

        List<CaseImg> caseImgList = caseImgService.getByCaseId(caseId, 1);
        for (CaseImg img : caseImgList) {
            downResource(img.getCaseId(),img.getImgUrl());
        }
        jsonObject.put(basePath+caseImg+caseId, ResultData.ok(caseImgList));


        FileUtil.writeString(JSON.toJSONString(jsonObject), FilePath.OFFLINE_PACKAGE_PATH+caseId+"/www/package/"+jsonDataName,"UTF-8");

    }




    //http://127.0.0.1:8080/offline.html?caseId=362&app=1&share=1#/show/summary

    static String batName = "start-browser.bat";
   public void cpIndexHtml(Integer caseId){
       log.info("down-offline-cpIndexHtml:{}",caseId);

       String caseOfflinePath = FilePath.OFFLINE_PACKAGE_PATH+caseId;
       FileUtil.copyContent(new File(FilePath.OFFLINE_TEMPLATE_PATH),new File(caseOfflinePath),true);

       String s = FileUtil.readString(caseOfflinePath + File.separator + batName, StandardCharsets.UTF_8);
       String s1 = s.replaceAll("@caseId", String.valueOf(caseId));
       FileUtil.writeString(s1, caseOfflinePath + File.separator + batName,"UTF-8");
   }


   @Autowired
    DownService downService;
   @Autowired
    UploadToOssUtil uploadToOssUtil;

   public void downSwkk(Integer caseId,String num,Integer type){

       String swkkPath = FilePath.OFFLINE_PACKAGE_PATH+caseId + "/www/swkk/"+num;
       String swkkZipPath = swkkPath +".zip";
       String swssPath = FilePath.OFFLINE_PACKAGE_PATH+caseId + "/www/swss/"+num;
       String swssZipPath = swssPath +".zip";

       Integer isObj = 0;
       if(type == 4 || type == 6){
           isObj =1;
       }
       try {

           DownVo downVo = downService.checkDownLoad(num, isObj,"offline");
           log.info("down:{}",downVo);
           if(downVo.getDownloadStatus() == 3 && StringUtils.isNotBlank(downVo.getDownloadUrl())){
               downZip(type,downVo.getDownloadUrl(),swkkZipPath,swkkPath,swssZipPath,swssPath);
           }else {
               DownVo down = downService.down(num, isObj,"offline");
               if(down.getDownloadStatus() == 1){
                   DownloadProcessVo downloadProcessVo = downService.downloadProcess(num, isObj);
                   while (downloadProcessVo.getStatus() != 1002 ){
                       downloadProcessVo =  downService.downloadProcess(num, isObj);
                       Thread.sleep(2000L);
                   }
                   downZip(type,downloadProcessVo.getUrl(),swkkZipPath,swkkPath,swssZipPath,swssPath);
               }
           }
       }catch (Exception e){
           log.info("下载场景离线包失败：{}",num,e);
       }

   }
    @Value("${upload.query-path}")
    private String queryPath;
   public void downZip(Integer type ,String uri,String kkzipPath,String kknumPath,String sszipPath,String ssNumPath){
       try {
           if(type != 2 && type != 5){ //深时点云
               if(uri.contains("?")){
                   uri = uri.split("[?]")[0];
               }
               ShellUtil.yunDownload(uri.replace(queryPath, ""), kkzipPath);

               ShellUtil.unZip(kkzipPath,kknumPath);
               FileUtil.del(kkzipPath);
           }else {
//               InputStream in = minIoUploadService.getObject(uri);
//               FileOutputStream out = new FileOutputStream(sszipPath);
//               byte[] buffer = new byte[1024];
//               int bytesRead;
//               while ((bytesRead = in.read(buffer)) != -1) {
//                   out.write(buffer, 0, bytesRead);
//               }
//               out.flush();
//               out.close();
//               in.close();
               ShellUtil.yunDownloadSs(uri.replace(queryPath, ""), sszipPath);
               ShellUtil.unZip(sszipPath,ssNumPath);
               FileUtil.del(sszipPath);
           }
       }catch (Exception e){
            log.info("下载场景离线包失败：{}",uri,e);
       }

   }
    public void downModel(Integer caseId, String modelGlbUrl) {
        String path = String.format(FilePath.OFFLINE_OSS_PATH,caseId);

        JSONArray jsonArray = JSONArray.parseArray(modelGlbUrl);
        for (Object object : jsonArray) {
            String res = (String) object;
            log.info("下载模型：{}",res);
            res = res.replace(queryPath, "");
            if(res.contains(".json") || res.contains(".shp")){
                res = new File(res).getParentFile().getPath();
            }
            ShellUtil.yunDownload(res, path +queryPath + res);
        }
    }
    public void downResources(Integer caseId,String urls) {

        JSONArray jsonArray = JSONArray.parseArray(urls);
        for (Object object : jsonArray) {
            String res = (String) object;
            downResource(caseId,res);
        }
    }
    public void downResources(Integer caseId,JSONArray urls) {
        for (Object object : urls) {
            String res = (String) object;
            downResource(caseId,res);
        }
    }

    public void downResource(Integer caseId,String url) {
        if(StringUtils.isBlank(url) ){
            return;
        }
        url = url.replace(queryPath, "");
        if( !uploadToOssUtil.existKey(url)){
            log.info("downResource文件不存在：{},{}",caseId,url);
            return;
        }
        String path = String.format(FilePath.OFFLINE_OSS_PATH,caseId);
        ShellUtil.yunDownload(url,path+queryPath+url);
    }

    public void downWordByTemplate(Integer caseId,XWPFTemplate template,String name){
        // 指定输出文件的路径
        String outputPath = String.format(FilePath.OFFLINE_RESOURCE_PACKAGE_PATH,caseId) + name;
        try {
            FileOutputStream out = new FileOutputStream(outputPath);
            template.write(out);
            out.close();
            log.info("文档已成功写入到: " + outputPath);
        } catch (Exception e) {
           log.info("写出文档失败：{},{}",caseId,name,e);
        }
    }



}