Accept a comma-separated list of ports to forward

README.md

@@ -67,6 +67,11 @@ OPTIONS
   -r, --remote=remote        git remote of app to use
 
 DESCRIPTION
+  Provide a port or comma-separated list of ports to forward.
+
+  For example, "4000,9000:9001" will forward port 4000 to port 4000 and
+  port 9000 to port 9001.
+
   Example:
 
        $ heroku ps:forward 8080 --app murmuring-headland-14719

commands/port.js

@@ -16,7 +16,12 @@ module.exports = function(topic, command) {
     topic: topic,
     command: command,
     description: 'Forward traffic on a local port to a dyno',
-    help: `Example:
+    help: `Provide a port or comma-separated list of ports to forward.
+
+    For example, "4000,9000:9001" will forward port 4000 to port 4000 and
+    port 9000 to port 9001.
+
+    Example:
 
     $ heroku ps:forward 8080 --app murmuring-headland-14719`,
     args: [{name: 'port', optional: false}],
@@ -31,24 +36,33 @@ module.exports = function(topic, command) {
 
 function * run(context, heroku) {
   yield exec.initFeature(context, heroku, function *(configVars) {
-    let remotePort = context.args.port;
-    let localPort = context.flags.localPort || remotePort;
+    const portMappings = context.args.port.split(',').map(function(portMapping) {
+      const ports = portMapping.split(':')
+      return [ports[0], ports[1] || ports[0]]
+    })
 
     yield exec.createSocksProxy(context, heroku, configVars, function(dynoIp, dynoName, socksPort) {
-      cli.log(`Listening on ${cli.color.white.bold(localPort)} and forwarding to ${cli.color.white.bold(`${dynoName}:${remotePort}`)}`)
+      portMappings.forEach(function(portMapping) {
+        const localPort = portMapping[0]
+        const remotePort = portMapping[1]
+
+        cli.log(`Listening on ${cli.color.white.bold(localPort)} and forwarding to ${cli.color.white.bold(`${dynoName}:${remotePort}`)}`)
+
+        net.createServer(function(connIn) {
+          socks.connect({
+            host: '0.0.0.0',
+            port: remotePort,
+            proxyHost: '127.0.0.1',
+            proxyPort: socksPort,
+            auths: [ socks.auth.None() ]
+          }, function(socket) {
+            connIn.pipe(socket);
+            socket.pipe(connIn);
+          });
+        }).listen(localPort);
+      })
+
       cli.log(`Use ${cli.color.magenta('CTRL+C')} to stop port fowarding`)
-      net.createServer(function(connIn) {
-        socks.connect({
-          host: '0.0.0.0',
-          port: remotePort,
-          proxyHost: '127.0.0.1',
-          proxyPort: socksPort,
-          auths: [ socks.auth.None() ]
-        }, function(socket) {
-          connIn.pipe(socket);
-          socket.pipe(connIn);
-        });
-      }).listen(localPort);
     });
   });
   return new Promise(resolve => {})