cr.ac.siua.tec.utils.impl.CEInclusionPDFGenerator.java Source code

Introduction

Here is the source code for cr.ac.siua.tec.utils.impl.CEInclusionPDFGenerator.java

Source

/*
   TRI-S - Web Service
   Developed by: Luis E. Ugalde Barrantes - Diego Ugalde ?vila. 2016.
   This code is licensed under the GNU GENERAL PUBLIC LICENSE (GPL) V3. See LICENSE file for details.
*/

package cr.ac.siua.tec.utils.impl;

import cr.ac.siua.tec.utils.PDFGenerator;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDDocumentCatalog;
import org.apache.pdfbox.pdmodel.interactive.form.PDAcroForm;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.Calendar;
import java.util.HashMap;
import java.util.Map;

@Component("CEInclusion")
public class CEInclusionPDFGenerator extends PDFGenerator {

    /**
     * Fills the PDF file (inclusion.pdf) with the ticket values and returns base64 encoded string.
     */
    @Override
    public String generate(HashMap<String, String> formValues) {
        String originalPdf = PDFGenerator.RESOURCES_PATH + "inclusion.pdf";
        try {
            PDDocument _pdfDocument = PDDocument.load(originalPdf);
            PDDocumentCatalog docCatalog = _pdfDocument.getDocumentCatalog();
            PDAcroForm acroForm = docCatalog.getAcroForm();
            formValues.remove("Queue");
            formValues.remove("Justificacin");

            //Set some fields manually.
            Calendar cal = Calendar.getInstance();
            int year = cal.get(Calendar.YEAR);
            int month = cal.get(Calendar.MONTH) + 1;
            if (month > 10 || month < 5)
                acroForm.getField("Semestre").setValue("I");
            else
                acroForm.getField("Semestre").setValue("II");
            if (month > 10 && acroForm.getField("Semestre").getValue().equals("I"))
                year++;
            acroForm.getField("Ao").setValue(String.valueOf(year));

            //Fills enrolled courses table.
            String enrolledCourses[] = formValues.get("Cursos matriculados").split("\n");
            for (int i = 0; i < enrolledCourses.length; i++) {
                String courseRow[] = enrolledCourses[i].split("-");
                acroForm.getField("Cdigo" + Integer.toString(i)).setValue(courseRow[0]);
                acroForm.getField("Curso" + Integer.toString(i)).setValue(courseRow[1]);
                acroForm.getField("Grupo" + Integer.toString(i)).setValue(courseRow[2]);
            }
            formValues.remove("Cursos matriculados");

            //Iterates through remaining custom fields.
            for (Map.Entry<String, String> entry : formValues.entrySet()) {
                acroForm.getField(entry.getKey()).setValue(entry.getValue());
            }

            return encodePDF(_pdfDocument);
        } catch (IOException e) {
            e.printStackTrace();
            System.out.println("Excepcin al llenar el PDF.");
            return null;
        }
    }
}