Refactor to allow for custom host (#481)

nuxt-modules · Mar 18, 2024 · 48f06d3 · 48f06d3
1 parent 5a7d614
commit 48f06d3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/storybook.ts b/src/storybook.ts
@@ -9,8 +9,9 @@ import { logger } from '@nuxt/kit'
 
 export async function setupStorybook(options: any, nuxt: Nuxt) {
   const STORYBOOK_ROUTE = options.storybookRoute || '/__storybook_route'
-  const STORYBOOK_PORT =  await getPort({ ports: [options.port || 6006 , 6007, 6008, 6009, 6010]})
-  const STORYBOOK_URL = 'http://localhost:'+ STORYBOOK_PORT
+  const STORYBOOK_PORT =  await getPort({ ports: [options.port || 6006, 6007, 6008, 6009, 6010]})
+  const STORYBOOK_HOST = options.storybookHost || 'http://localhost'
+  const STORYBOOK_URL = + STORYBOOK_HOST + STORYBOOK_PORT == 80 ? '' : `:${STORYBOOK_PORT}`
 
   const projectDir = resolve(nuxt.options.rootDir)
   const args = isStorybookConfigured(projectDir) ? 
@@ -53,7 +54,7 @@ export async function setupStorybook(options: any, nuxt: Nuxt) {
     } )
 
     const storybookProxy = {
-      target: `http://localhost:${STORYBOOK_PORT}`,
+      target: STORYBOOK_URL,
       changeOrigin: true,
       followRedirects: true,
       secure: false,