fix: Add the theme attribute explicitly

gorzelinski · Nov 25, 2024 · 810f9b3 · 810f9b3
1 parent d697a8a
commit 810f9b3
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/hooks/use-theme.tsx b/hooks/use-theme.tsx
@@ -10,7 +10,9 @@ export function useTheme() {
   const saveTheme = (theme: Theme) => {
     setTheme(theme)
     setThemeAttribute(theme)
-    setCookie(COOKIES.theme, theme)
+    setCookie(COOKIES.theme, theme, {
+      sameSite: 'lax'
+    })
   }
 
   const toggleTheme = (theme: Theme): void =>
@@ -20,7 +22,11 @@ export function useTheme() {
     const savedTheme = getThemeAttribute()
     setTheme(savedTheme)
 
-    if (!hasCookie(COOKIES.theme)) setCookie(COOKIES.theme, savedTheme)
+    if (!hasCookie(COOKIES.theme)) {
+      setCookie(COOKIES.theme, savedTheme, {
+        sameSite: 'lax'
+      })
+    }
 
     const listenOsThemeChange = (event: MediaQueryListEvent) => {
       saveTheme(event.matches ? 'light' : 'dark')