From e8435c2444698602dbd6e9bf945e7d17db38f862 Mon Sep 17 00:00:00 2001
From: douglasward <douglas+github@giantmonkey.de>
Date: Wed, 1 Dec 2021 10:02:16 +0100
Subject: [PATCH] allow defining the element used when hydrating svelte
 components (#221)

* allow defining the mounted element for used when hydrating svelte components

* make sure component mount element tags match

* fix up tests after adding default element div hydrate option

* revert bumping of match[0] index as it comes before the added element name match

Co-authored-by: Douglas Ward <douglas@giantmonkey.de>
---
 .../__tests__/inlineSvelteComponent.spec.ts   | 10 ++++-----
 .../__tests__/partialHydration.spec.ts        | 22 +++++++++----------
 src/partialHydration/inlineSvelteComponent.ts | 19 +++++++++++-----
 src/partialHydration/mountComponentsInHtml.ts | 12 +++++-----
 src/utils/__tests__/svelteComponent.spec.ts   | 10 ++++-----
 src/utils/svelteComponent.ts                  |  6 ++++-
 src/utils/types.ts                            |  1 +
 7 files changed, 46 insertions(+), 34 deletions(-)

diff --git a/src/partialHydration/__tests__/inlineSvelteComponent.spec.ts b/src/partialHydration/__tests__/inlineSvelteComponent.spec.ts
index a09c6cd3..f42e5d1c 100644
--- a/src/partialHydration/__tests__/inlineSvelteComponent.spec.ts
+++ b/src/partialHydration/__tests__/inlineSvelteComponent.spec.ts
@@ -8,7 +8,7 @@ test('#escapeHtml', () => {
 });
 
 test('#inlinePreprocessedSvelteComponent', () => {
-  const options = 'loading=lazy';
+  const options = '{"loading":"lazy"}';
   expect(
     inlinePreprocessedSvelteComponent({
       name: 'Home',
@@ -18,10 +18,10 @@ test('#inlinePreprocessedSvelteComponent', () => {
       options,
     }),
   ).toEqual(
-    `<div class="ejs-component" data-ejs-component="Home" data-ejs-props={JSON.stringify([object Object])} data-ejs-options={JSON.stringify(${options})} />`,
+    `<div class="ejs-component" data-ejs-component="Home" data-ejs-props={JSON.stringify([object Object])} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} />`,
   );
   expect(inlinePreprocessedSvelteComponent({})).toEqual(
-    `<div class="ejs-component" data-ejs-component="" data-ejs-props={JSON.stringify([object Object])} data-ejs-options={JSON.stringify({"loading":"lazy"})} />`,
+    `<div class="ejs-component" data-ejs-component="" data-ejs-props={JSON.stringify([object Object])} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} />`,
   );
 });
 
@@ -38,9 +38,9 @@ test('#inlineSvelteComponent', () => {
       options,
     }),
   ).toEqual(
-    `<div class="ejs-component" data-ejs-component="Home" data-ejs-props="{&quot;welcomeText&quot;:&quot;Hello World&quot;}" data-ejs-options="{&quot;loading&quot;:&quot;lazy&quot;}"></div>`,
+    `<div class="ejs-component" data-ejs-component="Home" data-ejs-props="{&quot;welcomeText&quot;:&quot;Hello World&quot;}" data-ejs-options="{&quot;loading&quot;:&quot;lazy&quot;,&quot;element&quot;:&quot;div&quot;}"></div>`,
   );
   expect(inlineSvelteComponent({})).toEqual(
-    `<div class="ejs-component" data-ejs-component="" data-ejs-props="{}" data-ejs-options="{&quot;loading&quot;:&quot;lazy&quot;}"></div>`,
+    `<div class="ejs-component" data-ejs-component="" data-ejs-props="{}" data-ejs-options="{&quot;loading&quot;:&quot;lazy&quot;,&quot;element&quot;:&quot;div&quot;}"></div>`,
   );
 });
diff --git a/src/partialHydration/__tests__/partialHydration.spec.ts b/src/partialHydration/__tests__/partialHydration.spec.ts
index a9735a87..bda87588 100644
--- a/src/partialHydration/__tests__/partialHydration.spec.ts
+++ b/src/partialHydration/__tests__/partialHydration.spec.ts
@@ -9,7 +9,7 @@ describe('#partialHydration', () => {
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({"loading":"lazy"})} />`,
+      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} />`,
     );
   });
 
@@ -17,11 +17,11 @@ describe('#partialHydration', () => {
     expect(
       (
         await partialHydration.markup({
-          content: '<DatePicker hydrate-client={{ a: "c" }} hydrate-options={{ loading: "lazy" }}/>',
+          content: '<DatePicker hydrate-client={{ a: "c" }} hydrate-options={{ "loading": "lazy" }}/>',
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "c" })} data-ejs-options={JSON.stringify({ loading: "lazy" })} />`,
+      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "c" })} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} />`,
     );
   });
 
@@ -29,11 +29,11 @@ describe('#partialHydration', () => {
     expect(
       (
         await partialHydration.markup({
-          content: '<DatePicker hydrate-client={{ a: "c" }} hydrate-options={{ timeout: 2000 }}/>',
+          content: '<DatePicker hydrate-client={{ a: "c" }} hydrate-options={{ "timeout": 2000 }}/>',
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "c" })} data-ejs-options={JSON.stringify({ timeout: 2000 })} />`,
+      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "c" })} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div","timeout":2000})} />`,
     );
   });
 
@@ -41,11 +41,11 @@ describe('#partialHydration', () => {
     expect(
       (
         await partialHydration.markup({
-          content: '<DatePicker hydrate-client={{ a: "b" }} hydrate-options={{ loading: "eager" }} />',
+          content: '<DatePicker hydrate-client={{ a: "b" }} hydrate-options={{ "loading": "eager" }} />',
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({ loading: "eager" })} />`,
+      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({"loading":"eager","element":"div"})} />`,
     );
   });
   it('eager, root margin, threshold', async () => {
@@ -53,11 +53,11 @@ describe('#partialHydration', () => {
       (
         await partialHydration.markup({
           content:
-            '<DatePicker hydrate-client={{ a: "b" }} hydrate-options={{ loading: "eager", rootMargin: "500px", threshold: 0 }} />',
+            '<DatePicker hydrate-client={{ a: "b" }} hydrate-options={{ "loading": "eager", "rootMargin": "500px", "threshold": 0 }} />',
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({ loading: "eager", rootMargin: "500px", threshold: 0 })} />`,
+      `<div class="ejs-component" data-ejs-component="DatePicker" data-ejs-props={JSON.stringify({ a: "b" })} data-ejs-options={JSON.stringify({"loading":"eager","element":"div","rootMargin":"500px","threshold":0})} />`,
     );
   });
   it('open string', async () => {
@@ -102,11 +102,11 @@ describe('#partialHydration', () => {
     expect(
       (
         await partialHydration.markup({
-          content: `<Clock hydrate-client={{}} hydrate-options={{ loading: 'eager', preload: true }} /><Block hydrate-client={{}} hydrate-options={{ loading: 'lazy' }} /><Alock hydrate-client={{}} hydrate-options={{ loading: 'lazy' }} />`,
+          content: `<Clock hydrate-client={{}} hydrate-options={{ "loading": "eager", "preload": true }} /><Block hydrate-client={{}} hydrate-options={{ "loading": "lazy" }} /><Alock hydrate-client={{}} hydrate-options={{ "loading": "lazy" }} />`,
         })
       ).code,
     ).toEqual(
-      `<div class="ejs-component" data-ejs-component="Clock" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({ loading: 'eager', preload: true })} /><div class="ejs-component" data-ejs-component="Block" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({ loading: 'lazy' })} /><div class="ejs-component" data-ejs-component="Alock" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({ loading: 'lazy' })} />`,
+      `<div class="ejs-component" data-ejs-component="Clock" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({"loading":"eager","element":"div","preload":true})} /><div class="ejs-component" data-ejs-component="Block" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} /><div class="ejs-component" data-ejs-component="Alock" data-ejs-props={JSON.stringify({})} data-ejs-options={JSON.stringify({"loading":"lazy","element":"div"})} />`,
     );
   });
 });
diff --git a/src/partialHydration/inlineSvelteComponent.ts b/src/partialHydration/inlineSvelteComponent.ts
index 5aa95ed8..671a63f3 100644
--- a/src/partialHydration/inlineSvelteComponent.ts
+++ b/src/partialHydration/inlineSvelteComponent.ts
@@ -1,5 +1,8 @@
-const defaultHydrationOptions = {
+import { HydrateOptions } from '../utils/types';
+
+const defaultHydrationOptions: HydrateOptions = {
   loading: 'lazy',
+  element: 'div',
 };
 
 export function escapeHtml(text: string): string {
@@ -22,19 +25,21 @@ export function inlinePreprocessedSvelteComponent({
   props = {},
   options = '',
 }: InputParamsInlinePreprocessedSvelteComponent): string {
-  const hydrationOptions = options.length > 0 ? options : JSON.stringify(defaultHydrationOptions);
+  const hydrationOptions =
+    options.length > 0 ? { ...defaultHydrationOptions, ...JSON.parse(options) } : defaultHydrationOptions;
+  const hydrationOptionsString = JSON.stringify(hydrationOptions);
 
   const replacementAttrs = {
     class: '"ejs-component"',
     'data-ejs-component': `"${name}"`,
     'data-ejs-props': `{JSON.stringify(${props})}`,
-    'data-ejs-options': `{JSON.stringify(${hydrationOptions})}`,
+    'data-ejs-options': `{JSON.stringify(${hydrationOptionsString})}`,
   };
   const replacementAttrsString = Object.entries(replacementAttrs).reduce(
     (out, [key, value]) => `${out} ${key}=${value}`,
     '',
   );
-  return `<div${replacementAttrsString} />`;
+  return `<${hydrationOptions.element}${replacementAttrsString} />`;
 }
 
 type InputParamsInlineSvelteComponent = {
@@ -42,6 +47,7 @@ type InputParamsInlineSvelteComponent = {
   props?: any;
   options?: {
     loading?: string; // todo: enum, can't get it working: 'lazy', 'eager', 'none'
+    element?: string; // default: 'div'
   };
 };
 
@@ -50,7 +56,8 @@ export function inlineSvelteComponent({
   props = {},
   options = {},
 }: InputParamsInlineSvelteComponent): string {
-  const hydrationOptions = Object.keys(options).length > 0 ? options : defaultHydrationOptions;
+  const hydrationOptions =
+    Object.keys(options).length > 0 ? { ...defaultHydrationOptions, ...options } : defaultHydrationOptions;
 
   const replacementAttrs = {
     class: '"ejs-component"',
@@ -63,5 +70,5 @@ export function inlineSvelteComponent({
     '',
   );
 
-  return `<div${replacementAttrsString}></div>`;
+  return `<${hydrationOptions.element}${replacementAttrsString}></${hydrationOptions.element}>`;
 }
diff --git a/src/partialHydration/mountComponentsInHtml.ts b/src/partialHydration/mountComponentsInHtml.ts
index 29316569..e1e0bde6 100644
--- a/src/partialHydration/mountComponentsInHtml.ts
+++ b/src/partialHydration/mountComponentsInHtml.ts
@@ -14,23 +14,23 @@ export default function mountComponentsInHtml({ page, html, hydrateOptions }): s
   let outputHtml = html;
   // sometimes svelte adds a class to our inlining.
   const matches = outputHtml.matchAll(
-    /<div class="ejs-component[^]*?" data-ejs-component="([A-Za-z]+)" data-ejs-props="({[^]*?})" data-ejs-options="({[^]*?})"><\/div>/gim,
+    /<(\S+) class="ejs-component[^]*?" data-ejs-component="([A-Za-z]+)" data-ejs-props="({[^]*?})" data-ejs-options="({[^]*?})"><\/\1>/gim,
   );
 
   for (const match of matches) {
-    const hydrateComponentName = match[1];
+    const hydrateComponentName = match[2];
     let hydrateComponentProps;
     let hydrateComponentOptions;
 
     try {
-      hydrateComponentProps = JSON.parse(replaceSpecialCharacters(match[2]));
+      hydrateComponentProps = JSON.parse(replaceSpecialCharacters(match[3]));
     } catch (e) {
-      throw new Error(`Failed to JSON.parse props for ${hydrateComponentName} ${match[2]}`);
+      throw new Error(`Failed to JSON.parse props for ${hydrateComponentName} ${match[3]}`);
     }
     try {
-      hydrateComponentOptions = JSON.parse(replaceSpecialCharacters(match[3]));
+      hydrateComponentOptions = JSON.parse(replaceSpecialCharacters(match[4]));
     } catch (e) {
-      throw new Error(`Failed to JSON.parse props for ${hydrateComponentName} ${match[3]}`);
+      throw new Error(`Failed to JSON.parse props for ${hydrateComponentName} ${match[4]}`);
     }
 
     if (hydrateOptions) {
diff --git a/src/utils/__tests__/svelteComponent.spec.ts b/src/utils/__tests__/svelteComponent.spec.ts
index d9b317db..150a2c50 100644
--- a/src/utils/__tests__/svelteComponent.spec.ts
+++ b/src/utils/__tests__/svelteComponent.spec.ts
@@ -83,7 +83,7 @@ describe('#svelteComponent', () => {
         render: () => ({
           head: '<head>',
           css: { code: '<old>' },
-          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy" }"></div></div>',
+          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy", "element": "div" }"></div></div>',
         }),
         _css: ['<css>', '<css2>'],
       }),
@@ -110,7 +110,7 @@ describe('#svelteComponent', () => {
     );
 
     expect(componentProps.page.componentsToHydrate[0]).toMatchObject({
-      hydrateOptions: { loading: 'lazy' },
+      hydrateOptions: { loading: 'lazy', element: 'div' },
       id: 'SwrzsrVDCd',
       name: 'datepickerSwrzsrVDCd',
       prepared: {},
@@ -125,7 +125,7 @@ describe('#svelteComponent', () => {
         render: () => ({
           head: '<head>',
           css: { code: '<old>' },
-          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy" }"></div></div>',
+          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy", "element": "div" }"></div></div>',
         }),
         _css: ['<css>', '<css2>'],
         _cssMap: ['<cssmap>', '<cssmap2>'],
@@ -189,7 +189,7 @@ describe('#svelteComponent', () => {
     );
     expect(props.page.svelteCss).toEqual([{ css: ['<css>', '<css2>'], cssMap: ['<cssmap>', '<cssmap2>'] }]);
     expect(props.page.componentsToHydrate[0]).toMatchObject({
-      hydrateOptions: { loading: 'lazy' },
+      hydrateOptions: { loading: 'lazy', element: 'div' },
       id: 'SwrzsrVDCd',
       name: 'datepickerSwrzsrVDCd',
       prepared: {},
@@ -204,7 +204,7 @@ describe('#svelteComponent', () => {
         render: () => ({
           head: '<head>',
           css: { code: '<old>' },
-          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy" }"></div></div>',
+          html: '<div class="svelte-datepicker"><div class="ejs-component" data-ejs-component="Datepicker" data-ejs-props="{ "a": "b" }" data-ejs-options="{ "loading": "lazy", "element": "div" }"></div></div>',
         }),
         _css: ['<css>', '<css2>'],
         _cssMap: ['<cssmap>', '<cssmap2>'],
diff --git a/src/utils/svelteComponent.ts b/src/utils/svelteComponent.ts
index 9c8b6535..1b1f1d1d 100644
--- a/src/utils/svelteComponent.ts
+++ b/src/utils/svelteComponent.ts
@@ -61,7 +61,11 @@ const svelteComponent =
         id,
       });
 
-      return `<div class="${cleanComponentName.toLowerCase()}-component" id="${uniqueComponentName}">${innerHtml}</div>`;
+      return `<${
+        hydrateOptions.element
+      } class="${cleanComponentName.toLowerCase()}-component" id="${uniqueComponentName}">${innerHtml}</${
+        hydrateOptions.element
+      }>`;
     } catch (e) {
       // console.log(e);
       page.errors.push(e);
diff --git a/src/utils/types.ts b/src/utils/types.ts
index 35376c64..22402ac1 100644
--- a/src/utils/types.ts
+++ b/src/utils/types.ts
@@ -174,6 +174,7 @@ export type HydrateOptions = {
   noPrefetch?: boolean;
   threshold?: number;
   rootMargin?: string;
+  element?: string;
 };
 
 export interface ComponentPayload {