Add --out flag for writing an output file.

davglass · Dec 4, 2014 · 7e80d50 · 7e80d50
1 parent 47bffc9
commit 7e80d50
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/bin/license-checker b/bin/license-checker
@@ -14,8 +14,16 @@ var path = require('path');
 var fs = require('fs');
 
 checker.init(args, function(json) {
-    if (args.json) return console.log(JSON.stringify(json, null, 2))
-    if (args.csv) return console.log(checker.asCSV(json))
+    var formattedOutput = '';
+    if (args.json) formattedOutput = JSON.stringify(json, null, 2);
+    else if (args.csv) formattedOutput = checker.asCSV(json);
+    else formattedOutput = checker.asTree(json);
 
-    checker.print(json);
+    if (args.out) {
+        var dir = path.dirname(args.out);
+        mkdirp.sync(dir);
+        fs.writeFileSync(args.out, formattedOutput, 'utf8');
+    } else {
+        console.log(formattedOutput);
+    }
 });
diff --git a/lib/args.js b/lib/args.js
@@ -9,6 +9,7 @@ var nopt = require('nopt'),
     known = {
         json: Boolean,
         csv: Boolean,
+        out: require('path'),
         unknown: Boolean,
         version: Boolean,
         color: Boolean,

diff --git a/lib/index.js b/lib/index.js
@@ -135,7 +135,11 @@ exports.init = function(options, callback) {
 };
 
 exports.print = function(sorted) {
-    console.log(treeify.asTree(sorted, true));
+    console.log(exports.asTree(sorted));
+};
+
+exports.asTree = function(sorted) {
+    return treeify.asTree(sorted, true);
 };
 
 exports.asCSV = function(sorted) {