Add ad enhancer to enhanceBlocks fn

Co-authored-by: Jamie B <[email protected]>
guardian · Sep 11, 2023 · dfc452d · dfc452d
1 parent 4a5593f
commit dfc452d
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/dotcom-rendering/src/lib/adPlaceholder.ts → dotcom-rendering/src/model/adPlaceholder.ts b/dotcom-rendering/src/lib/adPlaceholder.ts → dotcom-rendering/src/model/adPlaceholder.ts
@@ -102,3 +102,9 @@ const adPlaceholder = (elements: FEElement[]): FEElement[] => {
 
 	return elementsWithAds.elements;
 };
+
+export const enhanceAdPlaceholders = (blocks: Block[]): Block[] =>
+	blocks.map((b) => ({
+		...b,
+		elements: adPlaceholder(b.elements),
+	}));
diff --git a/dotcom-rendering/src/model/enhanceBlocks.ts b/dotcom-rendering/src/model/enhanceBlocks.ts
@@ -1,4 +1,5 @@
 import type { Newsletter } from '../types/content';
+import { enhanceAdPlaceholders } from './adPlaceholder';
 import { enhanceBlockquotes } from './enhance-blockquotes';
 import { enhanceDividers } from './enhance-dividers';
 import { enhanceDots } from './enhance-dots';
@@ -35,6 +36,16 @@ class BlockEnhancer {
 		return this;
 	}
 
+	enhanceAdPlaceholders() {
+		if (
+			!(this.format.design === 'LiveBlogDesign') &&
+			!(this.format.design === 'DeadBlogDesign')
+		) {
+			this.blocks = enhanceAdPlaceholders(this.blocks);
+		}
+		return this;
+	}
+
 	enhanceDividers() {
 		this.blocks = enhanceDividers(this.blocks);
 		return this;
@@ -106,5 +117,6 @@ export const enhanceBlocks = (
 		.enhanceNumberedLists()
 		.enhanceEmbeds()
 		.enhanceTweets()
-		.enhanceNewsletterSignup().blocks;
+		.enhanceNewsletterSignup()
+		.enhanceAdPlaceholders().blocks;
 };