hu.blackbelt.epsilon.runtime.execution.contexts.EclExecutionContext Maven / Gradle / Ivy

Go to download
package hu.blackbelt.epsilon.runtime.execution.contexts;

import hu.blackbelt.epsilon.runtime.execution.exceptions.ScriptExecutionException;
import lombok.Builder;
import lombok.Getter;
import lombok.NonNull;
import org.eclipse.epsilon.ecl.EclModule;
import org.eclipse.epsilon.ecl.trace.MatchTrace;
import org.eclipse.epsilon.eol.IEolModule;

import java.net.URI;
import java.util.List;
import java.util.Map;

public class EclExecutionContext extends EolExecutionContext {

    @Getter
    @NonNull
    private String exportMatchTrace;

    @Getter
    @NonNull
    private String useMatchTrace;

    @Builder.Default
    private EclModule eclModule = new EclModule();

    @Builder(builderMethodName = "eclExecutionContextBuilder")
    public EclExecutionContext(URI source, List parameters, String useMatchTrace, String exportMatchTrace, EclModule eclModule) {
        super(source, parameters);
        this.useMatchTrace = useMatchTrace;
        this.exportMatchTrace = exportMatchTrace;
        if (eclModule != null) {
            this.eclModule = eclModule;
        }
    }

    @Override
    public IEolModule getModule(Map context) throws ScriptExecutionException {
        if (useMatchTrace != null) {
            if (context.get(useMatchTrace) != null) {
                eclModule.getContext().setMatchTrace((MatchTrace)context.get(useMatchTrace));
            } else {
                eclModule.getContext().setMatchTrace(new MatchTrace());
            }
        }
        return eclModule;
    };
    
    @Override
    public void post(Map context) {
         if (exportMatchTrace != null) {
            context.put(
                exportMatchTrace,
                eclModule.getContext().getMatchTrace().getReduced());
        }
    }
}