net.sf.nakeduml.javageneration.hibernate.HibernateConfigGenerator Maven / Gradle / Ivy

Go to download
package net.sf.nakeduml.javageneration.hibernate;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import net.sf.nakeduml.feature.visit.VisitBefore;
import net.sf.nakeduml.javageneration.AbstractTextProducingVisitor;
import net.sf.nakeduml.javageneration.CharArrayTextSource;
import net.sf.nakeduml.javametamodel.OJClass;
import net.sf.nakeduml.javametamodel.OJPackage;
import net.sf.nakeduml.javametamodel.OJPathName;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedClass;
import net.sf.nakeduml.metamodel.models.INakedModel;
import net.sf.nakeduml.visitor.AbstractOJVisitor;

public class HibernateConfigGenerator extends AbstractTextProducingVisitor {
	public static final class OJVisitor extends AbstractOJVisitor {
		private final HashSet classes;

		private OJVisitor(HashSet classes) {
			this.classes = classes;
		}

		@VisitBefore(matchSubclasses = true)
		public void visitPackage(OJPackage p) {
			for (OJClass c : p.getClasses()) {
				if (c instanceof OJAnnotatedClass && ((OJAnnotatedClass) c).hasAnnotation(new OJPathName("javax.persistence.Table"))) {
					classes.add(c);
				}
			}
		}
	}

	public static final String RESOURCE_DIR = "gen-src";

	@VisitBefore
	public void generateConfig(INakedModel o) {
		HashMap vars = new HashMap();
		Set classesRecursively = getClassesRecursively();
		List sortedClasses = new ArrayList(classesRecursively);
		Collections.sort(sortedClasses, new Comparator() {
			@Override
			public int compare(OJClass o1, OJClass o2) {
				if (o1.getName().endsWith("Audit")) {
					return 1;
				} else {
					return -1;
				}
			}
		});
		vars.put("persistentClasses", sortedClasses);
		// TODO parameterize this!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
		vars.put("requiresAuditing", true);
		vars.put("config", this.config);
		processTemplate(o, "templates/Model/Jbpm4HibernateConfig.vsl", "${model.mappingInfo.javaName}.hibernate.config.xml",
				CharArrayTextSource.TEST_RESOURCE, vars);
	}

	private Set getClassesRecursively() {
		final HashSet classes = new HashSet();
		new OJVisitor(classes).startVisiting(super.javaModel);
		return classes;
	}
}