/**
* Copyright (c) 2013 Puppet Labs, Inc. and other contributors, as listed below.
* 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:
* Puppet Labs
*/
package com.puppetlabs.geppetto.pp.dsl.ui.labeling;
import java.util.Iterator;
import com.puppetlabs.geppetto.pp.PPPackage;
import com.puppetlabs.geppetto.pp.dsl.linking.PPQualifiedNameConverter;
import com.puppetlabs.geppetto.pp.pptp.PPTPPackage;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IStorage;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.jface.viewers.StyledString;
import org.eclipse.xtext.resource.IEObjectDescription;
import org.eclipse.xtext.resource.IReferenceDescription;
import org.eclipse.xtext.resource.IResourceDescription;
import org.eclipse.xtext.ui.label.DeclarativeLabelProvider;
import org.eclipse.xtext.ui.label.DefaultEditorImageUtil;
import org.eclipse.xtext.ui.resource.IStorage2UriMapper;
import org.eclipse.xtext.util.Pair;
import com.google.inject.Inject;
/**
* Provides labels for a IEObjectDescriptions and IResourceDescriptions.
*
* see http://www.eclipse.org/Xtext/documentation/latest/xtext.html#labelProvider
*/
public class PPDescriptionLabelProvider extends DeclarativeLabelProvider implements IIconNames {
@Inject
PPQualifiedNameConverter nameConverter;
@Inject
private IStorage2UriMapper storage2UriMapper;
@Inject
private DefaultEditorImageUtil imageUtil;
public String getClassLabel(EClass clazz) {
if(clazz.getEPackage() == PPTPPackage.eINSTANCE) {
switch(clazz.getClassifierID()) {
case PPTPPackage.TYPE:
return "Type";
case PPTPPackage.TP_VARIABLE:
return "Variable";
case PPTPPackage.PARAMETER:
return "Parameter";
case PPTPPackage.PROPERTY:
return "Property";
case PPTPPackage.FUNCTION:
return "Function";
}
// Should not really ever be displayed
return "PPTP-class(" + clazz.getName() + ")";
}
else if(clazz.getEPackage() == PPPackage.eINSTANCE) {
switch(clazz.getClassifierID()) {
case PPPackage.HOST_CLASS_DEFINITION:
return "Class";
case PPPackage.DEFINITION:
return "Definition";
case PPPackage.VARIABLE_EXPRESSION:
return "Variable";
case PPPackage.DEFINITION_ARGUMENT:
return "Parameter";
default:
// Should not really ever be displayed
return "PP-class (" + clazz.getName() + ")";
}
}
return clazz.getName();
}
public Object image(IEObjectDescription element) {
if(element.getEClass().getEPackage() == PPTPPackage.eINSTANCE) {
return pptpImage(element);
}
else if(element.getEClass().getEPackage() == PPPackage.eINSTANCE) {
return ppImage(element);
}
return element.getQualifiedName() + " - " + element.getEClass().getName();
}
/**
* This method is only invoked if the containerEObjectURI of the {@link IReferenceDescription} is null, i.e. the
* reference is owned by an element without any indexed container.
*
* @since 2.1
*/
public Object image(IReferenceDescription element) {
URI uri = element.getSourceEObjectUri();
if(uri != null) {
String fileName = uri.lastSegment();
return imageUtil.getDefaultEditorImageDescriptor(fileName);
}
return null;
}
protected Object ppImage(IEObjectDescription element) {
switch(element.getEClass().getClassifierID()) {
case PPPackage.DEFINITION:
return DEFINITION;
case PPPackage.HOST_CLASS_DEFINITION:
return CLASS;
case PPPackage.VARIABLE_EXPRESSION:
return VARIABLE;
case PPPackage.NODE_DEFINITION:
return NODE;
case PPPackage.DEFINITION_ARGUMENT:
return PARAMETER;
}
return null;
}
protected Object pptpImage(IEObjectDescription element) {
switch(element.getEClass().getClassifierID()) {
case PPTPPackage.FUNCTION:
return FUNCTION;
case PPTPPackage.TYPE:
return TYPE;
case PPTPPackage.TP_VARIABLE:
return VARIABLE;
case PPTPPackage.PROPERTY:
case PPTPPackage.PARAMETER:
return PARAMETER;
}
return null;
}
public Object text(IEObjectDescription element) {
EPackage epkg = element.getEClass().getEPackage();
if(epkg == PPTPPackage.eINSTANCE || epkg == PPPackage.eINSTANCE) {
StyledString bld = new StyledString(nameConverter.toString(element.getName()));
bld.append(" : " + getClassLabel(element.getEClass()), StyledString.DECORATIONS_STYLER);
return bld;
}
return super.text(element);
}
public Object text(IReferenceDescription element) {
StyledString bld = new StyledString("::");
bld.append(" Top Level Scope", StyledString.DECORATIONS_STYLER);
return bld;
}
public Object text(IResourceDescription element) {
Iterator<Pair<IStorage, IProject>> storages = storage2UriMapper.getStorages(element.getURI()).iterator();
if(storages.hasNext()) {
IStorage storage = storages.next().getFirst();
return storage.getFullPath().toString();
}
return null;
}
}