aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/ExportManagerImpl.java @ d90fc9de

package vldc.aswi.service;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.utils.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import vldc.aswi.domain.Configuration;
import vldc.aswi.model.ExportType;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
import vldc.aswi.utils.ExporterPDF;
import vldc.aswi.utils.ExporterXLSX;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.List;

/**
 * Manager for Export.
 */
@Service
@Slf4j
public class ExportManagerImpl implements ExportManager {

    /** Autowired sql query manager. */
    @Autowired
    private SqlQueryManager sqlQueryManager;

    /**
     * Export contingency table to specific export type.
     * @param newConfiguration - Exported configuration.
     * @param isNotRemoveEmpty - If blank rows / columns must be inherit or not.
     * @param response - Http response.
     * @param type - Export type.
     */
    @Override
    public void exportContingencyTable(Configuration newConfiguration, boolean isNotRemoveEmpty, HttpServletResponse response, ExportType type) {

        // Get contingency table rows.
        List<ContingencyTableRow> contingencyTableRows = this.sqlQueryManager.getContingencyTableRow(newConfiguration.getAssembly(),
                newConfiguration.getParametersInConfiguration(), isNotRemoveEmpty);

        try {
            File tmpFile = null;

            switch(type) {

                case PDF:

                    // Create temporary file.
                    tmpFile = File.createTempFile("outPDF", ".pdf");

                    // Export contingency table rows to pdf format.
                    ExporterPDF.export(tmpFile, contingencyTableRows);

                    // Set content type to pdf.
                    response.setContentType("application/pdf");
                    break;
                case XLSX:

                    // Create temporary file.
                    tmpFile = File.createTempFile("outXLSX", ".xlsx");

                    // Export contingency table rows to xlsx format.
                    ExporterXLSX.export(tmpFile, newConfiguration.getTableName(), contingencyTableRows);

                    // Set content type to xlsx.
                    response.setContentType("application/xlsx");
                    break;
            }

            // Force "Save As..." dialog.
            response.setHeader("Content-Disposition", "attachment; filename=" + tmpFile + "");
            response.setContentLength((int)tmpFile.length());

            // Copy temporary file content to response.
            InputStream in = new BufferedInputStream(new FileInputStream(tmpFile));
            ServletOutputStream out = response.getOutputStream();

            IOUtils.copy(in, out);
            in.close();

            out.flush();
            out.close();

            // Delete temporary file.
            tmpFile.delete();

            System.out.println("pdf successfully exported");
            // TODO: print success message - file created.

        } catch (Exception e) {
            System.out.println("pdf export FAILED 1");
            // TODO: print error message - file not created.
        }
    }
}