nbr-web-app/nbr-web-extract-data.js

NodeBot RPA Web Automation

const WebSocket = require('ws');

module.exports = function(RED) {
	//must be same with register
    function NbrWebExtractDataNode(config) {
        RED.nodes.createNode(this,config);
		// Node parameters
		this.scrape = config.scrape;
		this.variable = config.variable;
		this.waitbefore = config.waitbefore;
		this.waitafter = config.waitafter;

		var ws;
        var node = this;
		var flowContext = this.context().flow;
        node.on('input', function(msg,send,done) {
			node.status({fill:"blue",shape:"ring",text:"Extracting"});
			//prepare script parameters
			var t = {user:"admin",module:"WebDriver",action:"EXTRACTDATA",browser:"",scrape:"",variable:"",waitbefore:500,waitafter:500};
			t.browser = flowContext.get("nbr-web-session");
			t.scrape = node.scrape;
			if (node.variable != "")
			    t.variable = node.variable;
			if (node.waitbefore != "")
			    t.waitbefore = parseInt(node.waitbefore);
			if (node.waitafter != "")
			    t.waitafter = parseInt(node.waitafter);
			ws = new WebSocket('ws://127.0.0.1:7000',{
			  rejectUnauthorized: false
			});
			
			ws.onopen = () => {
				msg.payload = "Executing";
				ws.send(JSON.stringify(t));
			}
			
			ws.on('message', function incoming(data){
				var d = JSON.parse(data);
				msg.payload = d.value;
				if (d.status=="SUCCESS") {
					var r = {rowcount:d.rowcount};
					flowContext.set(node.variable, r);
					msg.payload = r;
					node.send(msg);
					node.status({fill:"green",shape:"ring",text:"done"});
				} else {
				    node.status({fill:"red",shape:"ring",text:"error"});
				    msg.payload = d.value;
					node.error(new Error(msg.payload));
				    done(new Error(msg.payload))
				}
			});
			
			ws.on('error', function(err) {
				console.error("WebSocket error:", err);
				node.status({fill:"red",shape:"ring",text:"error"});
				if (err.code == "ECONNREFUSED")
					node.error(new Error("Proxy server is not running"));
				else 
					node.error(new Error(err));
				done(new Error(err))
			});
        });
		node.on('close', function(done) {
			done();
		});
	}
    RED.nodes.registerType("nbr-web-extract-data",NbrWebExtractDataNode);
}