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

NodeBot RPA Web Automation

const WebSocket = require('ws');

module.exports = function(RED) {
	//must be same with register
    function NbrWebScrapDataNode(config) {
        RED.nodes.createNode(this,config);
		// Node parameters
		this.table = config.table;
		this.header = config.header;
		this.row = config.row;
		this.col = config.col;
		this.firstRecord = config.firstRecord;
		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:"Scraping"});
			//prepare script parameters
			var t = {user:"admin",module:"WebDriver",action:"SCRAPDATA",browser:"",table:"",header:"",row:"",col:"",firstRecord:false,variable:"",waitbefore:500,waitafter:500};
			t.browser = flowContext.get("nbr-web-session");
			t.table = node.table;
			t.row = node.row;
			t.col = node.col;
			if (node.header != "")
			    t.header = node.header;
			if (node.firstRecord != "")
			    t.firstRecord = node.firstRecord;
			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) {
				node.status({fill:"red",shape:"circle",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-scrap-data",NbrWebScrapDataNode);
}