105 lines
3.7 KiB
Java
105 lines
3.7 KiB
Java
package cn.bunny.service.impl;
|
||
|
||
import cn.bunny.core.vms.VmsCodeGeneratorService;
|
||
import cn.bunny.core.vms.VmsZipService;
|
||
import cn.bunny.domain.dto.VmsArgumentDto;
|
||
import cn.bunny.domain.vo.GeneratorVo;
|
||
import cn.bunny.domain.vo.VmsPathVo;
|
||
import cn.bunny.service.VmsService;
|
||
import cn.bunny.utils.ResourceFileUtil;
|
||
import lombok.RequiredArgsConstructor;
|
||
import org.springframework.http.HttpHeaders;
|
||
import org.springframework.http.HttpStatus;
|
||
import org.springframework.http.ResponseEntity;
|
||
import org.springframework.stereotype.Service;
|
||
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.UUID;
|
||
import java.util.stream.Collectors;
|
||
|
||
/**
|
||
* VMS服务主实现类，负责协调各子服务完成代码生成、资源管理和打包下载功能
|
||
*/
|
||
@Service
|
||
@RequiredArgsConstructor
|
||
public class VmsServiceImpl implements VmsService {
|
||
|
||
private final VmsCodeGeneratorService codeGeneratorService;
|
||
private final VmsZipService zipService;
|
||
|
||
@Override
|
||
public Map<String, List<GeneratorVo>> generator(VmsArgumentDto dto) {
|
||
return codeGeneratorService.generateCode(dto);
|
||
}
|
||
|
||
/**
|
||
* 获取VMS资源文件路径列表并按类型分组
|
||
*
|
||
* @return 按类型分组的VMS路径Map，key为类型，value为对应类型的VMS路径列表
|
||
* @throws RuntimeException 当获取资源路径失败时抛出
|
||
*/
|
||
public Map<String, List<VmsPathVo>> vmsResourcePathList() {
|
||
try {
|
||
// 1. 获取vms目录下所有相对路径文件列表
|
||
List<String> vmsRelativeFiles = ResourceFileUtil.getRelativeFiles("vms");
|
||
|
||
// 2. 处理文件路径并分组
|
||
return vmsRelativeFiles.stream()
|
||
.map(this::convertToVmsPathVo) // 转换为VO对象
|
||
.collect(Collectors.groupingBy(VmsPathVo::getType)); // 按类型分组
|
||
} catch (Exception e) {
|
||
throw new RuntimeException("Failed to get VMS resource paths: " + e.getMessage(), e);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 将文件路径字符串转换为VmsPathVo对象
|
||
*
|
||
* @param vmFile 文件相对路径字符串
|
||
* @return 转换后的VmsPathVo对象
|
||
*/
|
||
private VmsPathVo convertToVmsPathVo(String vmFile) {
|
||
// 分割文件路径
|
||
String[] filepathList = vmFile.split("/");
|
||
|
||
// 获取文件名（不含扩展名）
|
||
String filename = filepathList[filepathList.length - 1].replace(".vm", "");
|
||
|
||
/*
|
||
生成前端可用的唯一DOM元素ID
|
||
格式: "id-" + 无横线的UUID (例如: "id-550e8400e29b41d4a716446655440000")
|
||
|
||
用途:
|
||
1. 用于关联label标签和input元素的for属性
|
||
2. 确保列表项在前端有唯一标识
|
||
*/
|
||
String id = "id-" + UUID.randomUUID().toString().replace("-", "");
|
||
|
||
return VmsPathVo.builder()
|
||
.id(id)
|
||
.name(vmFile)
|
||
.label(filename)
|
||
.type(filepathList[0]) // 使用路径的第一部分作为类型
|
||
.build();
|
||
}
|
||
|
||
@Override
|
||
public ResponseEntity<byte[]> downloadByZip(VmsArgumentDto dto) {
|
||
// 创建ZIP文件
|
||
byte[] zipBytes = zipService.createZipFile(dto);
|
||
|
||
// 下载文件名称
|
||
String uuid = UUID.randomUUID().toString().split("-")[0];
|
||
String generateZipFilename = "code-" + uuid + ".zip";
|
||
|
||
// 设置响应头
|
||
HttpHeaders headers = new HttpHeaders();
|
||
headers.add("Content-Disposition", "attachment; filename=" + generateZipFilename);
|
||
headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
|
||
headers.add("Pragma", "no-cache");
|
||
headers.add("Expires", "0");
|
||
|
||
return new ResponseEntity<>(zipBytes, headers, HttpStatus.OK);
|
||
}
|
||
} |