org.jnario.feature.linking.FeatureLazyLinker Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.jnario.standalone Show documentation
The required libraries to execute Jnario specifications without Eclipse.
The newest version!
/*******************************************************************************
 * Copyright (c) 2012 BMW Car IT and others.
 * 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
 *******************************************************************************/
package org.jnario.feature.linking;

import java.util.List;

import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.xtext.diagnostics.IDiagnosticConsumer;
import org.eclipse.xtext.util.OnChangeEvictingCache;
import org.eclipse.xtext.util.concurrent.IUnitOfWork;
import org.eclipse.xtext.xbase.XBlockExpression;
import org.eclipse.xtext.xbase.XConstructorCall;
import org.eclipse.xtext.xbase.XExpression;
import org.eclipse.xtext.xbase.XStringLiteral;
import org.eclipse.xtext.xbase.XVariableDeclaration;
import org.eclipse.xtext.xbase.XbaseFactory;
import org.jnario.feature.feature.FeatureFile;
import org.jnario.feature.feature.Step;
import org.jnario.feature.feature.StepImplementation;
import org.jnario.feature.jvmmodel.FeatureJvmModelInferrer;
import org.jnario.feature.jvmmodel.StepArgumentsProvider;
import org.jnario.linking.JnarioLazyLinker;

import com.google.inject.Inject;

/**
 * @author Birgit Engelmann - Initial contribution and API
 * @author Sebastian Benz
 */
public class FeatureLazyLinker extends JnarioLazyLinker {
	
	@Inject StepArgumentsProvider stepArgumentsProvider;
	@Inject FeatureQuery featureQuery;
	@Inject OnChangeEvictingCache cache;
	
	@Override
	protected void beforeModelLinked(final EObject model,
			IDiagnosticConsumer diagnosticsConsumer) {
		super.beforeModelLinked(model, diagnosticsConsumer);
		cache.execWithoutCacheClear(model.eResource(), new IUnitOfWork.Void() {
			@Override
			public void process(Resource state) throws Exception {
				generateStepValues(model);
			}
		});
	}

	private void generateStepValues(EObject model) {
		if (!(model instanceof FeatureFile)) {
			return;
		}
		FeatureFile featureFile = (FeatureFile) model;
		generateArguments(featureQuery.allSteps(featureFile));		
	}
	
	private void generateArguments(Iterable steps) {
		for (Step step : steps) {
			if (step instanceof StepImplementation) {
				generateArguments(step);
			}else{
				step.setExpression(null);
			}
		}
	}

	private void generateArguments(Step step) {
		List arguments = stepArgumentsProvider.findStepArguments(step);
		if(arguments.isEmpty()){
			return;
		}
		XVariableDeclaration stepValuesDec = createVariableForStepArguments(arguments);
		EList expressions = ((XBlockExpression)step.getExpression()).getExpressions();
		if(argumentsVariableAlreadyDeclared(expressions)){
			return;
		}
		expressions.add(0, stepValuesDec);
	}

	public boolean argumentsVariableAlreadyDeclared(EList expressions) {
		if (expressions.get(0) instanceof XVariableDeclaration) {
			XVariableDeclaration variableDeclaration = (XVariableDeclaration) expressions.get(0);
			if(FeatureJvmModelInferrer.STEP_VALUES.equals(variableDeclaration.getName())){
				return true;
			}
		}
		return false;
	}
	
	private XVariableDeclaration createVariableForStepArguments(List arguments){
		XVariableDeclaration variableDec = XbaseFactory.eINSTANCE.createXVariableDeclaration();
		variableDec.setName(FeatureJvmModelInferrer.STEP_VALUES);
		XConstructorCall constructor = XbaseFactory.eINSTANCE.createXConstructorCall();
		for (String arg : arguments) {
			XStringLiteral stringLiteral = XbaseFactory.eINSTANCE.createXStringLiteral();
			stringLiteral.setValue(arg);
			constructor.getArguments().add(stringLiteral);
		}
		variableDec.setRight(constructor);
		return variableDec;
	}

}