From 78abacdadcaca62b9bc099369c5f89009116d07c Mon Sep 17 00:00:00 2001
From: toby <tlornewr@gmail.com>
Date: Sun, 12 Mar 2017 12:51:04 -0400
Subject: [PATCH] Add download button for displayFilesAsHTML files

+ Added bytes attribute for "file objects" in untar and unzip
+ Added download button on files displayed by displayFilesAsHTML
---
 src/js/core/Utils.js          | 31 +++++++++++++++++++++++++++----
 src/js/operations/Compress.js | 21 +++++++++++++--------
 2 files changed, 40 insertions(+), 12 deletions(-)

diff --git a/src/js/core/Utils.js b/src/js/core/Utils.js
index 9ef0b2b1..abee0c50 100755
--- a/src/js/core/Utils.js
+++ b/src/js/core/Utils.js
@@ -1028,20 +1028,43 @@ var Utils = {
         };
 
         var formatFile = function(file, i) {
+            var blob = new Blob(
+                [new Uint8Array(file.bytes)],
+                {type: "octet/stream"}
+            );
+            var blobUrl = window.URL.createObjectURL(blob);
+
+            var downloadAnchorElem = $("<a></a>")
+                .html("\u21B4")
+                .attr("href", blobUrl)
+                .attr("title", "Download '" + file.fileName + "'")
+                .attr("download", file.fileName);
+
+            var expandFileContentsElem = $("<a></a>")
+                .html("&#x1F50D")
+                .addClass("collapsed")
+                .attr("data-toggle", "collapse")
+                .attr("aria-expanded", "true")
+                .attr("aria-controls", "collapse" + i)
+                .attr("href", "#collapse" + i)
+                .attr("title", "Show/hide contents of '" + file.fileName + "'");
+
             var html = "<div class='panel panel-default'>" +
                        "<div class='panel-heading' role='tab' id='heading" + i + "'>" +
                        "<h4 class='panel-title'>" +
-                       "<a class='collapsed' role='button' data-toggle='collapse' " +
-                       "href='#collapse" + i + "' " +
-                       "aria-expanded='true' aria-controls='collapse" + i +"'>" +
+                       "<div>" +
                        file.fileName +
+                       " " +
+                       $(expandFileContentsElem).prop("outerHTML") +
+                       " " +
+                       $(downloadAnchorElem).prop("outerHTML") +
                        "<span class='pull-right'>" +
                        // These are for formatting when stripping HTML
                        "<span style='display: none'>\n</span>" +
                        file.size.toLocaleString() + " bytes" +
                        "<span style='display: none'>\n</span>" +
                        "</span>" +
-                       "</a>" +
+                       "</div>" +
                        "</h4>" +
                        "</div>" +
                        "<div id='collapse" + i + "' class='panel-collapse collapse' " +
diff --git a/src/js/operations/Compress.js b/src/js/operations/Compress.js
index 6e756264..b5f32736 100755
--- a/src/js/operations/Compress.js
+++ b/src/js/operations/Compress.js
@@ -309,9 +309,8 @@ var Compress = {
             files = [];
 
         filenames.forEach(function(fileName) {
-            var contents = unzip.decompress(fileName);
-
-            contents = Utils.byteArrayToUtf8(contents);
+            var bytes = unzip.decompress(fileName);
+            var contents = Utils.byteArrayToUtf8(bytes);
 
             var file = {
                 fileName: fileName,
@@ -320,6 +319,7 @@ var Compress = {
 
             var isDir = contents.length === 0 && fileName.endsWith("/");
             if (!isDir) {
+                file.bytes = bytes;
                 file.contents = contents;
             }
 
@@ -477,6 +477,13 @@ var Compress = {
             this.position = 0;
         };
 
+        Stream.prototype.getBytes = function(bytesToGet) {
+            var newPosition = this.position + bytesToGet;
+            var bytes = this.bytes.slice(this.position, newPosition);
+            this.position = newPosition;
+            return bytes;
+        };
+
         Stream.prototype.readString = function(numBytes) {
             var result = "";
             for (var i = this.position; i < this.position + numBytes; i++) {
@@ -535,11 +542,9 @@ var Compress = {
                     endPosition += 512 - (file.size % 512);
                 }
 
-                file.contents = "";
-
-                while (stream.position < endPosition) {
-                    file.contents += stream.readString(512);
-                }
+                file.bytes = stream.getBytes(file.size);
+                file.contents = Utils.byteArrayToUtf8(file.bytes);
+                stream.position = endPosition;
             } else if (file.type === "5") {
                 // Directory
                 files.push(file);