Wrap in requestAnimationFrame

zer0kool · Aug 21, 2024 · c11125d · c11125d
1 parent 3ca91ae
commit c11125d
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/theme.js b/theme.js
@@ -3,20 +3,21 @@ const themeMap = {
   dark: "light"
 };
 
-const theme = localStorage.getItem('theme')
-  || (tmp = Object.keys(themeMap)[0],
-      localStorage.setItem('theme', tmp),
-      tmp);
+const theme = localStorage.getItem('theme') || (tmp => {
+  localStorage.setItem('theme', tmp = Object.keys(themeMap)[0]);
+  return tmp;
+})();
 const bodyClass = document.body.classList;
 bodyClass.add(theme);
 
 function toggleTheme() {
   const current = localStorage.getItem('theme');
   const next = themeMap[current];
 
-  bodyClass.replace(current, next);
-  localStorage.setItem('theme', next);
+  requestAnimationFrame(() => { // Wrap in requestAnimationFrame
+    bodyClass.replace(current, next);
+    localStorage.setItem('theme', next);
+  });
 }
 
 document.getElementById('themeButton').onclick = toggleTheme;
-