Option wait for image loading

bauhausjs · May 5, 2018 · 53e7ec2 · 53e7ec2
1 parent f723ff3
commit 53e7ec2
Showing 1 changed file with 31 additions and 7 deletions.
diff --git a/lib/phantom-script.js b/lib/phantom-script.js
@@ -13,7 +13,8 @@ var cmdArgs = ["htmlPath",
              "paperWidth",
              "paperHeight",
              "renderDelay",
-             "runningsArgs"];
+             "runningsArgs",
+             "waitImgLoading"];
 
 args = cmdArgs.reduce(function (args, name, i) {
     args[name] = system.args[i + 1];
@@ -74,13 +75,36 @@ page.open(skeleton, function (status) {
     page.paperSize = paperSize(runnings, defaultFormat);
   }
 
-  /* Render the page */
-  setTimeout(function () {
-    page.render(args.pdfPath);
-    page.close();
-    phantom.exit(0);
-  }, parseInt(args.renderDelay, 10));
+  if(args.waitImgLoading) {
+
+    setInterval( function() {
+
+      pendingImages = page.evaluate(function() {
+          var images = document.images;
+          for (var i = 0; i < images.length; i++) {
+            if (!images[i].complete) {
+                return images[i].src;
+            }
+          }
+          return false
+      });
+
+      if(pendingImages) { return }
+      page.render(args.pdfPath);
+      page.close();
+      phantom.exit(0);
 
+    }, parseInt(args.renderDelay, 10));
+
+
+  } else {
+    /* Render the page */
+    setTimeout(function () {
+      page.render(args.pdfPath);
+      page.close();
+      phantom.exit(0);
+    }, parseInt(args.renderDelay, 10));
+  }
 });
 
 function paperSize(runnings, obj) {