Skip to content

Commit dd0434c

Browse files
authored
Merge pull request modelcontextprotocol#207 from samuel871211/perf_useTheme
perf: add `useMemo` to the return value of `useTheme`
2 parents 38fb710 + ed31f98 commit dd0434c

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

client/src/lib/useTheme.ts

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import { useCallback, useEffect, useState } from "react";
1+
import { useCallback, useEffect, useMemo, useState } from "react";
22

33
type Theme = "light" | "dark" | "system";
44

@@ -36,16 +36,14 @@ const useTheme = (): [Theme, (mode: Theme) => void] => {
3636
};
3737
}, [theme]);
3838

39-
return [
40-
theme,
41-
useCallback((newTheme: Theme) => {
42-
setTheme(newTheme);
43-
localStorage.setItem("theme", newTheme);
44-
if (newTheme !== "system") {
45-
document.documentElement.classList.toggle("dark", newTheme === "dark");
46-
}
47-
}, []),
48-
];
39+
const setThemeWithSideEffect = useCallback((newTheme: Theme) => {
40+
setTheme(newTheme);
41+
localStorage.setItem("theme", newTheme);
42+
if (newTheme !== "system") {
43+
document.documentElement.classList.toggle("dark", newTheme === "dark");
44+
}
45+
}, []);
46+
return useMemo(() => [theme, setThemeWithSideEffect], [theme]);
4947
};
5048

5149
export default useTheme;

0 commit comments

Comments
 (0)