sheetaki/src/util.js

34 lines
1.1 KiB
JavaScript

const XLSX = require('xlsx');
module.exports = function(req, body, url, res) {
/* try to read the buffer */
let wb;
try {
wb = XLSX.read(body, {type:'buffer'});
} catch(e) { return res.status(500).send(e.message || e); }
/* N parameter specifies worksheet index */
const N = url.query.N ? parseInt(url.query.N,10) : 0;
/* -1 -> return sheet names */
if(N < 0) switch(url.query.t || "csv") {
case "json": return res.status(200).send(JSON.stringify(wb.SheetNames.join("\n")));
default: return res.status(200).send(wb.SheetNames.join("\n"));
}
/* find worksheet */
if(N >= wb.SheetNames.length) return res.status(500).send(`Cannot find sheet ${N}`);
const ws = wb.Sheets[wb.SheetNames[N]];
/* result depends on "t" parameter:
* - json: AOA (sheet_to_json with header:1, raw: true)
* - html: HTML (using stream.to_html)
* default: CSV (using stream.to_csv)
*/
switch(url.query.t) {
case "json": return res.status(200).json(XLSX.utils.sheet_to_json(ws, {header:1, raw:true}));
case "html": return XLSX.stream.to_html(ws).pipe(res);
default: return XLSX.stream.to_csv(ws).pipe(res);
}
};