🩹 Await render hook

Render hooks can be asynchronous. This commit awaits its completion.
eclipsesource · Jan 18, 2025 · eec2a03 · eec2a03
1 parent 0699f5b
commit eec2a03
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/read-document.ts b/src/read-document.ts
@@ -35,7 +35,7 @@ export type DocumentDefinition = {
     modificationDate?: Date;
     relationship?: FileRelationShip;
   }[];
-  onRenderDocument?: (pdfDoc: PDFDocument) => void;
+  onRenderDocument?: (pdfDoc: PDFDocument) => void | Promise<void>;
 };
 
 export type Metadata = {

diff --git a/src/render/render-document.test.ts b/src/render/render-document.test.ts
@@ -72,8 +72,9 @@ describe('renderDocument', () => {
   it('calls custom render hook', async () => {
     const def = {
       content: [],
-      onRenderDocument: (pdfDoc: PDFDocument) => {
+      onRenderDocument: async (pdfDoc: PDFDocument) => {
         pdfDoc.setTitle('Test Title');
+        await new Promise((resolve) => setTimeout(resolve, 10));
       },
     };
 

diff --git a/src/render/render-document.ts b/src/render/render-document.ts
@@ -23,7 +23,8 @@ export async function renderDocument(def: DocumentDefinition, pages: Page[]): Pr
     });
   }
 
-  def.onRenderDocument?.(pdfDoc);
+  await def.onRenderDocument?.(pdfDoc);
+
   const idInfo = {
     creator: 'pdfmkr',
     time: new Date().toISOString(),