/**
 * 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,send,done) {
            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(),null,false).xml(); }
                            if (node.ret === "text") { pay2 = $(this).text(); }
                            if (node.ret === "attr") {
                                pay2 = Object.assign({},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: ""
                                };
                                send(new_msg);
                            }
                        }
                        if (node.as === "single") {
                            if (node.ret === "html") { pay.push( cheerio.load($(this).html().trim(),null,false).xml() ); }
                            if (node.ret === "text") { pay.push( $(this).text() ); }
                            if (node.ret === "attr") {
                                var attribs = Object.assign({},this.attribs);
                                pay.push( 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);
                        send(msg);
                    }
                    done();
                }
                catch (error) {
                    done(error.message);
                }
            }
            else { send(msg); done(); } // If no payload - just pass it on.
        });
    }
    RED.nodes.registerType("html",CheerioNode);
}
