Wait until stream finshed to write the pdf (#121)

diegomura · Jul 13, 2017 · a517d11 · a517d11
1 parent 61d41d0
commit a517d11
Showing 1 changed file with 14 additions and 5 deletions.
diff --git a/packages/react-pdf-node/src/index.js b/packages/react-pdf-node/src/index.js
@@ -10,12 +10,21 @@ export default {
     PDFRenderer.updateContainer(element, node, null);
 
     const output = await pdf(container).toBuffer();
-    output.pipe(fs.createWriteStream(filePath));
+    const stream = fs.createWriteStream(filePath);
+    output.pipe(stream);
 
-    if (callback) {
-      callback(output, filePath);
-    }
+    await new Promise((resolve, reject) => {
+      stream.on('finish', () => {
+        if (callback) {
+          callback(output, filePath);
+        }
+        resolve(output);
 
-    console.log(`📝  PDF successfuly exported on ${path.resolve(filePath)}`);
+        console.log(
+          `📝  PDF successfuly exported on ${path.resolve(filePath)}`,
+        );
+      });
+      stream.on('error', reject);
+    });
   },
 };