Make sure we load all data before downloading it

fix #980

umap/static/umap/js/umap.controls.js

@@ -986,7 +986,8 @@ L.U.Map.include({
         const status = this.permissions.getShareStatusDisplay()
         name.textContent = this.getDisplayName()
         // status is not set until map is saved once
-        if (status) share_status.textContent = L._('Visibility: {status}', {
+        if (status)
+          share_status.textContent = L._('Visibility: {status}', {
             status: status,
           })
       }
@@ -1129,10 +1130,23 @@ L.U.Map.include({
     toggleCaveat()
     const download = L.DomUtil.create('a', 'button', container)
     download.textContent = L._('Download data')
-    L.DomEvent.on(download, 'click', () => this.download(typeInput.value), this)
+    L.DomEvent.on(
+      download,
+      'click',
+      () => {
+        if (typeInput.value === 'umap') this.fullDownload()
+        else this.download(typeInput.value)
+      }
+    )
     this.ui.openPanel({ data: { html: container } })
   },
 
+  fullDownload: function () {
+    // Make sure all data is loaded before downloading
+    this.once('dataloaded', () => this.download())
+    this.loadDatalayers(true) // Force load
+  },
+
   download: function (mode) {
     const type = this.EXPORT_TYPES[mode || 'umap']
     const content = type.formatter(this)

umap/static/umap/js/umap.js

@@ -379,33 +379,41 @@ L.U.Map.include({
   },
 
   initDatalayers: function () {
-    let toload = (dataToload = seen = this.options.datalayers.length)
+    for (let j = 0; j < this.options.datalayers.length; j++) {
-    const self = this
+      this.createDataLayer(this.options.datalayers[j])
+    }
+    this.loadDatalayers()
+  },
+
+  loadDatalayers: function (force) {
+    force = force || L.Util.queryString('download') // In case we are in download mode, let's go strait to loading all data
+    let toload = (dataToload = total = this.datalayers_index.length)
     let datalayer
     const loaded = () => {
-      self.datalayersLoaded = true
+      this.datalayersLoaded = true
-      self.fire('datalayersloaded')
+      this.fire('datalayersloaded')
     }
     const decrementToLoad = () => {
       toload--
       if (toload === 0) loaded()
     }
     const dataLoaded = () => {
-      self.dataLoaded = true
+      this.dataLoaded = true
-      self.fire('dataloaded')
+      this.fire('dataloaded')
     }
     const decrementDataToLoad = () => {
       dataToload--
       if (dataToload === 0) dataLoaded()
     }
-    for (let j = 0; j < this.options.datalayers.length; j++) {
+    this.eachDataLayer(function (datalayer) {
-      datalayer = this.createDataLayer(this.options.datalayers[j])
+      if (force && !datalayer.hasDataLoaded()) datalayer.show()
-      if (datalayer.displayedOnLoad()) datalayer.onceLoaded(decrementToLoad)
+      if (datalayer.displayedOnLoad() || force) datalayer.onceLoaded(decrementToLoad)
       else decrementToLoad()
-      if (datalayer.displayedOnLoad()) datalayer.onceDataLoaded(decrementDataToLoad)
+      if (datalayer.displayedOnLoad() || force)
+        datalayer.onceDataLoaded(decrementDataToLoad)
       else decrementDataToLoad()
-    }
+    })
-    if (seen === 0) {
+    if (total === 0) {
       // no datalayer
       loaded()
       dataLoaded()