From 2aa0b81fb90ab95a4e806bb0673515d7ea347e71 Mon Sep 17 00:00:00 2001
From: Corban Riley <corbanbrook@gmail.com>
Date: Mon, 6 Nov 2023 12:32:30 -0500
Subject: [PATCH] Added root and scope props to ThemeProvider which allow
 ThemeProvider to attach theme selector to a specific root element and persist
 it under a specific scope name

---
 .../ThemeProvider/ThemeProvider.stories.tsx   | 73 +++++++++++++++++++
 .../ThemeProvider/ThemeProvider.tsx           | 26 ++++---
 2 files changed, 90 insertions(+), 9 deletions(-)

diff --git a/src/components/ThemeProvider/ThemeProvider.stories.tsx b/src/components/ThemeProvider/ThemeProvider.stories.tsx
index 150e4b46a..d880086f3 100644
--- a/src/components/ThemeProvider/ThemeProvider.stories.tsx
+++ b/src/components/ThemeProvider/ThemeProvider.stories.tsx
@@ -2,6 +2,11 @@ import { Meta } from '@storybook/react'
 
 import { Button } from '~/components/Button'
 
+import { Box } from '../Box'
+import { Card } from '../Card'
+import { Collapsible } from '../Collapsible'
+import { Text } from '../Text'
+
 import { ThemeProvider, useTheme } from './ThemeProvider'
 
 export default {
@@ -17,3 +22,71 @@ export const Default = () => {
 
   return <Button label="Toggle theme" onClick={toggleTheme} />
 }
+
+export const Nested = () => {
+  return (
+    <Card>
+      <Text variant="normal" color="text100" fontWeight="bold">
+        Root Application
+      </Text>
+
+      <div id="app1">
+        <ThemeProvider root="#app1" scope="application1" theme="light">
+          <Card background="backgroundPrimary" marginTop="4">
+            <Collapsible label="Nested Application 1">
+              <Text variant="normal" color="text100">
+                Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do
+                eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut
+                enim ad minim veniam, quis nostrud exercitation ullamco laboris
+                nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor
+                in reprehenderit in voluptate velit esse cillum dolore eu fugiat
+                nulla pariatur. Excepteur sint occaecat cupidatat non proident,
+                sunt in culpa qui officia deserunt mollit anim id est laborum.
+              </Text>
+
+              <Box marginTop="4">
+                <SetThemeButton />
+              </Box>
+
+              <div id="app2">
+                <ThemeProvider root="#app2" scope="application2" theme="dark">
+                  <Card background="backgroundPrimary" marginTop="4">
+                    <Collapsible label="Nested Application 2">
+                      <Text variant="normal" color="text100">
+                        Lorem ipsum dolor sit amet, consectetur adipiscing elit,
+                        sed do eiusmod tempor incididunt ut labore et dolore
+                        magna aliqua. Ut enim ad minim veniam, quis nostrud
+                        exercitation ullamco laboris nisi ut aliquip ex ea
+                        commodo consequat. Duis aute irure dolor in
+                        reprehenderit in voluptate velit esse cillum dolore eu
+                        fugiat nulla pariatur. Excepteur sint occaecat cupidatat
+                        non proident, sunt in culpa qui officia deserunt mollit
+                        anim id est laborum.
+                      </Text>
+
+                      <Box marginTop="4">
+                        <SetThemeButton />
+                      </Box>
+                    </Collapsible>
+                  </Card>
+                </ThemeProvider>
+              </div>
+            </Collapsible>
+          </Card>
+        </ThemeProvider>
+      </div>
+    </Card>
+  )
+}
+
+const SetThemeButton = () => {
+  const { theme, setTheme } = useTheme()
+
+  const themeLabel = theme === 'light' ? 'Dark' : 'Light'
+
+  const toggleTheme = () => {
+    setTheme(theme === 'light' ? 'dark' : 'light')
+  }
+
+  return <Button label={`Set ${themeLabel} Mode`} onClick={toggleTheme} />
+}
diff --git a/src/components/ThemeProvider/ThemeProvider.tsx b/src/components/ThemeProvider/ThemeProvider.tsx
index 39afab4aa..3826355a5 100644
--- a/src/components/ThemeProvider/ThemeProvider.tsx
+++ b/src/components/ThemeProvider/ThemeProvider.tsx
@@ -15,6 +15,9 @@ const DEFAULT_THEME = 'dark'
 const THEME_ATTR = 'data-theme'
 const STORAGE_KEY = '@sequence.theme'
 
+const getStorageKey = (scope?: string) =>
+  scope ? `${STORAGE_KEY}.${scope}` : STORAGE_KEY
+
 interface ThemeContextValue {
   theme: Theme
   setTheme: (mode: Theme) => void
@@ -22,10 +25,12 @@ interface ThemeContextValue {
 
 interface ThemeProviderProps {
   theme?: Theme
+  root?: string
+  scope?: string
 }
 
-const getTheme = (): Theme => {
-  const persistedTheme = localStorage.getItem(STORAGE_KEY) as Theme
+const getTheme = (scope?: string): Theme => {
+  const persistedTheme = localStorage.getItem(getStorageKey(scope)) as Theme
 
   if (THEMES.includes(persistedTheme)) {
     return persistedTheme
@@ -43,7 +48,9 @@ const getTheme = (): Theme => {
 const ThemeContext = createContext<ThemeContextValue | null>(null)
 
 export const ThemeProvider = (props: PropsWithChildren<ThemeProviderProps>) => {
-  const [theme, setTheme] = useState<Theme>(props.theme || getTheme())
+  const [theme, setTheme] = useState<Theme>(
+    props.theme || getTheme(props.scope)
+  )
 
   useEffect(() => {
     // Add is-apple class
@@ -60,12 +67,13 @@ export const ThemeProvider = (props: PropsWithChildren<ThemeProviderProps>) => {
 
   // Set the data-theme attribtute on the document root element
   useEffect(() => {
-    const root = document.querySelector(':root')
+    const rootEl = document.querySelector(props.root || ':root')
 
-    if (root) {
-      root.setAttribute(THEME_ATTR, theme)
+    if (rootEl) {
+      console.log('Found', props.root || ':root')
+      rootEl.setAttribute(THEME_ATTR, theme)
     }
-  }, [theme])
+  }, [theme, props.root])
 
   // Create the context value
   const value: ThemeContextValue = useMemo(() => {
@@ -74,14 +82,14 @@ export const ThemeProvider = (props: PropsWithChildren<ThemeProviderProps>) => {
       setTheme: (mode: Theme) => {
         if (THEMES.includes(mode)) {
           // Save to local storage
-          localStorage.setItem(STORAGE_KEY, mode)
+          localStorage.setItem(getStorageKey(props.scope), mode)
 
           // Set the theme state which will cause a re-render
           setTheme(mode)
         }
       },
     }
-  }, [theme])
+  }, [theme, props.scope])
 
   return (
     <ThemeContext.Provider value={value}>