e99aea4a0a135b111e60ef77427e3910482c8e3c
[hibernate4-maven-plugin] / src / main / java / de / juplo / plugins / hibernate / AbstractSchemaMojo.java
1 package de.juplo.plugins.hibernate;
2
3
4 import com.pyx4j.log4j.MavenLogAppender;
5 import java.io.File;
6 import java.io.FileInputStream;
7 import java.io.FileOutputStream;
8 import java.io.IOException;
9 import java.io.InputStream;
10 import java.net.MalformedURLException;
11 import java.net.URL;
12 import java.security.NoSuchAlgorithmException;
13 import java.time.ZonedDateTime;
14 import java.util.Collections;
15 import java.util.EnumSet;
16 import java.util.HashMap;
17 import java.util.HashSet;
18 import java.util.Iterator;
19 import java.util.LinkedHashSet;
20 import java.util.List;
21 import java.util.Map;
22 import java.util.Map.Entry;
23 import java.util.Properties;
24 import java.util.Set;
25 import java.util.regex.Matcher;
26 import java.util.regex.Pattern;
27 import javax.persistence.Embeddable;
28 import javax.persistence.Entity;
29 import javax.persistence.MappedSuperclass;
30 import javax.persistence.spi.PersistenceUnitTransactionType;
31 import org.apache.maven.artifact.Artifact;
32 import org.apache.maven.model.Resource;
33 import org.apache.maven.plugin.AbstractMojo;
34 import org.apache.maven.plugin.MojoExecutionException;
35 import org.apache.maven.plugin.MojoFailureException;
36 import org.apache.maven.project.MavenProject;
37 import org.hibernate.boot.MetadataBuilder;
38 import org.hibernate.boot.MetadataSources;
39 import org.hibernate.boot.cfgxml.internal.ConfigLoader;
40 import org.hibernate.boot.cfgxml.spi.LoadedConfig;
41 import org.hibernate.boot.cfgxml.spi.MappingReference;
42 import org.hibernate.boot.model.naming.ImplicitNamingStrategy;
43 import org.hibernate.boot.model.naming.PhysicalNamingStrategy;
44 import org.hibernate.boot.registry.BootstrapServiceRegistry;
45 import org.hibernate.boot.registry.BootstrapServiceRegistryBuilder;
46 import org.hibernate.boot.registry.StandardServiceRegistry;
47 import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
48 import org.hibernate.boot.registry.classloading.spi.ClassLoaderService;
49 import org.hibernate.boot.registry.classloading.spi.ClassLoadingException;
50 import org.hibernate.boot.registry.selector.spi.StrategySelector;
51 import org.hibernate.boot.spi.MetadataImplementor;
52 import org.hibernate.cfg.AvailableSettings;
53 import static org.hibernate.cfg.AvailableSettings.DIALECT;
54 import static org.hibernate.cfg.AvailableSettings.DRIVER;
55 import static org.hibernate.cfg.AvailableSettings.FORMAT_SQL;
56 import static org.hibernate.cfg.AvailableSettings.HBM2DDL_DELIMITER;
57 import static org.hibernate.cfg.AvailableSettings.HBM2DLL_CREATE_NAMESPACES;
58 import static org.hibernate.cfg.AvailableSettings.IMPLICIT_NAMING_STRATEGY;
59 import static org.hibernate.cfg.AvailableSettings.JPA_JDBC_DRIVER;
60 import static org.hibernate.cfg.AvailableSettings.JPA_JDBC_PASSWORD;
61 import static org.hibernate.cfg.AvailableSettings.JPA_JDBC_URL;
62 import static org.hibernate.cfg.AvailableSettings.JPA_JDBC_USER;
63 import static org.hibernate.cfg.AvailableSettings.PASS;
64 import static org.hibernate.cfg.AvailableSettings.PHYSICAL_NAMING_STRATEGY;
65 import static org.hibernate.cfg.AvailableSettings.SHOW_SQL;
66 import static org.hibernate.cfg.AvailableSettings.USER;
67 import static org.hibernate.cfg.AvailableSettings.URL;
68 import org.hibernate.engine.config.spi.ConfigurationService;
69 import org.hibernate.engine.jdbc.connections.spi.ConnectionProvider;
70 import org.hibernate.internal.util.config.ConfigurationException;
71 import org.hibernate.jpa.boot.internal.ParsedPersistenceXmlDescriptor;
72 import org.hibernate.jpa.boot.internal.PersistenceXmlParser;
73 import org.hibernate.tool.schema.TargetType;
74 import org.hibernate.tool.schema.internal.ExceptionHandlerCollectingImpl;
75 import org.hibernate.tool.schema.internal.exec.ScriptTargetOutputToFile;
76 import org.hibernate.tool.schema.spi.ExecutionOptions;
77 import org.hibernate.tool.schema.spi.SchemaManagementToolCoordinator;
78 import org.hibernate.tool.schema.spi.ScriptTargetOutput;
79 import org.hibernate.tool.schema.spi.TargetDescriptor;
80 import org.scannotation.AnnotationDB;
81
82
83 /**
84  * Baseclass with common attributes and methods.
85  *
86  * @phase process-classes
87  * @threadSafe
88  * @requiresDependencyResolution runtime
89  */
90 public abstract class AbstractSchemaMojo extends AbstractMojo
91 {
92   public final static String EXECUTE = "hibernate.schema.execute";
93   public final static String OUTPUTDIRECTORY = "project.build.outputDirectory";
94   public final static String SCAN_CLASSES = "hibernate.schema.scan.classes";
95   public final static String SCAN_DEPENDENCIES = "hibernate.schema.scan.dependencies";
96   public final static String SCAN_TESTCLASSES = "hibernate.schema.scan.test_classes";
97   public final static String TEST_OUTPUTDIRECTORY = "project.build.testOutputDirectory";
98   public final static String SKIPPED = "hibernate.schema.skipped";
99   public final static String SCRIPT = "hibernate.schema.script";
100
101   private final static Pattern SPLIT = Pattern.compile("[^,\\s]+");
102
103   private final Set<String> packages = new HashSet<String>();
104
105
106   /**
107    * The maven project.
108    * <p>
109    * Only needed internally.
110    *
111    * @parameter property="project"
112    * @required
113    * @readonly
114    */
115   private MavenProject project;
116
117   /**
118    * Build-directory.
119    * <p>
120    * Only needed internally.
121    *
122    * @parameter property="project.build.directory"
123    * @required
124    * @readonly
125    */
126   private String buildDirectory;
127
128
129   /** Parameters to configure the genaration of the SQL *********************/
130
131   /**
132    * Excecute the generated SQL.
133    * If set to <code>false</code>, only the SQL-script is created and the
134    * database is not touched.
135    * <p>
136    * <strong>Important:</strong>
137    * This configuration value can only be configured through the
138    * <code>pom.xml</code>, or by the definition of a system-property, because
139    * it is not known by Hibernate nor JPA and, hence, not picked up from
140    * their configuration!
141    *
142    * @parameter property="hibernate.schema.execute" default-value="true"
143    * @since 2.0
144    */
145   private Boolean execute;
146
147   /**
148    * Skip execution
149    * <p>
150    * If set to <code>true</code>, the execution is skipped.
151    * <p>
152    * A skipped execution is signaled via the maven-property
153    * <code>${hibernate.schema.skipped}</code>.
154    * <p>
155    * The execution is skipped automatically, if no modified or newly added
156    * annotated classes are found and the dialect was not changed.
157    * <p>
158    * <strong>Important:</strong>
159    * This configuration value can only be configured through the
160    * <code>pom.xml</code>, or by the definition of a system-property, because
161    * it is not known by Hibernate nor JPA and, hence, not picked up from
162    * their configuration!
163    *
164    * @parameter property="hibernate.schema.skip" default-value="${maven.test.skip}"
165    * @since 1.0
166    */
167   private boolean skip;
168
169   /**
170    * Force generation/execution
171    * <p>
172    * Force the generation and (if configured) the execution of the SQL, even if
173    * no modified or newly added annotated classes where found and the
174    * configuration was not changed.
175    * <p>
176    * <code>skip</code> takes precedence over <code>force</code>.
177    * <p>
178    * <strong>Important:</strong>
179    * This configuration value can only be configured through the
180    * <code>pom.xml</code>, or by the definition of a system-property, because
181    * it is not known by Hibernate nor JPA and, hence, not picked up from
182    * their configuration!
183    *
184    * @parameter property="hibernate.schema.force" default-value="false"
185    * @since 1.0
186    */
187   private boolean force;
188
189   /**
190    * Hibernate dialect.
191    *
192    * @parameter property="hibernate.dialect"
193    * @since 1.0
194    */
195   private String dialect;
196
197   /**
198    * Delimiter in output-file.
199    * <p>
200    * <strong>Important:</strong>
201    * This configuration value can only be configured through the
202    * <code>pom.xml</code>, or by the definition of a system-property, because
203    * it is not known by Hibernate nor JPA and, hence, not picked up from
204    * their configuration!
205    *
206    * @parameter property="hibernate.hbm2ddl.delimiter" default-value=";"
207    * @since 1.0
208    */
209   private String delimiter;
210
211   /**
212    * Show the generated SQL in the command-line output.
213    *
214    * @parameter property="hibernate.show_sql"
215    * @since 1.0
216    */
217   private Boolean show;
218
219   /**
220    * Format output-file.
221    *
222    * @parameter property="hibernate.format_sql"
223    * @since 1.0
224    */
225   private Boolean format;
226
227   /**
228    * Specifies whether to automatically create also the database schema/catalog.
229    *
230    * @parameter property="hibernate.hbm2dll.create_namespaces" default-value="false"
231    * @since 2.0
232    */
233   private Boolean createNamespaces;
234
235   /**
236    * Implicit naming strategy
237    *
238    * @parameter property="hibernate.implicit_naming_strategy"
239    * @since 2.0
240    */
241   private String implicitNamingStrategy;
242
243   /**
244    * Physical naming strategy
245    *
246    * @parameter property="hibernate.physical_naming_strategy"
247    * @since 2.0
248    */
249   private String physicalNamingStrategy;
250
251   /**
252    * Wether the project should be scanned for annotated-classes, or not
253    * <p>
254    * This parameter is intended to allow overwriting of the parameter
255    * <code>exclude-unlisted-classes</code> of a <code>persistence-unit</code>.
256    * If not specified, it defaults to <code>true</code>
257    *
258    * @parameter property="hibernate.schema.scan.classes"
259    * @since 2.0
260    */
261   private Boolean scanClasses;
262
263   /**
264    * Classes-Directory to scan.
265    * <p>
266    * This parameter defaults to the maven build-output-directory for classes.
267    * Additionally, all dependencies are scanned for annotated classes.
268    * <p>
269    * <strong>Important:</strong>
270    * This configuration value can only be configured through the
271    * <code>pom.xml</code>, or by the definition of a system-property, because
272    * it is not known by Hibernate nor JPA and, hence, not picked up from
273    * their configuration!
274    *
275    * @parameter property="project.build.outputDirectory"
276    * @since 1.0
277    */
278   private String outputDirectory;
279
280   /**
281    * Dependency-Scopes, that should be scanned for annotated classes.
282    * <p>
283    * By default, only dependencies in the scope <code>compile</code> are
284    * scanned for annotated classes. Multiple scopes can be seperated by
285    * white space or commas.
286    * <p>
287    * If you do not want any dependencies to be scanned for annotated
288    * classes, set this parameter to <code>none</code>.
289    * <p>
290    * The plugin does not scan for annotated classes in transitive
291    * dependencies. If some of your annotated classes are hidden in a
292    * transitive dependency, you can simply add that dependency explicitly.
293    *
294    * @parameter property="hibernate.schema.scan.dependencies" default-value="compile"
295    * @since 1.0.3
296    */
297   private String scanDependencies;
298
299   /**
300    * Whether to scan the test-branch of the project for annotated classes, or
301    * not.
302    * <p>
303    * If this parameter is set to <code>true</code> the test-classes of the
304    * artifact will be scanned for hibernate-annotated classes additionally.
305    * <p>
306    * <strong>Important:</strong>
307    * This configuration value can only be configured through the
308    * <code>pom.xml</code>, or by the definition of a system-property, because
309    * it is not known by Hibernate nor JPA and, hence, not picked up from
310    * their configuration!
311    *
312    * @parameter property="hibernate.schema.scan.test_classes" default-value="false"
313    * @since 1.0.1
314    */
315   private Boolean scanTestClasses;
316
317   /**
318    * Test-Classes-Directory to scan.
319    * <p>
320    * This parameter defaults to the maven build-output-directory for
321    * test-classes.
322    * <p>
323    * This parameter is only used, when <code>scanTestClasses</code> is set
324    * to <code>true</code>!
325    * <p>
326    * <strong>Important:</strong>
327    * This configuration value can only be configured through the
328    * <code>pom.xml</code>, or by the definition of a system-property, because
329    * it is not known by Hibernate nor JPA and, hence, not picked up from
330    * their configuration!
331    *
332    * @parameter property="project.build.testOutputDirectory"
333    * @since 1.0.2
334    */
335   private String testOutputDirectory;
336
337
338   /** Conection parameters *************************************************/
339
340   /**
341    * SQL-Driver name.
342    *
343    * @parameter property="hibernate.connection.driver_class"
344    * @since 1.0
345    */
346   private String driver;
347
348   /**
349    * Database URL.
350    *
351    * @parameter property="hibernate.connection.url"
352    * @since 1.0
353    */
354   private String url;
355
356   /**
357    * Database username
358    *
359    * @parameter property="hibernate.connection.username"
360    * @since 1.0
361    */
362   private String username;
363
364   /**
365    * Database password
366    *
367    * @parameter property="hibernate.connection.password"
368    * @since 1.0
369    */
370   private String password;
371
372
373   /** Parameters to locate configuration sources ****************************/
374
375   /**
376    * Path to a file or name of a ressource with hibernate properties.
377    * If this parameter is specified, the plugin will try to load configuration
378    * values from a file with the given path or a ressource on the classpath with
379    * the given name. If both fails, the execution of the plugin will fail.
380    * <p>
381    * If this parameter is not set the plugin will load configuration values
382    * from a ressource named <code>hibernate.properties</code> on the classpath,
383    * if it is present, but will not fail if there is no such ressource.
384    * <p>
385    * During ressource-lookup, the test-classpath takes precedence.
386    *
387    * @parameter
388    * @since 1.0
389    */
390   private String hibernateProperties;
391
392   /**
393    * Path to Hibernate configuration file (.cfg.xml).
394    * If this parameter is specified, the plugin will try to load configuration
395    * values from a file with the given path or a ressource on the classpath with
396    * the given name. If both fails, the execution of the plugin will fail.
397    * <p>
398    * If this parameter is not set the plugin will load configuration values
399    * from a ressource named <code>hibernate.cfg.xml</code> on the classpath,
400    * if it is present, but will not fail if there is no such ressource.
401    * <p>
402    * During ressource-lookup, the test-classpath takes precedence.
403    * <p>
404    * Settings in this file will overwrite settings in the properties file.
405    *
406    * @parameter
407    * @since 1.1.0
408    */
409   private String hibernateConfig;
410
411   /**
412    * Name of the persistence-unit.
413    * If this parameter is specified, the plugin will try to load configuration
414    * values from a persistence-unit with the specified name. If no such
415    * persistence-unit can be found, the plugin will throw an exception.
416    * <p>
417    * If this parameter is not set and there is only one persistence-unit
418    * available, that unit will be used automatically. But if this parameter is
419    * not set and there are multiple persistence-units available on,
420    * the class-path, the execution of the plugin will fail.
421    * <p>
422    * Settings in this file will overwrite settings in the properties or the
423    * configuration file.
424    *
425    * @parameter
426    * @since 1.1.0
427    */
428   private String persistenceUnit;
429
430   /**
431    * List of Hibernate-Mapping-Files (XML).
432    * Multiple files can be separated with white-spaces and/or commas.
433    *
434    * @parameter property="hibernate.mapping"
435    * @since 1.0.2
436    */
437   private String mappings;
438
439
440
441   public final void execute(String filename)
442     throws
443       MojoFailureException,
444       MojoExecutionException
445   {
446     if (skip)
447     {
448       getLog().info("Execution of hibernate-maven-plugin was skipped!");
449       project.getProperties().setProperty(SKIPPED, "true");
450       return;
451     }
452
453     ModificationTracker tracker;
454     try
455     {
456       tracker = new ModificationTracker(buildDirectory, filename, getLog());
457     }
458     catch (NoSuchAlgorithmException e)
459     {
460       throw new MojoFailureException("Digest-Algorithm MD5 is missing!", e);
461     }
462
463     final SimpleConnectionProvider connectionProvider =
464         new SimpleConnectionProvider(getLog());
465
466     try
467     {
468       /** Start extended logging */
469       MavenLogAppender.startPluginLog(this);
470
471       /** Load checksums for old mapping and configuration */
472       tracker.load();
473
474       /** Create the ClassLoader */
475       MutableClassLoader classLoader = createClassLoader();
476
477       /** Create a BootstrapServiceRegistry with the created ClassLoader */
478       BootstrapServiceRegistry bootstrapServiceRegitry =
479           new BootstrapServiceRegistryBuilder()
480               .applyClassLoader(classLoader)
481               .build();
482       ClassLoaderService classLoaderService =
483           bootstrapServiceRegitry.getService(ClassLoaderService.class);
484
485       Properties properties = new Properties();
486       ConfigLoader configLoader = new ConfigLoader(bootstrapServiceRegitry);
487
488       /** Loading and merging configuration */
489       properties.putAll(loadProperties(configLoader));
490       LoadedConfig config = loadConfig(configLoader);
491       if (config != null)
492         properties.putAll(config.getConfigurationValues());
493       ParsedPersistenceXmlDescriptor unit =
494           loadPersistenceUnit(classLoaderService, properties);
495       if (unit != null)
496         properties.putAll(unit.getProperties());
497
498       /** Overwriting/Completing configuration */
499       configure(properties, tracker);
500
501       /** Check configuration for modifications */
502       if(tracker.track(properties))
503         getLog().debug("Configuration has changed.");
504       else
505         getLog().debug("Configuration unchanged.");
506
507       /** Check, that the outputfile is writable */
508       final File output = getOutputFile(filename);
509       /** Check, if the outputfile is missing or was changed */
510       checkOutputFile(output, tracker);
511
512       /** Configure Hibernate */
513       final StandardServiceRegistry serviceRegistry =
514           new StandardServiceRegistryBuilder(bootstrapServiceRegitry)
515               .applySettings(properties)
516               .addService(ConnectionProvider.class, connectionProvider)
517               .build();
518       final MetadataSources sources = new MetadataSources(serviceRegistry);
519
520       /** Add the remaining class-path-elements */
521       completeClassPath(classLoader);
522
523       /** Apply mappings from hibernate-configuration, if present */
524       if (config != null)
525       {
526         for (MappingReference mapping : config.getMappingReferences())
527           mapping.apply(sources);
528       }
529
530       Set<String> classes;
531       if (unit == null)
532       {
533         /** No persistent unit: default behaviour */
534         if (scanClasses == null)
535           scanClasses = true;
536         Set<URL> urls = new HashSet<URL>();
537         if (scanClasses)
538           addRoot(urls, outputDirectory);
539         if (scanTestClasses)
540           addRoot(urls, testOutputDirectory);
541         addDependencies(urls);
542         classes = scanUrls(urls);
543       }
544       else
545       {
546         /** Follow configuration in persisten unit */
547         if (scanClasses == null)
548           scanClasses = !unit.isExcludeUnlistedClasses();
549         Set<URL> urls = new HashSet<URL>();
550         if (scanClasses)
551         {
552           /**
553            * Scan the root of the persiten unit and configured jars for
554            * annotated classes
555            */
556           urls.add(unit.getPersistenceUnitRootUrl());
557           for (URL url : unit.getJarFileUrls())
558             urls.add(url);
559         }
560         if (scanTestClasses)
561           addRoot(urls, testOutputDirectory);
562         classes = scanUrls(urls);
563         for (String className : unit.getManagedClassNames())
564           classes.add(className);
565         /**
566          * Add mappings from the default mapping-file
567          * <code>META-INF/orm.xml</code>, if present
568          */
569         boolean error = false;
570         InputStream is;
571         is = classLoader.getResourceAsStream("META-INF/orm.xml");
572         if (is != null)
573         {
574           getLog().info("Adding default JPA-XML-mapping from META-INF/orm.xml");
575           try
576           {
577             tracker.track("META-INF/orm.xml", is);
578             sources.addResource("META-INF/orm.xml");
579           }
580           catch (IOException e)
581           {
582             getLog().error("cannot read META-INF/orm.xml: " + e);
583             error = true;
584           }
585         }
586         else
587         {
588           getLog().debug("no META-INF/orm.xml found");
589         }
590         /**
591          * Add mappings from files, that are explicitly configured in the
592          * persistence unit
593          */
594         for (String mapping : unit.getMappingFileNames())
595         {
596           getLog().info("Adding explicitly configured mapping from " + mapping);
597           is = classLoader.getResourceAsStream(mapping);
598           if (is != null)
599           {
600             try
601             {
602               tracker.track(mapping, is);
603               sources.addResource(mapping);
604             }
605             catch (IOException e)
606             {
607               getLog().info("cannot read mapping-file " + mapping + ": " + e);
608               error = true;
609             }
610           }
611           else
612           {
613             getLog().error("cannot find mapping-file " + mapping);
614             error = true;
615           }
616         }
617         if (error)
618           throw new MojoFailureException(
619               "error, while reading mappings configured in persistence-unit \"" +
620               unit.getName() +
621               "\""
622               );
623       }
624
625       /** Add the configured/collected annotated classes */
626       for (String className : classes)
627         addAnnotated(className, sources, classLoaderService, tracker);
628
629       /** Add explicitly configured classes */
630       addMappings(sources, tracker);
631
632       /** Skip execution, if mapping and configuration is unchanged */
633       if (!tracker.modified())
634       {
635         getLog().info("Mapping and configuration unchanged.");
636         if (force)
637           getLog().info("Generation/execution is forced!");
638         else
639         {
640           getLog().info("Skipping schema generation!");
641           project.getProperties().setProperty(SKIPPED, "true");
642           return;
643         }
644       }
645
646
647       /** Truncate output file */
648       try
649       {
650         new FileOutputStream(output).getChannel().truncate(0).close();
651       }
652       catch (IOException e)
653       {
654         String error =
655             "Error while truncating " + output.getAbsolutePath() + ": "
656             + e.getMessage();
657         getLog().warn(error);
658         throw new MojoExecutionException(error);
659       }
660
661       /** Create a connection, if sufficient configuration infromation is available */
662       connectionProvider.open(classLoaderService, properties);
663
664       MetadataBuilder metadataBuilder = sources.getMetadataBuilder();
665
666       StrategySelector strategySelector =
667           serviceRegistry.getService(StrategySelector.class);
668
669       if (properties.containsKey(IMPLICIT_NAMING_STRATEGY))
670       {
671         metadataBuilder.applyImplicitNamingStrategy(
672             strategySelector.resolveStrategy(
673                 ImplicitNamingStrategy.class,
674                 properties.getProperty(IMPLICIT_NAMING_STRATEGY)
675                 )
676             );
677       }
678
679       if (properties.containsKey(PHYSICAL_NAMING_STRATEGY))
680       {
681         metadataBuilder.applyPhysicalNamingStrategy(
682             strategySelector.resolveStrategy(
683                 PhysicalNamingStrategy.class,
684                 properties.getProperty(PHYSICAL_NAMING_STRATEGY)
685                 )
686             );
687       }
688
689       /** Prepare the generation of the SQL */
690       Map settings = new HashMap();
691       settings.putAll(
692           serviceRegistry
693               .getService(ConfigurationService.class)
694               .getSettings()
695               );
696       ExceptionHandlerCollectingImpl handler =
697           new ExceptionHandlerCollectingImpl();
698       ExecutionOptions options =
699           SchemaManagementToolCoordinator
700               .buildExecutionOptions(settings, handler);
701       final EnumSet<TargetType> targetTypes = EnumSet.of(TargetType.SCRIPT);
702       if (execute)
703         targetTypes.add(TargetType.DATABASE);
704       TargetDescriptor target = new TargetDescriptor()
705       {
706         @Override
707         public EnumSet<TargetType> getTargetTypes()
708         {
709           return targetTypes;
710         }
711
712         @Override
713         public ScriptTargetOutput getScriptTargetOutput()
714         {
715           String charset =
716               (String)
717               serviceRegistry
718                   .getService(ConfigurationService.class)
719                   .getSettings()
720                   .get(AvailableSettings.HBM2DDL_CHARSET_NAME);
721           return new ScriptTargetOutputToFile(output, charset);
722         }
723       };
724
725       /**
726        * Change class-loader of current thread.
727        * This is necessary, because still not all parts of Hibernate 5 use
728        * the newly introduced ClassLoaderService and will fail otherwise!
729        */
730       Thread thread = Thread.currentThread();
731       ClassLoader contextClassLoader = thread.getContextClassLoader();
732       try
733       {
734         thread.setContextClassLoader(classLoader);
735         build((MetadataImplementor)metadataBuilder.build(), options, target);
736       }
737       finally
738       {
739         thread.setContextClassLoader(contextClassLoader);
740         for (Exception e : handler.getExceptions())
741           getLog().error(e.getMessage());
742         /** Track, the content of the generated script */
743         checkOutputFile(output, tracker);
744       }
745     }
746     catch (MojoExecutionException e)
747     {
748       tracker.failed();
749       throw e;
750     }
751     catch (MojoFailureException e)
752     {
753       tracker.failed();
754       throw e;
755     }
756     catch (RuntimeException e)
757     {
758       tracker.failed();
759       throw e;
760     }
761     finally
762     {
763       /** Remember mappings and configuration */
764       tracker.save();
765
766       /** Close the connection - if one was opened */
767       connectionProvider.close();
768
769       /** Stop Log-Capturing */
770       MavenLogAppender.endPluginLog(this);
771     }
772   }
773
774
775   abstract void build(
776       MetadataImplementor metadata,
777       ExecutionOptions options,
778       TargetDescriptor target
779       )
780     throws
781       MojoFailureException,
782       MojoExecutionException;
783
784
785   private MutableClassLoader createClassLoader() throws MojoExecutionException
786   {
787     try
788     {
789       getLog().debug("Creating ClassLoader for project-dependencies...");
790       LinkedHashSet<URL> urls = new LinkedHashSet<URL>();
791       File file;
792
793       file = new File(testOutputDirectory);
794       if (!file.exists())
795       {
796         getLog().info("Creating test-output-directory: " + testOutputDirectory);
797         file.mkdirs();
798       }
799       urls.add(file.toURI().toURL());
800
801       file = new File(outputDirectory);
802       if (!file.exists())
803       {
804         getLog().info("Creating output-directory: " + outputDirectory);
805         file.mkdirs();
806       }
807       urls.add(file.toURI().toURL());
808
809       return new MutableClassLoader(urls, getLog());
810     }
811     catch (Exception e)
812     {
813       getLog().error("Error while creating ClassLoader!", e);
814       throw new MojoExecutionException(e.getMessage());
815     }
816   }
817
818   private void completeClassPath(MutableClassLoader classLoader)
819       throws
820         MojoExecutionException
821   {
822     try
823     {
824       getLog().debug("Completing class-paths of the ClassLoader for project-dependencies...");
825       List<String> classpathFiles = project.getCompileClasspathElements();
826       if (scanTestClasses)
827         classpathFiles.addAll(project.getTestClasspathElements());
828       LinkedHashSet<URL> urls = new LinkedHashSet<URL>();
829       for (String pathElement : classpathFiles)
830       {
831         getLog().debug("Dependency: " + pathElement);
832         urls.add(new File(pathElement).toURI().toURL());
833       }
834       classLoader.add(urls);
835     }
836     catch (Exception e)
837     {
838       getLog().error("Error while creating ClassLoader!", e);
839       throw new MojoExecutionException(e.getMessage());
840     }
841   }
842
843   private Map loadProperties(ConfigLoader configLoader)
844       throws
845         MojoExecutionException
846   {
847     /** Try to read configuration from properties-file */
848     if (hibernateProperties == null)
849     {
850       try
851       {
852         return configLoader.loadProperties("hibernate.properties");
853       }
854       catch (ConfigurationException e)
855       {
856         getLog().debug(e.getMessage());
857         return Collections.EMPTY_MAP;
858       }
859     }
860     else
861     {
862       try
863       {
864         File file = new File(hibernateProperties);
865         if (file.exists())
866         {
867           getLog().info("Reading settings from file " + hibernateProperties + "...");
868           return configLoader.loadProperties(file);
869         }
870         else
871           return configLoader.loadProperties(hibernateProperties);
872       }
873       catch (ConfigurationException e)
874       {
875         getLog().error("Error while reading properties!", e);
876         throw new MojoExecutionException(e.getMessage());
877       }
878     }
879   }
880
881   private LoadedConfig loadConfig(ConfigLoader configLoader)
882       throws MojoExecutionException
883   {
884     /** Try to read configuration from configuration-file */
885     if (hibernateConfig == null)
886     {
887       try
888       {
889         return configLoader.loadConfigXmlResource("hibernate.cfg.xml");
890       }
891       catch (ConfigurationException e)
892       {
893         getLog().debug(e.getMessage());
894         return null;
895       }
896     }
897     else
898     {
899       try
900       {
901         File file = new File(hibernateConfig);
902         if (file.exists())
903         {
904           getLog().info("Reading configuration from file " + hibernateConfig + "...");
905           return configLoader.loadConfigXmlFile(file);
906         }
907         else
908         {
909           return configLoader.loadConfigXmlResource(hibernateConfig);
910         }
911       }
912       catch (ConfigurationException e)
913       {
914         getLog().error("Error while reading configuration!", e);
915         throw new MojoExecutionException(e.getMessage());
916       }
917     }
918   }
919
920   private void configure(Properties properties, ModificationTracker tracker)
921       throws MojoFailureException
922   {
923     /**
924      * Special treatment for the configuration-value "execute": if it is
925      * switched to "true", the genearation fo the schema should be forced!
926      */
927     if (tracker.check(EXECUTE, execute.toString()) && execute)
928       tracker.touch();
929
930     /**
931      * Configure the generation of the SQL.
932      * Overwrite values from properties-file if the configuration parameter is
933      * known to Hibernate.
934      */
935     configure(properties, dialect, DIALECT);
936     configure(properties, delimiter, HBM2DDL_DELIMITER);
937     configure(properties, format, FORMAT_SQL);
938     configure(properties, createNamespaces, HBM2DLL_CREATE_NAMESPACES);
939     configure(properties, implicitNamingStrategy, IMPLICIT_NAMING_STRATEGY);
940     configure(properties, physicalNamingStrategy, PHYSICAL_NAMING_STRATEGY);
941     tracker.track(OUTPUTDIRECTORY, outputDirectory); // << not reflected in hibernate configuration!
942     tracker.track(SCAN_DEPENDENCIES, scanDependencies); // << not reflected in hibernate configuration!
943     tracker.track(SCAN_TESTCLASSES, scanTestClasses.toString()); // << not reflected in hibernate configuration!
944     tracker.track(TEST_OUTPUTDIRECTORY, testOutputDirectory); // << not reflected in hibernate configuration!
945
946     /**
947      * Special treatment for the configuration-value "show": a change of its
948      * configured value should not lead to a regeneration of the database
949      * schama!
950      */
951     if (show == null)
952       show = Boolean.valueOf(properties.getProperty(SHOW_SQL));
953     else
954       properties.setProperty(SHOW_SQL, show.toString());
955
956     /**
957      * Configure the connection parameters.
958      * Overwrite values from properties-file.
959      */
960     configure(properties, driver, DRIVER, JPA_JDBC_DRIVER);
961     configure(properties, url, URL, JPA_JDBC_URL);
962     configure(properties, username, USER, JPA_JDBC_USER);
963     configure(properties, password, PASS, JPA_JDBC_PASSWORD);
964
965     if (properties.isEmpty())
966     {
967       getLog().error("No properties set!");
968       throw new MojoFailureException("Hibernate configuration is missing!");
969     }
970
971     getLog().info("Gathered hibernate-configuration (turn on debugging for details):");
972     for (Entry<Object,Object> entry : properties.entrySet())
973       getLog().info("  " + entry.getKey() + " = " + entry.getValue());
974   }
975
976   private void configure(
977       Properties properties,
978       String value,
979       String key,
980       String alternativeKey
981       )
982   {
983     configure(properties, value, key);
984
985     if (properties.containsKey(alternativeKey))
986     {
987       if (properties.containsKey(key))
988       {
989         getLog().warn(
990             "Ignoring property " + alternativeKey + "=\"" +
991             properties.getProperty(alternativeKey) +
992             "\" in favour for property " + key + "=\"" +
993             properties.getProperty(key) + "\""
994             );
995         properties.remove(alternativeKey);
996       }
997       else
998       {
999         value = properties.getProperty(alternativeKey);
1000         properties.remove(alternativeKey);
1001         getLog().info(
1002             "Using value \"" + value + "\" from property " + alternativeKey +
1003             " for property " + key
1004             );
1005         properties.setProperty(key, value);
1006       }
1007     }
1008   }
1009
1010   private void configure(Properties properties, String value, String key)
1011   {
1012     if (value != null)
1013     {
1014       if (properties.containsKey(key))
1015         getLog().info(
1016             "Overwriting property " + key + "=\"" +
1017             properties.getProperty(key) +
1018             "\" with value \"" + value + "\""
1019             );
1020       else
1021         getLog().debug("Using value \"" + value + "\" for property " + key);
1022       properties.setProperty(key, value);
1023     }
1024   }
1025
1026   private void configure(Properties properties, Boolean value, String key)
1027   {
1028     configure(properties, value == null ? null : value.toString(), key);
1029   }
1030
1031   private File getOutputFile(String filename)
1032       throws
1033         MojoExecutionException
1034   {
1035     File output = new File(filename);
1036
1037     if (!output.isAbsolute())
1038     {
1039       // Interpret relative file path relative to build directory
1040       output = new File(buildDirectory, filename);
1041     }
1042     getLog().debug("Output file: " + output.getPath());
1043
1044     // Ensure that directory path for specified file exists
1045     File outFileParentDir = output.getParentFile();
1046     if (null != outFileParentDir && !outFileParentDir.exists())
1047     {
1048       try
1049       {
1050         getLog().info(
1051             "Creating directory path for output file:" +
1052             outFileParentDir.getPath()
1053             );
1054         outFileParentDir.mkdirs();
1055       }
1056       catch (Exception e)
1057       {
1058         String error =
1059             "Error creating directory path for output file: " + e.getMessage();
1060         getLog().error(error);
1061         throw new MojoExecutionException(error);
1062       }
1063     }
1064
1065     try
1066     {
1067       output.createNewFile();
1068     }
1069     catch (IOException e)
1070     {
1071       String error = "Error creating output file: " + e.getMessage();
1072       getLog().error(error);
1073       throw new MojoExecutionException(error);
1074     }
1075
1076     if (!output.canWrite())
1077     {
1078       String error =
1079           "Output file " + output.getAbsolutePath() + " is not writable!";
1080       getLog().error(error);
1081       throw new MojoExecutionException(error);
1082     }
1083
1084     return output;
1085   }
1086
1087   private void checkOutputFile(File output, ModificationTracker tracker)
1088       throws
1089         MojoExecutionException
1090   {
1091     try
1092     {
1093       if (output.exists())
1094         tracker.track(SCRIPT, new FileInputStream(output));
1095       else
1096         tracker.track(SCRIPT, ZonedDateTime.now().toString());
1097     }
1098     catch (IOException e)
1099     {
1100       String error =
1101           "Error while checking the generated script: " + e.getMessage();
1102       getLog().error(error);
1103       throw new MojoExecutionException(error);
1104     }
1105   }
1106
1107   private void addMappings(MetadataSources sources, ModificationTracker tracker)
1108       throws MojoFailureException
1109   {
1110     getLog().debug("Adding explicitly configured mappings...");
1111     if (mappings != null)
1112     {
1113       try
1114       {
1115         for (String filename : mappings.split("[\\s,]+"))
1116         {
1117           // First try the filename as absolute/relative path
1118           File file = new File(filename);
1119           if (!file.exists())
1120           {
1121             // If the file was not found, search for it in the resource-directories
1122             for (Resource resource : project.getResources())
1123             {
1124               file = new File(resource.getDirectory() + File.separator + filename);
1125               if (file.exists())
1126                 break;
1127             }
1128           }
1129           if (file.exists())
1130           {
1131             if (file.isDirectory())
1132               // TODO: add support to read all mappings under a directory
1133               throw new MojoFailureException(file.getAbsolutePath() + " is a directory");
1134             if (tracker.track(filename, new FileInputStream(file)))
1135               getLog().debug("Found new or modified mapping-file: " + filename);
1136             else
1137               getLog().debug("Mapping-file unchanged: " + filename);
1138
1139             sources.addFile(file);
1140           }
1141           else
1142             throw new MojoFailureException("File " + filename + " could not be found in any of the configured resource-directories!");
1143         }
1144       }
1145       catch (IOException e)
1146       {
1147         throw new MojoFailureException("Cannot calculate MD5 sums!", e);
1148       }
1149     }
1150   }
1151
1152   private void addRoot(Set<URL> urls, String path) throws MojoFailureException
1153   {
1154     try
1155     {
1156       File dir = new File(outputDirectory);
1157       if (dir.exists())
1158       {
1159         getLog().info("Adding " + dir.getAbsolutePath() + " to the list of roots to scan...");
1160         urls.add(dir.toURI().toURL());
1161       }
1162     }
1163     catch (MalformedURLException e)
1164     {
1165       getLog().error("error while adding the project-root to the list of roots to scan!", e);
1166       throw new MojoFailureException(e.getMessage());
1167     }
1168   }
1169
1170   private void addDependencies(Set<URL> urls) throws MojoFailureException
1171   {
1172     try
1173     {
1174       if (scanDependencies != null)
1175       {
1176         Matcher matcher = SPLIT.matcher(scanDependencies);
1177         while (matcher.find())
1178         {
1179           getLog().info("Adding dependencies from scope " + matcher.group() + " to the list of roots to scan");
1180           for (Artifact artifact : project.getDependencyArtifacts())
1181           {
1182             if (!artifact.getScope().equalsIgnoreCase(matcher.group()))
1183               continue;
1184             if (artifact.getFile() == null)
1185             {
1186               getLog().warn("Cannot add dependency " + artifact.getId() + ": no JAR-file available!");
1187               continue;
1188             }
1189             getLog().info("Adding dependencies from scope " + artifact.getId() + " to the list of roots to scan");
1190             urls.add(artifact.getFile().toURI().toURL());
1191           }
1192         }
1193       }
1194     }
1195     catch (MalformedURLException e)
1196     {
1197       getLog().error("Error while adding dependencies to the list of roots to scan!", e);
1198       throw new MojoFailureException(e.getMessage());
1199     }
1200   }
1201
1202   private Set<String> scanUrls(Set<URL> scanRoots)
1203       throws
1204         MojoFailureException
1205   {
1206     try
1207     {
1208       AnnotationDB db = new AnnotationDB();
1209       for (URL root : scanRoots)
1210         db.scanArchives(root);
1211
1212       Set<String> classes = new HashSet<String>();
1213       if (db.getAnnotationIndex().containsKey(Entity.class.getName()))
1214         classes.addAll(db.getAnnotationIndex().get(Entity.class.getName()));
1215       if (db.getAnnotationIndex().containsKey(MappedSuperclass.class.getName()))
1216         classes.addAll(db.getAnnotationIndex().get(MappedSuperclass.class.getName()));
1217       if (db.getAnnotationIndex().containsKey(Embeddable.class.getName()))
1218         classes.addAll(db.getAnnotationIndex().get(Embeddable.class.getName()));
1219
1220       return classes;
1221     }
1222     catch (Exception e)
1223     {
1224       getLog().error("Error while scanning!", e);
1225       throw new MojoFailureException(e.getMessage());
1226     }
1227   }
1228
1229   private void addAnnotated(
1230       String name,
1231       MetadataSources sources,
1232       ClassLoaderService classLoaderService,
1233       ModificationTracker tracker
1234       )
1235       throws
1236         MojoFailureException,
1237         MojoExecutionException
1238   {
1239     try
1240     {
1241       getLog().info("Adding annotated resource: " + name);
1242       String packageName = null;
1243
1244       boolean error = false;
1245       try
1246       {
1247         Class<?> annotatedClass = classLoaderService.classForName(name);
1248         String resourceName = annotatedClass.getName();
1249         resourceName =
1250             resourceName.substring(
1251                 resourceName.lastIndexOf(".") + 1,
1252                 resourceName.length()
1253                 ) + ".class";
1254         InputStream is = annotatedClass.getResourceAsStream(resourceName);
1255         if (is != null)
1256         {
1257           if (tracker.track(name, is))
1258             getLog().debug("New or modified class: " + name);
1259           else
1260             getLog().debug("Unchanged class: " + name);
1261           sources.addAnnotatedClass(annotatedClass);
1262           packageName = annotatedClass.getPackage().getName();
1263         }
1264         else
1265         {
1266           getLog().error("cannot find ressource " + resourceName + " for class " + name);
1267           error = true;
1268         }
1269       }
1270       catch(ClassLoadingException e)
1271       {
1272         packageName = name;
1273       }
1274       if (error)
1275       {
1276         throw new MojoExecutionException("error while inspecting annotated class " + name);
1277       }
1278
1279       while (packageName != null)
1280       {
1281         if (packages.contains(packageName))
1282           return;
1283         String resource = packageName.replace('.', '/') + "/package-info.class";
1284         InputStream is = classLoaderService.locateResourceStream(resource);
1285         if (is == null)
1286         {
1287           // No compiled package-info available: no package-level annotations!
1288           getLog().debug("Package " + packageName + " is not annotated.");
1289         }
1290         else
1291         {
1292           if (tracker.track(packageName, is))
1293             getLog().debug("New or modified package: " + packageName);
1294           else
1295            getLog().debug("Unchanged package: " + packageName);
1296           getLog().info("Adding annotations from package " + packageName);
1297           sources.addPackage(packageName);
1298         }
1299         packages.add(packageName);
1300         int i = packageName.lastIndexOf('.');
1301         if (i < 0)
1302           packageName = null;
1303         else
1304           packageName = packageName.substring(0,i);
1305       }
1306     }
1307     catch (Exception e)
1308     {
1309       getLog().error("Error while adding the annotated class " + name, e);
1310       throw new MojoFailureException(e.getMessage());
1311     }
1312   }
1313
1314   private ParsedPersistenceXmlDescriptor loadPersistenceUnit(
1315       ClassLoaderService classLoaderService,
1316       Properties properties
1317       )
1318       throws
1319         MojoFailureException
1320   {
1321     PersistenceXmlParser parser =
1322         new PersistenceXmlParser(
1323             classLoaderService,
1324             PersistenceUnitTransactionType.RESOURCE_LOCAL
1325              );
1326
1327     Map<String, ParsedPersistenceXmlDescriptor> units =
1328         parser.doResolve(properties);
1329
1330     if (persistenceUnit == null)
1331     {
1332       Iterator<String> names = units.keySet().iterator();
1333       if (!names.hasNext())
1334       {
1335         getLog().info("Found no META-INF/persistence.xml.");
1336         return null;
1337       }
1338
1339       String name = names.next();
1340       if (!names.hasNext())
1341       {
1342           getLog().info("Using persistence-unit " + name);
1343           return units.get(name);
1344       }
1345
1346       StringBuilder builder = new StringBuilder();
1347       builder.append("No name provided and multiple persistence units found: ");
1348       builder.append(name);
1349       while(names.hasNext())
1350       {
1351         builder.append(", ");
1352         builder.append(names.next());
1353       }
1354       builder.append('.');
1355       throw new MojoFailureException(builder.toString());
1356     }
1357
1358     if (units.containsKey(persistenceUnit))
1359     {
1360       getLog().info("Using configured persistence-unit " + persistenceUnit);
1361       return units.get(persistenceUnit);
1362     }
1363
1364     throw new MojoFailureException("Could not find persistence-unit " + persistenceUnit);
1365   }
1366 }