From c3ac8d9728ad15224dca2a6e8b99177ecf07661e Mon Sep 17 00:00:00 2001
From: xingo xu <xingoxu@users.noreply.github.com>
Date: Thu, 6 Jun 2024 02:20:05 +0900
Subject: [PATCH] feat: light/dark modes switch immediately with the system
 theme (#95)

---
 src/components/LightDarkSwitch.svelte | 54 +++++++++++++++++++++------
 src/types/config.ts                   |  6 ++-
 src/utils/setting-utils.ts            | 24 +++++++-----
 3 files changed, 62 insertions(+), 22 deletions(-)

diff --git a/src/components/LightDarkSwitch.svelte b/src/components/LightDarkSwitch.svelte
index 602fbf0ed..abd982437 100644
--- a/src/components/LightDarkSwitch.svelte
+++ b/src/components/LightDarkSwitch.svelte
@@ -1,22 +1,52 @@
 <script lang="ts">
-
-import Icon from "@iconify/svelte"
-import {i18n} from '@i18n/translation'
+import type { LIGHT_DARK_MODE } from '@/types/config.ts'
+import {
+  AUTO_MODE,
+  DARK_MODE,
+  LIGHT_MODE,
+} from '@constants/constants.ts'
 import I18nKey from '@i18n/i18nKey'
-import {setTheme, getStoredTheme} from '../utils/setting-utils.ts'
-import {onMount} from "svelte";
-import {AUTO_MODE, DARK_MODE, LIGHT_MODE} from "@constants/constants.ts";
+import { i18n } from '@i18n/translation'
+import Icon from '@iconify/svelte'
+import {
+  applyThemeToDocument,
+  getStoredTheme,
+  setTheme,
+} from '@utils/setting-utils.ts'
+import { onMount } from 'svelte'
 
-const seq = [LIGHT_MODE, DARK_MODE, AUTO_MODE]
-let mode = AUTO_MODE
+const seq: LIGHT_DARK_MODE[] = [
+  LIGHT_MODE,
+  DARK_MODE,
+  AUTO_MODE,
+]
+let mode: LIGHT_DARK_MODE = AUTO_MODE
 
 onMount(() => {
-    mode = getStoredTheme()
+  mode = getStoredTheme()
+  const darkModePreference = window.matchMedia(
+    '(prefers-color-scheme: dark)',
+  )
+  const changeThemeWhenSchemeChanged: Parameters<
+    typeof darkModePreference.addEventListener<'change'>
+  >[1] = e => {
+    applyThemeToDocument(mode)
+  }
+  darkModePreference.addEventListener(
+    'change',
+    changeThemeWhenSchemeChanged,
+  )
+  return () => {
+    darkModePreference.removeEventListener(
+      'change',
+      changeThemeWhenSchemeChanged,
+    )
+  }
 })
 
-function switchScheme(newMode: string) {
-    mode = newMode
-    setTheme(newMode)
+function switchScheme(newMode: LIGHT_DARK_MODE) {
+  mode = newMode
+  setTheme(newMode)
 }
 
 function toggleScheme() {
diff --git a/src/types/config.ts b/src/types/config.ts
index 1dd5c9f6b..ec4cd6a95 100644
--- a/src/types/config.ts
+++ b/src/types/config.ts
@@ -1,3 +1,5 @@
+import type { LIGHT_MODE, DARK_MODE, AUTO_MODE } from "@constants/constants"
+
 export type SiteConfig = {
   title: string
   subtitle: string
@@ -17,7 +19,7 @@ export type SiteConfig = {
 }
 
 export type Favicon = {
-  src: string,
+  src: string
   theme?: 'light' | 'dark'
   sizes?: string
 }
@@ -54,3 +56,5 @@ export type LicenseConfig = {
   name: string
   url: string
 }
+
+export type LIGHT_DARK_MODE = typeof LIGHT_MODE | typeof DARK_MODE | typeof AUTO_MODE
diff --git a/src/utils/setting-utils.ts b/src/utils/setting-utils.ts
index 5ef906bea..647190101 100644
--- a/src/utils/setting-utils.ts
+++ b/src/utils/setting-utils.ts
@@ -1,4 +1,5 @@
 import {AUTO_MODE, DARK_MODE, DEFAULT_THEME, LIGHT_MODE} from "@constants/constants.ts";
+import type { LIGHT_DARK_MODE } from '@/types/config'
 
 export function getDefaultHue(): number {
   const fallback = '250'
@@ -20,25 +21,30 @@ export function setHue(hue: number): void {
   r.style.setProperty('--hue', hue)
 }
 
-export function setTheme(theme: string): void {
-  localStorage.setItem('theme', theme)
+
+export function applyThemeToDocument(theme: LIGHT_DARK_MODE) {
   switch (theme) {
     case LIGHT_MODE:
-      document.documentElement.classList.remove('dark');
+      document.documentElement.classList.remove('dark')
       break
     case DARK_MODE:
-      document.documentElement.classList.add('dark');
+      document.documentElement.classList.add('dark')
       break
     case AUTO_MODE:
       if (window.matchMedia('(prefers-color-scheme: dark)').matches) {
-        document.documentElement.classList.add('dark');
+        document.documentElement.classList.add('dark')
       } else {
-        document.documentElement.classList.remove('dark');
+        document.documentElement.classList.remove('dark')
       }
       break
   }
 }
 
-export function getStoredTheme(): string {
-  return localStorage.getItem('theme') || DEFAULT_THEME
-}
\ No newline at end of file
+export function setTheme(theme: LIGHT_DARK_MODE): void {
+  localStorage.setItem('theme', theme)
+  applyThemeToDocument(theme)
+}
+
+export function getStoredTheme(): LIGHT_DARK_MODE {
+  return localStorage.getItem('theme') as LIGHT_DARK_MODE || DEFAULT_THEME
+}