feat: Allow for custom prerender injection selector

preactjs · Dec 16, 2023 · 204f633 · 204f633
1 parent a82389f
commit 204f633
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/demo/src/index.tsx b/demo/src/index.tsx
@@ -34,6 +34,7 @@ export async function prerender() {
 	return {
 		html,
 		links,
+		selector: "#app",
 		head: {
 			lang: "en",
 			title: "Prerendered Preact App",

diff --git a/src/prerender.ts b/src/prerender.ts
@@ -329,8 +329,21 @@ export function PrerenderPlugin({
 					}
 				}
 
-				// Inject pre-rendered HTML into the start of <body>:
-				htmlDoc.querySelector("body")?.insertAdjacentHTML("afterbegin", body);
+				if (result.selector) {
+					const el = htmlDoc.querySelector(result.selector);
+					if (!el)
+						throw new Error(
+							`Unable to detect prerender selector "${result.selector}" in input HTML template`,
+						);
+					el.insertAdjacentHTML("afterbegin", body);
+				} else {
+					const el = htmlDoc.querySelector("body");
+					if (!el)
+						throw new Error(
+							"No prerender selector was specified and <body> does not exist in input HTML template",
+						);
+					el.insertAdjacentHTML("afterbegin", body);
+				}
 
 				// Add generated HTML to compilation:
 				if (route.url === "/")