namuol · thaggie · Mar 10, 2016 · Mar 16, 2016
diff --git a/Makefile b/Makefile
@@ -1,4 +1,4 @@
-.PHONY: pass fail test
+.PHONY: pass fail mix test
 
 TAP = ./node_modules/tape/bin/tape
 MIN = ./bin/tap-difflet
@@ -9,4 +9,10 @@ pass:
 fail:
 	@$(TAP) test/fail.js | $(MIN)
 
+mix:
+	@$(TAP) test/mix.js | $(MIN)
+
+mixp:
+	@$(TAP) test/mix.js | $(MIN) -p 
+
 test: pass fail
diff --git a/README.md b/README.md
@@ -15,6 +15,12 @@ npm install tap-difflet --save-dev
 tape test/*.js | tap-difflet
 ~~~
 
+For pessimistic usage specify -p or --pessimist - this will only output the errors and the summary.
+
+~~~ text
+tape test/*.js | tap-difflet -p
+~~~
+
 ## Output
 
 ![tap-difflet](http://i.imgur.com/8uFAvXU.png)

diff --git a/bin/tap-difflet b/bin/tap-difflet
@@ -19,7 +19,74 @@ var tap = parser();
 var out = through();
 var dup = duplexer(tap, out);
 
+
+/* Pessimism */
+
+function userSpecifiedPessimistic (args) {
+  for (var ai=2; ai < args.length; ++ai) {
+    if (args[ai] === '-p' || args[ai] === '--pessimist') {
+      return true;
+    }
+  }
+
+  return false;
+};
+
+
+if (userSpecifiedPessimistic(process.argv)) {
+  var anAssertFailed = true;
+  var originalOut = out;
+  var buffer = null;
+  tap.on('comment', function(comment) {
+    if (buffer && anAssertFailed) {
+      for (var bi=0; bi<buffer.length; ++bi) {
+        originalOut.push(buffer[bi]);
+      }
+    }
+
+    buffer = null;
+    anAssertFailed = false;
+    out = originalOut;
+
+    if (getCommentType(comment) === CommentType.OTHER) {
+      buffer = [];
+      out = buffer;
+    }
+  });
+
+  tap.on('assert', function(res) {
+    if (!res.ok) {
+      anAssertFailed = true;
+    }
+  });
+}
+
 /* Helpers */
+var CommentType = {
+  OTHER: 1,
+  TESTS: 2,
+  PASS: 3,
+  FAIL: 4,
+  OK: 5
+};
+
+function getCommentType(comment) {
+  if (/^tests\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.TESTS;
+  }
+  else if (/^pass\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.PASS;
+  }
+  else if (/^fail\s+[0-9]+$/gi.test(comment)) {
+    return CommentType.FAIL;
+  }
+  else if (/^ok$/gi.test(comment)) {
+    return CommentType.OK;
+  }
+  else {
+    return CommentType.OTHER;
+  }
+};
 
 function output(str) {
   out.push(str);
@@ -31,25 +98,26 @@ var timer   = hirestime();
 var errors  = [];
 
 tap.on('comment', function(comment) {
-    if (/^tests\s+[0-9]+$/gi.test(comment)) {
-        output('\n');
-        comment = chalk.white.bold(comment);
-    }
-    else if (/^pass\s+[0-9]+$/gi.test(comment)) {
-        comment = chalk.green.bold(comment);
-    }
-    else if (/^fail\s+[0-9]+$/gi.test(comment)) {
-        comment = chalk.red.bold(comment);
-    }
-    else if (/^ok$/gi.test(comment)) {
-        return;
-    }
-    else {
-        output('\n');
-        comment = chalk.white.bold(comment);
-    }
+  var commentType = getCommentType(comment);
+  if (commentType === CommentType.TESTS) {
+    output('\n');
+    comment = chalk.white.bold(comment);
+  }
+  else if (commentType === CommentType.PASS) {
+    comment = chalk.green.bold(comment);
+  }
+  else if (commentType === CommentType.FAIL) {
+    comment = chalk.red.bold(comment);
+  }
+  else if (commentType === CommentType.OK) {
+    return;
+  }
+  else {
+    output('\n');
+    comment = chalk.white.bold(comment);
+  }
 
-    output('  ' + comment + '\n');
+  output('  ' + comment + '\n');
 });
 
 tap.on('assert', function(res) {
@@ -90,10 +158,10 @@ tap.on('results', function(res) {
 
 tap.on('diag', function (diag) {
   var expected, actual, at,
-      gotExpected = true,
-      gotActual = true,
-      gotAt = true,
-      str = '';
+    gotExpected = true,
+    gotActual = true,
+    gotAt = true,
+    str = '';
 
   if (diag.hasOwnProperty('expected')) {
     expected = diag.expected;
@@ -120,15 +188,15 @@ tap.on('diag', function (diag) {
 
   if (gotActual && gotExpected) {
     if (typeof expected !== typeof actual ||
-        typeof expected === "object" && (!actual || !expected)) {
+        typeof expected === 'object' && (!actual || !expected)) {
       str = 'Expected ' + typeof expected + ' but got ' + typeof actual;
-    } else if (typeof expected === "string") {
+    } else if (typeof expected === 'string') {
       if (str.indexOf('\n') >= 0) {
         str = ansidiff.lines(expected, actual);
       } else {
         str = ansidiff.chars(expected, actual);
       }
-    } else if (typeof expected === "object") {
+    } else if (typeof expected === 'object') {
       str = objdiff(expected, actual);
     } else {
       str = chalk.white('Expected ') + chalk.bold(''+expected) + chalk.white(' but got ') + chalk.bold(''+actual);

diff --git a/test/mix.js b/test/mix.js
@@ -0,0 +1,2 @@
+require ('./pass');
+require ('./fail');