Merge pull request #21 from eGavr/logging

Refactoring

Author: eGavr

lib/cli.js

@@ -66,21 +66,22 @@ module.exports = require('coa').Cmd()
             .then(function (source) {
                 if (opts.clean) return tocClean(source);
 
-                if (source.indexOf(TOC_COMMENT) === -1) return 'NO TOC COMMENT';
+                if (source.indexOf(TOC_COMMENT) === -1) {
+                    log.noTocCommentErr();
+                    process.exit(1);
+                }
 
                 var options = _.pick(opts, ['maxDepth', 'bullet']);
                 return tocInsert(source, options);
+            }, function () {
+                log.noFileErr(args.source);
+                process.exit(1);
             })
             .then(function (res) {
-                if (res === 'NO TOC COMMENT') return log.noTocCommentErr();
-
                 return vfs.write(target, res)
                     .then(function () {
                         opts.clean ? log.successCleanMsg() : log.successInsertMsg();
                     });
-            })
-            .fail(function (err) {
-                console.log(err);
             });
     })
     .run(process.argv.slice(2));

lib/log.js

@@ -11,7 +11,12 @@ module.exports = {
     },
 
     noTocCommentErr: function () {
-        console.log('A TOC generation ' + 'failed'.bold.red +
+        console.error('A TOC generation ' + 'failed'.bold.red +
             '. Can not find the HTML comment ' + '<!-- TOC -->'.bold.red + '.');
+    },
+
+    noFileErr: function (filepath) {
+        console.error('A TOC generation ' + 'failed'.bold.red +
+            '. Can not open file ' + filepath.underline + '.');
     }
 };