@galaxyproject/gx-it-proxy/lib/main.js

A dynamic reverse proxy for Interactive Tools in Galaxy

#!/usr/bin/env node
/*
Inspiration taken from
	https://github.com/jupyter/multiuser-server/blob/master/multiuser/js/main.js
*/
const { Command } = require("commander");
const packageInfo = require("../package");
const { DynamicProxy } = require("./proxy");
const { mapFor } = require("./mapper");

const program = new Command();

program
  .version(packageInfo.version)
  .option("--ip <n>", "Public-facing IP of the proxy", "localhost")
  .option("--port <n>", "Public-facing port of the proxy", parseInt)
  .option("--sessions <file>", "Routes file to monitor")
  .option(
    "--pollingInterval <n>",
    "Polling interval for PostgreSQL databases (in ms, defaults to 5000, " +
      "use 0 to disable polling)",
    parseInt,
    5000,
  )
  .option("--proxyPathPrefix <path>", "Path-based proxy prefix")
  .option("--forwardIP <n>", "Forward all requests to IP")
  .option("--forwardPort <n>", "Forward all requests to port", parseInt)
  .option(
    "--reverseProxy",
    "Cause the proxy to rewrite location blocks with its own port",
  )
  .option("--verbose");

const main = function (argv_) {
  const argv = argv_ || process.argv;
  program.parse(argv);
  const options = program.opts();

  let sessions = null;
  if (options.sessions) {
    sessions = mapFor(options.sessions, options.pollingInterval);
  }

  const dynamicProxyOptions = {
    sessionMap: sessions,
    proxyPathPrefix: options.proxyPathPrefix,
    verbose: options.verbose,
    port: options.port,
  };

  if (options.reverseProxy) {
    dynamicProxyOptions.reverseProxy = true;
  }

  if (options.forwardIP) {
    dynamicProxyOptions.forwardIP = options["forwardIP"];
  }

  if (options.forwardPort) {
    dynamicProxyOptions.forwardPort = options["forwardPort"];
  }

  const dynamicProxy = new DynamicProxy(dynamicProxyOptions);
  const listen = {
    port: options.port || 8000,
    ip: options.ip,
  };
  dynamicProxy.listen(listen);
  return dynamicProxy;
};

exports.main = main;

if (require.main === module) {
  main();
}