doctoc.js

#!/usr/bin/env node

'use strict';

var path      =  require('path')
  , fs        =  require('fs')
  , file      =  require('./lib/file')
  , transform =  require('./lib/transform')
  , argv      =  process.argv
  , files;

function cleanPath(path) {
  var homeExpanded = (path.indexOf('~') === 0) ? process.env.HOME + path.substr(1) : path;

  // Escape all spaces
  return homeExpanded.replace(/\s/g, '\\ ');
}

function transformAndSave(files, mode) {
  console.log('\n==================\n');
  
  var transformed = files
    .map(function (x) {
      var content = fs.readFileSync(x.path, 'utf8')
        , result = transform(content, mode);
      result.path = x.path;
      return result;
    });
  var changed = transformed.filter(function (x) { return x.transformed; })
    , unchanged = transformed.filter(function (x) { return !x.transformed; });

  unchanged.forEach(function (x) {
    console.log('"%s" is up to date', x.path);
  });

  changed.forEach(function (x) { 
    console.log('"%s" will be updated', x.path);
    fs.writeFileSync(x.path, x.data, 'utf8'); 
  });
}

var modes = [ 
  [ 'bitbucket', 'bitbucket.org' ]
, [ 'nodejs'   , 'nodejs.org'    ]
, [ 'github'   , 'github.com'    ]
, [ 'gitlab'   , 'gitlab.com'    ]
, [ 'ghost'    , 'ghost.org'     ]
];

var mode = 'github.com'

if (argv.length < 3) {
  console.error('Usage: doctoc [mode] <path> (where path is some path to a directory (i.e. .) or a file (i.e. README.md) )');
  console.error('\nAvailable modes are:');
  for (var i = 0; i < modes.length; i++) {
    console.error('  --%s\t%s', modes[i][0], modes[i][1]);
  }
  console.error('Defaults to \'github.com\'.')
  process.exit(0);
}

for (var i = 0; i < modes.length; i++) {
  var idx = argv.indexOf('--' + modes[i][0]);
  if (~idx) {
    mode = modes[i][1];
    argv.splice(idx, 1);
    break;
  }
}

var target = cleanPath(argv[2])
  , stat = fs.statSync(target)

if (stat.isDirectory()) {
  console.log ('\nDocToccing "%s" and its sub directories for %s.', target, mode);
  files = file.findMarkdownFiles(target);
} else {
  console.log ('\nDocToccing single file "%s" for %s.', target, mode);
  files = [{ path: target }];
}

transformAndSave(files, mode);

console.log('\nEverything is OK.');