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);
    }
}
