com.h3xstream.retirejs.vuln.TemplateBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of retirejs-core Show documentation

There is a newer version: 3.0.4

package com.h3xstream.retirejs.vuln;

import com.esotericsoftware.minlog.Log;
import com.github.mustachejava.DefaultMustacheFactory;
import com.github.mustachejava.Mustache;
import com.github.mustachejava.MustacheFactory;

import java.io.*;
import java.util.List;

public class TemplateBuilder {

    public static String buildDescription(String templateFile, String detectedLibrary,String detectedVersion,List urls,String aboveVersion,String belowVersion) {
        InputStream tpl = TemplateBuilder.class.getResourceAsStream(templateFile);

        if(tpl == null) return errorResult(null);

        try {
            //Build the model mapped to the template
            DescriptionModel model = new DescriptionModel();
            model.setDetectedLibrary(detectedLibrary);
            model.setDetectedVersion(detectedVersion);
            model.getLinks().addAll(urls);
            model.setAboveVersion(aboveVersion == null || "".equals(aboveVersion) ? "*" : aboveVersion);
            model.setBelowVersion(belowVersion);

            //Compile the template
            MustacheFactory mf = new DefaultMustacheFactory();
            Mustache mustache = mf.compile(new InputStreamReader(tpl), "");

            //OutputStream to String
            ByteArrayOutputStream outBuffer = new ByteArrayOutputStream();
            mustache.execute(new PrintWriter(outBuffer), model).flush();
            return outBuffer.toString();
        } catch (IOException e) {
           return errorResult(e);
        }
    }

    private static String errorResult(Exception e) {
        Log.error("Unable to generate the description." + (e != null? e.getMessage():""));
        return "An error occurs while loading description template.";
    }
}