DtoGrammarLabelProvider.java

/**
 * Copyright (c) 2011 - 2015, Lunifera GmbH (Gross Enzersdorf), Loetz KG (Heidelberg)
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *         Florian Pirchner - Initial implementation
 */
package org.lunifera.dsl.dto.xtext.ui.labeling;

import static com.google.common.collect.Iterables.filter;

import java.util.Iterator;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.xtext.common.types.JvmParameterizedTypeReference;
import org.eclipse.xtext.common.types.JvmVisibility;
import org.eclipse.xtext.xbase.ui.labeling.XbaseLabelProvider;
import org.lunifera.dsl.semantic.common.types.LClass;
import org.lunifera.dsl.semantic.common.types.LDataType;
import org.lunifera.dsl.semantic.common.types.LEnum;
import org.lunifera.dsl.semantic.common.types.LEnumLiteral;
import org.lunifera.dsl.semantic.common.types.LFeature;
import org.lunifera.dsl.semantic.common.types.LPackage;
import org.lunifera.dsl.semantic.dto.LDto;
import org.lunifera.dsl.semantic.dto.LDtoAttribute;
import org.lunifera.dsl.semantic.dto.LDtoInheritedAttribute;
import org.lunifera.dsl.semantic.dto.LDtoInheritedReference;
import org.lunifera.dsl.semantic.dto.LDtoMapper;
import org.lunifera.dsl.semantic.dto.LDtoOperation;
import org.lunifera.dsl.semantic.dto.LDtoReference;

//git@github.com/lunifera/lunifera-dsl.git
import com.google.inject.Inject;

/**
 * Provides labels for a EObjects.
 */
@SuppressWarnings("restriction")
public class DtoGrammarLabelProvider extends XbaseLabelProvider {

	@Inject
	private DtoGrammarImages images;

	@Inject
	public DtoGrammarLabelProvider(AdapterFactoryLabelProvider delegate) {
		super(delegate);
	}

	public ImageDescriptor image(LDto element) {
		return images.forClass(JvmVisibility.PUBLIC, 0);
	}

	public ImageDescriptor image(LEnum element) {
		return images.forEnum(JvmVisibility.PUBLIC);
	}

	public ImageDescriptor image(LEnumLiteral element) {
		return images.forEnumLiteral();
	}

	public String image(LDtoMapper element) {
		return "converter.gif";
	}

	public String image(LDtoInheritedReference element) {
		return "inheritedfeature.gif";
	}

	public ImageDescriptor image(LDtoReference element) {
		if (!element.isCascading()) {
			return images.forField(JvmVisibility.PUBLIC, 0);
		} else {
			return images.forCascading(JvmVisibility.PUBLIC);
		}
	}

	public String image(LDtoInheritedAttribute element) {
		return "inheritedfeature.gif";
	}

	public ImageDescriptor image(LDtoOperation element) {
		return images.forOperation(JvmVisibility.PUBLIC, 0);
	}

	public ImageDescriptor image(LDtoAttribute element) {
		if (!element.isCascading()) {
			return images.forField(JvmVisibility.PUBLIC, 0);
		} else {
			return images.forCascading(JvmVisibility.PUBLIC);
		}
	}

	public ImageDescriptor image(LDataType element) {
		return images.forDatatype();
	}

	public ImageDescriptor image(LPackage element) {
		return images.forPackage();
	}

	public ImageDescriptor image(JvmParameterizedTypeReference typeRef) {
		return images.forTypeParameter(0);
	}

	public String text(LClass element) {
		return element.getName();
	}

	public String text(LFeature element) {
//		JvmField inferredField = getFirstOrNull(
//				associations.getJvmElements(element), JvmField.class);
//		if (inferredField != null) {
//			JvmTypeReference type = inferredField.getType();
//			if (type != null) {
//				return element.getName() + " : " + type.getSimpleName();
//			}
//		}
		return element.getName();
	}

	public String text(LDtoMapper element) {
		return "Mapper";
	}

	public String text(LDtoInheritedAttribute element) {
//		JvmField inferredField = getFirstOrNull(
//				associations.getJvmElements(element), JvmField.class);
//		if (inferredField != null) {
//			JvmTypeReference type = inferredField.getType();
//			if (type != null) {
//				return extension.inheritedFeature(element).getName() + " : "
//						+ type.getSimpleName();
//			}
//		}
		return element.getInheritedFeature().getName();
	}

	public String text(LDtoInheritedReference element) {
//		JvmField inferredField = getFirstOrNull(
//				associations.getJvmElements(element), JvmField.class);
//		if (inferredField != null) {
//			JvmTypeReference type = inferredField.getType();
//			if (type != null) {
//				return extension.inheritedFeature(element).getName() + " : "
//						+ type.getSimpleName();
//			}
//		}
		return element.getInheritedFeature().getName();
	}

	public String text(JvmParameterizedTypeReference typeRef) {
		return typeRef.getType().getSimpleName();
	}

	protected <T> T getFirstOrNull(Iterable<EObject> elements, Class<T> type) {
		Iterator<T> iterator = filter(elements, type).iterator();
		return iterator.hasNext() ? iterator.next() : null;
	}
}