Merge pull request #23 from Comcast/master

Added a promise to writeImageToFile
vmpowerio · Jul 24, 2017 · 87ca437 · 87ca437
2 parents 6eb6ce0 + c186e10
commit 87ca437
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -145,8 +145,17 @@ class ChartjsNode {
     writeImageToFile(imageType, filePath) {
         return this.getImageBuffer(imageType)
         .then(buffer => {
-            var out = fs.createWriteStream(filePath);
-            return out.write(buffer);
+            return new BbPromise((resolve,reject) => {
+                var out = fs.createWriteStream(filePath);
+                out.on('finish', () => {
+                    return resolve();
+                });
+                out.on('error', () => {
+                    return reject();
+                });
+                out.write(buffer);
+                out.end();
+            });
         });
     }
     /**