feat(bin): add bin to render pdfs / pngs / etc

Author: anulman

bin/render-vendor

@@ -0,0 +1,38 @@
+#! /usr/bin/env node
+
+const _ = require('lodash');
+const program = require('commander');
+const { version } = require('../package.json');
+
+const COMMANDS = {
+  kill: 'kill spawned renderers',
+  list: 'list spawned renderers'
+}
+
+let urls = [];
+let isCommand = process.argv[0].endsWith('node') &&
+  process.argv[1].endsWith('render-vendor') &&
+  Object.keys(COMMANDS).includes(process.argv[2]) ||
+  Object.keys(COMMANDS).includes(process.argv[1]);
+
+program.version(version);
+
+_.forEach(COMMANDS, (description, name) => {
+  program.command(name, description)
+});
+
+program
+  .option('-f, --format [value]', 'output format; default = \'pdf\'')
+  .arguments('[urls...]')
+  .action((_urls) => {
+    urls = [...urls, ..._urls];
+  })
+  .parse(process.argv);
+
+if (!isCommand) {
+  if (urls.length === 0) {
+    process.exit(0);
+  }
+
+  require('./render-vendor-print').call(program, urls);
+}

bin/render-vendor-kill

@@ -0,0 +1,19 @@
+#! /usr/bin/env node
+
+const childProcess = require('child_process');
+let killScriptComponents = [
+  'ps aux',
+  'grep render-vendor',
+  'grep phantomjs',
+  'grep -v grep',
+  'awk \'{ print $2 }\'',
+  'xargs kill'
+]
+
+try {
+  childProcess.execSync(killScriptComponents.join(' | '));
+} catch (err) {
+  // nop; nothing found
+}
+
+process.exit(0);

bin/render-vendor-list

@@ -0,0 +1,19 @@
+#! /usr/bin/env node
+
+const childProcess = require('child_process');
+let listScriptComponents = [
+  'ps aux',
+  'grep render-vendor',
+  'grep phantomjs',
+  'grep -v grep',
+]
+
+try {
+  let stdout = childProcess.execSync(listScriptComponents.join(' | '));
+
+  console.log(stdout.toString());
+} catch (err) {
+  // nop; nothing found
+}
+
+process.exit(0);

bin/render-vendor-print

@@ -0,0 +1,53 @@
+#! /usr/bin/env node
+
+const Listr = require('listr');
+const { Renderer } = require('../renderer');
+
+module.exports = function(urls) {
+  process.on('uncaughtException', () => {
+    Renderer.destroy();
+    process.exit(1);
+  });
+
+  let tasks = new Listr([{
+    title: `Booting a ${Renderer.rendererConstructor.name}`,
+    task: () => Renderer.renderer.boot()
+  }, {
+    title: 'Rendering webpages',
+    task: (ctx, task) => new Listr(urls.map((url) => {
+      return makeRenderUrlTaskObjectFor.call(this, url, task);
+    }), {
+      concurrent: true,
+      exitOnError: false
+    })
+  }, {
+    title: `Shutting down ${Renderer.renderer.constructor.name}`,
+    task: () => Renderer.destroy()
+  }], {
+    exitOnError: false
+  });
+
+  tasks.run()
+    .then(() => process.exit(0))
+    .catch(() => process.exit(1));
+}
+
+function makeRenderUrlTaskObjectFor(url, loadTask) {
+  return {
+    title: `${url}: Loading...`,
+    task: (ctx, task) => Renderer.load(url)
+      .then((page) => {
+        let format = this.format || 'pdf';
+
+        task.title = `${url}: Rendering ${format}...`;
+
+        return page.render({ format });
+      })
+      .then((filepath) => {
+        loadTask.title = [
+          loadTask.title,
+          `   ✔ ${url}: Rendered to ${filepath}`
+        ].join('\n');
+      })
+  };
+}