55 lines
1.9 KiB
JavaScript
55 lines
1.9 KiB
JavaScript
var XLSX = require('xlsx');
|
|
var URL = require('url');
|
|
var request = require('request');
|
|
var micro = require('micro');
|
|
|
|
function do_url(req, url, res) {
|
|
request(url.query.url, {encoding:null}, function(err, response, body) {
|
|
if(err) return micro.send(res, 500, err);
|
|
switch(response.statusCode) {
|
|
case 200: break;
|
|
case 404: return micro.send(res, 404, "Cannot find " + url.query.url);
|
|
default: return micro.send(res, 500, "Unrecognized status code " + response.statusCode);
|
|
}
|
|
var wb = XLSX.read(body, {type:'buffer'});
|
|
var N = url.query.N ? parseInt(url.query.N,10) : 0;
|
|
if(N < 0) {
|
|
switch(url.query.t || "csv") {
|
|
case "json": return micro.send(res, 200, JSON.stringify(wb.SheetNames.join("\n")));
|
|
default: return micro.send(res, 200, wb.SheetNames.join("\n"));
|
|
}
|
|
}
|
|
if(N >= wb.SheetNames.length) return micro.send(res, 500, "Cannot find sheet " + N);
|
|
var ws = wb.Sheets[wb.SheetNames[N]];
|
|
switch(url.query.t) {
|
|
case "json": return micro.send(res, 200, JSON.stringify(XLSX.utils.sheet_to_json(ws, {header:1, raw:true})));
|
|
default: XLSX.stream.to_csv(ws).pipe(res);
|
|
}
|
|
});
|
|
}
|
|
|
|
var msg = [
|
|
'request /data?url=<url>&N=<idx> to convert the data at [URL] to CSV',
|
|
'',
|
|
'parameters:',
|
|
'- url=<url> the url to request',
|
|
'- N=<idx> the sheet index to use (-1 for sheet list)',
|
|
'- t=<type> export type: "json" for json',
|
|
'',
|
|
'examples: ',
|
|
'- /data?url=https://obamawhitehouse.archives.gov/sites/default/files/omb/budget/fy2014/assets/receipts.xls',
|
|
'',
|
|
'js-xlsx version: ' + XLSX.version
|
|
].join("\n");
|
|
module.exports = function(req, res) {
|
|
res.setHeader('Access-Control-Allow-Origin', '*');
|
|
var url = URL.parse(req.url, true);
|
|
if(url.pathname == "/") return msg;
|
|
var mode = -1;
|
|
if(url.query.url) mode = 0;
|
|
if(mode == -1) return micro.send(res, 500, "Must issue command");
|
|
switch(mode) {
|
|
case 0: return do_url(req, url, res);
|
|
}
|
|
};
|