Fixed #54

Author: MASNathan

demo/shot.php

@@ -41,6 +41,9 @@
 $fileLocation = 'test';
 $screen->save($fileLocation);
 
+header("Cache-Control: no-cache, no-store, must-revalidate"); // HTTP 1.1.
+header("Pragma: no-cache"); // HTTP 1.0.
+header("Expires: 0"); // Proxies.
 header('Content-Type:' . $screen->getImageType()->getMimeType());
 header('Content-Length: ' . filesize($screen->getImageLocation()));
 readfile($screen->getImageLocation());

templates/screen-capture.php

@@ -14,18 +14,24 @@
 page.settings.resourceTimeout = <?php echo $timeout ?>;
 <?php endif ?>
 
-page.open('<?php echo $url ?>', function () {
+page.open('<?php echo $url ?>', function (status) {
+    if (status !== 'success') {
+        console.log('Unable to load the address!');
+        phantom.exit(1);
+    }
 
     <?php if (isset($includedJsScripts)) : ?>
         <?php foreach ($includedJsScripts as $script) : ?>
             page.injectJs('<?php echo $script ?>');
         <?php endforeach ?>
     <?php endif ?>
 
-    /* This will set the page background color white */
     page.evaluate(function() {
         <?php if (isset($backgroundColor)) : ?>
-            document.body.bgColor = '<?php echo $backgroundColor ?>';
+            /* This will set the page background color */
+            if (document && document.body) {
+                document.body.bgColor = '<?php echo $backgroundColor ?>';
+            }
         <?php endif ?>
 
         <?php if (isset($includedJsSnippets)) : ?>