1 files changed, 15 insertions, 0 deletions
diff --git a/lib/output/website.js b/lib/output/website.js
index 929c82f..30e0e8c 100644
--- a/lib/output/website.js
+++ b/lib/output/website.js
@@ -174,6 +174,21 @@ WebsiteOutput.prototype.onPage = function(page) {
     });
 };
 
+// Finish generation, create ebook using ebook-convert
+WebsiteOutput.prototype.finish = function() {
+    var that = this;
+
+    return Promise()
+    .then(function() {
+        return WebsiteOutput.super_.prototype.finish.apply(that);
+    })
+
+    // Copy assets from plugins
+    .then(function() {
+        return that.plugins.copyResources('website', that.resolve('gitbook'));
+    });
+};
+
 // ----- Utilities ----
 
 // Render a template using nunjucks