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

89 lines
4.0 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) {
"use strict";
2014-05-04 00:32:04 +02:00
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;
for (var t = 0; t < node.template.length; t++) {
node.template[t] = node.template[t].trim(); // remove leading and trailing whitespace
if (node.template[t].charAt(0) === '"' && node.template[t].charAt(node.template[t].length -1) === '"') {
// remove leading and trialing quotes (if they exist) - and remove whitepace again.
node.template[t] = node.template[t].substr(1,node.template[t].length -2).trim();
}
}
2014-05-04 00:32:04 +02:00
this.on("input", function(msg) {
if (msg.hasOwnProperty("payload")) {
if (typeof msg.payload == "object") { // convert to csv
try {
2014-06-30 21:44:01 +02:00
var ou = "";
for (var t in node.template) {
if (msg.payload.hasOwnProperty(node.template[t])) {
if (msg.payload[node.template[t]].indexOf(node.sep) != -1) {
ou += node.quo + msg.payload[node.template[t]] + node.quo + node.sep;
2014-05-04 00:32:04 +02:00
}
2014-06-30 21:44:01 +02:00
else if (msg.payload[node.template[t]].indexOf(node.quo) != -1) {
msg.payload[node.template[t]] = msg.payload[node.template[t]].replace(/"/g, '""');
ou += node.quo + msg.payload[node.template[t]] + node.quo + node.sep;
2014-05-04 00:32:04 +02:00
}
2014-06-30 21:44:01 +02:00
else { ou += msg.payload[node.template[t]] + node.sep; }
}
}
2014-06-30 21:44:01 +02:00
msg.payload = ou.slice(0,-1);
2014-05-04 00:32:04 +02:00
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) {
2014-06-30 21:44:01 +02:00
if ( node.template[j] && (node.template[j] !== "") ) { o[node.template[j]] = k[j]; }
2014-05-04 00:32:04 +02:00
j += 1;
k[j] = "";
}
else {
k[j] += msg.payload[i];
}
}
2014-06-30 21:44:01 +02:00
if ( node.template[j] && (node.template[j] !== "") ) { o[node.template[j]] = k[j]; }
2014-05-04 00:32:04 +02:00
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);
}