feat: read cookie from server

nuxt-modules · Nov 25, 2023 · c8eab90 · c8eab90
1 parent 957d9d4
commit c8eab90
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/runtime/plugin.server.ts b/src/runtime/plugin.server.ts
@@ -1,8 +1,8 @@
 import { reactive, ref } from 'vue'
 
 import type { ColorModeInstance } from './types'
-import { defineNuxtPlugin, isVue2, isVue3, useHead, useState, useRouter } from '#imports'
-import { preference, hid, script, dataValue } from '#color-mode-options'
+import { defineNuxtPlugin, isVue2, isVue3, useHead, useState, useRouter, useRequestHeaders } from '#imports'
+import { preference, hid, script, dataValue, storage, storageKey } from '#color-mode-options'
 
 const addScript = (head) => {
   head.script = head.script || []
@@ -45,6 +45,13 @@ export default defineNuxtPlugin((nuxtApp) => {
   }
 
   if (isVue3) {
+    if (storage === 'cookie') {
+      const { cookie } = useRequestHeaders(['cookie'])
+      const [, value] = cookie?.split('; ').map(s => s.split('=')).find(([k]) => k === storageKey) ?? []
+      if (value) {
+        colorMode.preference = value
+      }
+    }
     useHead({ htmlAttrs })
   }