From 034af5364523c49c2a48b49eb20b65cda18592d1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timo=20Westk=C3=A4mper?= <timo.westkamper@mysema.com>
Date: Wed, 30 Nov 2011 21:17:00 +0200
Subject: [PATCH] #59 : added more options

---
 .../maven/AbstractMetaDataExportMojo.java     | 34 ++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/querydsl-maven-plugin/src/main/java/com/mysema/query/maven/AbstractMetaDataExportMojo.java b/querydsl-maven-plugin/src/main/java/com/mysema/query/maven/AbstractMetaDataExportMojo.java
index 7fdc628ce..949de8ded 100644
--- a/querydsl-maven-plugin/src/main/java/com/mysema/query/maven/AbstractMetaDataExportMojo.java
+++ b/querydsl-maven-plugin/src/main/java/com/mysema/query/maven/AbstractMetaDataExportMojo.java
@@ -127,6 +127,16 @@ public class AbstractMetaDataExportMojo extends AbstractMojo{
      */
     private String namingStrategyClass;
 
+    /**
+     * @parameter
+     */
+    private String beanSerializerClass;
+    
+    /**
+     * @parameter
+     */
+    private String serializerClass;
+    
     /**
      * serialize beans as well
      *
@@ -159,6 +169,11 @@ public class AbstractMetaDataExportMojo extends AbstractMojo{
      * @parameter
      */
     private String[] customTypes;
+    
+    /**
+     * @parameter default-value=false
+     */
+    private boolean createScalaSources;
 
     @Override
     public void execute() throws MojoExecutionException, MojoFailureException {
@@ -194,6 +209,7 @@ public class AbstractMetaDataExportMojo extends AbstractMojo{
         if (beanSuffix != null) {
             exporter.setBeanSuffix(beanSuffix);
         }
+        exporter.setCreateScalaSources(createScalaSources);
         exporter.setPackageName(packageName);
         exporter.setBeanPackageName(beanPackageName);
         exporter.setInnerClassesForKeys(innerClassesForKeys);
@@ -203,7 +219,23 @@ public class AbstractMetaDataExportMojo extends AbstractMojo{
         exporter.setTableNamePattern(tableNamePattern);
         exporter.setColumnAnnotations(columnAnnotations);
         exporter.setValidationAnnotations(validationAnnotations);
+        if (serializerClass != null) {
+            try {
+                exporter.setSerializerClass((Class)Class.forName(serializerClass));
+            } catch (ClassNotFoundException e) {
+                getLog().error(e);
+                throw new MojoExecutionException(e.getMessage(), e);
+            }
+        }
         if (exportBeans) {
+            if (beanSerializerClass != null) {
+                try {
+                    exporter.setBeanSerializerClass((Class)Class.forName(beanSerializerClass));
+                } catch (ClassNotFoundException e) {
+                    getLog().error(e);
+                    throw new MojoExecutionException(e.getMessage(), e);
+                }
+            }
             exporter.setBeanSerializer(new BeanSerializer());
         }
         String sourceEncoding = (String)project.getProperties().get("project.build.sourceEncoding");
@@ -248,7 +280,7 @@ public class AbstractMetaDataExportMojo extends AbstractMojo{
             throw new MojoExecutionException(e.getMessage(), e);
         }
     }
-
+    
     protected boolean isForTest() {
         return false;
     }