From 2baba05e09e51d58e4d18836a4553d98bcabeb4b Mon Sep 17 00:00:00 2001
From: Joel Collins <joel@jtcollins.net>
Date: Thu, 28 Nov 2019 17:15:29 +0000
Subject: [PATCH] Draft of zip downloader

---
 .../galleryComponents/zipDownloader.vue       | 147 ++++++++++++++++++
 .../viewComponents/galleryDisplay.vue         |  25 ++-
 2 files changed, 171 insertions(+), 1 deletion(-)
 create mode 100644 src/components/viewComponents/galleryComponents/zipDownloader.vue

diff --git a/src/components/viewComponents/galleryComponents/zipDownloader.vue b/src/components/viewComponents/galleryComponents/zipDownloader.vue
new file mode 100644
index 00000000..8dd2d01f
--- /dev/null
+++ b/src/components/viewComponents/galleryComponents/zipDownloader.vue
@@ -0,0 +1,147 @@
+<template>
+  <div v-if="zipBuilderUri">
+    <div v-show="downloadReady">
+      <button
+        :disabled="isDownloading"
+        type="button"
+        class="uk-button uk-button-default uk-form-small"
+        @click="downloadWithAxios()"
+      >
+        {{ isDownloading ? "Downloading..." : "Download" }}
+      </button>
+    </div>
+    <div v-show="!downloadReady">
+      <taskSubmitter
+        v-if="zipBuilderUri"
+        :can-terminate="false"
+        :submit-url="zipBuilderUri"
+        :submit-label="'Create ZIP'"
+        :submit-data="captureIds"
+        @submit="onSubmit"
+        @response="onResponse"
+        @error="onError"
+      >
+      </taskSubmitter>
+    </div>
+  </div>
+</template>
+
+<script>
+import axios from "axios";
+import taskSubmitter from "../../genericComponents/taskSubmitter";
+
+// Export main app
+export default {
+  name: "ZipDownloader",
+
+  components: {
+    taskSubmitter
+  },
+
+  props: {
+    captureIds: {
+      type: Array,
+      required: true
+    }
+  },
+
+  data: function() {
+    return {
+      isDownloading: false,
+      downloadReady: false,
+      downloadUrl: null,
+      zipBuilderUri: null,
+      zipGetterUri: null,
+      lastSessionId: null
+    };
+  },
+
+  computed: {
+    pluginsUri: function() {
+      return `${this.$store.getters.baseUri}/api/v2/plugins`;
+    }
+  },
+
+  mounted() {
+    this.updateZipperUri();
+  },
+
+  methods: {
+    updateZipperUri: function() {
+      axios
+        .get(this.pluginsUri) // Get a list of plugins
+        .then(response => {
+          var plugins = response.data;
+          // if ZipBuilderPlugin is enabled
+          if ("ZipBuilderPlugin" in plugins) {
+            // Get plugin action link
+            var builderLink = plugins.ZipBuilderPlugin.views.build.links.self;
+            var getterLink = plugins.ZipBuilderPlugin.views.get.links.self;
+            // Store plugin action URI
+            this.zipBuilderUri = `${this.$store.getters.baseUri}${builderLink}`;
+            this.zipGetterUri = `${this.$store.getters.baseUri}${getterLink}`;
+          }
+        })
+        .catch(error => {
+          this.modalError(error); // Let mixin handle error
+        });
+    },
+
+    resetZipper: function() {
+      this.downloadReady = false;
+    },
+
+    forceFileDownload(response) {
+      const url = window.URL.createObjectURL(new Blob([response.data]));
+      const link = document.createElement("a");
+      link.href = url;
+      link.setAttribute("download", `${this.lastSessionId}.zip`); //or any other extension
+      document.body.appendChild(link);
+      link.click();
+    },
+
+    downloadWithAxios() {
+      this.isDownloading = true;
+      axios
+        .get(this.downloadUrl, { responseType: "arraybuffer" })
+        .then(response => {
+          this.forceFileDownload(response);
+          this.deleteLastZip();
+          this.resetZipper();
+        })
+        .catch(error => {
+          this.modalError(error); // Let mixin handle error
+        })
+        .finally(() => {
+          this.isDownloading = false;
+        });
+    },
+
+    deleteLastZip() {
+      axios
+        .delete(this.downloadUrl)
+        .then(response => {
+          console.log(response);
+        })
+        .catch(error => {
+          this.modalError(error); // Let mixin handle error
+        });
+    },
+
+    onResponse: function(response) {
+      this.lastSessionId = response.return;
+      this.downloadUrl = `${this.zipGetterUri}/${this.lastSessionId}`;
+      this.downloadReady = true;
+    },
+
+    onSubmit: function(submitData) {
+      console.log("SUBMITTED");
+      console.log(submitData);
+    },
+
+    onError: function(error) {
+      this.modalError(error); // Let mixin handle error
+    }
+  }
+};
+</script>
diff --git a/src/components/viewComponents/galleryDisplay.vue b/src/components/viewComponents/galleryDisplay.vue
index 7d4b889d..d5bfffbb 100644
--- a/src/components/viewComponents/galleryDisplay.vue
+++ b/src/components/viewComponents/galleryDisplay.vue
@@ -52,6 +52,10 @@
           </li>
         </ul>
       </div>
+
+      <div class="uk-navbar-right">
+        <ZipDownloader :capture-ids="Object.keys(filteredCaptures)" />
+      </div>
     </nav>
 
     <div
@@ -92,13 +96,16 @@ import axios from "axios";
 import captureCard from "./galleryComponents/captureCard.vue";
 import scanCard from "./galleryComponents/scanCard.vue";
 
+import ZipDownloader from "./galleryComponents/zipDownloader";
+
 // Export main app
 export default {
   name: "GalleryDisplay",
 
   components: {
     captureCard,
-    scanCard
+    scanCard,
+    ZipDownloader
   },
 
   data: function() {
@@ -225,6 +232,22 @@ export default {
       return this.filterCaptureList(this.itemList, this.checkedTags);
     },
 
+    filteredCaptures: function() {
+      var captures = {};
+
+      for (var item of this.filteredItems) {
+        if ("captureList" in item) {
+          for (var capture of item.captureList) {
+            captures[capture.metadata.id] = capture;
+          }
+        } else {
+          captures[item.metadata.id] = item;
+        }
+      }
+
+      return captures;
+    },
+
     sortedItems: function() {
       // Sort filteredItems using sortCaptureList function
       return this.sortCaptureList(this.filteredItems);