Add `opts.host` to customize host of resource URLs

lib/swagger.js

@@ -193,7 +193,7 @@ function addRoute(app, uri, doc, opts) {
       // NOTE header names (keys) are always all-lowercase
       var proto = headers['x-forwarded-proto'] || opts.protocol || req.protocol;
       var prefix = opts.omitProtocolInBaseUrl ? '//' : proto + '://';
-      var host = headers['x-forwarded-host'] || headers.host;
+      var host = headers['x-forwarded-host'] || opts.host || headers.host;
       doc.basePath = prefix + host + initialPath;
     }
     res.status(200).send(doc);

test/swagger.test.js

@@ -119,6 +119,17 @@ describe('swagger definition', function() {
         done();
       });
     });
+
+    it('supports opts.header', function(done) {
+      var app = givenAppWithSwagger({ host: 'example.com:8080' });
+      getAPIDeclaration(app, 'products')
+        .end(function(err, res) {
+          if (err) return done(err);
+          var baseUrl = url.parse(res.body.basePath);
+          expect(baseUrl.host).to.equal('example.com:8080');
+          done();
+        });
+    });
   });
 
   describe('Model definition attributes', function() {