fb8479988c
- added`callback`option so we can get CSV data in case we do not want
to download
- added options:
- `quoted` for cases when we have line breaks in content
- `separator` for cases when we have commas in content
59 lines
1.7 KiB
JavaScript
59 lines
1.7 KiB
JavaScript
/*
|
|
inspired by https://www.npmjs.com/package/react-csv-downloader
|
|
now removed from Github
|
|
*/
|
|
|
|
const newLine = '\r\n';
|
|
const appendLine = (content, row, { separator, quoted }) => {
|
|
const line = row.map(data => {
|
|
if (!quoted) return data;
|
|
// quote data
|
|
data = typeof data === 'string' ? data.replace(/"/g, '"') : data;
|
|
return `"${data}"`;
|
|
});
|
|
content.push(line.join(separator));
|
|
};
|
|
|
|
const defaults = {
|
|
separator: ',',
|
|
quoted: false
|
|
};
|
|
|
|
export default function csv(columns, datas, options, noHeader = false) {
|
|
options = Object.assign({}, defaults, options);
|
|
let columnOrder;
|
|
const content = [];
|
|
const column = [];
|
|
|
|
if (columns) {
|
|
columnOrder = columns.map(v => {
|
|
if (typeof v === 'string') return v;
|
|
if (!noHeader) {
|
|
column.push(typeof v.title !== 'undefined' ? v.title : v.key);
|
|
}
|
|
return v.key;
|
|
});
|
|
if (column.length > 0) appendLine(content, column, options);
|
|
} else {
|
|
columnOrder = [];
|
|
datas.forEach(v => {
|
|
if (!Array.isArray(v)) {
|
|
columnOrder = columnOrder.concat(Object.keys(v));
|
|
}
|
|
});
|
|
if (columnOrder.length > 0) {
|
|
columnOrder = columnOrder.filter((value, index, self) => self.indexOf(value) === index);
|
|
if (!noHeader) appendLine(content, columnOrder, options);
|
|
}
|
|
}
|
|
|
|
if (Array.isArray(datas)) {
|
|
datas.forEach(row => {
|
|
if (!Array.isArray(row)) {
|
|
row = columnOrder.map(k => (typeof row[k] !== 'undefined' ? row[k] : ''));
|
|
}
|
|
appendLine(content, row, options);
|
|
});
|
|
}
|
|
return content.join(newLine);
|
|
}
|