Handle empty files

Author: Will O'Brien

index.js

@@ -9,8 +9,9 @@ module.exports = function(opts){
   var title = opts.title || "";
 
   var addToc = function (file){
-    var withToc = doctoc(file.contents.toString(), null, null, title);
-    file.contents = new Buffer(withToc.data, 'utf8');
+    var mdWithToc = doctoc(file.contents.toString(), null, null, title).data;
+    mdWithToc = mdWithToc || file.contents.toString();
+    file.contents = new Buffer(mdWithToc, 'utf8');
     return file;
   };
 

test/fixture/empty.md

@@ -47,4 +47,23 @@ it('should allow title change', function (cb) {
   stream.end();
 });
 
+it('should empty files', function (cb) {
 
+  var stream = gdoctoc();
+
+  stream.on('data', function (file) {
+    assert.equal(file.relative, 'empty.md');
+    assert.equal(file.path, 'test/fixture/empty.md');
+    assert.equal(file.contents.toString(), fs.readFileSync('./test/fixture/empty.md', 'utf8'));
+    cb();
+  });
+
+  stream.write(new gutil.File({
+    cwd: 'test',
+    base: 'test/fixture',
+    path: 'test/fixture/empty.md',
+    contents: fs.readFileSync('./test/fixture/empty.md')
+  }));
+
+  stream.end();
+});