1 files changed, 8 insertions, 3 deletions
diff --git a/lib/generate/ebook/index.js b/lib/generate/ebook/index.js
index 4ecccec..c74ffcd 100644
--- a/lib/generate/ebook/index.js
+++ b/lib/generate/ebook/index.js
@@ -11,6 +11,12 @@ var stringUtils = require("../../utils/string");
 
 var Generator = function() {
     BaseGenerator.apply(this, arguments);
+
+    // eBook format
+    this.ebookFormat = this.options.extension || path.extname(this.options.output).replace("\.", "") || "pdf";
+
+    // Styles to use
+    this.styles = ["ebook", this.ebookFormat];
 };
 util.inherits(Generator, BaseGenerator);
 
@@ -20,7 +26,6 @@ Generator.prototype.finish = function() {
     return BaseGenerator.prototype.finish.apply(this)
     .then(function() {
         var d = Q.defer();
-        var format = that.options.extension || path.extname(that.options.output).replace("\.", "") || "pdf";
 
         if (!that.options.cover && fs.existsSync(path.join(that.options.output, "cover.jpg"))) {
             that.options.cover = path.join(that.options.output, "cover.jpg");
@@ -44,7 +49,7 @@ Generator.prototype.finish = function() {
             "--breadth-first": true
         };
 
-        if (format == "pdf") {
+        if (that.ebookFormat == "pdf") {
             var pdfOptions = that.options.pdf;
 
             _.extend(_options, {
@@ -64,7 +69,7 @@ Generator.prototype.finish = function() {
         var command = [
             "ebook-convert",
             path.join(that.options.output, "SUMMARY.html"),
-            path.join(that.options.output, "index."+format),
+            path.join(that.options.output, "index."+that.ebookFormat),
             stringUtils.optionsToShellArgs(_options)
         ].join(" ");