diff --git a/src/main/java/com/youlai/boot/platform/codegen/model/vo/CodegenPreviewVO.java b/src/main/java/com/youlai/boot/platform/codegen/model/vo/CodegenPreviewVO.java index f9435b14..18d1def3 100644 --- a/src/main/java/com/youlai/boot/platform/codegen/model/vo/CodegenPreviewVO.java +++ b/src/main/java/com/youlai/boot/platform/codegen/model/vo/CodegenPreviewVO.java @@ -16,4 +16,10 @@ public class CodegenPreviewVO { @Schema(description = "生成文件内容") private String content; + @Schema(description = "文件范围(frontend/backend)") + private String scope; + + @Schema(description = "文件语言(扩展名)") + private String language; + } diff --git a/src/main/java/com/youlai/boot/platform/codegen/service/impl/CodegenServiceImpl.java b/src/main/java/com/youlai/boot/platform/codegen/service/impl/CodegenServiceImpl.java index 6c2c6d44..0161033b 100644 --- a/src/main/java/com/youlai/boot/platform/codegen/service/impl/CodegenServiceImpl.java +++ b/src/main/java/com/youlai/boot/platform/codegen/service/impl/CodegenServiceImpl.java @@ -168,6 +168,8 @@ public class CodegenServiceImpl implements CodegenService { // 文件名 UserController.java String fileName = getFileName(entityName, templateName, extension); previewVo.setFileName(fileName); + previewVo.setScope(resolveScope(templateName)); + previewVo.setLanguage(resolveLanguage(fileName)); /* 2. 生成文件路径 */ // 包名:com.youlai.boot @@ -198,6 +200,17 @@ public class CodegenServiceImpl implements CodegenService { return list; } + private String resolveScope(String templateName) { + return switch (templateName) { + case "API", "API_TYPES", "VIEW" -> "frontend"; + default -> "backend"; + }; + } + + private String resolveLanguage(String fileName) { + return FileNameUtil.extName(fileName).toLowerCase(); + } + /** * 生成文件名。 *