Add callback on Document and node mounter (#69)

Fixes #63
diegomura · May 22, 2017 · 18f8473 · 18f8473
1 parent c671601
commit 18f8473
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 4 deletions.
diff --git a/examples/fractals/index.js b/examples/fractals/index.js
@@ -26,4 +26,6 @@ const doc = (
   </Document>
 );
 
-ReactPDF.render(doc, `${__dirname}/output.pdf`);
+ReactPDF.render(doc, `${__dirname}/output.pdf`, () => {
+  console.log('Fractals rendered!');
+});
diff --git a/examples/fractals/output.pdf b/examples/fractals/output.pdf
diff --git a/packages/react-pdf-node/src/index.js b/packages/react-pdf-node/src/index.js
@@ -3,10 +3,10 @@ import path from 'path';
 import { PDFRenderer, createElement, pdf } from '@react-pdf/core';
 
 export default {
-  render(element, filePath) {
+  render(element, filePath, callback) {
     const container = createElement('ROOT');
-
     const node = PDFRenderer.createContainer(container);
+
     PDFRenderer.updateContainer(element, node, null);
 
     const output = pdf(container).toBuffer();
@@ -19,6 +19,10 @@ export default {
       fs.write(fd, output, 0, output.length, null, function(err) {
         if (err) throw new Error(`PDF-react 'Error writing file: ${err}'`);
         fs.close(fd, function() {
+          if (callback) {
+            callback(output, filePath);
+          }
+
           console.log(
             `📝  PDF successfuly exported on ${path.resolve(filePath)}`,
           );

diff --git a/packages/react-pdf/src/index.js b/packages/react-pdf/src/index.js
@@ -11,7 +11,13 @@ const Document = 'DOCUMENT';
 
 const pdf = input => {
   function parse(input) {
-    return input.render();
+    const result = input.render();
+
+    if (input.props.onRender) {
+      input.props.onRender();
+    }
+
+    return result;
   }
 
   function toBlob() {