~nishiwei/ard-uav-server.git

			@@ -2,40 +2,74 @@

			import cn.hutool.core.bean.BeanUtil;
			import cn.hutool.core.collection.CollectionUtil;
			import cn.hutool.core.io.FileUtil;
			import com.ard.oss.service.impl.OssServiceContext;
			import com.ard.route.constant.FileTypeConstants;
			import com.ard.route.domain.RoutePointReq;
			import com.ard.route.domain.UavRouteReq;
			import com.ard.route.domain.WaypointHeadingReq;
			import com.ard.route.domain.WaypointTurnReq;
			import com.ard.route.domain.kml.*;
			import com.ard.route.domain.djuav.RoutePointReq;
			import com.ard.route.domain.djuav.UavRouteReq;
			import com.ard.route.domain.djuav.WaypointHeadingReq;
			import com.ard.route.domain.djuav.WaypointTurnReq;
			import com.ard.route.domain.djuav.kml.*;
			import com.ard.route.domain.route.ArdRouteInfo;
			import com.ard.route.domain.vo.KmzInfoVo;
			import com.ard.route.domain.vo.KmzResultVo;
			import com.ard.route.enums.kml.ExitOnRCLostEnums;
			import com.ard.route.service.IArdRouteInfoService;
			import com.ard.route.util.InitializationUtils;
			import com.ard.route.util.RouteFileUtils;
			import com.ard.route.service.UavRouteService;
			import com.thoughtworks.xstream.XStream;
			import com.thoughtworks.xstream.io.xml.DomDriver;
			import org.apache.commons.compress.archivers.ArchiveEntry;
			import org.apache.commons.compress.archivers.ArchiveInputStream;
			import org.apache.commons.compress.archivers.zip.ZipArchiveInputStream;
			import org.apache.commons.lang3.StringUtils;
			import org.springframework.beans.BeanUtils;
			import org.springframework.mock.web.MockMultipartFile;
			import org.springframework.stereotype.Service;
			import org.springframework.web.multipart.MultipartFile;

			import java.io.File;
			import javax.annotation.Resource;
			import java.io.*;
			import java.util.ArrayList;
			import java.util.List;
			import java.util.Objects;
			import java.util.zip.ZipOutputStream;

			import static com.ard.route.util.RouteFileUtils.*;
			import static com.ard.route.util.ToolUtils.*;

			/**
			* Author:songjian
			* Author:xia
			* Date: 2024/12/22 10:36
			**/
			@Service
			public class UavRouteServiceImpl implements UavRouteService {

			@Resource
			private OssServiceContext ossService;

			@Resource
			private IArdRouteInfoService ardRouteInformationService;

			InitializationUtils init = new InitializationUtils();

			/**
			* 编辑kmz文件
			*/
			@Override
			public void updateKmz(UavRouteReq uavRouteReq) {
			// TODO 替换本地文件路径！！！
			File file = FileUtil.file("/Users/songjian/Project/IdeaProjects/dj-uav/file/kmz/航线kmz文件.kmz");
			try (ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(FileUtil.getInputStream(file))) {
			public KmzResultVo updateKmz(UavRouteReq uavRouteReq,String fileName){
			KmzResultVo result = new KmzResultVo();
			// 校验修改的文件是否存在
			ArdRouteInfo routeInformation = ardRouteInformationService.selectArdRouteInfoByName(fileName);
			if ( routeInformation == null ){
			result.setData("修改航线文件失败，请检查文件是否存在");
			return result;
			}

			// 获取文件
			String minioUrlName = subStringMinioUrlName(routeInformation.getPath());
			try (InputStream inputStream = ossService.getObject(kmzPath,minioUrlName);
			ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){
			ArchiveEntry entry;
			KmlInfo kmlInfo = new KmlInfo();
			KmlInfo wpmlInfo = new KmlInfo();
			@@ -51,9 +85,13 @@
			this.handleRouteUpdate(wpmlInfo, uavRouteReq, FileTypeConstants.WPML, kmlParams);
			}
			}
			RouteFileUtils.buildKmz("更新航线kmz文件", kmlInfo, wpmlInfo);
			// 删除文件
			ossService.deleteObject(kmzPath, minioUrlName);
			return buildKmzMethod(routeInformation.getName(), kmlInfo, wpmlInfo);
			} catch (Exception e) {
			e.printStackTrace();
			result.setData("更新航线kmz文件失败");
			return result;
			}
			}

			@@ -98,11 +136,107 @@
			}
			}

			/**
			* 生成kmz文件(带航点)
			*/
			@Override
			public void buildKmz(UavRouteReq uavRouteReq) {
			public KmzResultVo buildKmz(UavRouteReq uavRouteReq, String kmzName) {
			// 赋值默认参数
			init.initialization(uavRouteReq);
			KmlParams kmlParams = new KmlParams();
			BeanUtils.copyProperties(uavRouteReq, kmlParams);
			kmlParams.setRoutePointList(BeanUtil.copyToList(uavRouteReq.getRoutePointList(), RoutePointInfo.class));
			RouteFileUtils.buildKmz("航线kmz文件AAAAAA", kmlParams);
			return buildKmz(kmzName, kmlParams);
			}

			/**
			* 生成航线 KMZ 文件
			*
			* @param fileName 文件名
			* @param kmlParams 参数对象
			* @return 本地文件路径
			*/
			public KmzResultVo buildKmz(String fileName, KmlParams kmlParams) {
			KmlInfo kmlInfo = buildKml(kmlParams);
			KmlInfo wpmlInfo = buildWpml(kmlParams);
			return buildKmzMethod(fileName, kmlInfo, wpmlInfo);
			}

			/**
			* 生成航线 KMZ 文件
			*
			* @param fileName 文件名
			* @param kmlInfo kml 文件信息
			* @param wpmlInfo wpml 文件信息
			* @return 本地文件路径
			*/
			public KmzResultVo buildKmzMethod(String fileName, KmlInfo kmlInfo, KmlInfo wpmlInfo) {
			XStream xStream = new XStream(new DomDriver());
			xStream.processAnnotations(KmlInfo.class);
			xStream.addImplicitCollection(KmlActionGroup.class, "action");
			xStream.addImplicitCollection(KmlPlacemark.class, "actionGroup");

			String kml = XML_HEADER + xStream.toXML(kmlInfo);
			String wpml = XML_HEADER + xStream.toXML(wpmlInfo);
			KmzResultVo result = new KmzResultVo();
			try (ByteArrayOutputStream baos = new ByteArrayOutputStream();
			ZipOutputStream zipOutputStream = new ZipOutputStream(baos)) {
			zipOutputStream.setLevel(0);

			// 创建 wpmz 目录中的 template.kml 文件条目
			buildZipFile("wpmz/template.kml", zipOutputStream, kml);
			// 创建 wpmz 目录中的 waylines.wpml 文件条目
			buildZipFile("wpmz/waylines.wpml", zipOutputStream, wpml);

			zipOutputStream.finish();
			zipOutputStream.flush();

			// 创建MultipartFile对象
			MultipartFile multipartFile = new MockMultipartFile(
			fileName + ".kmz",
			fileName + ".kmz",
			"application/vnd.google-earth.kmz",
			new ByteArrayInputStream(baos.toByteArray())
			);
			// 上传文件
			String kmzFilePath = ossService.upload(kmzPath,multipartFile);
			result.setPath(kmzFilePath);
			return result;
			} catch (Exception e) {
			throw new RuntimeException("KMZ文件生成或上传失败", e);
			}
			}

			/**
			* 解析kmz文件
			*/
			@Override
			public KmzInfoVo parseKmz(String fileUrl){
			KmzInfoVo kmzInfoVO = new KmzInfoVo();
			try (InputStream inputStream = ossService.getObject(kmzPath,fileUrl);
			ArchiveInputStream archiveInputStream = new ZipArchiveInputStream(inputStream)){
			ArchiveEntry entry;
			while (!Objects.isNull(entry = archiveInputStream.getNextEntry())) {
			String name = entry.getName();
			if (name.toLowerCase().endsWith(".kml")) {
			kmzInfoVO.setKmlInfo(RouteFileUtils.parseKml(archiveInputStream));
			} else if (name.toLowerCase().endsWith(".wpml")) {
			kmzInfoVO.setWpmlInfo(RouteFileUtils.parseKml(archiveInputStream));
			}
			}
			return kmzInfoVO;
			}catch (Exception e) {
			e.printStackTrace();
			}
			return null;
			}

			/**
			* KmzInfoVo实体类转换为UavRouteReq实体类
			*/
			@Override
			public UavRouteReq kmzInfoVoToUavRouteReq(KmzInfoVo kmzInfoVo) {
			return kmzInfoVoToUavRouteReq(kmzInfoVo);
			}

			}