kreta99 / qs-custom-forms Goto Github PK

	options.uri = 'https://' + hostname + ':4242/qrs/ProxyService/' + proxyid + '?xrfkey=1234567891234567';
		request(options, function (error, response, body) {
		if(error){
			console.log("Error: " + error);
		}
		else{				
			
			
			px.push({"id":body.id,"name":body.serverNodeConfiguration.name});
			
			// Once everything completed
			if (pxCnt===px.length){
				console.log("Select proxy:");
				for (let r in px ){					
					console.log(" "+r+" - " + px[r].name + " ("+px[r].id+")");										
				}	
								
				var standard_input = process.stdin;
				standard_input.setEncoding('utf-8');
				standard_input.on('data', function (data) {
										
				var selPrx=parseInt(data);
					
					
					if(isNaN(selPrx) || selPrx>pxCnt)
					{
						console.log('Pick only one of the selected numbers and hit enter');
						 return
					} else {
						//Stop reading input
						process.stdin.pause();
						console.log("Processing proxy..."+px[selPrx].name);
						getProxyProp(px[selPrx].id);
					}						
				});	
			}							
		}
	});
}	

	options.uri = 'https://' + hostname + ':4242/qrs/ProxyService/' + pid + '?xrfkey=1234567891234567';
	
	request(options, function (error, response, body) {
		if(error){
			console.log("Error: " + error);
		}
		else{
				
			setCustomForms(body);
			
		}
	});

//set new base64 encoded forms in Proxy properties structure
if(isRestore){
	body.settings.formAuthenticationPageTemplate = "";
	body.settings.loggedOutPageTemplate = "";
	body.settings.errorPageTemplate = "";
} else {
	var login_form_file = ".\\login_custom.html"; //file will be base64 encoded by this script!
	var logout_form_file = ".\\logout_custom.html"; //file will be base64 encoded by this script!
	var error_form_file = ".\\error_custom.html"; //file will be base64 encoded by this script!
	var login_form = fs.readFileSync(login_form_file);
	var logout_form = fs.readFileSync(logout_form_file);
	var error_form = fs.readFileSync(error_form_file);
	body.settings.formAuthenticationPageTemplate = Buffer.from(login_form).toString('base64');
	body.settings.loggedOutPageTemplate = Buffer.from(logout_form).toString('base64');
	body.settings.errorPageTemplate = Buffer.from(error_form).toString('base64');
	
}

options.method = 'PUT';
options.body = body;

request(options, function (error, response, body) {
	if(error){
		console.log("Error: " + error);
	}
	else{			
		for (prop in body.settings){ 
				if (prop=="virtualProxies"){
					console.log(prop);			
					console.log(body.settings[prop]);						
				}
				if (prop=="formAuthenticationPageTemplate"){ 
					console.log(prop);
					console.log('\x1b[33m%s\x1b[0m', body.settings[prop].substr(0,100));  //yellow
									
					
				}
			}
		if(isRestore){
			console.log('\x1b[36m%s\x1b[0m',"*************** ORIGINAL login, logout and error forms restored ***************");
			
		} else {
			console.log('\x1b[36m%s\x1b[0m',"*************** CUSTOM login, logout and error forms SET ***************");			
		}
		
		
	
	}
});	

//set new base64 encoded forms in Proxy properties structure
	body.settings.formAuthenticationPageTemplate = Buffer.from(login_form).toString('base64');
	body.settings.loggedOutPageTemplate = Buffer.from(logout_form).toString('base64');
	body.settings.errorPageTemplate = Buffer.from(error_form).toString('base64');

	for (pid in body){
		var proxyid = body[pid].id;	
	
		options.uri = 'https://' + hostname + ':4242/qrs/ProxyService/' + proxyid + '?xrfkey=1234567891234567';
		
		request(options, function (error, response, body) {
			if(error){
				console.log("Error: " + error);
			}
			else{
					
				setCustomForms(body);
				//console.log("Set for proxy:"+proxyid);
			}
		});
	}

function getProxyProp(body) {
			var proxyid = "27402a61-2364-4edc-a0b3-07477169d119";//proxy id for emergencies when nothing else works body[0].id;