From fb828c9cf4304b857798d3d0dd4aedf125342e53 Mon Sep 17 00:00:00 2001
From: QiShaoXuan <qishaoxuan777@gmail.com>
Date: Tue, 27 Sep 2022 17:19:23 +0800
Subject: [PATCH] feat: add system theme judgment

---
 src/styles/themeProvider.tsx | 39 +++++++++++++++++++++++++++++++++---
 src/styles/types.ts          |  2 +-
 src/styles/utils.ts          | 39 ++++++++++++++++++++++++++++++++++++
 3 files changed, 76 insertions(+), 4 deletions(-)
 create mode 100644 src/styles/utils.ts

diff --git a/src/styles/themeProvider.tsx b/src/styles/themeProvider.tsx
index 383ae2dd74..27f94ca682 100644
--- a/src/styles/themeProvider.tsx
+++ b/src/styles/themeProvider.tsx
@@ -3,10 +3,11 @@ import {
   Global,
   css,
 } from '@emotion/react';
-import { createContext, useState } from 'react';
+import { createContext, useEffect, useRef, useState } from 'react';
 import type { PropsWithChildren } from 'react';
 import { ThemeMode, ThemeProviderProps, ThemeProviderValue } from './types';
 import { lightTheme, darkTheme, globalThemeConstant } from './theme';
+import { SystemTheme, LocalStorageThemeMode } from '@/styles/utils';
 
 export const ThemeContext = createContext<ThemeProviderValue>({
   mode: 'light',
@@ -15,15 +16,47 @@ export const ThemeContext = createContext<ThemeProviderValue>({
 });
 
 export const ThemeProvider = ({
-  defaultMode = 'light',
+  defaultThemeMode = 'light',
   children,
 }: PropsWithChildren<ThemeProviderProps>) => {
-  const [mode, setMode] = useState<ThemeMode>(defaultMode);
+  const { current: localStorageThemeMode } = useRef(
+    new LocalStorageThemeMode()
+  );
+
+  const [mode, setMode] = useState<ThemeMode>(defaultThemeMode);
   const theme = mode === 'dark' ? darkTheme : lightTheme;
+
   const changeMode = (themeMode: ThemeMode) => {
     setMode(themeMode);
+    // Remember the theme mode which user selected for next time
+    localStorageThemeMode.set(themeMode);
   };
 
+  useEffect(() => {
+    const systemTheme = new SystemTheme();
+
+    // TODO: System theme mode and user theme mode which selected need to be prioritized
+    // If user has selected a theme mode, use it
+    const selectedThemeMode = localStorageThemeMode.get();
+    if (selectedThemeMode) {
+      defaultThemeMode !== selectedThemeMode && setMode(selectedThemeMode);
+    } else {
+      // If user has not selected a theme mode, use system theme
+      const systemThemeMode = systemTheme.get();
+      defaultThemeMode !== systemThemeMode && setMode(systemThemeMode);
+    }
+
+    // When system theme changed, change the theme mode
+    systemTheme.onChange(() => {
+      // TODO: There may be should be provided a way to let user choose whether to
+      setMode(systemTheme.get());
+    });
+
+    return () => {
+      systemTheme.dispose();
+    };
+  }, []);
+
   return (
     <ThemeContext.Provider value={{ mode, changeMode, theme }}>
       <Global
diff --git a/src/styles/types.ts b/src/styles/types.ts
index e3a9ce0476..ccfb39290a 100644
--- a/src/styles/types.ts
+++ b/src/styles/types.ts
@@ -1,7 +1,7 @@
 export type ThemeMode = 'light' | 'dark';
 
 export type ThemeProviderProps = {
-  defaultMode?: ThemeMode;
+  defaultThemeMode?: ThemeMode;
 };
 
 export type ThemeProviderValue = {
diff --git a/src/styles/utils.ts b/src/styles/utils.ts
new file mode 100644
index 0000000000..bdc0ef8f18
--- /dev/null
+++ b/src/styles/utils.ts
@@ -0,0 +1,39 @@
+import { ThemeMode } from '@/styles/types';
+
+export class SystemTheme {
+  media: MediaQueryList = window.matchMedia('(prefers-color-scheme: light)');
+  eventList: Array<(e: Event) => void> = [];
+  eventHandler = (e: Event) => {
+    this.eventList.forEach(fn => fn(e));
+  };
+
+  constructor() {
+    this.media.addEventListener('change', this.eventHandler);
+  }
+
+  get = (): ThemeMode => {
+    if (typeof window === 'undefined') {
+      return 'light';
+    }
+    return this.media.matches ? 'light' : 'dark';
+  };
+
+  onChange = (callback: () => void) => {
+    this.eventList.push(callback);
+  };
+
+  dispose = () => {
+    this.eventList = [];
+    this.media.removeEventListener('change', this.eventHandler);
+  };
+}
+
+export class LocalStorageThemeMode {
+  name = 'Affine-theme-mode';
+  get = (): ThemeMode | null => {
+    return localStorage.getItem(this.name) as ThemeMode | null;
+  };
+  set = (mode: ThemeMode) => {
+    localStorage.setItem(this.name, mode);
+  };
+}