org.jvnet.hyperjaxb3.ejb.strategy.mapping.EmbeddedMapping Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hyperjaxb3-ejb-plugin Show documentation
There is a newer version: 0.6.2
Show newest version
package org.jvnet.hyperjaxb3.ejb.strategy.mapping;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.jvnet.jaxb2_commons.util.OutlineUtils;

import com.sun.java.xml.ns.persistence.orm.AttributeOverride;
import com.sun.java.xml.ns.persistence.orm.Basic;
import com.sun.java.xml.ns.persistence.orm.Column;
import com.sun.java.xml.ns.persistence.orm.EmbeddableAttributes;
import com.sun.java.xml.ns.persistence.orm.Embedded;
import com.sun.tools.xjc.Options;
import com.sun.tools.xjc.model.CClassInfo;
import com.sun.tools.xjc.model.CPropertyInfo;
import com.sun.tools.xjc.model.CTypeInfo;
import com.sun.tools.xjc.outline.ClassOutline;
import com.sun.tools.xjc.outline.FieldOutline;
import com.sun.tools.xjc.outline.Outline;

public class EmbeddedMapping extends PropertyMapping implements
		FieldOutlineMapping {

	public Embedded process(Mapping context, FieldOutline fieldOutline,
			Options options) throws Exception {

		final Embedded embedded = context.getCustomizing().getEmbedded(
				fieldOutline);

		createEmbedded$Name(context, fieldOutline, embedded);

		createEmbedded$AttributeOverride(context, fieldOutline, embedded);
		// createBasic$Column(context, fieldOutline, basic);

		// if (basic.getLob() == null && basic.getTemporal() == null
		// && basic.getEnumerated() == null) {
		// if (isTemporal(fieldOutline)) {
		// basic.setTemporal(getTemporalType(fieldOutline));
		// } else if (isEnumerated(fieldOutline)) {
		// basic.setEnumerated(getEnumerated(fieldOutline));
		// } else if (isLob(fieldOutline)) {
		// basic.setLob(getLob(fieldOutline));
		// }
		//
		// }

		return embedded;
	}

	public void createEmbedded$Name(Mapping context, FieldOutline fieldOutline,
			final Embedded embedded) {
		embedded.setName(OutlineUtils.getPropertyName(fieldOutline));
	}

	public void createEmbedded$AttributeOverride(Mapping context,
			FieldOutline fieldOutline, final Embedded embedded)
			throws Exception {
		final CPropertyInfo propertyInfo = fieldOutline.getPropertyInfo();

		final Collection types = propertyInfo.ref();

		assert types.size() == 1;

		final CTypeInfo typeInfo = types.iterator().next();

		assert typeInfo instanceof CClassInfo;

		final CClassInfo classInfo = (CClassInfo) typeInfo;

		final Outline outline = fieldOutline.parent().parent();

		final ClassOutline classOutline = outline.getClazz(classInfo);

		final Options options = outline.getModel().options;

		final Map attributeOverrides = new HashMap();
		for (final AttributeOverride attributeOverride : embedded
				.getAttributeOverride()) {
			attributeOverrides.put(attributeOverride.getName(),
					attributeOverride);
		}

		final EmbeddableAttributes embeddableAttributes = context
				.getEmbeddableAttributesMapping().process(context,
						classOutline, options);

		for (final Basic basic : embeddableAttributes.getBasic()) {
			final String name = basic.getName();
			final AttributeOverride attributeOverride;
			if (!attributeOverrides.containsKey(name)) {
				attributeOverride = new AttributeOverride();
				attributeOverride.setName(name);
				attributeOverrides.put(name, attributeOverride);
				embedded.getAttributeOverride().add(attributeOverride);
			} else {
				attributeOverride = attributeOverrides.get(name);
			}

			if (attributeOverride.getColumn() == null) {
				final Column column = new Column();
				basic.getColumn().copyTo(column);
				column.setName(null);
				attributeOverride.setColumn(column);
			}
		}

		final FieldOutline[] fieldOutlines = classOutline.getDeclaredFields();
		for (final FieldOutline childFieldOutline : fieldOutlines) {
			final AttributeOverride attributeOverride = attributeOverrides
					.get(OutlineUtils.getPropertyName(childFieldOutline));

			if (attributeOverride != null) {
				final Column column = attributeOverride.getColumn();
				if (column.getName() == null
						|| "##default".equals(column.getName())) {
					column.setName(context.getNaming().getEmbedded$Column$Name(
							context, fieldOutline, childFieldOutline));
				}
			}
		}
	}
}