download-selected

[arvados-workbench2.git] / src / views-components / context-menu / actions / download-action.tsx

diff --git a/src/views-components/context-menu/actions/download-action.tsx b/src/views-components/context-menu/actions/download-action.tsx
index 1f6979d0c78e5296cb5b436873e9846683afe183..2ec7fe1c735a6496861a4b4acd8c872453342ea4 100644 (file)
--- a/src/views-components/context-menu/actions/download-action.tsx
+++ b/src/views-components/context-menu/actions/download-action.tsx
@@ -3,27 +3,55 @@
 // SPDX-License-Identifier: AGPL-3.0
 
 import * as React from "react";
-import { ListItemIcon, ListItemText, Button, ListItem } from "@material-ui/core";
+import { ListItemIcon, ListItemText, ListItem } from "@material-ui/core";
 import { DownloadIcon } from "../../../components/icon/icon";
+import * as JSZip from "jszip";
+import * as FileSaver from 'file-saver';
+import axios from 'axios';
 
-export const DownloadAction = (props: { href?: string, download?: string, onClick?: () => void }) => {
-    const targetProps = props.download ? {} : { target: '_blank' };
+export const DownloadAction = (props: { href?: any, download?: any, onClick?: () => void, kind?: string, currentCollectionUuid?: string; }) => {
     const downloadProps = props.download ? { download: props.download } : {};
-    return props.href
+
+    const createZip = (fileUrls: string[], download: string[]) => {
+        const zip = new JSZip();
+        let id = 1;
+        fileUrls.map((href: string) => {
+            axios.get(href).then(response => response).then(({ data }: any) => {
+                const splittedByDot = href.split('.');
+                if (splittedByDot[splittedByDot.length - 1] !== 'json') {
+                    if (fileUrls.length === id) {
+                        zip.file(download[id-1], data);
+                        zip.generateAsync({ type: 'blob' }).then((content) => {
+                            FileSaver.saveAs(content, `download-${props.currentCollectionUuid}.zip`);
+                        });
+                    } else {
+                        zip.file(download[id-1], data);
+                        zip.generateAsync({ type: 'blob' });
+                    }
+                } else {
+                    zip.file(download[id-1], JSON.stringify(data));
+                    zip.generateAsync({ type: 'blob' });
+                }
+                id++;
+            });
+        });
+    };
+
+    return props.href || props.kind === 'files'
         ? <a
             style={{ textDecoration: 'none' }}
-            href={props.href}
+            href={props.kind === 'files' ? null : props.href}
             onClick={props.onClick}
-            {...targetProps}
             {...downloadProps}>
-            <ListItem button>
-                <ListItemIcon>
-                    <DownloadIcon />
-                </ListItemIcon>
+            <ListItem button onClick={() => props.kind === 'files' ? createZip(props.href, props.download) : undefined}>
+                {props.kind !== 'files' ?
+                    <ListItemIcon>
+                        <DownloadIcon />
+                    </ListItemIcon> : <span />}
                 <ListItemText>
-                    Download
-            </ListItemText>
+                    {props.kind === 'files' ? 'Download selected' : 'Download'}
+                </ListItemText>
             </ListItem>
-        </a >
+        </a>
         : null;
 };
\ No newline at end of file