custom watermark support

bpampuch · Nov 24, 2024 · ece4927 · ece4927
1 parent fd9e562
commit ece4927
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 12 deletions.
diff --git a/examples/pdfs/sections.pdf b/examples/pdfs/sections.pdf
diff --git a/examples/sections.js b/examples/sections.js
@@ -8,6 +8,7 @@ var docDefinition = {
 	header: function () { return 'default header'; },
 	footer: function () { return 'default footer'; },
 	background: function() { return { text:'global background', alignment: 'right' }; },
+	watermark: 'default watermark',
 	content: [
 		{
 			section: [
@@ -19,6 +20,7 @@ var docDefinition = {
 			header: function (currentPage, pageCount) { return 'header: ' + currentPage.toString() + ' of ' + pageCount; },
 			footer: function (currentPage, pageCount) { return 'footer: ' + currentPage.toString() + ' of ' + pageCount; },
 			background: function() { return { text:'SECTION 2 background', alignment: 'right' }; },
+			watermark: 'SECTION 2 watermark',
 			pageOrientation: 'landscape',
 			section: [
 				'SECTION 2',
@@ -29,6 +31,7 @@ var docDefinition = {
 			header: null,
 			footer: null,
 			background: null,
+			watermark: null,
 			pageSize: 'A7',
 			pageOrientation: 'portrait',
 			section: [
@@ -37,6 +40,7 @@ var docDefinition = {
 			]
 		},
 		{
+			watermark: 'inherit',
 			pageSize: 'A6',
 			pageOrientation: 'portrait',
 			pageMargins: 5,
@@ -46,6 +50,7 @@ var docDefinition = {
 			]
 		},
 		{
+			watermark: 'watermark for inherit',
 			pageSize: 'A6',
 			pageOrientation: 'landscape',
 			pageMargins: 10,
@@ -55,6 +60,7 @@ var docDefinition = {
 			]
 		},
 		{
+			watermark: 'inherit',
 			pageSize: 'inherit',
 			pageOrientation: 'inherit',
 			pageMargins: 'inherit',

diff --git a/src/LayoutBuilder.js b/src/LayoutBuilder.js
@@ -203,9 +203,7 @@ class LayoutBuilder {
 
 		this.processNode(docStructure);
 		this.addHeadersAndFooters(header, footer);
-		if (watermark != null) {
-			this.addWatermark(watermark, pdfDocument, defaultStyle);
-		}
+		this.addWatermark(watermark, pdfDocument, defaultStyle);
 
 		return { pages: this.writer.context().pages, linearNodeList: this.linearNodeList };
 	}
@@ -276,17 +274,26 @@ class LayoutBuilder {
 	}
 
 	addWatermark(watermark, pdfDocument, defaultStyle) {
-		if (isString(watermark)) {
-			watermark = { 'text': watermark };
-		}
-
-		if (!watermark.text) { // empty watermark text
-			return;
-		}
-
 		let pages = this.writer.context().pages;
 		for (let i = 0, l = pages.length; i < l; i++) {
-			pages[i].watermark = getWatermarkObject({ ...watermark }, pages[i].pageSize, pdfDocument, defaultStyle);
+			let pageWatermark = watermark;
+			if (pages[i].customProperties['watermark'] || pages[i].customProperties['watermark'] === null) {
+				pageWatermark = pages[i].customProperties['watermark'];
+			}
+
+			if (pageWatermark === undefined || pageWatermark === null) {
+				continue;
+			}
+
+			if (isString(pageWatermark)) {
+				pageWatermark = { 'text': pageWatermark };
+			}
+
+			if (!pageWatermark.text) { // empty watermark text
+				continue;
+			}
+
+			pages[i].watermark = getWatermarkObject({ ...pageWatermark }, pages[i].pageSize, pdfDocument, defaultStyle);
 		}
 
 		function getWatermarkObject(watermark, pageSize, pdfDocument, defaultStyle) {