Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.util;

import com.spice.profitmandi.common.enumuration.ContentType;
import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVPrinter;
import org.apache.commons.csv.CSVRecord;
import org.apache.commons.io.output.ByteArrayOutputStream;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.apache.tika.Tika;
import org.springframework.web.multipart.MultipartFile;

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

public class FileUtil {

        private static final Logger LOGGER = LogManager.getLogger(FileUtil.class);

        public static ContentType detectFileType(File file) throws ProfitMandiBusinessException {
                Tika tika = new Tika();

                // detecting the file type using detect method
                String filetype;
                try {
                        filetype = tika.detect(file);
                } catch (IOException e) {
                        e.printStackTrace();
                        throw new ProfitMandiBusinessException("Content-Type", file.getName(), "DCMNT_1001");
                }
                if (filetype.contains("pdf")) {
                        return ContentType.PDF;
                } else if (filetype.contains("jpg") | filetype.contains("jpeg")) {
                        return ContentType.JPEG;
                } else if (filetype.contains("png")) {
                        return ContentType.PNG;
                } else {
                        throw new ProfitMandiBusinessException("Content-Type", file.getName(), "DCMNT_1002");
                }
        }

        public static void main(String[] args) {
                // detectFileType(new File("/hsps-docs/1499163441532"));
                // System.out.println("he");
        }

        public static ByteArrayOutputStream getCSVByteStream(List<String> headers, List<List<?>> rows) throws Exception {
                try (ByteArrayOutputStream baos = new ByteArrayOutputStream();
                                PrintStream writer = new PrintStream(baos);
                                CSVPrinter csvPrinter = new CSVPrinter(writer,
                                                CSVFormat.EXCEL.withTrim().withHeader(headers.toArray(new String[] {})));) {
                        for (List<?> row : rows) {
                                csvPrinter.printRecord(row);
                        }
                        return baos;
                }
        }

        public static ByteArrayOutputStream getCSVByteStreamWithMultiHeaders(List<List<String>> headers, List<List<?>> rows) throws Exception {
                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                try (PrintStream writer = new PrintStream(baos);
                         CSVPrinter csvPrinter = new CSVPrinter(writer, CSVFormat.EXCEL.withTrim())) {

                        // Print all header rows manually
                        for (List<String> headerRow : headers) {
                                csvPrinter.printRecord(headerRow);
                        }

                        // Print all data rows
                        for (List<?> row : rows) {
                                csvPrinter.printRecord(row);
                        }
                }
                return baos;
        }


        public static List<CSVRecord> readFile(MultipartFile file) throws IOException, ProfitMandiBusinessException {

                CSVParser parser = new CSVParser(new InputStreamReader(file.getInputStream()), CSVFormat.DEFAULT);
                List<CSVRecord> records = parser.getRecords();
                LOGGER.info("records" + records);
                LOGGER.info("parser" + parser);
                if (records.size() < 2) {
                        parser.close();
                        throw new ProfitMandiBusinessException("Uploaded File", "", "No records Found");
                }
                // Remove header
                records.remove(0);
                parser.close();
                return records;

        }

        public static List<CSVRecord> readFile(File file) throws IOException, ProfitMandiBusinessException {
                FileReader filereader = new FileReader(file);
                CSVParser parser = new CSVParser(filereader, CSVFormat.DEFAULT);
                List<CSVRecord> records = parser.getRecords();
                LOGGER.info("records" + records);
                LOGGER.info("parser" + parser);
                if (records.size() < 2) {
                        parser.close();
                        throw new ProfitMandiBusinessException("Uploaded File", "", "No records Found");
                }
                // Remove header
                records.remove(0);
                parser.close();
                return records;

        }
}