Download external images in ebook format

1 files changed, 10 insertions, 0 deletions

diff --git a/lib/utils/logger.js b/lib/utils/logger.js
index 064c196..4c6af79 100644
--- a/lib/utils/logger.js
+++ b/lib/utils/logger.js
@@ -84,6 +84,16 @@ module.exports = function(_write, logLevel) {
 		logger[levelKey].ln = _.partial(logger.logLn, level);
 		logger[levelKey].ok = _.partial(logger.ok, level);
 		logger[levelKey].fail = _.partial(logger.fail, level);
+		logger[levelKey].promise = function(p) {
+			return p.
+			then(function(st) {
+				logger[levelKey].ok();
+				return st;
+			}, function(err) {
+				logger[levelKey].fail();
+				throw err;
+			});
+		}
 	});
 
 	return logger;