mirror of https://github.com/node-red/node-red.git
85 lines
3.7 KiB
JavaScript
85 lines
3.7 KiB
JavaScript
/**
|
|
* Copyright JS Foundation and other contributors, http://js.foundation
|
|
*
|
|
* 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.
|
|
**/
|
|
|
|
module.exports = function(RED) {
|
|
"use strict";
|
|
var cheerio = require('cheerio');
|
|
|
|
function CheerioNode(n) {
|
|
RED.nodes.createNode(this,n);
|
|
this.property = n.property||"payload";
|
|
this.outproperty = n.outproperty||this.property||"payload";
|
|
this.tag = n.tag;
|
|
this.ret = n.ret || "html";
|
|
this.as = n.as || "single";
|
|
var node = this;
|
|
this.on("input", function(msg) {
|
|
var value = RED.util.getMessageProperty(msg,node.property);
|
|
if (value !== undefined) {
|
|
var tag = node.tag;
|
|
if (msg.hasOwnProperty("select")) { tag = node.tag || msg.select; }
|
|
try {
|
|
var $ = cheerio.load(value);
|
|
var pay = [];
|
|
var count = 0;
|
|
$(tag).each(function() {
|
|
count++;
|
|
});
|
|
var index = 0;
|
|
$(tag).each(function() {
|
|
if (node.as === "multi") {
|
|
var pay2 = null;
|
|
if (node.ret === "html") { pay2 = cheerio.load($(this).html().trim()).xml(); }
|
|
if (node.ret === "text") { pay2 = $(this).text(); }
|
|
if (node.ret === "attr") { pay2 = this.attribs; }
|
|
//if (node.ret === "val") { pay2 = $(this).val(); }
|
|
/* istanbul ignore else */
|
|
if (pay2) {
|
|
var new_msg = RED.util.cloneMessage(msg);
|
|
RED.util.setMessageProperty(new_msg,node.outproperty,pay2);
|
|
new_msg.parts = {
|
|
id: msg._msgid,
|
|
index: index,
|
|
count: count,
|
|
type: "string",
|
|
ch: ""
|
|
};
|
|
node.send(new_msg);
|
|
}
|
|
}
|
|
if (node.as === "single") {
|
|
if (node.ret === "html") { pay.push( cheerio.load($(this).html().trim()).xml() ); }
|
|
if (node.ret === "text") { pay.push( $(this).text() ); }
|
|
if (node.ret === "attr") { pay.push( this.attribs ); }
|
|
//if (node.ret === "val") { pay.push( $(this).val() ); }
|
|
}
|
|
index++;
|
|
});
|
|
if (node.as === "single") { // Always return an array - even if blank
|
|
RED.util.setMessageProperty(msg,node.outproperty,pay);
|
|
node.send(msg);
|
|
}
|
|
}
|
|
catch (error) {
|
|
node.error(error.message,msg);
|
|
}
|
|
}
|
|
else { node.send(msg); } // If no payload - just pass it on.
|
|
});
|
|
}
|
|
RED.nodes.registerType("html",CheerioNode);
|
|
}
|