From 84247644ffb35156b151ab3a685ccedcae6494a4 Mon Sep 17 00:00:00 2001
From: ONS-Tom <Tom.Cooling@ons.gov.uk>
Date: Mon, 20 Nov 2017 11:17:27 +0000
Subject: [PATCH] Feature/export to csv (#12)

* Add export to csv functionality

- Using utils/export
- Copied from the business-index-ui

* Add config/export with name of CSV

* Add export to JSON button to ResultsTable
---
 src/components/ResultsTable.js | 12 ++++++++-
 src/config/export.js           |  5 ++++
 src/utils/export.js            | 47 ++++++++++++++++++++++++++++++++++
 3 files changed, 63 insertions(+), 1 deletion(-)
 create mode 100644 src/config/export.js
 create mode 100644 src/utils/export.js

diff --git a/src/components/ResultsTable.js b/src/components/ResultsTable.js
index ff2b475..2c18206 100644
--- a/src/components/ResultsTable.js
+++ b/src/components/ResultsTable.js
@@ -5,6 +5,7 @@ import 'react-table/react-table.css';
 import SummaryTable from '../components/SummaryTable';
 import ChildRefTable from '../components/ChildRefTable';
 import { employmentBands, legalStatusBands, tradingStatusBands, turnoverBands } from '../utils/convertBands';
+import { downloadCSV, downloadJSON } from '../utils/export';
 
 const ResultsTable = ({ results, showFilter, showPagination, defaultPageSize, convertBands }) => {
   return (
@@ -63,7 +64,16 @@ const ResultsTable = ({ results, showFilter, showPagination, defaultPageSize, co
           );
         }}
       />
-      <br /><br />
+      <br />
+      <div className="sdc-isolation">
+        <div className="summary">
+          <h2 className="summary__title saturn" id="">Export Results</h2>
+          <button className="btn btn--secondary" onClick={() => downloadCSV(results)}>Save as CSV</button>
+          &nbsp;
+          <button className="btn btn--secondary" onClick={() => downloadJSON(results)}>Save as JSON</button>
+        </div>
+      </div>
+      <br />
       <SummaryTable
         title="Useful Information"
         items={[
diff --git a/src/config/export.js b/src/config/export.js
new file mode 100644
index 0000000..f26b1ba
--- /dev/null
+++ b/src/config/export.js
@@ -0,0 +1,5 @@
+const exportCsv = {
+  NAME: 'BusinessIndexResults',
+};
+
+export default exportCsv;
diff --git a/src/utils/export.js b/src/utils/export.js
new file mode 100644
index 0000000..beab999
--- /dev/null
+++ b/src/utils/export.js
@@ -0,0 +1,47 @@
+import config from '../config/export';
+
+const { NAME } = config;
+
+export function exportCSV(header, results) {
+  const columnNames = ['id', 'businessName', 'postCode', 'industryCode', 'legalStatus', 'tradingStatus', 'turnover', 'employmentBands', 'companyNo'];
+  let CSV = '';
+  CSV += `${header}\r\n`; // Firstly, insert the header
+  // Go through the results, putting them in the CSV string
+  for (let i = 0; i < results.length; i += 1) {
+    let row = '';
+    // If there is missing data, insert an empty string
+    for (let j = 0; j < columnNames.length; j += 1) {
+      if (results[i][columnNames[j]] === undefined) {
+        row += '"",';
+      } else {
+        row += `"${results[i][columnNames[j]]}",`;
+      }
+    }
+    CSV += `${row}\r\n`;
+  }
+  return CSV;
+}
+
+export function downloadCSV(results) {
+  const header = 'UBRN,Business Name,PostCode,Industry Code,Legal Status,Trading Status,Turnover,Employment,Company Reference Number';
+  const csv = exportCSV(header, results, NAME);
+  const uri = `data:text/csv;charset=utf-8,${escape(csv)}`;
+  const link = document.createElement('a');
+  link.href = uri;
+  const filename = `${NAME}.csv`;
+  link.download = filename;
+  // Below append/remove child is to ensure the download button works on
+  // Firefox, link.click()
+  document.body.appendChild(link);
+  link.click();
+  document.body.removeChild(link);
+}
+
+export function downloadJSON(results) {
+  const jsonStr = JSON.stringify(results, null, 2);
+  const uri = `data:text/json;charset=utf-8,${escape(jsonStr)}`;
+  const link = document.createElement('a');
+  link.href = uri;
+  link.download = `${NAME}.json`;
+  link.click();
+}