dev.galasa.artifact.internal.VelocitySkeletonProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dev.galasa.artifact.manager Show documentation

A Galasa project

There is a newer version: 0.38.0

Show newest version

/*
 * Copyright contributors to the Galasa project
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package dev.galasa.artifact.internal;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.SequenceInputStream;
import java.util.Map;
import java.util.Map.Entry;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.velocity.VelocityContext;
import org.apache.velocity.app.Velocity;

import dev.galasa.artifact.ISkeletonProcessor;
import dev.galasa.artifact.SkeletonProcessorException;
import dev.galasa.framework.spi.IFramework;

public class VelocitySkeletonProcessor implements ISkeletonProcessor {

    private static final Log logger = LogFactory.getLog(VelocitySkeletonProcessor.class);

    public VelocitySkeletonProcessor(IFramework framework) {
    }

    @Override
    public InputStream processSkeleton(InputStream skeleton, Map parameters)
            throws SkeletonProcessorException {

        return processVelocitySkeleton(skeleton, parameters);
    }

    private InputStream processVelocitySkeleton(InputStream skeleton, Map parameters)
            throws SkeletonProcessorException {

        logger.info("Processing skeleton with Velocity");

        InputStream safeEOF = new ByteArrayInputStream(" ".getBytes());
        InputStream streamPlus = new SequenceInputStream(skeleton, safeEOF);
        InputStreamReader ir = new InputStreamReader(streamPlus);

        try {
            Velocity.init();
        } catch (Exception e) {
            throw new SkeletonProcessorException("Error attempting to initialise velocity", e);
        }

        VelocityContext context = new VelocityContext();

        // Supplied parameters will override our defaults
        for (Entry entry : parameters.entrySet()) {
            context.put(entry.getKey(), entry.getValue());
        }

        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        OutputStreamWriter ow = new OutputStreamWriter(baos);

        try {
            Velocity.evaluate(context, ow, "VelocityRenderer", ir);
            ow.close();
        } catch (Exception e) {
            throw new SkeletonProcessorException("Error attempting to process skeleton with velocity", e);
        }

        return new ByteArrayInputStream(baos.toByteArray());
    }
}