com.castlemon.maven.output.impl.HTMLOutput Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of maven-usage Show documentation
Examines a Maven repo to determine where a specified project is used
The newest version!
package com.castlemon.maven.output.impl;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.castlemon.maven.domain.RunData;
import com.castlemon.maven.output.Output;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

@Component
public class HTMLOutput implements Output {

    private static final Logger LOGGER = LoggerFactory.getLogger(HTMLOutput.class);

    public void writeData(RunData runData) {
        File checkDir = new File(runData.getOutputDirectory());
        if (!checkDir.exists()) {
            LOGGER.error("Output directory does not exist: " + runData.getOutputDirectory());
            return;
        }
        try {
            setupResources(runData.getOutputDirectory());
            Configuration cfg = new Configuration(Configuration.VERSION_2_3_23);
            cfg.setClassForTemplateLoading(this.getClass(), "/templates");
            cfg.setDefaultEncoding("UTF-8");
            Map context = new HashMap();
            context.put("runData", runData);
            Template template = cfg.getTemplate("usagereport.ftl");
            File file = new File(runData.getOutputDirectory() + File.separator + "usagereport.html");
            Writer out = new OutputStreamWriter(new FileOutputStream(file), StandardCharsets.UTF_8);
            template.process(context, out);
        } catch (IOException e) {
            LOGGER.error("I/O Error when writing HTML file", e);
        } catch (TemplateException e) {
            LOGGER.error("Template error when writing HTML file", e);
        }
    }

    private void setupResources(String outputDir) throws IOException {
        setupCSSResources(outputDir);
        setupJSResources(outputDir);
        setupFontResources(outputDir);
    }

    private void setupCSSResources(String outputDir) throws IOException {
        File cssDir = makeDir("css", outputDir);
        copyFile(cssDir, "bootstrap.min.css", "css");
        copyFile(cssDir, "dataTables.bootstrap.min.css", "css");
    }

    private void setupJSResources(String outputDir) throws IOException {
        File jsDir = makeDir("js", outputDir);
        copyFile(jsDir, "dataTables.bootstrap.min.js", "js");
        copyFile(jsDir, "jquery-1.12.0.min.js", "js");
        copyFile(jsDir, "jquery.dataTables.min.js", "js");
    }

    private void setupFontResources(String outputDir) throws IOException {
        File jsDir = makeDir("fonts", outputDir);
        copyFile(jsDir, "glyphicons-halflings-regular.woff2", "fonts");
        copyFile(jsDir, "glyphicons-halflings-regular.woff", "fonts");
        copyFile(jsDir, "glyphicons-halflings-regular.ttf", "fonts");
    }

    private void copyFile(File cssDir, String fileName, String resourceDir) throws IOException {
        InputStream resourceArchiveInputStream = null;
        FileOutputStream outStream = null;
        String filePath = resourceDir + "/" + fileName;
        try {
            resourceArchiveInputStream = HTMLOutput.class.getResourceAsStream(filePath);
            if (resourceArchiveInputStream == null) {
                resourceArchiveInputStream = HTMLOutput.class.getResourceAsStream("/" + filePath);
            }
            File file = new File(cssDir, fileName);
            outStream = new FileOutputStream(file);
            IOUtils.copy(resourceArchiveInputStream, outStream);
        } finally {
            IOUtils.closeQuietly(resourceArchiveInputStream);
            IOUtils.closeQuietly(outStream);
        }
    }

    private File makeDir(String dirName, String outputDir) {
        File newDir = new File(outputDir + File.separator + dirName);
        newDir.mkdir();
        return newDir;
    }

}