net.sf.nakeduml.jaxb.JaxbImplementor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metamodel Show documentation
A uml code generator and execution engine
The newest version!
package net.sf.nakeduml.jaxb;

import java.util.Collections;

import net.sf.nakeduml.feature.visit.VisitAfter;
import net.sf.nakeduml.javageneration.AbstractJavaProducingVisitor;
import net.sf.nakeduml.javageneration.NakedStructuralFeatureMap;
import net.sf.nakeduml.javametamodel.OJPathName;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedClass;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedOperation;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotationValue;
import net.sf.nakeduml.metamodel.core.INakedEntity;
import net.sf.nakeduml.metamodel.core.INakedInterface;
import net.sf.nakeduml.metamodel.core.INakedProperty;

public class JaxbImplementor extends AbstractJavaProducingVisitor {

	@VisitAfter(matchSubclasses = true)
	public void visitClass(INakedEntity c) {
		//TODO this should be a numl webservice step
		if (hasOJClass(c) && !(c instanceof INakedInterface)) {
			OJAnnotatedClass owner = findJavaClass(c);
			addXmlRootElement(owner);
			for (INakedProperty p : c.getEffectiveAttributes()) {
				if (p.getNakedBaseType() instanceof INakedInterface) {
					addXmlAnyElement(owner, c, p);
				}
			}
		}	
	}
	
	private void addXmlAnyElement(OJAnnotatedClass clazz, INakedEntity c, INakedProperty p) {
		NakedStructuralFeatureMap map = new NakedStructuralFeatureMap(p);
		OJAnnotatedOperation oper = (OJAnnotatedOperation)clazz.findOperation(map.getter(), Collections.EMPTY_LIST);
		oper.addAnnotationIfNew(new OJAnnotationValue(new OJPathName("javax.xml.bind.annotation.XmlAnyElement")));
	}

	private void addXmlRootElement(OJAnnotatedClass owner) {
		owner.addAnnotationIfNew(new OJAnnotationValue(new OJPathName("javax.xml.bind.annotation.XmlRootElement")));
	}	
	
	@VisitAfter(matchSubclasses = true)
	public void visitClass(INakedProperty np) {
		if (!np.isDerived() && np.getNakedBaseType() instanceof INakedEntity && !np.isInverse() && hasOJClass(np.getOwner())) {
			NakedStructuralFeatureMap map = new NakedStructuralFeatureMap(np);
			OJAnnotatedClass owner = findJavaClass(np.getOwner());
			OJAnnotatedOperation o = (OJAnnotatedOperation)owner.findOperation(map.getter(), Collections.EMPTY_LIST);
			JaxbAnnotator.addXmlTransient(o);
		}
	}	
}