jpa/plugins/org.eclipse.jpt.ui/src/org/eclipse/jpt/ui/internal/java/structure/JavaPersistentAttributeItemProvider.java

package org.eclipse.jpt.ui.internal.java.structure;

import org.eclipse.emf.common.notify.AdapterFactory;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.edit.provider.IEditingDomainItemProvider;
import org.eclipse.emf.edit.provider.IItemLabelProvider;
import org.eclipse.emf.edit.provider.IStructuredItemContentProvider;
import org.eclipse.emf.edit.provider.ITreeItemContentProvider;
import org.eclipse.emf.edit.provider.ItemProviderAdapter;
import org.eclipse.emf.edit.provider.ViewerNotification;
import org.eclipse.jpt.core.internal.IAttributeMapping;
import org.eclipse.jpt.core.internal.IPersistentAttribute;
import org.eclipse.jpt.core.internal.JpaCorePackage;
import org.eclipse.jpt.core.internal.mappings.IBasic;
import org.eclipse.jpt.core.internal.mappings.IEmbedded;
import org.eclipse.jpt.core.internal.mappings.IEmbeddedId;
import org.eclipse.jpt.core.internal.mappings.IId;
import org.eclipse.jpt.core.internal.mappings.IManyToMany;
import org.eclipse.jpt.core.internal.mappings.IManyToOne;
import org.eclipse.jpt.core.internal.mappings.IOneToMany;
import org.eclipse.jpt.core.internal.mappings.IOneToOne;
import org.eclipse.jpt.core.internal.mappings.ITransient;
import org.eclipse.jpt.core.internal.mappings.IVersion;
import org.eclipse.jpt.ui.internal.mappings.JpaUiMappingsImages;

public class JavaPersistentAttributeItemProvider extends ItemProviderAdapter
	implements IEditingDomainItemProvider, 
		IStructuredItemContentProvider,
		ITreeItemContentProvider, 
		IItemLabelProvider
{
	public JavaPersistentAttributeItemProvider(AdapterFactory adapterFactory) {
		super(adapterFactory);
	}
	
	@Override
	public Object getImage(Object object) {
		IAttributeMapping mapping = ((IPersistentAttribute) object).getMapping();
		
		if (mapping instanceof IBasic) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.BASIC);
		}
		else if (mapping instanceof IId) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.ID);
		}
		else if (mapping instanceof IVersion) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.VERSION);
		}
		else if (mapping instanceof IEmbedded) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.EMBEDDED);
		}
		else if (mapping instanceof IEmbeddedId) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.EMBEDDED_ID);
		}
		else if (mapping instanceof IOneToOne) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.ONE_TO_ONE);
		}
		else if (mapping instanceof IOneToMany) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.ONE_TO_MANY);
		}
		else if (mapping instanceof IManyToOne) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.MANY_TO_ONE);
		}
		else if (mapping instanceof IManyToMany) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.MANY_TO_MANY);
		}
		else if (mapping instanceof ITransient) {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.TRANSIENT);
		}
		else {
			return JpaUiMappingsImages.getImage(JpaUiMappingsImages.NULL_ATTRIBUTE_MAPPING);
		}
	}
	
	@Override
	public String getText(Object object) {
		return ((IPersistentAttribute) object).getName();
	}
	
	@Override
	public void notifyChanged(Notification notification) {
		updateChildren(notification);

		switch (notification.getFeatureID(IPersistentAttribute.class)) {
		case JpaCorePackage.IPERSISTENT_ATTRIBUTE__MAPPING_KEY:
			fireNotifyChanged(new ViewerNotification(notification, notification
					.getNotifier(), false, true));
			return;
		}
		super.notifyChanged(notification);
	}
}