Subversion Repositories SmartDukaan

package com.spice.profitmandi.common.util;

import java.io.File;

import java.io.IOException;

import java.io.PrintStream;

import java.util.List;

import org.apache.commons.csv.CSVFormat;

import org.apache.commons.csv.CSVPrinter;

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 com.spice.profitmandi.common.enumuration.ContentType;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

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;

		}

	}

}