From bb181b9dcd25f0cde8a76cdf3e693fd15a7db808 Mon Sep 17 00:00:00 2001
From: Benjamin Coe <ben@npmjs.com>
Date: Sat, 2 Jan 2016 16:08:54 -0800
Subject: [PATCH] add --check-coverage shorthand for checking coverage
 threshold fixes #105

---
 bin/nyc.js | 61 +++++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 47 insertions(+), 14 deletions(-)

diff --git a/bin/nyc.js b/bin/nyc.js
index 0132a7945..b411685b6 100755
--- a/bin/nyc.js
+++ b/bin/nyc.js
@@ -87,6 +87,27 @@ if (process.env.NYC_CWD) {
       type: 'boolean',
       describe: 'cache instrumentation results for improved performance'
     })
+    .option('check-coverage', {
+      type: 'boolean',
+      default: false,
+      describe: 'check whether coverage is within thresholds provided'
+    })
+    .option('branches', {
+      default: 0,
+      description: 'what % of branches must be covered?'
+    })
+    .option('functions', {
+      default: 0,
+      description: 'what % of functions must be covered?'
+    })
+    .option('lines', {
+      default: 90,
+      description: 'what % of lines must be covered?'
+    })
+    .option('statements', {
+      default: 0,
+      description: 'what % of statements must be covered?'
+    })
     .help('h')
     .alias('h', 'help')
     .version(require('../package.json').version)
@@ -102,18 +123,7 @@ if (process.env.NYC_CWD) {
 
     report(argv)
   } else if (~argv._.indexOf('check-coverage')) {
-    foreground(
-      process.execPath,
-      [
-        require.resolve('istanbul/lib/cli'),
-        'check-coverage',
-        '--lines=' + argv.lines,
-        '--functions=' + argv.functions,
-        '--branches=' + argv.branches,
-        '--statements=' + argv.statements,
-        path.resolve(process.cwd(), './.nyc_output/*.json')
-      ]
-    )
+    checkCoverage(argv)
   } else if (argv._.length) {
     // wrap subprocesses and execute argv[1]
     var nyc = (new NYC())
@@ -132,8 +142,15 @@ if (process.env.NYC_CWD) {
     sw([__filename], env)
 
     foreground(nyc.mungeArgs(argv), function (done) {
-      if (!argv.silent) report(argv)
-      return done()
+      if (argv.checkCoverage) {
+        checkCoverage(argv, function (done) {
+          if (!argv.silent) report(argv)
+          return done()
+        })
+      } else {
+        if (!argv.silent) report(argv)
+        return done()
+      }
     })
   } else {
     // I don't have a clue what you're doing.
@@ -148,3 +165,19 @@ function report (argv) {
     reporter: argv.reporter
   })).report()
 }
+
+function checkCoverage (argv, cb) {
+  foreground(
+    process.execPath,
+    [
+      require.resolve('istanbul/lib/cli'),
+      'check-coverage',
+      '--lines=' + argv.lines,
+      '--functions=' + argv.functions,
+      '--branches=' + argv.branches,
+      '--statements=' + argv.statements,
+      path.resolve(process.cwd(), './.nyc_output/*.json')
+    ],
+    cb
+  )
+}