net.sf.nakeduml.javageneration.auditing.AuditEntryMassageOriginalClasses 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.javageneration.auditing;

import java.util.List;

import net.sf.nakeduml.feature.visit.VisitBefore;
import net.sf.nakeduml.javageneration.NakedStructuralFeatureMap;
import net.sf.nakeduml.javageneration.util.OJUtil;
import net.sf.nakeduml.javametamodel.OJBlock;
import net.sf.nakeduml.javametamodel.OJIfStatement;
import net.sf.nakeduml.javametamodel.OJOperation;
import net.sf.nakeduml.javametamodel.OJPathName;
import net.sf.nakeduml.javametamodel.OJSimpleStatement;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedClass;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedField;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotatedOperation;
import net.sf.nakeduml.javametamodel.annotation.OJAnnotationValue;
import net.sf.nakeduml.javametamodel.annotation.OJClassValue;
import net.sf.nakeduml.metamodel.core.INakedEntity;
import net.sf.nakeduml.metamodel.core.INakedProperty;

public class AuditEntryMassageOriginalClasses extends AbstractJavaProducingVisitorForAudit {


	@VisitBefore(matchSubclasses = true)
	public void visitClasses(INakedEntity entity) {
		if (isPersistent(entity) && hasOJClass(entity)) {
			OJPathName path = OJUtil.classifierPathname(entity);
			OJAnnotatedClass ojClass = (OJAnnotatedClass) this.javaModel.findIntfOrCls(path);
			if (ojClass != null) {
				addMakeAuditCopyIdOnly(entity, ojClass);
				addMakeAuditCopyWithoutParentMethod(ojClass);
				addMakeAuditCopyMethod(entity, ojClass);
//				addEntityListenerAnnotation(ojClass);
			}
		}
	}

	private void addMakeAuditCopyIdOnly(INakedEntity entity, OJAnnotatedClass ojClass) {
		ojClass.addToImports(new OJPathName("net.sf.nakeduml.util.AuditId"));
		
		OJAnnotatedOperation makeAuditCopyIdOnly = new OJAnnotatedOperation();
		makeAuditCopyIdOnly.setName("makeAuditCopyIdOnly");
		OJPathName pathName = ojClass.getPathName();
		String remove = pathName.getNames().remove(pathName.getNames().size() - 1);
		pathName.getNames().add(remove + "_Audit");
		makeAuditCopyIdOnly.setReturnType(pathName);
		makeAuditCopyIdOnly.setAbstract(ojClass.isAbstract());

		if (!ojClass.isAbstract()) {
			OJBlock body = makeAuditCopyIdOnly.getBody();
			OJAnnotatedField result = new OJAnnotatedField();
			result.setName("result");
			result.setType(pathName);
			result.setInitExp("new " + pathName + "()");
			body.addToLocals(result);

			String rootClassName = entity.getMappingInfo().getJavaName().toString();
			OJSimpleStatement setId = new OJSimpleStatement("result.setId(new  AuditId(getId(), getObjectVersion()))");
			body.addToStatements(setId);
			body.addToStatements("result.setOriginal(new "+rootClassName + "())");
			body.addToStatements("result.getOriginal().setId(getId())");
			body.addToStatements("return result");
		}

		ojClass.addToOperations(makeAuditCopyIdOnly);

	}

	private void addMakeAuditCopyMethod(INakedEntity entity, OJAnnotatedClass ojClass) {
		OJOperation operation = new OJAnnotatedOperation();
		operation.setName("makeAuditCopy");
		OJPathName pathName = ojClass.getPathName();
		String remove = pathName.getNames().remove(pathName.getNames().size() - 1);
		pathName.getNames().add(remove + "_Audit");
		operation.setReturnType(new OJPathName("net.sf.nakeduml.util.Audited"));
		operation.setOwner(ojClass);
		operation.setAbstract(ojClass.isAbstract());
		if (!ojClass.isAbstract()) {
			OJBlock body = operation.getBody();
			OJAnnotatedField result = new OJAnnotatedField();
			result.setName("result");
			result.setType(pathName);
			result.setInitExp("makeAuditCopyWithoutParent()");
			body.addToLocals(result);

			List attributes = entity.getEffectiveAttributes();
			for (INakedProperty property : attributes) {
				NakedStructuralFeatureMap map = new NakedStructuralFeatureMap(property);
				if (!property.isDerived() && !property.isComposite() && isPersistent((property).getNakedBaseType()) && (map.isManyToOne() || map.isOne())) {
					OJIfStatement ifStatement = new OJIfStatement();
					ifStatement.setCondition(map.getter() + "() != null");
					OJBlock then = new OJBlock();
					OJSimpleStatement thenStatement = new OJSimpleStatement("result." + map.setter() + "(" + map.getter() + "().makeAuditCopyIdOnly())");
					then.addToStatements(thenStatement);
					ifStatement.setThenPart(then);
					body.addToStatements(ifStatement);
				}
			}

			String rootClassName = entity.getMappingInfo().getJavaName().toString();
			body.addToStatements("result.setOriginal(new "+rootClassName + "())");
			body.addToStatements("result.getOriginal().setId(getId())");

			body.addToStatements("return result");
		}
	}

	private void addEntityListenerAnnotation(OJAnnotatedClass c) {
		OJAnnotationValue entityListener = new OJAnnotationValue(new OJPathName("javax.persistence.EntityListeners"));
		entityListener.addClassValue(new OJClassValue(new OJPathName("util.NakedUmlEJB3AuditListener")));
		c.addAnnotationIfNew(entityListener);
	}

	private void addMakeAuditCopyWithoutParentMethod(OJAnnotatedClass c) {
		OJOperation operation = new OJAnnotatedOperation();
		operation.setName("makeAuditCopyWithoutParent");
		OJPathName pathName = c.getPathName();
		String remove = pathName.getNames().remove(pathName.getNames().size() - 1);
		pathName.getNames().add(remove + "_Audit");
		operation.setReturnType(pathName);
		c.addToImports(pathName);
		operation.setOwner(c);
		operation.setAbstract(c.isAbstract());
		if (!c.isAbstract()) {
			OJBlock body = operation.getBody();
			OJAnnotatedField result = new OJAnnotatedField();
			result.setName("result");
			result.setType(pathName);
			result.setInitExp("new " + pathName + "()");
			body.addToLocals(result);
			// body.addToStatements("result.set"+c.getName()+"(this)");
			// body.addToStatements("result.setOriginal(this)");
			body.addToStatements("result.copyShallowState((" + c.getPathName() + ")this,result)");
			body.addToStatements("return result");
		}
		c.addToImplementedInterfaces(new OJPathName("net.sf.nakeduml.util.Auditable"));
	}

}