1
0
mirror of https://github.com/node-red/node-red.git synced 2023-10-10 13:36:53 +02:00
node-red/nodes/core/parsers/70-CSV.js

79 lines
3.5 KiB
JavaScript
Raw Normal View History

/**
* Copyright 2014 IBM Corp.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
**/
2014-05-04 00:32:04 +02:00
module.exports = function(RED) {
function CSVNode(n) {
RED.nodes.createNode(this,n);
this.template = n.temp.split(",");
this.sep = n.sep || ',';
this.sep = this.sep.replace("\\n","\n").replace("\\r","\r").replace("\\t","\t");
this.quo = '"';
var node = this;
this.on("input", function(msg) {
if (msg.hasOwnProperty("payload")) {
if (typeof msg.payload == "object") { // convert to csv
try {
var o = "";
for (var i in node.template) {
if (msg.payload.hasOwnProperty(node.template[i])) {
if (msg.payload[node.template[i]].indexOf(node.sep) != -1) {
o += node.quo + msg.payload[node.template[i]] + node.quo + node.sep;
}
else if (msg.payload[node.template[i]].indexOf(node.quo) != -1) {
msg.payload[node.template[i]] = msg.payload[node.template[i]].replace(/"/g, '""');
o += node.quo + msg.payload[node.template[i]] + node.quo + node.sep;
}
else { o += msg.payload[node.template[i]] + node.sep; }
}
}
2014-05-04 00:32:04 +02:00
msg.payload = o.slice(0,-1);
node.send(msg);
}
2014-05-04 00:32:04 +02:00
catch(e) { node.log(e); }
}
2014-05-04 00:32:04 +02:00
else if (typeof msg.payload == "string") { // convert to object
try {
var f = true;
var j = 0;
var k = [""];
var o = {};
for (var i = 0; i < msg.payload.length; i++) {
if (msg.payload[i] === node.quo) {
f = !f;
if (msg.payload[i-1] === node.quo) { k[j] += '\"'; }
}
else if ((msg.payload[i] === node.sep) && f) {
if ( node.template[j] && (node.template[j] != "") ) { o[node.template[j]] = k[j]; }
j += 1;
k[j] = "";
}
else {
k[j] += msg.payload[i];
}
}
2014-05-04 00:32:04 +02:00
if ( node.template[j] && (node.template[j] != "") ) { o[node.template[j]] = k[j]; }
msg.payload = o;
node.send(msg);
}
2014-05-04 00:32:04 +02:00
catch(e) { node.log(e); }
}
2014-05-04 00:32:04 +02:00
else { node.log("This node only handles csv strings or js objects."); }
}
2014-05-04 00:32:04 +02:00
});
}
RED.nodes.registerType("csv",CSVNode);
}