Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.utils;

import in.shop2020.dtrapi.Storage.Mongo;

import java.io.OutputStream;

import java.lang.reflect.Type;

import java.util.HashMap;

import java.util.Map;

import com.google.gson.Gson;

import com.google.gson.reflect.TypeToken;

import com.itextpdf.text.Document;

import com.itextpdf.text.Element;

import com.itextpdf.text.Font;

import com.itextpdf.text.Image;

import com.itextpdf.text.Paragraph;

import com.itextpdf.text.pdf.PdfContentByte;

import com.itextpdf.text.pdf.PdfImportedPage;

import com.itextpdf.text.pdf.PdfPCell;

import com.itextpdf.text.pdf.PdfPTable;

import com.itextpdf.text.pdf.PdfReader;

import com.itextpdf.text.pdf.PdfWriter;

public class PdfGenerator {

	private static final Map<String, String> titles = new HashMap<String, String>();

	static {

		titles.put("bEntityDoc", "Business Document");

		titles.put("gstDoc", "GST Doc");

		titles.put("panDoc", "Pan Doc");

		titles.put("itrDoc", "ITR DOC");

		titles.put("angleDoc1", "Angle Doc1");

		titles.put("angleDoc2", "Angle Doc2");

		titles.put("angleDoc3", "Angle Doc3");

		titles.put("angleDoc4", "Angle Doc4");

		titles.put("angleDoc5", "Angle Doc5");

		titles.put("ownershipDoc", "OwnerShip Doc");

		titles.put("chequeCopy", "Cheque");

		titles.put("insuranceDoc", "Insurance Doc");

		titles.put("loanDoc", "Loan Doc");

		titles.put("sanctionDoc", "Sanction Doc");

	}

	private static final Font FONT_TITLE = new Font(Font.FontFamily.HELVETICA, 18, Font.BOLD);

	private static Font FONT_BOLD = new Font(Font.FontFamily.TIMES_ROMAN, 10, Font.BOLD);

	public static void generateAndWrite(OutputStream outputStream, int fofoId) throws Exception {

		// String responseString = HttpClientUtil.doGet("localhost:8080",

		// fofoId);

		Document document = new Document();

		PdfWriter pdfWriter = PdfWriter.getInstance(document, outputStream);

		document.open();

		PdfContentByte cb = pdfWriter.getDirectContent();

		String jsonString = Mongo.getFofoFormJsonStringByFofoId(fofoId);

		Map<String, String> titlePathMap = new HashMap<String, String>();

		Type stringStringMap = new TypeToken<Map<String, String>>() {

		}.getType();

		Gson gson = new Gson();

		Map<String, String> map = gson.fromJson(jsonString, stringStringMap);

		Paragraph paragraphTitle = new Paragraph(map.get("registeredBusinessName"), FONT_TITLE);

		paragraphTitle.setAlignment(Element.ALIGN_CENTER);

		document.add(paragraphTitle);

		System.out.println(map);

		for (String key : map.keySet()) {

			String value = map.get(key).toString();

			if (value.startsWith("/hsps-docs/")) {

				titlePathMap.put(titles.get(key), value);

			}

		}

		PdfPTable table = new PdfPTable(1);

		PdfPCell title;

		PdfPTable innerTable;

		for (Map.Entry<String, String> entry : titlePathMap.entrySet()) {

			Image image = null;

			try {

				image = Image.getInstance(entry.getValue());

			} catch (Exception e) {

				e.printStackTrace();

				try {

					PdfReader reader = new PdfReader(entry.getValue());

					for (int i = 1; i <= reader.getNumberOfPages(); i++) {

						PdfImportedPage page = pdfWriter.getImportedPage(reader, i);

						cb.addTemplate(page, 0, 0);

						document.newPage();

					}

				} catch (Exception ex) {

					ex.printStackTrace();

					document.add(new Paragraph(entry.getKey() + " is missing or password encrypted."));

				}

				continue;

			}

			innerTable = new PdfPTable(1);

			float scaler = (Math.min((document.getPageSize().getWidth()) / image.getWidth(), 1)) * 100;

			title = new PdfPCell(new Paragraph(entry.getKey(), FONT_BOLD));

			title.setHorizontalAlignment(Element.ALIGN_CENTER);

			innerTable.addCell(title);

			image.scalePercent(scaler);

			innerTable.addCell(image);

			table.addCell(innerTable);

		}

		document.add(table);

		document.close();

	}

}