From 17a60f2403c23bd21196d9ac70d6490e4652cec5 Mon Sep 17 00:00:00 2001
From: DK Jang <dkjang@broadinstitute.org>
Date: Wed, 27 Nov 2024 09:56:09 -0500
Subject: [PATCH] DK byor tsv download (#798)

* temp save

* Download section data in TSV implemented
---
 .../researchPortal/ResearchDataTable.vue      |   9 ++
 src/utils/uiUtils.js                          | 131 ++++++++++++++++++
 2 files changed, 140 insertions(+)

diff --git a/src/components/researchPortal/ResearchDataTable.vue b/src/components/researchPortal/ResearchDataTable.vue
index 24f54dc48..2510b90af 100644
--- a/src/components/researchPortal/ResearchDataTable.vue
+++ b/src/components/researchPortal/ResearchDataTable.vue
@@ -73,6 +73,12 @@
 									>
 										CSV
 									</div>
+									<div
+										class="convert-2-csv btn-sm"
+										@click="convertJson2Tsv(filteredData, pageID + sectionId + '_filtered')"
+									>
+										TSV
+									</div>
 									<div
 										class="convert-2-csv btn-sm"
 										@click="saveJson(filteredData, pageID + sectionId + '_filtered')"
@@ -1176,6 +1182,9 @@ export default Vue.component("research-data-table", {
 			//next convert json to csv
 			this.utils.uiUtils.saveByorCsv(jsonData, FILENAME);
 		},
+		convertJson2Tsv(DATA, FILENAME) {
+			this.utils.uiUtils.saveByorTsv(DATA, FILENAME);
+		},
 		saveJson(DATA, FILENAME) {
 			this.utils.uiUtils.saveJson(DATA, FILENAME);
 		},
diff --git a/src/utils/uiUtils.js b/src/utils/uiUtils.js
index 162dc9af8..94c2c5ca3 100644
--- a/src/utils/uiUtils.js
+++ b/src/utils/uiUtils.js
@@ -324,6 +324,136 @@ let saveByorCsv = function (DATA, FILENAME) {
     document.body.removeChild(downloadLink);
 };
 
+let saveByorTsv = function (DATA, FILENAME) {
+    let topRows = [];
+    let features = {};
+    const downloadFilename = FILENAME || "download";
+
+    DATA.map((d) => {
+        for (let [key, value] of Object.entries(d)) {
+            if (key != "features") {
+                if (typeof value == "object" && !value.length) {
+                    for (let [vKey, vValue] of Object.entries(value)) {
+                        let headerKey = key + "__" + vKey;
+                        if (!topRows.includes(headerKey)) {
+                            topRows.push(headerKey);
+                        }
+                    }
+                } else {
+                    if (!topRows.includes(key)) {
+                        topRows.push(key);
+                    }
+                }
+            } else {
+                for (let [fKey, fValue] of Object.entries(value)) {
+                    if (!features[fKey]) {
+                        features[fKey] = [];
+                    }
+
+                    for (let [fVKey, fVValue] of Object.entries(fValue[0])) {
+                        if (!features[fKey].includes(fVKey)) {
+                            features[fKey].push(fVKey);
+                        }
+                    }
+                }
+            }
+        }
+    });
+
+    let csvData = [];
+
+    if (Object.keys(features).length > 0) {
+        DATA.map((d) => {
+            for (let [fKey, fValue] of Object.entries(features)) {
+                for (let i = 0; i < d.features[fKey].length; i++) {
+                    let dArr = [];
+
+                    topRows.map((t) => {
+                        let path = t.split("__");
+                        if (path.length == 2) {
+                            let cValue =
+                                d[path[0]][path[1]] !== null
+                                    ? d[path[0]][path[1]] === 0
+                                        ? 0
+                                        : d[path[0]][path[1]]
+                                    : "";
+                            dArr.push(cValue);
+                        } else {
+                            let cValue =
+                                d[path[0]] !== null
+                                    ? d[path[0]] === 0
+                                        ? 0
+                                        : d[path[0]]
+                                    : "";
+                            dArr.push(cValue);
+                        }
+                    });
+
+                    for (let [fRKey, fRValue] of Object.entries(features)) {
+                        if (fRKey == fKey) {
+                            features[fRKey].map((cKey) => {
+                                dArr.push(d.features[fRKey][i][cKey]);
+                            });
+                        } else {
+                            features[fRKey].map((cKey) => {
+                                dArr.push("");
+                            });
+                        }
+                    }
+                    csvData.push(dArr);
+                }
+            }
+        });
+    } else if (Object.keys(features).length == 0) {
+        DATA.map((d) => {
+            let dArr = [];
+            topRows.map((t) => {
+                let path = t.split("__");
+                if (path.length == 2) {
+                    let cValue =
+                        d[path[0]][path[1]] !== null
+                            ? d[path[0]][path[1]] === 0
+                                ? 0
+                                : d[path[0]][path[1]]
+                            : "";
+                    dArr.push(cValue);
+                } else {
+                    let cValue =
+                        d[path[0]] !== null
+                            ? d[path[0]] === 0
+                                ? 0
+                                : d[path[0]]
+                            : "";
+                    dArr.push(cValue);
+                }
+            });
+            csvData.push(dArr);
+        });
+    }
+
+    let header = topRows;
+
+    for (let [fKey, fValue] of Object.entries(features)) {
+        fValue.map((fCKey) => {
+            header.push("feature__" + fKey + "__" + fCKey);
+        });
+    }
+
+    let csv = [
+        header.join("\t"), // header row first
+        ...csvData.map((row) => row.join("\t")),
+    ].join("\r\n");
+
+    let downloadLink = document.createElement("a");
+    let blob = new Blob(["\ufeff", csv]);
+    let url = URL.createObjectURL(blob);
+    downloadLink.href = url;
+    downloadLink.download = downloadFilename + ".tsv"; //Name the file here
+    document.body.appendChild(downloadLink);
+    downloadLink.click();
+    document.body.removeChild(downloadLink);
+};
+
 let convertJson2Csv = function (DATA, FILENAME) {
     const items = DATA;
     const downloadFilename = FILENAME || "download";
@@ -644,6 +774,7 @@ export default {
     convertJson2Csv,
     convertJson2Tsv,
     saveByorCsv,
+    saveByorTsv,
     saveJson,
     getAxisTicks,
     showTabContent,