Fix use of cover.jpg in ebook

2 files changed, 5 insertions, 5 deletions

diff --git a/bin/gitbook.js b/bin/gitbook.js
index f79efae..c5a51a4 100755
--- a/bin/gitbook.js
+++ b/bin/gitbook.js
@@ -92,7 +92,7 @@ build.command(prog.command('pdf [source_dir]'))
 
 build.command(prog.command('ebook [source_dir]'))
 .description('Build a gitbook as a eBook')
-.option('-c, --cover <path>', 'Cover image, default is cover.png if exists')
+.option('-c, --cover <path>', 'Cover image, default is cover.jpg if exists')
 .action(function(dir, options) {
     var ext = options.output ? path.extname(options.output) : "epub";
 
diff --git a/lib/generate/ebook/index.js b/lib/generate/ebook/index.js
index 4a69ca6..aaf4d35 100644
--- a/lib/generate/ebook/index.js
+++ b/lib/generate/ebook/index.js
@@ -19,10 +19,6 @@ var Generator = function() {
     this.options = _.defaults(this.options, {
         extension: "epub"
     });
-
-    if (!this.options.cover && fs.existsSync(path.join(this.options.input, "cover.png"))) {
-        this.options.cover = path.join(this.options.input, "cover.png")
-    }
 };
 util.inherits(Generator, BaseGenerator);
 
@@ -37,6 +33,10 @@ Generator.prototype.finish = function() {
             "--cover": that.options.cover
         };
 
+        if (!that.options.cover && fs.existsSync(path.join(that.options.output, "cover.jpg"))) {
+            that.options.cover = path.join(that.options.output, "cover.jpg");
+        }
+
         var command = [
             "ebook-convert",
             path.join(that.options.output, "index.html"),