fix bug in error checking between snapshot and downloadImage

Author: timelyportfolio

src/components/modebar/buttons.js

@@ -55,17 +55,15 @@ modeBarButtons.toImage = {
             return;
         }
 
-        if(gd._snapshotInProgress) {
-            Lib.notifier('Snapshotting is still in progress - please hold', 'long');
-            return;
-        }
-
-        gd._snapshotInProgress = true;
         Lib.notifier('Taking snapshot - this may take a few seconds', 'long');
 
-        downloadImage(gd).catch(function() {
-            Lib.notifier('Sorry there was a problem downloading your snapshot', 'long');
-        });
+        downloadImage(gd)
+          .then(function(filename){
+              Lib.notifier('Snapshot succeeded - ' + filename);
+          })
+          .catch(function(err) {
+              Lib.notifier('Sorry there was a problem downloading your snapshot', 'long');
+          });
     }
 };
 

src/snapshot/download.js

@@ -30,8 +30,7 @@ function downloadImage(gd, opts) {
 
     return new Promise(function(resolve,reject) {
         if(gd._snapshotInProgress) {
-            reject(new Error('Snapshotting is unavailable in Internet Explorer. ' +
-                     'Consider exporting your images using the Plotly Cloud'));
+            reject(new Error('Snapshotting already in progress.'));
         }
 
         gd._snapshotInProgress = true;