From 6215310fa14f51f05cff9aad8466a68325438cd3 Mon Sep 17 00:00:00 2001
From: ayan4m1 <andrew@bulletlogic.com>
Date: Sat, 13 Jan 2024 05:25:27 -0500
Subject: [PATCH] add theme context for switching dark mode and remembering
 with localStorage

---
 src/components/header.js        | 48 +++++++++------------------------
 src/components/layout.js        |  9 +++----
 src/components/themeProvider.js | 43 +++++++++++++++++++++++++++++
 src/hooks/useDarkMode.js        | 24 -----------------
 src/hooks/useMediaQuery.js      | 11 +++++---
 src/hooks/useThemeContext.js    |  7 +++++
 6 files changed, 73 insertions(+), 69 deletions(-)
 create mode 100644 src/components/themeProvider.js
 delete mode 100644 src/hooks/useDarkMode.js
 create mode 100644 src/hooks/useThemeContext.js

diff --git a/src/components/header.js b/src/components/header.js
index 05007ff..14b48c6 100644
--- a/src/components/header.js
+++ b/src/components/header.js
@@ -1,34 +1,22 @@
-import { useSpring, animated } from '@react-spring/web';
-import { Link, useStaticQuery, graphql } from 'gatsby';
-import { Container, Navbar, Nav, NavDropdown } from 'react-bootstrap';
-
-import usePrefersReducedMotion from 'hooks/usePrefersReducedMotion';
 import { FontAwesomeIcon } from '@fortawesome/react-fontawesome';
 import {
   faFileArchive,
   faGamepad,
   faHeart,
+  faMoon,
   faNewspaper,
   faRectangleList,
+  faSun,
   faTable
 } from '@fortawesome/free-solid-svg-icons';
 import { Fragment } from 'react';
+import { Link, useStaticQuery, graphql } from 'gatsby';
+import { Container, Navbar, Nav, NavDropdown, Button } from 'react-bootstrap';
 
-const bobbleSpring = {
-  from: { y: 4 },
-  to: [{ y: -4 }, { y: 4 }],
-  config: {
-    mass: 0.1,
-    tension: 20,
-    friction: 0.1
-  },
-  loop: true,
-  reset: false
-};
+import { useThemeContext } from 'hooks/useThemeContext';
 
 export default function Header() {
-  const [springs] = useSpring(() => bobbleSpring);
-  const disableMotion = usePrefersReducedMotion();
+  const { darkMode, toggleDarkMode } = useThemeContext();
   const data = useStaticQuery(graphql`
     query {
       site {
@@ -97,23 +85,13 @@ export default function Header() {
               </Nav.Link>
             </NavDropdown>
           </Nav>
-          <Nav className="ms-auto text-light">
-            {disableMotion ? (
-              <p style={{ textShadow: '1px 1px 4px rgba(0,0,0,0.6)' }}>
-                More Coming Soon &trade;
-              </p>
-            ) : (
-              <animated.p
-                className="mb-0"
-                style={{
-                  position: 'relative',
-                  textShadow: '1px 1px 4px rgba(0,0,0,0.6)',
-                  ...springs
-                }}
-              >
-                More Coming Soon &trade;
-              </animated.p>
-            )}
+          <Nav
+            className="ms-lg-auto text-light d-flex align-items-center justify-content-end"
+            style={{ flexDirection: 'row' }}
+          >
+            <Button variant="info" onClick={toggleDarkMode}>
+              <FontAwesomeIcon icon={darkMode ? faMoon : faSun} fixedWidth />
+            </Button>
           </Nav>
         </Navbar.Collapse>
       </Container>
diff --git a/src/components/layout.js b/src/components/layout.js
index f9da415..d9f06bb 100644
--- a/src/components/layout.js
+++ b/src/components/layout.js
@@ -1,25 +1,22 @@
 import PropTypes from 'prop-types';
-import { Fragment } from 'react';
 import Snowfall from 'react-snowfall';
 import { getDate, getMonth } from 'date-fns';
 
 import Header from 'components/header';
-import useDarkMode from 'hooks/useDarkMode';
 import usePrefersReducedMotion from 'hooks/usePrefersReducedMotion';
+import ThemeProvider from './themeProvider';
 
 export default function Layout({ children }) {
-  useDarkMode();
-
   const disableMotion = usePrefersReducedMotion();
   const isChristmas =
     getMonth(Date.now()) === 11 && [24, 25, 26].includes(getDate(Date.now()));
 
   return (
-    <Fragment>
+    <ThemeProvider>
       {!disableMotion && isChristmas && <Snowfall />}
       <Header />
       <main className="mt-3 mb-2">{children}</main>
-    </Fragment>
+    </ThemeProvider>
   );
 }
 
diff --git a/src/components/themeProvider.js b/src/components/themeProvider.js
new file mode 100644
index 0000000..3db8d6d
--- /dev/null
+++ b/src/components/themeProvider.js
@@ -0,0 +1,43 @@
+import PropTypes from 'prop-types';
+import { useCallback, useEffect } from 'react';
+import useLocalStorageState from 'use-local-storage-state';
+
+import useMediaQuery from 'hooks/useMediaQuery';
+import { ThemeContext } from 'hooks/useThemeContext';
+
+const mediaQuery = '(prefers-color-scheme: dark)';
+
+export default function ThemeProvider({ children }) {
+  const [darkMode, setDarkMode] = useLocalStorageState('darkMode', {
+    defaultValue: useMediaQuery(mediaQuery)
+  });
+  const toggleDarkMode = useCallback(
+    () => setDarkMode((val) => !val),
+    [setDarkMode]
+  );
+
+  useEffect(() => {
+    if (darkMode) {
+      document.body.classList.remove('light-mode');
+      document.body.classList.add('dark-mode');
+    } else {
+      document.body.classList.remove('dark-mode');
+      document.body.classList.add('light-mode');
+    }
+  }, [darkMode]);
+
+  return (
+    <ThemeContext.Provider
+      value={{
+        darkMode,
+        toggleDarkMode
+      }}
+    >
+      {children}
+    </ThemeContext.Provider>
+  );
+}
+
+ThemeProvider.propTypes = {
+  children: PropTypes.node.isRequired
+};
diff --git a/src/hooks/useDarkMode.js b/src/hooks/useDarkMode.js
deleted file mode 100644
index 1eb0965..0000000
--- a/src/hooks/useDarkMode.js
+++ /dev/null
@@ -1,24 +0,0 @@
-import { useCallback, useEffect, useState } from 'react';
-import useMediaQuery from './useMediaQuery';
-
-const mediaQuery = '(prefers-color-scheme: dark)';
-
-export default function useDarkMode() {
-  const [state, setState] = useState(useMediaQuery(mediaQuery));
-  const toggle = useCallback(() => setState((prevVal) => !prevVal), []);
-
-  useEffect(() => {
-    if (state) {
-      document.body.classList.remove('light-mode');
-      document.body.classList.add('dark-mode');
-    } else {
-      document.body.classList.remove('dark-mode');
-      document.body.classList.add('light-mode');
-    }
-  }, [state]);
-
-  return {
-    state,
-    toggle
-  };
-}
diff --git a/src/hooks/useMediaQuery.js b/src/hooks/useMediaQuery.js
index 6ccedd5..7b34ab4 100644
--- a/src/hooks/useMediaQuery.js
+++ b/src/hooks/useMediaQuery.js
@@ -3,6 +3,12 @@ import { useState, useEffect } from 'react';
 const queryMatches = (query) =>
   typeof window !== 'undefined' ? window.matchMedia(query).matches : false;
 
+const bindListener = (object, event, handler) => {
+  object.addEventListener(event, handler);
+
+  return () => object.removeEventListener(event, handler);
+};
+
 export default function useMediaQuery(query) {
   const [matches, setMatches] = useState(queryMatches(query));
 
@@ -11,12 +17,9 @@ export default function useMediaQuery(query) {
       return;
     }
 
-    const mediaQueryList = window.matchMedia(query);
     const listener = (event) => setMatches(event.matches);
 
-    mediaQueryList.addEventListener('change', listener);
-
-    return () => mediaQueryList.removeEventListener('change', listener);
+    return bindListener(window.matchMedia(query), 'change', listener);
   }, [query]);
 
   return matches;
diff --git a/src/hooks/useThemeContext.js b/src/hooks/useThemeContext.js
new file mode 100644
index 0000000..ba6e43a
--- /dev/null
+++ b/src/hooks/useThemeContext.js
@@ -0,0 +1,7 @@
+import { createContext, useContext } from 'react';
+
+export const ThemeContext = createContext({
+  darkMode: false
+});
+
+export const useThemeContext = () => useContext(ThemeContext);