From 3514badae47b6f52b8ad231f8682a3c733f2f18b Mon Sep 17 00:00:00 2001
From: dpiercey <dpiercey@ebay.com>
Date: Mon, 15 Apr 2024 10:54:08 -0700
Subject: [PATCH] feat: support client reorder attribute

---
 README.md                                     | 13 +++++
 .../renders.expected/loading.0.html           | 16 ++++++
 .../renders.expected/loading.1.html           | 32 ++++++++++++
 .../renders.expected/loading.2.html           | 26 ++++++++++
 .../ssr-stream-client-reorder/embed.marko     | 11 ++++
 .../ssr-stream-client-reorder/index.marko     | 18 +++++++
 .../micro-frame/__tests__/server.test.ts      |  5 ++
 src/components/micro-frame/marko-tag.json     |  7 +++
 .../micro-frame-component/node.marko          | 51 +++++++++++++------
 .../micro-frame-component/web.component.ts    | 15 ++++--
 10 files changed, 173 insertions(+), 21 deletions(-)
 create mode 100644 src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.0.html
 create mode 100644 src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.1.html
 create mode 100644 src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.2.html
 create mode 100644 src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/embed.marko
 create mode 100644 src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/index.marko

diff --git a/README.md b/README.md
index 48f6617..c61c09a 100644
--- a/README.md
+++ b/README.md
@@ -214,6 +214,19 @@ Optional `style` attribute which works the same way as [Marko style attribute](h
 <micro-frame src="..." style=["display:block", null, { marginRight: 16 }]/>
 ```
 
+## `client-reorder`
+
+Similar to the [<await> tag client-reorder attribute](https://markojs.com/docs/core-tags/#await) this tells the micro-frame to avoid blocking content later in the document.
+
+> Note when this is used the micro-frame will be buffered instead of streamed and inserted once it's ready.
+
+```marko
+<!--
+This example will disable the default 30s timeout.
+-->
+<micro-frame src="..." timeout=0/>
+```
+
 # Communicating between host and child
 
 Communicating with the embedded application happens primarily in one of two ways, either you want to do a full reload of and get new HTML, or you want to orchestrate a client side rendered update.
diff --git a/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.0.html b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.0.html
new file mode 100644
index 0000000..cc7aeeb
--- /dev/null
+++ b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.0.html
@@ -0,0 +1,16 @@
+<div>
+  Host app
+</div>
+<div
+  data-src="embed"
+  id="GENERATED-0"
+>
+  <span
+    id="GENERATED-1"
+  >
+    Loading...
+  </span>
+</div>
+<div>
+  Host footer
+</div>
\ No newline at end of file
diff --git a/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.1.html b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.1.html
new file mode 100644
index 0000000..a935c68
--- /dev/null
+++ b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.1.html
@@ -0,0 +1,32 @@
+<div>
+  Host app
+</div>
+<div
+  data-src="embed"
+  id="GENERATED-0"
+>
+  <span
+    id="GENERATED-1"
+  >
+    Loading...
+  </span>
+</div>
+<div>
+  Host footer
+</div>
+<div
+  id="GENERATED-2"
+  style="display:none"
+>
+  <div>
+    <h1>
+      Hello
+    </h1>
+    <h2>
+      World
+    </h2>
+  </div>
+</div>
+<script>
+  function $af(d,a,e,l,g,h,k,b,f,c){c=$af;if(a&&!c[a])(c[a+="$"]||(c[a]=[])).push(d);else{e=document;l=e.getElementById("af"+d);g=e.getElementById("afph"+d);h=e.createDocumentFragment();k=l.childNodes;b=0;for(f=k.length;b&lt;f;b++)h.appendChild(k.item(0));g&&g.parentNode.replaceChild(h,g);c[d]=1;if(a=c[d+"$"])for(b=0,f=a.length;b&lt;f;b++)c(a[b])}};$af(0)
+</script>
\ No newline at end of file
diff --git a/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.2.html b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.2.html
new file mode 100644
index 0000000..54d1417
--- /dev/null
+++ b/src/components/micro-frame/__tests__/__snapshots__/ssr-client-reorder-loading/renders.expected/loading.2.html
@@ -0,0 +1,26 @@
+<div>
+  Host app
+</div>
+<div
+  data-src="embed"
+  id="GENERATED-0"
+>
+  <div>
+    <h1>
+      Hello
+    </h1>
+    <h2>
+      World
+    </h2>
+  </div>
+</div>
+<div>
+  Host footer
+</div>
+<div
+  id="GENERATED-1"
+  style="display:none"
+/>
+<script>
+  function $af(d,a,e,l,g,h,k,b,f,c){c=$af;if(a&&!c[a])(c[a+="$"]||(c[a]=[])).push(d);else{e=document;l=e.getElementById("af"+d);g=e.getElementById("afph"+d);h=e.createDocumentFragment();k=l.childNodes;b=0;for(f=k.length;b&lt;f;b++)h.appendChild(k.item(0));g&&g.parentNode.replaceChild(h,g);c[d]=1;if(a=c[d+"$"])for(b=0,f=a.length;b&lt;f;b++)c(a[b])}};$af(0)
+</script>
\ No newline at end of file
diff --git a/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/embed.marko b/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/embed.marko
new file mode 100644
index 0000000..cb13320
--- /dev/null
+++ b/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/embed.marko
@@ -0,0 +1,11 @@
+import { wait } from "../../../../../__tests__/queue";
+<esbuild-assets/>
+
+<div>
+  <h1>Hello</h1>
+  <await(wait())>
+    <@then>
+      <h2>World</h2>
+    </@then>
+  </await>
+</div>
diff --git a/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/index.marko b/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/index.marko
new file mode 100644
index 0000000..99135bc
--- /dev/null
+++ b/src/components/micro-frame/__tests__/fixtures/ssr-stream-client-reorder/index.marko
@@ -0,0 +1,18 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>Simple Example</title>
+  <esbuild-assets/>
+</head>
+<body>
+  <div>Host app</div>
+  <micro-frame client-reorder src="embed">
+    <@loading>
+      Loading...
+    </@loading>
+  </micro-frame>
+  <div>Host footer</div>
+</body>
+</html>
diff --git a/src/components/micro-frame/__tests__/server.test.ts b/src/components/micro-frame/__tests__/server.test.ts
index cc70faf..0bdb0a5 100644
--- a/src/components/micro-frame/__tests__/server.test.ts
+++ b/src/components/micro-frame/__tests__/server.test.ts
@@ -121,6 +121,11 @@ describe(
   fixture(path.join(__dirname, "fixtures/ssr-stream-loading"))
 );
 
+describe(
+  "ssr client-reorder loading",
+  fixture(path.join(__dirname, "fixtures/ssr-stream-client-reorder"))
+);
+
 describe("csr 404", fixture(path.join(__dirname, "fixtures/csr-404")));
 
 describe("ssr 404", fixture(path.join(__dirname, "fixtures/ssr-404")));
diff --git a/src/components/micro-frame/marko-tag.json b/src/components/micro-frame/marko-tag.json
index 7eab30b..f7c9793 100644
--- a/src/components/micro-frame/marko-tag.json
+++ b/src/components/micro-frame/marko-tag.json
@@ -68,6 +68,13 @@
       }
     ]
   },
+  "@client-reorder": {
+    "autocomplete": [
+      {
+        "description": "Wether the micro-frame should should block/buffer content below the micro-frame."
+      }
+    ]
+  },
   "@class": {
     "autocomplete": [
       {
diff --git a/src/node_modules/@internal/micro-frame-component/node.marko b/src/node_modules/@internal/micro-frame-component/node.marko
index 3b3b2b4..c94a0eb 100644
--- a/src/node_modules/@internal/micro-frame-component/node.marko
+++ b/src/node_modules/@internal/micro-frame-component/node.marko
@@ -7,7 +7,7 @@ static const strictSSL = process.env.NODE_TLS_REJECT_UNAUTHORIZED !== "0";
 static function internalFetch(url, options) {
   return fetch(url, { ...options, ca, cachePath, strictSSL });
 }
-static async function fetchStream(input, out) {
+static async function fetchBody(input, out, buffer) {
   const global = out.global;
   let origin;
   let incomingHeaders;
@@ -79,6 +79,8 @@ static async function fetchStream(input, out) {
 
   if (!res.ok) throw new Error(res.statusText);
 
+  if (buffer) return res.text();
+
   if (!res.body || !res.body[Symbol.asyncIterator]) {
     throw new Error("Response body must be a stream.");
   }
@@ -87,25 +89,42 @@ static async function fetchStream(input, out) {
 }
 
 <div id=component.id class=input.class style=input.style data-src=input.src>
-  <if(input.loading)>
-    <${input.loading}/>
-    <!-- output a comment used as a marker to detect where the loading content starts so it can be removed -->$!{`<!--${component.id}-->`}
-  </if>
   <!--
   We put the streamed html in a preserved fragment.
   This allows Marko to avoid diffing that section.
   -->
   $ out.bf("@_", component, true);
-  <await(fetchStream(input, out)) timeout=input.timeout catch=input.catch>
-    <@then|iter|>
-      <ssr-wait
-        id=component.id
-        iter=iter
-        timeout=input.timeout
-        loading=input.loading
-        catch=input.catch
-      />
-    </@then>
-  </await>
+  <if(input.clientReorder)>
+    <await(fetchBody(input, out, true))
+      client-reorder
+      placeholder=input.loading
+      timeout=input.timeout
+      catch=input.catch
+    >
+      <@then|html|>
+        $!{html}
+      </@then>
+    </await>
+  </if>
+  <else>
+    <if(input.loading)>
+      <${input.loading}/>
+      <!-- output a comment used as a marker to detect where the loading content starts so it can be removed -->$!{`<!--${component.id}-->`}
+    </if>
+    <await(fetchBody(input, out, false))
+      timeout=input.timeout
+      catch=input.catch
+    >
+      <@then|iter|>
+        <ssr-wait
+          id=component.id
+          iter=iter
+          timeout=input.timeout
+          loading=input.loading
+          catch=input.catch
+        />
+      </@then>
+    </await>
+  </else>
   $ out.ef();
 </div>
diff --git a/src/node_modules/@internal/micro-frame-component/web.component.ts b/src/node_modules/@internal/micro-frame-component/web.component.ts
index 8cd5a19..d6b506f 100644
--- a/src/node_modules/@internal/micro-frame-component/web.component.ts
+++ b/src/node_modules/@internal/micro-frame-component/web.component.ts
@@ -64,17 +64,22 @@ export = {
         ? this.input.fetch(this.src, options, fetch)
         : fetch(this.src, options));
       if (!res.ok) throw new Error(res.statusText);
-      const reader = res.body!.getReader();
-      const decoder = new TextDecoder();
       writable = getWritableDOM(
         this.el,
         // references the start of the preserved Marko fragment.
         this.el.lastChild!.previousSibling
       );
 
-      let value: Uint8Array | undefined;
-      while ((value = (await reader.read()).value)) {
-        writable.write(decoder.decode(value));
+      if (this.input.clientReorder) {
+        writable.write(await res.text());
+      } else {
+        const reader = res.body!.getReader();
+        const decoder = new TextDecoder();
+
+        let value: Uint8Array | undefined;
+        while ((value = (await reader.read()).value)) {
+          writable.write(decoder.decode(value));
+        }
       }
 
       await writable.close();