From 10d3b29b160db02541baa5472e9274978718f6ed Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Wed, 29 Aug 2018 08:35:37 +0100
Subject: [PATCH 1/3] refactor: simplify file adding

License: MIT
Signed-off-by: Henrique Dias <hacdias@gmail.com>
---
 src/bundles/files.js   | 42 ++++++++++++++++--------------------------
 src/lib/dnd-backend.js |  6 +++---
 src/lib/files.js       |  2 +-
 3 files changed, 20 insertions(+), 30 deletions(-)

diff --git a/src/bundles/files.js b/src/bundles/files.js
index f48ba47b4..c8f6f96dd 100644
--- a/src/bundles/files.js
+++ b/src/bundles/files.js
@@ -228,38 +228,28 @@ export default (opts = {}) => {
     doFilesWrite: make(actions.WRITE, async (ipfs, root, rawFiles, id, { dispatch }) => {
       const { streams, totalSize } = await filesToStreams(rawFiles)
 
-      const updateProgress = (progress) => {
-        dispatch({ type: 'FILES_WRITE_UPDATED', payload: { id: id, progress } })
+      const updateProgress = (sent) => {
+        dispatch({ type: 'FILES_WRITE_UPDATED', payload: { id: id, progress: sent / totalSize * 100 } })
       }
 
       updateProgress(0)
 
-      let sent = 0
-
-      for (const file of streams) {
-        const dir = join(root, dirname(file.name))
-        await ipfs.files.mkdir(dir, { parents: true })
-        let alreadySent = 0
-
-        const res = await ipfs.add(file.content, {
-          pin: false,
-          // eslint-disable-next-line
-          progress: (bytes) => {
-            sent = sent + bytes - alreadySent
-            alreadySent = bytes
-            updateProgress(sent / totalSize * 100)
-          }
-        })
-
-        const src = `/ipfs/${res[0].hash}`
-        const dst = join(root, file.name)
-        await ipfs.files.cp([src, dst])
-
-        sent = sent - alreadySent + file.size
-        updateProgress(sent / totalSize * 100)
+      const res = await ipfs.files.add(streams, {
+        pin: false,
+        wrapWithDirectory: true,
+        progress: updateProgress
+      })
+
+      for (const { path, hash } of res) {
+        // Only go for direct children
+        if (path.indexOf('/') === -1 && path !== '') {
+          const src = `/ipfs/${hash}`
+          const dst = join(root, path)
+          await ipfs.files.cp([src, dst])
+        }
       }
 
-      updateProgress(100)
+      updateProgress(totalSize)
     }),
 
     doFilesDelete: make(actions.DELETE, (ipfs, files) => {
diff --git a/src/lib/dnd-backend.js b/src/lib/dnd-backend.js
index 8b1fa1136..972b0b49e 100644
--- a/src/lib/dnd-backend.js
+++ b/src/lib/dnd-backend.js
@@ -25,7 +25,7 @@ async function scanFiles (item, root = '') {
     const file = await getFileFromEntry(item)
 
     return [{
-      name: join(root, file.webkitRelativePath || file.name),
+      path: join(root, file.webkitRelativePath || file.name),
       content: fileReader(file),
       size: file.size
     }]
@@ -51,13 +51,13 @@ async function scan (files) {
     if (file.getAsEntry || file.webkitGetAsEntry) {
       entries.push({
         entry: getAsEntry(file),
-        name: file.name
+        path: file.name
       })
     } else {
       totalSize += file.size
 
       streams.push({
-        name: file.webkitRelativePath || file.name,
+        path: file.webkitRelativePath || file.name,
         content: fileReader(file),
         size: file.size
       })
diff --git a/src/lib/files.js b/src/lib/files.js
index 479b71cb2..861eaf034 100644
--- a/src/lib/files.js
+++ b/src/lib/files.js
@@ -28,7 +28,7 @@ export async function filesToStreams (files) {
     }
 
     streams.push({
-      name: file.webkitRelativePath || file.name,
+      path: file.webkitRelativePath || file.name,
       content: stream,
       size: file.size
     })

From 93e21153cc69214ad1bb6fbe774b943403e62136 Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Wed, 29 Aug 2018 11:52:01 +0100
Subject: [PATCH 2/3] feat: add error when gets partial response

License: MIT
Signed-off-by: Henrique Dias <hacdias@gmail.com>
---
 src/bundles/files.js | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/bundles/files.js b/src/bundles/files.js
index c8f6f96dd..0a39931c4 100644
--- a/src/bundles/files.js
+++ b/src/bundles/files.js
@@ -240,6 +240,11 @@ export default (opts = {}) => {
         progress: updateProgress
       })
 
+      if (res.length !== streams.length + 1) {
+        // See https://github.com/ipfs/js-ipfs-api/issues/797
+        throw new Error('Unable to finish upload: IPFS API returned a partial response. Try adding a smaller tree.')
+      }
+
       for (const { path, hash } of res) {
         // Only go for direct children
         if (path.indexOf('/') === -1 && path !== '') {

From c2b048884c01a05306b29e0644e24a2898c71c8b Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Wed, 29 Aug 2018 11:56:49 +0100
Subject: [PATCH 3/3] fix: cumulative size of selected files

License: MIT
Signed-off-by: Henrique Dias <hacdias@gmail.com>
---
 src/files/files-list/FilesList.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/files/files-list/FilesList.js b/src/files/files-list/FilesList.js
index 90df578e8..cf3d900f6 100644
--- a/src/files/files-list/FilesList.js
+++ b/src/files/files-list/FilesList.js
@@ -66,7 +66,13 @@ class FileList extends React.Component {
 
   get selectedMenu () {
     const unselectAll = () => this.toggleAll(false)
-    const size = this.selectedFiles.reduce((a, b) => a + (b.size || b.cumulativeSize), 0)
+    const size = this.selectedFiles.reduce((a, b) => {
+      if (b.cumulativeSize) {
+        return a + b.cumulativeSize
+      }
+
+      return a + b.size
+    }, 0)
     const show = this.state.selected.length !== 0
 
     return (