From 7bcb7cb8160bcff5e3299f6f480c4da0df0d1c0f Mon Sep 17 00:00:00 2001
From: Seung Park <seungpark@users.noreply.github.com>
Date: Wed, 1 May 2024 16:29:43 -0400
Subject: [PATCH] DOP-4566: set init values for tabs and tab selectors during
 build time (#1080)

---
 src/components/DocumentBody.js        | 53 +++++++++++++----------
 src/components/DocumentBodyPreview.js | 39 +++++++++--------
 src/components/RootProvider.js        | 24 +++++------
 src/components/Tabs/TabSelectors.js   |  2 +-
 src/components/Tabs/index.js          | 21 +++-------
 src/components/Tabs/tab-context.js    | 60 ++++++++++++++++++++++++++-
 src/layouts/index.js                  |  7 +++-
 src/layouts/preview-layout.js         |  1 -
 8 files changed, 134 insertions(+), 73 deletions(-)

diff --git a/src/components/DocumentBody.js b/src/components/DocumentBody.js
index da5f1c682..e814161b8 100644
--- a/src/components/DocumentBody.js
+++ b/src/components/DocumentBody.js
@@ -23,6 +23,7 @@ import DocsLandingSD from './StructuredData/DocsLandingSD';
 import BreadcrumbSchema from './StructuredData/BreadcrumbSchema';
 import { InstruqtProvider } from './Instruqt/instruqt-context';
 import { SuspenseHelper } from './SuspenseHelper';
+import { TabProvider } from './Tabs/tab-context';
 
 // lazy load the unified footer to improve page load speed
 const LazyFooter = lazy(() => import('./Footer'));
@@ -103,28 +104,30 @@ const DocumentBody = (props) => {
 
   return (
     <>
-      <InstruqtProvider hasLabDrawer={page?.options?.instruqt}>
-        <Widgets
-          location={location}
-          pageOptions={page?.options}
-          pageTitle={pageTitle}
-          slug={slug}
-          isInPresentationMode={isInPresentationMode}
-          template={template}
-        >
-          <ImageContextProvider images={props.data?.pageImage?.images ?? []}>
-            <FootnoteContext.Provider value={{ footnotes }}>
-              <div id="template-container">
-                <Template {...props} useChatbot={useChatbot}>
-                  {pageNodes.map((child, index) => (
-                    <ComponentFactory key={index} metadata={metadata} nodeData={child} page={page} slug={slug} />
-                  ))}
-                </Template>
-              </div>
-            </FootnoteContext.Provider>
-          </ImageContextProvider>
-        </Widgets>
-      </InstruqtProvider>
+      <TabProvider selectors={page?.options?.selectors}>
+        <InstruqtProvider hasLabDrawer={page?.options?.instruqt}>
+          <Widgets
+            location={location}
+            pageOptions={page?.options}
+            pageTitle={pageTitle}
+            slug={slug}
+            isInPresentationMode={isInPresentationMode}
+            template={template}
+          >
+            <ImageContextProvider images={props.data?.pageImage?.images ?? []}>
+              <FootnoteContext.Provider value={{ footnotes }}>
+                <div id="template-container">
+                  <Template {...props} useChatbot={useChatbot}>
+                    {pageNodes.map((child, index) => (
+                      <ComponentFactory key={index} metadata={metadata} nodeData={child} page={page} slug={slug} />
+                    ))}
+                  </Template>
+                </div>
+              </FootnoteContext.Provider>
+            </ImageContextProvider>
+          </Widgets>
+        </InstruqtProvider>
+      </TabProvider>
       {!isInPresentationMode && (
         <div data-testid="consistent-footer" id="footer-container">
           <SuspenseHelper fallback={null}>
@@ -148,6 +151,12 @@ DocumentBody.propTypes = {
   pageContext: PropTypes.shape({
     slug: PropTypes.string.isRequired,
   }),
+  data: PropTypes.shape({
+    page: PropTypes.shape({
+      children: PropTypes.array,
+      options: PropTypes.object,
+    }).isRequired,
+  }).isRequired,
 };
 
 export default DocumentBody;
diff --git a/src/components/DocumentBodyPreview.js b/src/components/DocumentBodyPreview.js
index 930e98c61..bac112855 100644
--- a/src/components/DocumentBodyPreview.js
+++ b/src/components/DocumentBodyPreview.js
@@ -12,6 +12,7 @@ import SEO from './SEO';
 import FootnoteContext from './Footnote/footnote-context';
 import ComponentFactory from './ComponentFactory';
 import { InstruqtProvider } from './Instruqt/instruqt-context';
+import { TabProvider } from './Tabs/tab-context';
 
 // Identify the footnotes on a page and all footnote_reference nodes that refer to them.
 // Returns a map wherein each key is the footnote name, and each value is an object containing:
@@ -102,24 +103,26 @@ const DocumentBody = (props) => {
       metadata={metadata}
     >
       <SEO pageTitle={pageTitle} siteTitle={siteTitle} />
-      <InstruqtProvider hasLabDrawer={page?.options?.instruqt}>
-        <Widgets
-          location={location}
-          pageOptions={page?.options}
-          pageTitle={pageTitle}
-          slug={slug}
-          template={template}
-          isInPresentationMode={isInPresentationMode}
-        >
-          <FootnoteContext.Provider value={{ footnotes }}>
-            <Template {...props} useChatbot={useChatbot}>
-              {pageNodes.map((child, index) => (
-                <ComponentFactory key={index} metadata={metadata} nodeData={child} page={page} slug={slug} />
-              ))}
-            </Template>
-          </FootnoteContext.Provider>
-        </Widgets>
-      </InstruqtProvider>
+      <TabProvider selectors={page?.options?.selectors}>
+        <InstruqtProvider hasLabDrawer={page?.options?.instruqt}>
+          <Widgets
+            location={location}
+            pageOptions={page?.options}
+            pageTitle={pageTitle}
+            slug={slug}
+            template={template}
+            isInPresentationMode={isInPresentationMode}
+          >
+            <FootnoteContext.Provider value={{ footnotes }}>
+              <Template {...props} useChatbot={useChatbot}>
+                {pageNodes.map((child, index) => (
+                  <ComponentFactory key={index} metadata={metadata} nodeData={child} page={page} slug={slug} />
+                ))}
+              </Template>
+            </FootnoteContext.Provider>
+          </Widgets>
+        </InstruqtProvider>
+      </TabProvider>
       <footer style={{ width: '100%', height: '568px', backgroundColor: '#061621' }}></footer>
     </Layout>
   );
diff --git a/src/components/RootProvider.js b/src/components/RootProvider.js
index ffce1b088..cc14e48bf 100644
--- a/src/components/RootProvider.js
+++ b/src/components/RootProvider.js
@@ -4,29 +4,25 @@ import { VersionContextProvider } from '../context/version-context';
 import { TocContextProvider } from '../context/toc-context';
 import { HeaderContextProvider } from './Header/header-context';
 import { SidenavContextProvider } from './Sidenav';
-import { TabProvider } from './Tabs/tab-context';
 import { ContentsProvider } from './Contents/contents-context';
 
-const RootProvider = ({ children, headingNodes, selectors, slug, repoBranches, remoteMetadata, project }) => {
+const RootProvider = ({ children, headingNodes, slug, repoBranches, remoteMetadata }) => {
   return (
-    <TabProvider selectors={selectors}>
-      <ContentsProvider headingNodes={headingNodes}>
-        <HeaderContextProvider>
-          <VersionContextProvider repoBranches={repoBranches} slug={slug}>
-            <TocContextProvider remoteMetadata={remoteMetadata}>
-              <SidenavContextProvider>{children}</SidenavContextProvider>
-            </TocContextProvider>
-          </VersionContextProvider>
-        </HeaderContextProvider>
-      </ContentsProvider>
-    </TabProvider>
+    <ContentsProvider headingNodes={headingNodes}>
+      <HeaderContextProvider>
+        <VersionContextProvider repoBranches={repoBranches} slug={slug}>
+          <TocContextProvider remoteMetadata={remoteMetadata}>
+            <SidenavContextProvider>{children}</SidenavContextProvider>
+          </TocContextProvider>
+        </VersionContextProvider>
+      </HeaderContextProvider>
+    </ContentsProvider>
   );
 };
 
 RootProvider.propTypes = {
   children: PropTypes.oneOfType([PropTypes.arrayOf(PropTypes.node), PropTypes.node]).isRequired,
   headingNodes: PropTypes.arrayOf(PropTypes.object),
-  selectors: PropTypes.object,
 };
 
 export default RootProvider;
diff --git a/src/components/Tabs/TabSelectors.js b/src/components/Tabs/TabSelectors.js
index 436b5fe54..c4c31679a 100644
--- a/src/components/Tabs/TabSelectors.js
+++ b/src/components/Tabs/TabSelectors.js
@@ -20,7 +20,7 @@ const getLabel = (name) => {
   }
 };
 
-const makeChoices = ({ name, iconMapping, options }) =>
+export const makeChoices = ({ name, iconMapping, options }) =>
   Object.entries(options).map(([tabId, title]) => ({
     text: getPlaintext(title),
     value: tabId,
diff --git a/src/components/Tabs/index.js b/src/components/Tabs/index.js
index 96abf5c4d..2f4a93be3 100644
--- a/src/components/Tabs/index.js
+++ b/src/components/Tabs/index.js
@@ -89,27 +89,18 @@ const Tabs = ({ nodeData: { children, options = {} }, page, ...rest }) => {
   const { activeTabs, selectors, setActiveTab } = useContext(TabContext);
   const tabIds = children.map((child) => getTabId(child));
   const tabsetName = options.tabset || generateAnonymousTabsetName(tabIds);
-  const [activeTab, setActiveTabIndex] = useState(0);
+  const [activeTab, setActiveTabIndex] = useState(() => {
+    // activeTabIdx at build time should be -1 if tabsetName !== drivers
+    // since no local storage to read, and no default tabs
+    const activeTabIdx = tabIds.indexOf(activeTabs?.[tabsetName]);
+    return activeTabIdx > -1 ? activeTabIdx : 0;
+  });
 
   const scrollAnchorRef = useRef();
-  const previousTabsetChoice = activeTabs[tabsetName];
   // Hide tabset if it includes the :hidden: option, or if it is controlled by a dropdown selector
   const isHidden = options.hidden || Object.keys(selectors).includes(tabsetName);
   const isProductLanding = page?.options?.template === 'product-landing';
 
-  useEffect(() => {
-    if (!previousTabsetChoice || !tabIds.includes(previousTabsetChoice)) {
-      const index = children.findIndex((item) => item.options.tabid === 'nodejs');
-      if (index > -1) {
-        // Set first tab to nodejs if no tab was previously selected
-        setActiveTab({ name: tabsetName, value: getTabId(children[index]) });
-      } else {
-        // Set first tab as active if no tab was previously selected and cant find nodejs
-        setActiveTab({ name: tabsetName, value: getTabId(children[0]) });
-      }
-    }
-  }, []); // eslint-disable-line react-hooks/exhaustive-deps
-
   useEffect(() => {
     const index = tabIds.indexOf(activeTabs[tabsetName]);
     if (index !== -1) {
diff --git a/src/components/Tabs/tab-context.js b/src/components/Tabs/tab-context.js
index 3f779a32b..c7ecbe144 100644
--- a/src/components/Tabs/tab-context.js
+++ b/src/components/Tabs/tab-context.js
@@ -1,3 +1,10 @@
+/**
+ * Context that contains language selector tab data
+ * Stores which tab is currently active by reading local storage
+ * or by setting default values, and allows
+ * child components to read and update
+ */
+
 import React, { useEffect, useReducer } from 'react';
 import IconC from '../icons/C';
 import IconCompass from '../icons/Compass';
@@ -19,6 +26,8 @@ import IconObjectiveC from '../icons/ObjectiveC';
 import IconJavascript from '../icons/Javascript';
 import IconTypescript from '../icons/Typescript';
 import IconDart from '../icons/Dart';
+import { isBrowser } from '../../utils/is-browser';
+import { makeChoices } from './TabSelectors';
 
 const DRIVER_ICON_MAP = {
   c: IconC,
@@ -64,8 +73,57 @@ const reducer = (prevState, { name, value }) => {
   };
 };
 
+const initActiveTabs = (choicesPerSelector, localActiveTabs) => {
+  // all tabbed content is read from browser local storage
+  // if there is no browser, wait for client side local storage
+  // hidden content should be handled from tab components
+  if (!isBrowser) {
+    return {};
+  }
+  // get default tabs based on availability
+  const defaultRes = Object.keys(choicesPerSelector || {}).reduce((res, selectorKey) => {
+    const nodeOptionIdx = choicesPerSelector[selectorKey].findIndex((tab) => tab.value === 'nodejs');
+    // NOTE: default tabs should be specified here
+    if (selectorKey === 'drivers' && nodeOptionIdx > -1) {
+      res[selectorKey] = 'nodejs';
+    } else {
+      res[selectorKey] = choicesPerSelector[selectorKey][0].value;
+    }
+    return res;
+  }, {});
+
+  // get local active tabs
+  const activeTabsRes = Object.keys(localActiveTabs || {}).reduce((res, activeTabKey) => {
+    if (typeof localActiveTabs[activeTabKey] === 'string') {
+      res[activeTabKey] = localActiveTabs[activeTabKey];
+    }
+    return res;
+  }, {});
+
+  // tabs initialize with default tabs overwritten by local storage tabs
+  const initialTabs = { ...defaultRes, ...activeTabsRes };
+  setLocalValue('activeTabs', initialTabs);
+  return initialTabs;
+};
+
 const TabProvider = ({ children, selectors = {} }) => {
-  const [activeTabs, setActiveTab] = useReducer(reducer, getLocalValue('activeTabs') || defaultContextValue.activeTabs);
+  // convert selectors to tab options first here, then set init values
+  // selectors are determined at build time
+
+  const choicesPerSelector = Object.keys(selectors).reduce((res, selector) => {
+    res[selector] = makeChoices({
+      name: selector,
+      options: selectors[selector],
+      ...(selector === 'drivers' && { iconMapping: DRIVER_ICON_MAP }),
+    });
+    return res;
+  }, {});
+
+  const [activeTabs, setActiveTab] = useReducer(
+    reducer,
+    getLocalValue('activeTabs'),
+    initActiveTabs.bind(null, choicesPerSelector)
+  );
 
   useEffect(() => {
     setLocalValue('activeTabs', activeTabs);
diff --git a/src/layouts/index.js b/src/layouts/index.js
index 25af705fd..10ff7662d 100644
--- a/src/layouts/index.js
+++ b/src/layouts/index.js
@@ -103,7 +103,6 @@ const DefaultLayout = ({ children, data: { page }, pageContext: { slug, repoBran
         slug={slug}
         repoBranches={repoBranches}
         headingNodes={page?.ast?.options?.headings}
-        selectors={page?.ast?.options?.selectors}
         remoteMetadata={remoteMetadata}
         project={project}
       >
@@ -140,6 +139,12 @@ DefaultLayout.propTypes = {
     slug: PropTypes.string,
     template: PropTypes.string,
   }).isRequired,
+  data: PropTypes.shape({
+    page: PropTypes.shape({
+      children: PropTypes.array,
+      options: PropTypes.object,
+    }).isRequired,
+  }).isRequired,
 };
 
 export default DefaultLayout;
diff --git a/src/layouts/preview-layout.js b/src/layouts/preview-layout.js
index d92a04ed6..476fbc8ba 100644
--- a/src/layouts/preview-layout.js
+++ b/src/layouts/preview-layout.js
@@ -153,7 +153,6 @@ const DefaultLayout = ({
           repoBranches={repoBranches}
           associatedReposInfo={associatedReposInfo}
           headingNodes={page?.options?.headings}
-          selectors={page?.options?.selectors}
           isAssociatedProduct={isAssociatedProduct}
         >
           <GlobalGrid>