agendash-standalone-koa.js

#!/usr/bin/env node
"use strict";
const { Agenda, DataSource } = require("@cawstudios/agenda");
const agendash = require("../app");
const Koa = require("koa");
const { Command } = require("commander");

const program = new Command();
program
  .option("-t, --dataSourceType <type>", "Data source type (mongo or postgres)", /^(mongo|postgres)$/i)
  .option("-d, --db <connection>", "DB connection string")
  .option("-c, --collection <name>", "Mongo collection name", "agendaJobs")
  .option("-p, --port <number>", "Server port", parseInt, 3000)
  .option("-t, --title <string>", "Page title", "Agendash")
  .option("-p, --path <path>", "Path to bind Agendash to", "/")
  .parse(process.argv);

const options = program.opts();
// Validation
if (!options.dataSourceType) {
  console.error("--dataSourceType required (mongo or postgres)");
  process.exit(1);
}

if (!options.db) {
  console.error("--db required");
  process.exit(1);
}

if (!options.path.startsWith("/")) {
  console.error("--path must begin with /");
  process.exit(1);
}

// DB Configuration
const dbConfig = {
  [DataSource.POSTGRES]: {
    dataSource: DataSource.POSTGRES,
    dataSourceOptions: { 
      db: {
        connectionString: options.db
      }
    }
  },
  [DataSource.MONGO]: {
    dataSource: DataSource.MONGO,
    dataSourceOptions: { db: { address: options.db, collection: options.collection } }
  }
}[options.dataSourceType];

if (!dbConfig) {
  console.error("Unsupported data source");
  process.exit(1);
}

const init = async () => {
  const agenda = new Agenda();
  await agenda.database(dbConfig);

  const app = new Koa();
  const middlewares = agendash(agenda, {
    middleware: "koa",
  });
  for (const middleware of middlewares) {
    app.use(middleware);
  }

  await app.listen(program.port);
  console.log("Server running on port %s", program.port);
};

process.on("unhandledRejection", (error) => {
  console.log(error);
  process.exit(1);
});

init();