cli.js

#!/usr/bin/env node

var fs = require('fs');
var toc = require('./index.js');
var utils = require('./lib/utils');
var args = utils.minimist(process.argv.slice(2), {
  boolean: ['i', 'json']
});

if (args._.length != 1) {
  console.error('Usage: markdown-toc [--json] [-i] <input> \n\
\n\
  input:  The markdown file to parse for table of contents,\n\
          or "-" to read from stdin.\n\
\n\
  --json: Print the TOC in json format\n\
\n\
  -i:     Edit the <input> file directly, adding the TOC at <!-- toc -->\n\
          (The default is to print the TOC to stdout.)\n\
');
  process.exit(1);
}

if (args.i && args.json) {
  console.error('markdown-toc: you cannot use both --json and -i');
  process.exit(1);
}

if (args.i && args._[0] === "-") {
  console.error('markdown-toc: you cannot use -i with "-" (stdin) for input');
  process.exit(1);
}

var input = process.stdin;
if (args._[0] !== '-') input = fs.createReadStream(args._[0]);

input.pipe(utils.concat(function (input) {
  if (args.i) {
    var newMarkdown = toc.insert(input.toString());
    fs.writeFileSync(args._[0], newMarkdown);
  } else {
    var parsed = toc(input.toString());
    output(parsed);
  }
}))

input.on('error', function onErr(err) {
  console.error(err);
  process.exit(1);
})

function output (parsed) {
  if (args.json) return console.log(JSON.stringify(parsed.json, null, '  '));
  process.stdout.write(parsed.content);
}