diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/controller/GeneratorController.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/controller/GeneratorController.java index b2af5d4..b9c94d6 100644 --- a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/controller/GeneratorController.java +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/controller/GeneratorController.java @@ -1,6 +1,7 @@ package com.agileboot.codegen.controller; import com.agileboot.codegen.dto.CodegenRequest; +import com.agileboot.codegen.dto.TemplatePreviewRequest; import com.agileboot.codegen.service.IGeneratorService; import com.agileboot.codegen.service.ITemplateRepositoryService; import com.agileboot.codegen.service.ITemplateUsageLogService; @@ -103,4 +104,95 @@ public class GeneratorController { return R.fail("生成代码失败: " + e.getMessage()); } } + + @GetMapping("/template/preview") + public R previewTemplateGet(@RequestParam("templateSource") String templateSource, + @RequestParam("templateId") String templateId, + @RequestParam(value = "tableSql", required = false) String tableSql, + @RequestParam(value = "templateContent", required = false) String templateContent, + @RequestParam(value = "authorName", required = false) String authorName) { + try { + log.info("开始预览模板(GET)，模板来源: {}, 模板ID: {}", templateSource, templateId); + + // 参数验证 + if (templateSource == null || templateId == null) { + return R.fail("模板来源和模板ID不能为空"); + } + + // 如果没有提供tableSql，使用默认的示例SQL + if (tableSql == null || tableSql.trim().isEmpty()) { + tableSql = "CREATE TABLE `user` (\n" + + " `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '主键ID',\n" + + " `username` varchar(50) NOT NULL COMMENT '用户名',\n" + + " `password` varchar(100) NOT NULL COMMENT '密码',\n" + + " `email` varchar(100) DEFAULT NULL COMMENT '邮箱',\n" + + " `phone` varchar(20) DEFAULT NULL COMMENT '手机号',\n" + + " `status` tinyint(1) NOT NULL DEFAULT '1' COMMENT '状态(0-禁用 1-启用)',\n" + + " `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间',\n" + + " `update_time` datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',\n" + + " PRIMARY KEY (`id`),\n" + + " UNIQUE KEY `uk_username` (`username`)\n" + + ") ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='用户表';"; + log.info("使用默认示例SQL进行模板预览"); + } + + // 构建选项参数 + Map options = new HashMap<>(); + options.put("tableSql", tableSql); + if (authorName != null && !authorName.trim().isEmpty()) { + options.put("authorName", authorName); + } + + // 调用生成服务预览单个模板 + String previewResult = generatorService.previewSingleTemplate( + templateSource, + templateId, + templateContent, + options); + + log.info("模板预览完成(GET)，模板来源: {}, 模板ID: {}", templateSource, templateId); + return R.ok("预览成功", previewResult); + + } catch (Exception e) { + log.error("预览模板失败(GET)", e); + return R.fail("预览模板失败: " + e.getMessage()); + } + } + + @PostMapping("/template/preview") + public R previewTemplate(@RequestBody TemplatePreviewRequest request) { + try { + log.info("开始预览模板，模板来源: {}, 模板ID: {}", request.getTemplateSource(), request.getTemplateId()); + + // 参数验证 + if (request.getTemplateSource() == null || request.getTemplateId() == null) { + return R.fail("模板来源和模板ID不能为空"); + } + + if (request.getTableSql() == null || request.getTableSql().trim().isEmpty()) { + return R.fail("表SQL语句不能为空"); + } + + // 合并请求参数 + Map options = new HashMap<>(); + if (request.getOptions() != null) { + options.putAll(request.getOptions()); + } + options.put("tableSql", request.getTableSql()); + + // 调用生成服务预览单个模板 + String previewResult = generatorService.previewSingleTemplate( + request.getTemplateSource(), + request.getTemplateId(), + request.getTemplateContent(), + options); + + log.info("模板预览完成，模板来源: {}, 模板ID: {}", request.getTemplateSource(), request.getTemplateId()); + return R.ok("预览成功", previewResult); + + } catch (Exception e) { + log.error("预览模板失败", e); + return R.fail("预览模板失败: " + e.getMessage()); + } + } } diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/dto/TemplatePreviewRequest.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/dto/TemplatePreviewRequest.java new file mode 100644 index 0000000..319b4dc --- /dev/null +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/dto/TemplatePreviewRequest.java @@ -0,0 +1,40 @@ +package com.agileboot.codegen.dto; + +import lombok.Data; + +import java.util.Map; + +/** + * 模板预览请求DTO + * + * @author system + * @date 2025-09-27 + */ +@Data +public class TemplatePreviewRequest { + + /** + * 模板来源(system-系统模板, user-用户模板) + */ + private String templateSource; + + /** + * 模板ID + */ + private String templateId; + + /** + * 表SQL语句 + */ + private String tableSql; + + /** + * 生成选项 + */ + private Map options; + + /** + * 模板内容(当templateSource为content时使用) + */ + private String templateContent; +} diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/GeneratorServiceImpl.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/GeneratorServiceImpl.java index 21b6618..327903e 100644 --- a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/GeneratorServiceImpl.java +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/GeneratorServiceImpl.java @@ -141,4 +141,109 @@ public class GeneratorServiceImpl implements IGeneratorService { return "// 生成失败: " + e.getMessage(); } } + + @Override + public String previewSingleTemplate(String templateSource, String templateId, String templateContent, Map options) { + try { + // 1. 解析SQL获取表结构信息 + String tableSql = MapUtil.getString(options, "tableSql"); + String ignorePrefix = MapUtil.getString(options, "ignorePrefix", ""); + + if (StringUtils.isBlank(tableSql)) { + log.error("SQL语句为空"); + return "// 错误: SQL语句不能为空"; + } + + ClassInfo classInfo = TableParseUtil.parseTableSql(tableSql, ignorePrefix); + if (classInfo == null) { + log.error("解析SQL失败"); + return "// 错误: 解析SQL失败"; + } + + // 2. 设置生成参数 + options.put("classInfo", classInfo); + options.put("tableName", classInfo.getTableName()); + options.put("authorName", MapUtil.getString(options, "authorName", "Wol")); + + // 3. 根据模板来源生成代码 + String generatedCode; + + switch (templateSource.toLowerCase()) { + case "user": + // 用户自定义模板：从数据库获取模板内容 + UserTemplate userTemplate = userTemplateMapper.selectById(Long.valueOf(templateId)); + if (userTemplate != null && StringUtils.isNotBlank(userTemplate.getTemplateContent())) { + generatedCode = FreemarkerUtil.generateByContent(userTemplate.getTemplateContent(), options); + } else { + log.warn("用户模板不存在或内容为空: templateId={}", templateId); + generatedCode = "// 用户模板不存在或内容为空: " + templateId; + } + break; + + case "system": + // 系统模板：需要根据templateId查找对应的模板文件 + generatedCode = generateSystemTemplate(templateId, options); + break; + + case "content": + // 直接使用提供的模板内容 + if (StringUtils.isNotBlank(templateContent)) { + generatedCode = FreemarkerUtil.generateByContent(templateContent, options); + } else { + generatedCode = "// 错误: 模板内容为空"; + } + break; + + default: + generatedCode = "// 错误: 不支持的模板来源: " + templateSource; + break; + } + + return generatedCode; + + } catch (Exception e) { + log.error("预览模板失败: templateSource={}, templateId={}", templateSource, templateId, e); + return "// 预览失败: " + e.getMessage(); + } + } + + /** + * 生成系统模板代码 + */ + private String generateSystemTemplate(String templateId, Map options) { + try { + // 从系统模板配置中查找对应的模板信息 + String templateConfig = getTemplateConfig(); + JSONArray templateGroups = JSONArray.parseArray(templateConfig); + + for (Object groupObj : templateGroups) { + JSONObject group = (JSONObject) groupObj; + String groupName = group.getString("group"); + JSONArray templates = group.getJSONArray("templates"); + + for (Object templateObj : templates) { + JSONObject template = (JSONObject) templateObj; + String id = template.getString("id"); + String name = template.getString("name"); + + if (templateId.equals(id)) { + // 找到对应的模板，生成代码 + String templatePath = groupName + "/" + name + ".ftl"; + if (FreemarkerUtil.templateExists(templatePath)) { + return FreemarkerUtil.generateByTemplate(templatePath, options); + } else { + log.warn("系统模板文件不存在: {}", templatePath); + return "// 系统模板文件不存在: " + templatePath; + } + } + } + } + + return "// 系统模板不存在: " + templateId; + + } catch (Exception e) { + log.error("生成系统模板失败: templateId={}", templateId, e); + return "// 生成系统模板失败: " + e.getMessage(); + } + } } diff --git a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/IGeneratorService.java b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/IGeneratorService.java index 09c39f5..dbe2ec3 100644 --- a/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/IGeneratorService.java +++ b/agileboot-system/wol-codegenerator/src/main/java/com/agileboot/codegen/service/IGeneratorService.java @@ -25,4 +25,15 @@ public interface IGeneratorService { * @return 生成的代码Map */ Map getResultByParams(Map options, String templateConfig); + + /** + * 预览单个模板 + * + * @param templateSource 模板来源(system/user/content) + * @param templateId 模板ID + * @param templateContent 模板内容(当templateSource为content时使用) + * @param options 生成参数 + * @return 生成的代码内容 + */ + String previewSingleTemplate(String templateSource, String templateId, String templateContent, Map options); }