Reporter

/*
 * Licensed to the Apache Software Foundation (ASF) under one   *
 * or more contributor license agreements.  See the NOTICE file *
 * distributed with this work for additional information        *
 * regarding copyright ownership.  The ASF licenses this file   *
 * to you under the Apache License, Version 2.0 (the            *
 * "License"); you may not use this file except in compliance   *
 * with the License.  You may obtain a copy of the License at   *
 *                                                              *
 *   http://www.apache.org/licenses/LICENSE-2.0                 *
 *                                                              *
 * Unless required by applicable law or agreed to in writing,   *
 * software distributed under the License is distributed on an  *
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY       *
 * KIND, either express or implied.  See the License for the    *
 * specific language governing permissions and limitations      *
 * under the License.                                           *
 */
package org.apache.rat;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.apache.commons.io.function.IOSupplier;
import org.apache.rat.api.RatException;
import org.apache.rat.license.LicenseSetFactory.LicenseFilter;
import org.apache.rat.report.RatReport;
import org.apache.rat.report.claim.ClaimStatistic;
import org.apache.rat.report.xml.XmlReportFactory;
import org.apache.rat.report.xml.writer.IXmlWriter;
import org.apache.rat.report.xml.writer.impl.base.XmlWriter;
import org.w3c.dom.Document;

/**
 * Class that executes the report as defined in a ReportConfiguration and stores
 * the result for later handleing.
 */
public class Reporter {

    /**
     * Format used for listing license families
     */
    private static final String LICENSE_FAMILY_FORMAT = "\t%s: %s%n";

    /**
     * Format used for listing licenses.
     */
    private static final String LICENSE_FORMAT = "%s:\t%s%n\t\t%s%n";

    private final Document document;
    private final ClaimStatistic statistic;
    private final ReportConfiguration configuration;

    /**
     * Create the reporter.
     * @param configuration the configuration to use.
     * @throws RatException on error.
     */
    public Reporter(ReportConfiguration configuration) throws RatException {
        this.configuration = configuration;
        try {
            if (configuration.getReportable() != null) {
                ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
                Writer outputWriter = new OutputStreamWriter(outputStream, StandardCharsets.UTF_8);
                try (IXmlWriter writer = new XmlWriter(outputWriter)) {
                    statistic = new ClaimStatistic();
                    RatReport report = XmlReportFactory.createStandardReport(writer, statistic, configuration);
                    report.startReport();
                    configuration.getReportable().run(report);
                    report.endReport();

                    InputStream inputStream = new ByteArrayInputStream(outputStream.toByteArray());
                    document = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(inputStream);
                }
            } else {
                document = DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument();
                statistic = new ClaimStatistic();
            }
        } catch (Exception e) {
            throw RatException.makeInstance(e);
        }
    }

    /**
     * Get the claim statistics from the run.
     * @return the claim statistics.
     */
    public ClaimStatistic getClaimsStatistic() {
        return statistic;
    }

    /**
     * Outputs the report using the stylesheet and output specified in the configuraiton.
     * @throws RatException on error.
     */
    public void output() throws RatException {
        if (configuration.isStyleReport()) {
            output(configuration.getStyleSheet(), configuration.getOutput());
        } else {
            output(null, configuration.getOutput());
        }
    }

    /**
     * Outputs the report to the specified output useing the optional stylesheet.
     * @param stylesheet the style sheet to use for XSLT formatting, may be null for XML output.
     * @param output the output stream to write to.
     * @throws RatException one error.
     */
    public void output(IOSupplier<InputStream> stylesheet, IOSupplier<OutputStream> output) throws RatException {
        InputStream styleIn = null;
        TransformerFactory tf = TransformerFactory.newInstance();
        Transformer transformer;
        try (OutputStream out=output.get()){
            if (stylesheet != null) {
                styleIn = stylesheet.get();
                transformer = tf.newTransformer(new StreamSource(styleIn));
            } else {
                transformer = tf.newTransformer();
                transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "no");
                transformer.setOutputProperty(OutputKeys.METHOD, "xml");
                transformer.setOutputProperty(OutputKeys.INDENT, "yes");
                transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
                transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
            }

            transformer.transform(new DOMSource(document),
                    new StreamResult(new OutputStreamWriter(out, "UTF-8")));
        } catch (TransformerException | IOException e) {
            throw new RatException(e);
        } finally {
            if (styleIn != null) {
                try {
                    styleIn.close();
                } catch (IOException e) {
                    configuration.getLog().error("Error closing stylesheet", e);
                }
            }
        }
    }

    /**
     * lists the license families information on the configured output stream.
     *
     * @param configuration The configuration for the system
     * @throws IOException if PrintWriter can not be retrieved from configuration.
     */
    public static void listLicenseFamilies(ReportConfiguration configuration, LicenseFilter filter) throws IOException {
        try (PrintWriter pw = configuration.getWriter().get()) {
            pw.format("Families (%s):%n", filter);
            configuration.getLicenseFamilies(filter)
                    .forEach(x -> pw.format(LICENSE_FAMILY_FORMAT, x.getFamilyCategory(), x.getFamilyName()));
            pw.println();
        }
    }

    /**
     * lists the licenses on the configured output stream.
     *
     * @param configuration The configuration for the system
     * @throws IOException if PrintWriter can not be retrieved from configuration.
     */
    public static void listLicenses(ReportConfiguration configuration, LicenseFilter filter) throws IOException {
        try (PrintWriter pw = configuration.getWriter().get()) {
            pw.format("Licenses (%s):%n", filter);
            configuration.getLicenses(filter)
                    .forEach(lic -> pw.format(LICENSE_FORMAT, lic.getLicenseFamily().getFamilyCategory(),
                            lic.getLicenseFamily().getFamilyName(), lic.getNote()));
            pw.println();
        }
    }

}