Add additional 'undefined' checks

davglass · Aug 12, 2015 · ea4d629 · ea4d629
1 parent af043a2
commit ea4d629
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -172,7 +172,7 @@ exports.asTree = function(sorted) {
 exports.asCSV = function(sorted, customFormat) {
     var text = [ ], textArr = [ ], lineArr = [ ];
 
-    if (Object.keys(customFormat).length > 0) {
+    if (customFormat && Object.keys(customFormat).length > 0) {
         textArr = [ ];
         textArr.push('"module name"');
 
@@ -191,7 +191,7 @@ exports.asCSV = function(sorted, customFormat) {
         lineArr = [ ];
 
         //Grab the custom keys from the custom format
-        if (Object.keys(customFormat).length > 0) {
+        if (customFormat && Object.keys(customFormat).length > 0) {
             lineArr.push('"' + key + '"');
             Object.keys(customFormat).forEach(function forEachCallback(item) {
                 lineArr.push('"' + module[item] + '"');
@@ -220,7 +220,7 @@ exports.asCSV = function(sorted, customFormat) {
 exports.asMarkDown = function(sorted, customFormat) {
 
     var text = [];
-    if (Object.keys(customFormat).length > 0) {
+    if (customFormat && Object.keys(customFormat).length > 0) {
 
         Object.keys(sorted).forEach(function sortedCallback(sortedItem) {