From 8ac17b8ed032044cda83aa64d64365afcc8337a5 Mon Sep 17 00:00:00 2001
From: Michal <mwargan@gmail.com>
Date: Tue, 16 Apr 2024 00:27:19 +0000
Subject: [PATCH] feat: added theme setup and persistence

---
 src/components/PageFooter.vue              | 18 ++---
 src/eventBus/listeners/broadcastChannel.ts | 11 ++++
 src/main.ts                                |  3 +
 src/router/index.ts                        |  5 ++
 src/themes/__tests__/useTheme.spec.ts      | 59 +++++++++++++++++
 src/themes/useTheme.ts                     | 76 ++++++++++++++++++++++
 6 files changed, 158 insertions(+), 14 deletions(-)
 create mode 100644 src/themes/__tests__/useTheme.spec.ts
 create mode 100644 src/themes/useTheme.ts

diff --git a/src/components/PageFooter.vue b/src/components/PageFooter.vue
index 73017e4..57b4f2a 100644
--- a/src/components/PageFooter.vue
+++ b/src/components/PageFooter.vue
@@ -1,23 +1,12 @@
 <script setup lang="ts">
 import i18n, { SUPPORT_LOCALES, setI18nLanguage } from "@/locales/i18n";
-import { eventTypes, useEventsBus } from "@/eventBus/events";
+import theme, { setTheme } from "@/themes/useTheme";
 
 const appName = import.meta.env.VITE_APP_NAME;
 
-const $bus = useEventsBus();
-
 const handleLocaleChange = (locale: string) => {
   setI18nLanguage(i18n, locale);
 };
-
-const setDarkMode = (value: string) => {
-  if (value === "dark" || value === "light") {
-    document.documentElement.setAttribute("data-theme", value);
-  } else {
-    document.documentElement.removeAttribute("data-theme");
-  }
-  $bus.$emit(eventTypes.changed_theme, value);
-};
 </script>
 <template>
   <footer>
@@ -26,10 +15,11 @@ const setDarkMode = (value: string) => {
       <li>
         <select
           name="dark-mode"
-          @change="setDarkMode(($event.target as HTMLSelectElement).value)"
+          @change="setTheme(($event.target as HTMLSelectElement).value)"
           aria-label="Dark Mode toggle"
+          :value="theme"
         >
-          <option value="auto">{{ $t("Auto") }}</option>
+          <option value="system">{{ $t("Auto") }}</option>
           <option value="light">{{ $t("Light") }}</option>
           <option value="dark">{{ $t("Dark") }}</option>
         </select>
diff --git a/src/eventBus/listeners/broadcastChannel.ts b/src/eventBus/listeners/broadcastChannel.ts
index 02f3a7c..da7d6a0 100644
--- a/src/eventBus/listeners/broadcastChannel.ts
+++ b/src/eventBus/listeners/broadcastChannel.ts
@@ -1,6 +1,7 @@
 import i18n, { setI18nLanguage } from "@/locales/i18n";
 import router from "@/router";
 import { eventTypes } from "../events";
+import { setTheme } from "@/themes/useTheme";
 
 /**
  * Be careful here - its quite easy to accidentally set up an infinite loop
@@ -24,6 +25,10 @@ bc.onmessage = (event) => {
   if (event.data.type === eventTypes.changed_locale) {
     setI18nLanguage(i18n, event.data.data, false);
   }
+
+  if (event.data.type === eventTypes.changed_theme) {
+    setTheme(event.data.data, false);
+  }
 };
 
 export default {
@@ -45,4 +50,10 @@ export default {
       data: e,
     });
   },
+  changed_theme: (e: string) => {
+    bc.postMessage({
+      type: eventTypes.changed_theme,
+      data: e,
+    });
+  },
 } as Record<eventTypes, any>;
diff --git a/src/main.ts b/src/main.ts
index 71103f6..3db41df 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -12,6 +12,7 @@ import "./eventBus/listeners/index";
 import VueGtagPlugin from "vue-gtag";
 import "./assets/main.css";
 import i18n, { SUPPORT_LOCALES } from "./locales/i18n";
+import { ThemePlugin } from "./themes/useTheme";
 import { gatePlugin } from "@m-media/vue3-gate-keeper";
 
 import gates from "./router/gates";
@@ -71,4 +72,6 @@ app.use(
 
 app.use(EventsPlugin);
 
+app.use(ThemePlugin);
+
 app.mount("#app");
diff --git a/src/router/index.ts b/src/router/index.ts
index eb9ce22..e01a4c3 100644
--- a/src/router/index.ts
+++ b/src/router/index.ts
@@ -52,6 +52,11 @@ const router = createRouter({
       name: "Examples",
       component: () => import("../views/Examples/DealsView.vue"),
     },
+    {
+      path: "/examples/single-product",
+      name: "Examples",
+      component: () => import("../views/Examples/SingleProductView.vue"),
+    },
     // Add a catch-all 404 page
     {
       path: "/:pathMatch(.*)*",
diff --git a/src/themes/__tests__/useTheme.spec.ts b/src/themes/__tests__/useTheme.spec.ts
new file mode 100644
index 0000000..21c4298
--- /dev/null
+++ b/src/themes/__tests__/useTheme.spec.ts
@@ -0,0 +1,59 @@
+import { beforeEach, describe, expect, it } from "vitest";
+
+import { getCurrentTheme, setBestGuessTheme, setTheme } from "../useTheme"; // Assuming this is the correct path to your module
+
+// We need to get jest to mock the window.matchMedia function
+
+describe("Theme functionality", () => {
+  beforeEach(() => {
+    // Reset localStorage and document theme attribute before each test
+    localStorage.clear();
+    document.documentElement.removeAttribute("data-theme");
+  });
+
+  it("should set and get dark mode correctly", () => {
+    setTheme("dark");
+    expect(document.documentElement.getAttribute("data-theme")).toEqual("dark");
+    expect(localStorage.getItem("theme")).toEqual("dark");
+    expect(getCurrentTheme()).toEqual("dark");
+
+    setTheme("light");
+    expect(document.documentElement.getAttribute("data-theme")).toEqual(
+      "light"
+    );
+    expect(localStorage.getItem("theme")).toEqual("light");
+    expect(getCurrentTheme()).toEqual("light");
+
+    setTheme("system");
+    // data-theme should be removed when set to system
+    expect(document.documentElement.getAttribute("data-theme")).toBeNull();
+    expect(localStorage.getItem("theme")).toEqual("system");
+    expect(getCurrentTheme()).toEqual("system");
+  });
+
+  it("should set best guess theme correctly", () => {
+    // Ensure localStorage is empty
+    expect(localStorage.getItem("theme")).toBeNull();
+
+    // Simulate no localStorage or document theme attribute set
+    setBestGuessTheme();
+    expect(document.documentElement.getAttribute("data-theme")).toBeNull();
+    expect(localStorage.getItem("theme")).toEqual("system");
+
+    // Simulate localStorage set
+    localStorage.setItem("theme", "dark");
+    setBestGuessTheme();
+    expect(document.documentElement.getAttribute("data-theme")).toEqual("dark");
+    expect(localStorage.getItem("theme")).toEqual("dark");
+
+    // Simulate document theme attribute set
+    // Remove localStorage
+    localStorage.removeItem("theme");
+    document.documentElement.setAttribute("data-theme", "light");
+    setBestGuessTheme();
+    expect(document.documentElement.getAttribute("data-theme")).toEqual(
+      "light"
+    );
+    expect(localStorage.getItem("theme")).toEqual("light");
+  });
+});
diff --git a/src/themes/useTheme.ts b/src/themes/useTheme.ts
new file mode 100644
index 0000000..8c88ac3
--- /dev/null
+++ b/src/themes/useTheme.ts
@@ -0,0 +1,76 @@
+import $bus, { eventTypes } from "@/eventBus/events";
+import { ref } from "vue";
+
+/**
+ * The supported themes. Dark, light, and system.
+ *
+ */
+export const SUPPORT_THEMES = ["system", "dark", "light"];
+
+export const getBestGuessTheme = () => {
+  return (
+    localStorage.getItem("theme") ??
+    // Read the data-theme attribute from the html element
+    document.documentElement.getAttribute("data-theme") ??
+    // Use the first supported theme
+    SUPPORT_THEMES[0]
+  );
+};
+
+const currentTheme = ref(getBestGuessTheme());
+
+export const setTheme = (value: string, emit = true) => {
+  if (value === "system") {
+    document.documentElement.removeAttribute("data-theme");
+  } else {
+    document.documentElement.setAttribute("data-theme", value);
+  }
+  localStorage.setItem("theme", value);
+  currentTheme.value = value;
+  if (emit) {
+    $bus.$emit(eventTypes.changed_theme, value);
+  }
+};
+
+export const getCurrentTheme = () => {
+  return document.documentElement.getAttribute("data-theme") ?? "system";
+};
+
+/**
+ * Set the current app theme to the best-guessed theme
+ */
+export function setBestGuessTheme() {
+  let theme = getBestGuessTheme();
+  // If the theme is not supported, fallback to system
+  if (!SUPPORT_THEMES.includes(theme)) {
+    theme = SUPPORT_THEMES[0];
+  }
+
+  setTheme(theme);
+}
+
+/**
+ * Setup theme
+ */
+export function setupTheme() {
+  setBestGuessTheme();
+
+  window
+    .matchMedia("(prefers-color-scheme: dark)")
+    .addEventListener("change", (e: MediaQueryListEvent) => {
+      setTheme(e.matches ? "dark" : "light", false);
+    });
+}
+
+/**
+ * Export as Vue3 plugin
+ *
+ */
+export const ThemePlugin = {
+  install: () => {
+    // Install by running the setup function
+    setupTheme();
+  },
+};
+
+export default currentTheme;