fix(table): throwing error when header row was enabled on a table wit…

…h multiple columns
schoero · Nov 16, 2023 · b096a51 · b096a51
1 parent 5462c27
commit b096a51
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/pdf/table.ts b/src/pdf/table.ts
@@ -195,6 +195,8 @@ export class Table {
     const tableAlign = this.data.align ? this.data.align : undefined;
     const tableVerticalAlign = this.data.verticalAlign ? this.data.verticalAlign : "top";
 
+    const headerRowIndex = this.data.rows.findIndex(row => !!row.header);
+
     const autoRowHeights: number[] = [];
 
     for(let layer: TableLayer = 0; layer < Object.keys(TableLayer).length; layer++){
@@ -314,10 +316,10 @@ export class Table {
               rowY = doc.y;
 
               // Insert header
-              const headerRow = this.data.rows.find(row => row.header);
+              const headerRow = this.data.rows[headerRowIndex];
               if(headerRow !== undefined){
                 this.data.rows.splice(rowIndex, 0, headerRow);
-                autoRowHeights.splice(rowIndex, 0, autoRowHeights[this.data.rows.indexOf(headerRow)]);
+                autoRowHeights.splice(rowIndex, 0, autoRowHeights[headerRowIndex]);
                 rowIndex--;
                 continue rowLoop;
               }
@@ -328,7 +330,7 @@ export class Table {
           // Switch page before overflowing rows and header rows
           if(layer > TableLayer.PageInjection){
             if(
-              !!row.header && rowY !== doc.page.margins.top ||
+              !!row.header && rowY !== (doc.page.margins.top ?? 0) && rowIndex !== headerRowIndex ||
               rowY + rowHeight >= doc.page.height - doc.page.margins.bottom
             ){
               doc.switchToPage(this.getCurrentPage(doc) + 1);