From 8e25babd9ea1196f2f66dedc18618356da1f4335 Mon Sep 17 00:00:00 2001
From: winwiz1 <winwiz@gmail.com>
Date: Sun, 24 Oct 2021 21:42:43 +1100
Subject: [PATCH] Client: Improve SSR

---
 client/src/components/ComponentB.tsx          |  3 +-
 client/src/components/Navigation.tsx          | 46 ++++++++++++++++--
 client/src/entrypoints/first.tsx              |  4 --
 client/src/entrypoints/second.tsx             |  4 --
 .../src/utils/postprocess/postProcessSSR.ts   | 47 +++++++------------
 5 files changed, 60 insertions(+), 44 deletions(-)

diff --git a/client/src/components/ComponentB.tsx b/client/src/components/ComponentB.tsx
index bab3b44..4b077f4 100644
--- a/client/src/components/ComponentB.tsx
+++ b/client/src/components/ComponentB.tsx
@@ -9,7 +9,6 @@ import { Helmet } from "react-helmet";
 import { Header, Icon, Container, Menu } from "semantic-ui-react";
 import { BaseComponent } from "./BaseComponent";
 import { StructuredData } from "./StructuredData";
-import * as SPAs from "../../config/spa.config";
 import { getTitle, getCanonical } from "../utils/misc";
 
 const cssNav = css({
@@ -64,7 +63,7 @@ const Navigation: React.FC = _props => {
         <Menu.Item>
           <Menu.Header>Go back to</Menu.Header>
           <Menu.Menu>
-            <Menu.Item href={`/${SPAs.getNames()[0]}`}>
+            <Menu.Item href="/">
               First SPA
               <Icon
                 name="object group outline"
diff --git a/client/src/components/Navigation.tsx b/client/src/components/Navigation.tsx
index fe41eff..ee1bd0d 100644
--- a/client/src/components/Navigation.tsx
+++ b/client/src/components/Navigation.tsx
@@ -9,35 +9,73 @@ import { NavLink } from "react-router-dom";
 import { Menu, Icon } from "semantic-ui-react";
 import * as SPAs from "../../config/spa.config";
 import styles from "../css/navigation.module.css";
+import { isServer } from "../utils/postprocess/misc";
 
 const cssStyle: Record<string, string> = {
   menu: styles["menu"],
 };
 
+const pathArray = ["/", "/a", "/lighthouse", "/namelookup"];
+
+const getIndex = () => {
+  if (isServer()) {
+    return -1;
+  }
+  const path = window.location.pathname;
+  return pathArray.findIndex((item) => item === path);
+}
+
 export const Navigation: React.FC = _props => {
+  React.useEffect(() => {
+    const idx = getIndex();
+    const menuItem = document.querySelector(`div.menu>a:nth-child(${idx + 1})`);
+    menuItem?.classList.add("active");
+    if (idx > 0) {
+      const menuItem = document.querySelector("div.menu>a:nth-child(1)");
+      menuItem?.classList.remove("active");
+    }
+  }, []);  // run once upon initial rendering
+
   return (
     <nav className={cssStyle.menu}>
       <Menu
         vertical
         size="large"
         className="nav_menu"
+        defaultActiveIndex={0}
       >
         <Menu.Item>
           <Menu.Header>First SPA</Menu.Header>
           <Menu.Menu>
-            <Menu.Item header as={NavLink} exact to="/">
+            <Menu.Item
+              header
+              as={NavLink}
+              exact to={pathArray[0]}
+            >
               Overview
               <Icon name="file alternate outline" size="large" />
             </Menu.Item>
-            <Menu.Item header as={NavLink} to="/a">
+            <Menu.Item
+              header
+              as={NavLink}
+              to={pathArray[1]}
+            >
               ComponentA
               <Icon name="cube" size="large" />
             </Menu.Item>
-            <Menu.Item header as={NavLink} to="/lighthouse">
+            <Menu.Item
+              header
+              as={NavLink}
+              to={pathArray[2]}
+            >
               Lighthouse
               <Icon name="tachometer alternate" size="large" />
             </Menu.Item>
-            <Menu.Item header as={NavLink} to="/namelookup">
+            <Menu.Item
+              header
+              as={NavLink}
+              to={pathArray[3]}
+            >
               NameLookup
               <Icon name="search" size="large" />
             </Menu.Item>
diff --git a/client/src/entrypoints/first.tsx b/client/src/entrypoints/first.tsx
index b4222fe..35982aa 100644
--- a/client/src/entrypoints/first.tsx
+++ b/client/src/entrypoints/first.tsx
@@ -13,7 +13,6 @@
  */
 import * as React from "react";
 import * as ReactDOM from "react-dom";
-import { Helmet } from "react-helmet";
 import { Router, Route, Switch } from "react-router-dom";
 import { ComponentA } from "../components/ComponentA";
 import { Lighthouse } from "../components/Lighthouse";
@@ -43,9 +42,6 @@ const First: React.FC = _props => {
     <>
       <Router history={getHistory()}>
         <ErrorBoundary>
-          <Helmet>
-            <meta charSet="utf-8" />
-         </Helmet>
           <div className="welcome">
             <h2>Welcome to {SPAs.appTitle}</h2>
           </div>
diff --git a/client/src/entrypoints/second.tsx b/client/src/entrypoints/second.tsx
index 7383dc8..06c9f3c 100644
--- a/client/src/entrypoints/second.tsx
+++ b/client/src/entrypoints/second.tsx
@@ -13,7 +13,6 @@
  */
 import * as React from "react";
 import * as ReactDOM from "react-dom";
-import { Helmet } from "react-helmet";
 import { ComponentB } from "../components/ComponentB";
 import { ErrorBoundary } from "../components/ErrorBoundary";
 // import { renderToString } from "react-dom/server";
@@ -26,9 +25,6 @@ const Second: React.FC = _props => {
   return (
     <>
       <ErrorBoundary>
-        <Helmet>
-          <meta charSet="utf-8" />
-        </Helmet>
         <div className="welcome">
           <h2>Welcome to {SPAs.appTitle}</h2>
         </div>
diff --git a/client/src/utils/postprocess/postProcessSSR.ts b/client/src/utils/postprocess/postProcessSSR.ts
index bf6d9d9..0e44518 100644
--- a/client/src/utils/postprocess/postProcessSSR.ts
+++ b/client/src/utils/postprocess/postProcessSSR.ts
@@ -2,7 +2,6 @@ import * as fs from "fs";
 import * as path from "path";
 import { promisify } from "util";
 import { JSDOM } from "jsdom";
-import { CallbackWrapper } from "./misc"
 import * as SPAs from "../../../config/spa.config";
 
 export async function postProcess(workDir: string, filePattern: string): Promise<void> {
@@ -23,39 +22,27 @@ export async function postProcess(workDir: string, filePattern: string): Promise
 
 async function postProcessBody(workDir: string, htmlFile: string, ssrFile: string): Promise<void> {
   const readFile = promisify(fs.readFile);
+  const writeFile = promisify(fs.writeFile);
+
   const htmlFilePath = path.join(workDir, htmlFile);
   const ssrFilePath = path.join(workDir, ssrFile);
 
-  const dataHtml = await readFile(htmlFilePath);
+  const jsdom = await JSDOM.fromFile(htmlFilePath);
+
+  if (!jsdom) {
+    throw "JSDOM creation failure";
+  }
+
   const dataSsr = (await readFile(ssrFilePath)).toString();
-  const reReact = /^\s*<div\s+id="app-root">/;
-  const ar: string[] = dataHtml.toString().replace(/\r\n?/g, "\n").split("\n");
-
-  const out = ar.map(str => {
-    if (reReact.test(str)) {
-      str += "\n";
-      str += dataSsr;
-    }
-    str += "\n";
-    return str;
-  });
-
-  const stream = fs.createWriteStream(htmlFilePath);
-  stream.on("error", err => {
-    console.error(`Failed to write to file ${htmlFilePath}, error: ${err}`)
-  });
-  out.forEach(str => { stream.write(str); });
-
-  stream.end();
-
-  let cbWrapper: CallbackWrapper|undefined;
-  const waitForBufferFlush = new Promise<void>((resolve) => { cbWrapper = new CallbackWrapper(resolve); });
-
-  stream.on("close", function() {
-    cbWrapper?.invoke();
-  });
-
-  await waitForBufferFlush;
+  const fragment = JSDOM.fragment(dataSsr);
+  const appRoot = jsdom.window.document.querySelector("div[id='app-root']");
+
+  if (!appRoot) {
+    throw "JSDOM - 'app-root' not found";
+  }
+
+  appRoot.appendChild(fragment);
+  await writeFile(htmlFilePath, jsdom.serialize());
 }
 
 async function postProcessHeader(workDir: string, htmlFile: string): Promise<void> {