Add --qr to show QRCode for root url

This is useful for pulling the root url up on your phone

Author: greggman

bin/servez

@@ -19,13 +19,28 @@ const log = {
   },
 };
 
+function genQRCode(s) {
+  const qr = QrCode.encodeText(s, Ecc.MEDIUM);
+
+  const lines = [];
+  for (let y = 0; y < qr.size; y++) {
+    const line = [];
+    for (let x = 0; x < qr.size; x++) {
+      line.push(c[qr.getModule(x, y) ? 'bgBlack' : 'bgWhite'](' '));
+    }
+    lines.push(line.join(''));
+  }
+  return lines.join('\n');
+}
+
 const optionSpec = {
   options: [
     { option: 'help', alias: 'h', type: 'Boolean',  description: 'displays help' },
     { option: 'port', alias: 'p', type: 'Int',      description: 'port', default: '8080' },
     { option: 'version',          type: 'Boolean',  description: 'print version' },
     { option: 'scan',             type: 'Boolean',  description: 'scan for open port', default: 'true', },
     { option: 'dirs',             type: 'Boolean',  description: 'show directory listing', default: 'true', },
+    { option: 'qr',               type: 'Boolean',  description: 'print QR Code for root url' },
     { option: 'cors',             type: 'Boolean',  description: 'send CORS headers', default: 'true', },
     { option: 'local',            type: 'Boolean',  description: 'local machine only', default: 'false', },
     { option: 'index',            type: 'Boolean',  description: 'serve index.html for directories', default: 'true', },
@@ -76,6 +91,8 @@ if (args.version) {
 
 const fs = require('fs');
 const path = require('path');
+const {QrCode, Ecc} = require('../lib/qrcodegen');
+const hosts = [];
 
 const root = path.resolve(args._[0] || process.cwd());
 try {
@@ -97,11 +114,29 @@ process.stdin.destroy(); // this allows control-c to not print "Terminate Batch?
 process.title = `servez ${root.split(/\\|\//g).slice(-3).join(path.sep)}`;
 
 const commands = {
-  log(data) {
-    console.log(...data);
+  log(args) {
+    console.log(...args);
+  },
+  error(args) {
+    console.error(c.red(args.join(' ')));
+  },
+  host(args) {
+    const localRE = /\D0\.0\.0\.0.\D|\D127\.0\.0\.|\Wlocalhost\W/
+    const [data] = args;
+    const {root} = data;
+    if (!localRE.test(root)) {
+      hosts.push(root);
+    }
   },
-  error(data) {
-    console.error(c.red([...data].join(' ')));
+  start(data) {
+    if (args.qr) {
+      for (const host of hosts) {
+        log.info(`--------------\nQR code for: ${host}`);
+        log.info(genQRCode(host));
+        log.info('');
+      }
+    }
+    log.info('press CTRL-C to stop the server.');
   },
 };
 

lib/qrcodegen.js

@@ -19,8 +19,8 @@ function sendCmd(cmd, data) {
 
 c.enabled = useColors;
 const logger = {
-  log: (...args) => sendCmd('log', [...args]),
-  error: (...args) => sendCmd('error', [...args]),
+  log: (...args) => sendCmd('log', args),
+  error: (...args) => sendCmd('error', args),
   c,
 };
 
@@ -29,8 +29,7 @@ const server = new Servez(Object.assign({
   dataDir,
   logger,
 }, args));
-server.on('start', () => {
-  logger.log('press CTRL-C to stop the server.');
-});
+server.on('host', (...args) => sendCmd('host', args));
+server.on('start', (...args) => sendCmd('start', args));