update getRoot function to handle port properly

Author: JoshMcCullough

service/src/express.js

@@ -14,8 +14,19 @@ const express = require("express")
 
 const app = express();
 app.use(function(req, res, next) {
-  req.getRoot = function() {
-    return req.protocol + "://" + req.get('host');
+  req.getRoot = function () {
+    const port = Number(req.headers['x-forwarded-port'] || req.socket.localPort);
+    const protocol = req.protocol;
+    const hostname = req.hostname;
+    let includePort = false;
+
+    // Only include port if non-standard for protocol
+    if ((protocol === 'http' && port !== 80) ||
+      (protocol === 'https' && port !== 443)) {
+      includePort = true;
+    }
+
+    return protocol + '://' + hostname + (includePort ? (':' + port) : '');
   };
 
   req.getPath = function() {