Move pagebreaks to an external plugin

Author: eKoopmans

src/index.js

@@ -2,6 +2,7 @@ import 'es6-promise/auto';
 
 import Worker from './worker.js';
 import './plugin/jspdf-plugin.js';
+import './plugin/pagebreaks.js';
 import './plugin/hyperlinks.js';
 
 /**

src/plugin/pagebreaks.js

@@ -0,0 +1,18 @@
+import Worker from '../worker.js';
+
+var orig = {
+  toContainer: Worker.prototype.toContainer
+};
+
+Worker.prototype.toContainer = function toContainer() {
+  return orig.toContainer.call(this).then(function toContainer_pagebreak() {
+    // Enable page-breaks.
+    var pageBreaks = source.querySelectorAll('.html2pdf__page-break');
+    var pxPageHeight = this.prop.pageSize.inner.px.height;
+    Array.prototype.forEach.call(pageBreaks, function pageBreak_loop(el) {
+      el.style.display = 'block';
+      var clientRect = el.getBoundingClientRect();
+      el.style.height = pxPageHeight - (clientRect.top % pxPageHeight) + 'px';
+    }, this);
+  });
+};

src/worker.js

@@ -122,15 +122,6 @@ Worker.prototype.toContainer = function toContainer() {
     this.prop.container.appendChild(source);
     this.prop.overlay.appendChild(this.prop.container);
     document.body.appendChild(this.prop.overlay);
-
-    // Enable page-breaks.
-    var pageBreaks = source.querySelectorAll('.html2pdf__page-break');
-    var pxPageHeight = this.prop.pageSize.inner.px.height;
-    Array.prototype.forEach.call(pageBreaks, function pageBreak_loop(el) {
-      el.style.display = 'block';
-      var clientRect = el.getBoundingClientRect();
-      el.style.height = pxPageHeight - (clientRect.top % pxPageHeight) + 'px';
-    }, this);
   });
 };