Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-12 04:18:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use hook with first render (#2481)

Browse Source
This commit is contained in:
Himself65
2023-05-22 15:58:13 +08:00
committed by GitHub
parent 5fbfabb3b2
commit 2eaaeef4a7
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
apps/web/src/providers/theme-provider.tsx
View File

@@ -4,18 +4,16 @@ import { memo, useRef } from 'react';
const themes = ['dark', 'light'];
// a workaround to sync theme to electron
let firstRender = true;
const DesktopThemeSync = memo(function DesktopThemeSync() {
const { theme } = useTheme();
const lastThemeRef = useRef(theme);
if (lastThemeRef.current !== theme || firstRender) {
const onceRef = useRef(false);
if (lastThemeRef.current !== theme || !onceRef.current) {
if (environment.isDesktop && theme) {
window.apis?.ui.handleThemeChange(theme as 'dark' | 'light' | 'system');
}
lastThemeRef.current = theme;
firstRender = false;
onceRef.current = true;
}
return null;
});