Flush callback called. Closes #3.

Author: drinchev

index.js

@@ -34,7 +34,7 @@ module.exports = function (options) {
   return through2.obj(function document(file, enc, cb) {
     files.push(file);
     cb();
-  }, function () {
+  }, function (cb) {
     var that = this;
     documentation(files.map(function(file) {
       return file.path;
@@ -53,6 +53,6 @@ module.exports = function (options) {
       } else {
         console.error('unknown type detected');
       }
-    }));
+    }).on('finish', function() { cb(); } ));
   });
 };

test/test.js

@@ -25,3 +25,13 @@ test('gulp-documentation md custom name', function(t) {
             t.end();
         });
 });
+
+test('gulp-documentation exit callback', function(t) {
+    t.plan(1);    
+    gulp.src(path.join(__dirname, '/fixture.js'))
+        .pipe(documentation({ format: 'md' }))
+        .on('end', function() { t.pass(); })
+        .resume();
+    
+});
+