SprayCompiler.java

package org.eclipselabs.spray.generator.graphiti.util;

import org.eclipse.xtext.common.types.JvmGenericType;
import org.eclipse.xtext.xbase.XAbstractFeatureCall;
import org.eclipse.xtext.xbase.XExpression;
import org.eclipse.xtext.xbase.compiler.IAppendable;
import org.eclipse.xtext.xbase.compiler.ImportManager;
import org.eclipse.xtext.xbase.compiler.StringBuilderBasedAppendable;
import org.eclipse.xtext.xbase.compiler.XbaseCompiler;
import org.eclipse.xtext.xbase.typing.ITypeProvider;
import org.eclipselabs.spray.mm.spray.Text;

import com.google.inject.Inject;

@SuppressWarnings("restriction")
public class SprayCompiler extends XbaseCompiler {
    @Inject
    private ITypeProvider typeProvider;
    
    private String metaClassVariable;
    
    public String compile(Text operation, ImportManager importManager) {
        StringBuilderBasedAppendable appendable = new StringBuilderBasedAppendable(importManager);
        IAppendable result = compile(operation.getValue(), appendable, typeProvider.getExpectedType(operation.getValue()));
        return result.toString();
    }

    @Override
    protected boolean isVariableDeclarationRequired(XExpression expr, IAppendable b) {
        if (expr instanceof XAbstractFeatureCall 
                && ((XAbstractFeatureCall)expr).getFeature() instanceof JvmGenericType) {
            return false;
        }
        return super.isVariableDeclarationRequired(expr,b);
    }

    @Override
    protected String getVarName(Object ex, IAppendable appendable) {
        if(ex instanceof JvmGenericType) {
            return metaClassVariable != null ? metaClassVariable : "this";
        }
        return super.getVarName(ex, appendable);
    }

    public String getMetaClassVariable() {
        return metaClassVariable;
    }

    public void setMetaClassVariable(String metaClassVariable) {
        this.metaClassVariable = metaClassVariable;
    }
    
    

}