fix(v2): set theme color mode for SSR (#4383)

facebook · Mar 10, 2021 · 6afa83c · 6afa83c
1 parent 02cd5d3
commit 6afa83c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/docusaurus-theme-classic/src/theme/hooks/useTheme.ts b/packages/docusaurus-theme-classic/src/theme/hooks/useTheme.ts
@@ -21,9 +21,9 @@ const coerceToTheme = (theme) => {
   return theme === themes.dark ? themes.dark : themes.light;
 };
 
-const getInitialTheme = () => {
+const getInitialTheme = (defaultMode) => {
   if (!ExecutionEnvironment.canUseDOM) {
-    return themes.light; // SSR: we don't care
+    return coerceToTheme(defaultMode);
   }
   return coerceToTheme(document.documentElement.getAttribute('data-theme'));
 };
@@ -38,9 +38,9 @@ const storeTheme = (newTheme) => {
 
 const useTheme = (): useThemeReturns => {
   const {
-    colorMode: {disableSwitch, respectPrefersColorScheme},
+    colorMode: {defaultMode, disableSwitch, respectPrefersColorScheme},
   } = useThemeConfig();
-  const [theme, setTheme] = useState(getInitialTheme);
+  const [theme, setTheme] = useState(getInitialTheme(defaultMode));
 
   const setLightTheme = useCallback(() => {
     setTheme(themes.light);