Load the font files found under resources/pdf-fonts

Co-authored-by: Alex Gonzalez <[email protected]> Use them as substitution fonts
codeforamerica · Nov 28, 2023 · 16b6409 · 16b6409
1 parent eb1990c
commit 16b6409
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 3 deletions.
diff --git a/src/main/java/formflow/library/pdf/PDFFormFiller.java b/src/main/java/formflow/library/pdf/PDFFormFiller.java
@@ -5,12 +5,18 @@
 import com.lowagie.text.pdf.PdfReader;
 import com.lowagie.text.pdf.PdfStamper;
 import lombok.extern.slf4j.Slf4j;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.core.io.ClassPathResource;
 import org.springframework.stereotype.Component;
 
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.net.URI;
 import java.util.Collection;
+import java.util.Objects;
+import java.util.Set;
+import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
 @Slf4j
@@ -29,7 +35,7 @@ public File fill(String pdfTemplatePath, Collection<PdfField> fields, boolean fl
       File outputFile = File.createTempFile("Filled_", ".pdf");
       PdfStamper pdfStamper = new PdfStamper(reader, new FileOutputStream(outputFile));
       AcroFields acroFields = pdfStamper.getAcroFields();
-      configureSubstituteFonts(acroFields);
+      configureSubstitutionFonts(acroFields);
       for (PdfField pdfField : fields) {
         acroFields.setField(pdfField.name(), pdfField.value());
       }
@@ -57,8 +63,9 @@ public File fill(String pdfTemplatePath, Collection<PdfField> fields) {
    * Configure the fonts to be used if the default fonts do not support the characters in the values to be filled
    * @param acroFields the `AcroFields` object (the form) in the PDF template
    */
-  private static void configureSubstituteFonts(AcroFields acroFields) {
-    Stream.of("NotoSans-Regular.ttf", "NotoSansSC-Regular.ttf").forEach( fontResource -> {
+  private static void configureSubstitutionFonts(AcroFields acroFields) {
+    Set<String> files = getFontFiles();
+    files.forEach(fontResource -> {
       BaseFont font = null;
       try {
         font = BaseFont.createFont(fontResource, BaseFont.IDENTITY_H, BaseFont.EMBEDDED);
@@ -68,4 +75,23 @@ private static void configureSubstituteFonts(AcroFields acroFields) {
       acroFields.addSubstitutionFont(font);
     });
   }
+
+  /**
+   * Get the list of font files to be used as substitution fonts from resources/pdf-fonts
+   * @return Set of filenames with paths
+   */
+  @NotNull
+  private static Set<String> getFontFiles() {
+    URI directoryURI = null;
+    try {
+      directoryURI = new ClassPathResource("pdf-fonts").getURI();
+    } catch (IOException e) {
+      log.warn("Failed to find pdf-fonts directory: %s", e);
+    }
+
+    return Stream.of(Objects.requireNonNull(new File(directoryURI).listFiles()))
+            .filter(file -> !file.isDirectory())
+            .map(File::getPath)
+            .collect(Collectors.toSet());
+  }
 }
diff --git a/src/main/resources/NotoSans-Regular.ttf → .../resources/pdf-fonts/NotoSans-Regular.ttf b/src/main/resources/NotoSans-Regular.ttf → .../resources/pdf-fonts/NotoSans-Regular.ttf
diff --git a/src/main/resources/NotoSansSC-Regular.ttf → ...esources/pdf-fonts/NotoSansSC-Regular.ttf b/src/main/resources/NotoSansSC-Regular.ttf → ...esources/pdf-fonts/NotoSansSC-Regular.ttf
diff --git a/src/test/java/formflow/library/pdf/PDFFormFillerTest.java b/src/test/java/formflow/library/pdf/PDFFormFillerTest.java
@@ -4,6 +4,8 @@
 import com.lowagie.text.pdf.PdfReader;
 import com.lowagie.text.pdf.parser.PdfTextExtractor;
 import org.junit.jupiter.api.Test;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.ActiveProfiles;
 
 import java.io.File;
 import java.io.IOException;
@@ -14,6 +16,8 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
+@ActiveProfiles("test")
+@SpringBootTest(properties = {"form-flow.path=flows-config/test-flow.yaml"})
 class PDFFormFillerTest {
 
   private final PDFFormFiller pdfFormFiller = new PDFFormFiller();