package org.springblade.mdm.commons.support;

import com.itextpdf.text.DocumentException;
import com.itextpdf.text.pdf.AcroFields;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;

import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Map;

/**
 * 模板生成pdf
 */
public class TemplatedPdfWriter {
	private byte[] templateBytes;

	public TemplatedPdfWriter(byte[] templateBytes){
		this.templateBytes = templateBytes;
	}

	public void setTemplateBytes(byte[] templateBytes){
		this.templateBytes = templateBytes;
	}
	public void write(Map<String,String> data, OutputStream outputStream) throws DocumentException, IOException {

		PdfReader reader = null;
		PdfStamper pdfStamper = null;

		reader = new PdfReader(this.templateBytes);
		pdfStamper =  new PdfStamper(reader, outputStream);
		// 使用中文字体
		BaseFont bf = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
		ArrayList<BaseFont> fontList = new ArrayList<>();
		fontList.add(bf);
		AcroFields fields = pdfStamper.getAcroFields();
		fields.setSubstitutionFonts(fontList);
		fillData(fields, data);//染
		//必须要调用这个，否则文档不会生成的
		pdfStamper.setFormFlattening(true);
		pdfStamper.close();
	}

	void fillData(AcroFields fields, Map<String, String> data) throws DocumentException, IOException {

		for (String key : data.keySet()) {
			String value = data.get(key);
			// 为字段赋值,注意字段名称是区分大小写的fields.setField(key,value);
			fields.setField(key,value,true);

			//form.setFieldProperty("fieldName", "clrflags", PdfAnnotation.FLAGS_HIDDEN, null);
			//fields.setFieldRichValue()
		}

	}

}