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 7bb5e96e69..09ea0f4d11 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 @@ -12,6 +12,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 74edc40101..b81dadb36d 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 @@ -21,6 +21,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 495a4cefe1..e6968b436a 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 @@ -21,6 +21,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 75b15884aa..35bef2f786 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 @@ -23,6 +23,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");