diff --git a/.verb.md b/.verb.md index bdd7611..6075df0 100644 --- a/.verb.md +++ b/.verb.md @@ -9,10 +9,10 @@ Assuming you want to add a TOC to README.md: ``` Usage: markdown-toc [options] - input: The Markdown file to parse for table of contents, + input: The Markdown files to parse for table of contents, or "-" to read from stdin. - -i: Edit the file directly, injecting the TOC at ; + -i: Edit the files directly, injecting the TOC at ; (Without this flag, the default is to print the TOC to stdout.) --json: Print the TOC in JSON format diff --git a/README.md b/README.md index b8766c4..9cc0e82 100644 --- a/README.md +++ b/README.md @@ -38,10 +38,10 @@ $ npm install --save markdown-toc ``` Usage: markdown-toc [options] - input: The Markdown file to parse for table of contents, + input: The Markdown files to parse for table of contents, or "-" to read from stdin. - -i: Edit the file directly, injecting the TOC at ; + -i: Edit the files directly, injecting the TOC at ; (Without this flag, the default is to print the TOC to stdout.) --json: Print the TOC in JSON format diff --git a/cli.js b/cli.js index 87dd4c0..db1fcb2 100755 --- a/cli.js +++ b/cli.js @@ -12,14 +12,14 @@ var args = utils.minimist(process.argv.slice(2), { } }); -if (args._.length !== 1) { +if (args._.length === 0) { console.error([ 'Usage: markdown-toc [options] ', '', - ' input: The Markdown file to parse for table of contents,', + ' input: The Markdown files to parse for table of contents,', ' or "-" to read from stdin.', '', - ' -i: Edit the file directly, injecting the TOC at ', + ' -i: Edit the files directly, injecting the TOC at ', ' (Without this flag, the default is to print the TOC to stdout.)', '', ' --json: Print the TOC in JSON format', @@ -54,22 +54,23 @@ if (args.i && args._[0] === '-') { process.exit(1); } -var input = process.stdin; -if (args._[0] !== '-') input = fs.createReadStream(args._[0]); +args._.forEach(function(filepath) { + var input = (filepath === '-') ? process.stdin : fs.createReadStream(filepath); -input.pipe(utils.concat(function(input) { - if (args.i) { - var newMarkdown = toc.insert(input.toString(), args); - fs.writeFileSync(args._[0], newMarkdown); - } else { - var parsed = toc(input.toString(), args); - output(parsed); - } -})); + input.pipe(utils.concat(function(input) { + if (args.i) { + var newMarkdown = toc.insert(input.toString(), args); + fs.writeFileSync(filepath, newMarkdown); + } else { + var parsed = toc(input.toString(), args); + output(parsed); + } + })); -input.on('error', function onErr(err) { - console.error(err); - process.exit(1); + input.on('error', function onErr(err) { + console.error(err); + process.exit(1); + }); }); function output(parsed) { diff --git a/test/test.js b/test/test.js index e93e41b..c12dc59 100644 --- a/test/test.js +++ b/test/test.js @@ -427,3 +427,43 @@ describe('toc.insert', function() { assert.equal(strip(toc.insert(str, { linkify: false })), read('test/expected/insert-no-links.md')); }); }); + +describe('cli.js', function() { + describe('when provided two Markdon files', function() { + var hook; + beforeEach(function(){ + hook = captureStream(process.stdout); + }); + it('should build a ToC for both files"', function() { + process.argv= ["node", "cli.js", "test/fixtures/basic.md", "test/fixtures/levels.md"]; + require('../cli'); + setTimeout(function() { + hook.unhook(); + assert.equal(hook.captured(), `- [AAA](#aaa) +- [BBB](#bbb) +- [CCC](#ccc) +- [DDD](#ddd)- [AAA](#aaa) + * [a.1](#a1) + + [a.2](#a2) + - [a.3](#a3)`); + }); + }); + }); +}); + +function captureStream(stream){ + var oldWrite = stream.write; + var buf = ''; + stream.write = function(chunk, encoding, callback){ + buf += chunk.toString(); // chunk is a String or Buffer + oldWrite.apply(stream, arguments); + } + return { + unhook: function unhook() { + stream.write = oldWrite; + }, + captured: function() { + return buf; + } + }; +}