Show port number even if not defined as option

vercel · Jan 15, 2017 · 6469086 · 6469086
1 parent 770f535
commit 6469086
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/bin/serve.js b/bin/serve.js
@@ -7,6 +7,7 @@ const updateNotifier = require('update-notifier')
 // Ours
 const pkg = require('../package')
 
+// Support for keywords "async" and "await"
 asyncToGen({
   excludes: null
 })
@@ -17,4 +18,5 @@ if (!process.env.NOW) {
   updateNotifier({pkg}).notify()
 }
 
+// Load package core with async/await support
 require('../lib')
diff --git a/lib/render.js b/lib/render.js
@@ -9,7 +9,7 @@ const fs = require('fs-promise')
 // Ours
 const prepareView = require('./view')
 
-module.exports = async (flags, current, dir, ignoredFiles) => {
+module.exports = async (port, current, dir, ignoredFiles) => {
   let files = []
   const subPath = path.relative(current, dir)
 
@@ -93,7 +93,7 @@ module.exports = async (flags, current, dir, ignoredFiles) => {
   }
 
   const details = {
-    port: flags.port,
+    port,
     files,
     assetDir: process.env.ASSET_DIR,
     directory,

diff --git a/lib/server.js b/lib/server.js
@@ -61,7 +61,8 @@ module.exports = async (req, res, flags, current, ignoredFiles) => {
 
     if (!fs.existsSync(indexPath)) {
       // Try to render the current directory's content
-      const renderedDir = await renderDirectory(flags, current, related, ignoredFiles)
+      const port = flags.port || req.socket.localPort
+      const renderedDir = await renderDirectory(port, current, related, ignoredFiles)
 
       // If it works, send the directory listing to the user
       if (renderedDir) {