diff --git a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/Extension.java b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/Extension.java index e9351d3abb..f5b2885dd1 100644 --- a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/Extension.java +++ b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/Extension.java @@ -29,6 +29,7 @@ public class Extension { public String revengFile = null; public Boolean generateAnnotations = true; public Boolean useGenerics = true; + public String templatePath = null; public Extension(Project project) {} diff --git a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateDaoTask.java b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateDaoTask.java index a81b71e34e..d75b6ba729 100644 --- a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateDaoTask.java +++ b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateDaoTask.java @@ -38,6 +38,11 @@ void doWork() { File outputFolder = getOutputFolder(); hbmExporter.getProperties().put(ExporterConstants.METADATA_DESCRIPTOR, createJdbcDescriptor()); hbmExporter.getProperties().put(ExporterConstants.DESTINATION_FOLDER, outputFolder); + String templatePath = getExtension().templatePath; + if (templatePath != null) { + getLogger().lifecycle("Setting template path to: " + templatePath); + hbmExporter.getProperties().put(ExporterConstants.TEMPLATE_PATH, new String[] { templatePath }); + } getLogger().lifecycle("Starting DAO export to directory: " + outputFolder + "..."); hbmExporter.start(); getLogger().lifecycle("DAO export finished"); diff --git a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateHbmTask.java b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateHbmTask.java index c9fc7b3a60..af6394c8e6 100644 --- a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateHbmTask.java +++ b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateHbmTask.java @@ -38,6 +38,11 @@ void doWork() { File outputFolder = getOutputFolder(); hbmExporter.getProperties().put(ExporterConstants.METADATA_DESCRIPTOR, createJdbcDescriptor()); hbmExporter.getProperties().put(ExporterConstants.DESTINATION_FOLDER, outputFolder); + String templatePath = getExtension().templatePath; + if (templatePath != null) { + getLogger().lifecycle("Setting template path to: " + templatePath); + hbmExporter.getProperties().put(ExporterConstants.TEMPLATE_PATH, new String[] { templatePath }); + } getLogger().lifecycle("Starting HBM export to directory: " + outputFolder + "..."); hbmExporter.start(); getLogger().lifecycle("HBM export finished"); diff --git a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateJavaTask.java b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateJavaTask.java index e86c7eceb8..3cbf6c498d 100644 --- a/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateJavaTask.java +++ b/gradle/plugin/src/main/java/org/hibernate/tool/gradle/task/GenerateJavaTask.java @@ -40,6 +40,11 @@ void doWork() { File outputFolder = getOutputFolder(); pojoExporter.getProperties().put(ExporterConstants.METADATA_DESCRIPTOR, createJdbcDescriptor()); pojoExporter.getProperties().put(ExporterConstants.DESTINATION_FOLDER, outputFolder); + String templatePath = getExtension().templatePath; + if (templatePath != null) { + getLogger().lifecycle("Setting template path to: " + templatePath); + pojoExporter.getProperties().put(ExporterConstants.TEMPLATE_PATH, new String[] { templatePath }); + } getLogger().lifecycle("Starting Java export to directory: " + outputFolder + "..."); pojoExporter.start(); getLogger().lifecycle("Java export finished");