1 files changed, 57 insertions, 0 deletions

diff --git a/server/middlewares/pods.ts b/server/middlewares/pods.ts
new file mode 100644
index 000000000..e405f265e
--- /dev/null
+++ b/server/middlewares/pods.ts
@@ -0,0 +1,57 @@
+'use strict'
+
+const constants = require('../initializers/constants')
+
+function setBodyHostsPort (req, res, next) {
+  if (!req.body.hosts) return next()
+
+  for (let i = 0; i < req.body.hosts.length; i++) {
+    const hostWithPort = getHostWithPort(req.body.hosts[i])
+
+    // Problem with the url parsing?
+    if (hostWithPort === null) {
+      return res.sendStatus(500)
+    }
+
+    req.body.hosts[i] = hostWithPort
+  }
+
+  return next()
+}
+
+function setBodyHostPort (req, res, next) {
+  if (!req.body.host) return next()
+
+  const hostWithPort = getHostWithPort(req.body.host)
+
+  // Problem with the url parsing?
+  if (hostWithPort === null) {
+    return res.sendStatus(500)
+  }
+
+  req.body.host = hostWithPort
+
+  return next()
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  setBodyHostsPort,
+  setBodyHostPort
+}
+
+// ---------------------------------------------------------------------------
+
+function getHostWithPort (host) {
+  const splitted = host.split(':')
+
+  // The port was not specified
+  if (splitted.length === 1) {
+    if (constants.REMOTE_SCHEME.HTTP === 'https') return host + ':443'
+
+    return host + ':80'
+  }
+
+  return host
+}