feat: add support nitro prefix env

atinux · Aug 13, 2024 · 58ebf85 · 58ebf85
1 parent 9979f0d
commit 58ebf85
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/module.ts b/src/module.ts
@@ -50,9 +50,11 @@ export default defineNuxtModule<ModuleOptions>({
 
     // Runtime Config
     const runtimeConfig = nuxt.options.runtimeConfig
+    const envSessionPassword = `${runtimeConfig.nitro?.envPrefix || 'NUXT_'}SESSION_PASSWORD`
+
     runtimeConfig.session = defu(runtimeConfig.session, {
       name: 'nuxt-session',
-      password: process.env.NUXT_SESSION_PASSWORD || '',
+      password: process.env[envSessionPassword] || '',
       cookie: {
         sameSite: 'lax',
       },
@@ -64,10 +66,10 @@ export default defineNuxtModule<ModuleOptions>({
       // Add it to .env
       const envPath = join(nuxt.options.rootDir, '.env')
       const envContent = await readFile(envPath, 'utf-8').catch(() => '')
-      if (!envContent.includes('NUXT_SESSION_PASSWORD')) {
+      if (!envContent.includes(envSessionPassword)) {
         await writeFile(
           envPath,
-          `${envContent ? envContent + '\n' : envContent}NUXT_SESSION_PASSWORD=${runtimeConfig.session.password}`,
+          `${envContent ? envContent + '\n' : envContent}${envSessionPassword}=${runtimeConfig.session.password}`,
           'utf-8',
         )
       }

diff --git a/src/runtime/server/utils/session.ts b/src/runtime/server/utils/session.ts
@@ -95,8 +95,11 @@ let sessionConfig: SessionConfig
 
 function _useSession(event: H3Event) {
   if (!sessionConfig) {
+    const runtimeConfig = useRuntimeConfig(event)
+    const envSessionPassword = `${runtimeConfig.nitro?.envPrefix || 'NUXT_'}SESSION_PASSWORD`
+
     // @ts-expect-error hard to define with defu
-    sessionConfig = defu({ password: process.env.NUXT_SESSION_PASSWORD }, useRuntimeConfig(event).session)
+    sessionConfig = defu({ password: process.env[envSessionPassword] }, runtimeConfig.session)
   }
   return useSession<UserSession>(event, sessionConfig)
 }