HBX-3155: Gradle: Add support for template path

Author: fanste

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) {}
 

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");

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");

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");