package org.jboss.windup.config.operation.graph;
import org.jboss.windup.config.GraphRewrite;
import org.jboss.windup.config.Variables;
import org.jboss.windup.config.operation.GraphOperation;
import org.jboss.windup.config.operation.Iteration;
import org.jboss.windup.config.operation.iteration.AbstractIterationOperation;
import org.jboss.windup.graph.model.WindupVertexFrame;
import org.jboss.windup.graph.service.GraphService;
import org.ocpsoft.rewrite.context.EvaluationContext;
/**
* Adds the given type to a {@link WindupVertexFrame} in the graph.
*
* @author <a href="mailto:lincolnbaxter@gmail.com">Lincoln Baxter, III</a>
*
*/
public class TypeOperation extends AbstractIterationOperation<WindupVertexFrame>
{
private Class<? extends WindupVertexFrame> newType;
private TypeOperation(String variableName, Class<? extends WindupVertexFrame> newType)
{
super(variableName);
this.newType = newType;
}
private TypeOperation(Class<? extends WindupVertexFrame> newType)
{
super();
this.newType = newType;
}
public static GraphOperation addType(String variableName, Class<? extends WindupVertexFrame> newType)
{
return new TypeOperation(variableName, newType);
}
/**
* Add type to the variable being iterated with the default name.
* @param newType Type to be added
* @return
*/
public static GraphOperation addType(Class<? extends WindupVertexFrame> newType)
{
return new TypeOperation(newType);
}
@Override
public void perform(GraphRewrite event, EvaluationContext context, WindupVertexFrame payload)
{
Variables varStack = Variables.instance(event);
WindupVertexFrame newFrame = GraphService.addTypeToModel(event.getGraphContext(), payload, newType);
Iteration.setCurrentPayload(varStack, getVariableName(), newFrame);
}
}