light_wallet.js

/*jslint node: true */
"use strict";
var db = require('./db.js');
var conf = require('./conf.js');
var myWitnesses = require('./my_witnesses.js');
var network = require('./network.js');
var walletGeneral = require('./wallet_general.js');
var light = require('./light.js');
var eventBus = require('./event_bus.js');

var RECONNECT_TO_LIGHT_VENDOR_PERIOD = 60 * 1000;
var changeLightVender = false;
var timerFlag;

function setLightVendorHost(light_vendor_host) {
	network.light_vendor_url = conf.WS_PROTOCOL + light_vendor_host; // for now, light vendor is also a hub
	if (conf.bLight) {
		timerFlag && clearInterval(timerFlag);
		timerFlag = setInterval(reconnectToLightVendor, RECONNECT_TO_LIGHT_VENDOR_PERIOD);
		connecteToLightVendor();
	}
}

function connecteToLightVendor() {
	if (!conf.bLight)
		return;
	if (!network.light_vendor_url)
		return console.log('refreshLightClientHistory called too early: light_vendor_url not set yet');
	network.findOutboundPeerOrConnect(network.light_vendor_url, function (err, ws) {
		var finish = function (err) {
			if (err) {
				if (!changeLightVender) {
					eventBus.emit('refresh_light_timeout', function (msg) {
						console.log("changeLightVender: " + msg);
						if (!msg) {
							setTimeout(connecteToLightVendor, 5000);
							return;
						};
						changeLightVender = true;
						setLightVendorHost(msg);
					});
				}
			}
			if (ws)
				ws.bRefreshingHistory = false;
		};
		if (err)
			return finish("refreshLightClientHistory: " + err);
		console.log('refreshLightClientHistory connected');
	})
}

function reconnectToLightVendor() {
	network.findOutboundPeerOrConnect(network.light_vendor_url, function (err, ws) {
		if (err)
			return console.log("reconnectToLightVendor: " + err);
		if (ws.bLightVendor)
			return console.log("already connected to light vendor");
		if (ws.bRefreshingHistory)
			return console.log("already refreshing history");
		refreshLightClientHistory();
	});
}

function readListOfUnstableUnits(handleUnits) {
	db.query("SELECT unit FROM units WHERE is_stable=0", function (rows) {
		var arrUnits = rows.map(function (row) { return row.unit; });
		handleUnits(arrUnits);
	});
}


function prepareRequestForHistory(handleResult) {
	myWitnesses.readMyWitnesses(function (arrWitnesses) {
		if (arrWitnesses.length === 0) // first start, witnesses not set yet
			return handleResult(null);
		var objHistoryRequest = { witnesses: arrWitnesses };
		walletGeneral.readMyAddresses(function (arrAddresses) {
			if (arrAddresses.length > 0)
				objHistoryRequest.addresses = arrAddresses;
			readListOfUnstableUnits(function (arrUnits) {
				if (arrUnits.length > 0)
					objHistoryRequest.requested_joints = arrUnits;
				if (!objHistoryRequest.addresses && !objHistoryRequest.requested_joints)
					return handleResult(null);
				if (!objHistoryRequest.addresses)
					return handleResult(objHistoryRequest);
				objHistoryRequest.last_stable_mci = 0;
				var strAddressList = arrAddresses.map(db.escape).join(', ');
				db.query(
					"SELECT unit FROM unit_authors JOIN units USING(unit) WHERE is_stable=1 AND address IN(" + strAddressList + ") \n\
					UNION \n\
					SELECT unit FROM outputs JOIN units USING(unit) WHERE is_stable=1 AND address IN("+ strAddressList + ")",
					function (rows) {
						if (rows.length)
							objHistoryRequest.known_stable_units = rows.map(function (row) { return row.unit; });
						handleResult(objHistoryRequest);
					}
				);
				/*db.query(
					"SELECT MAX(main_chain_index) AS last_stable_mci FROM units JOIN unit_authors USING(unit) WHERE is_stable=1 AND address IN(?)",
					[arrAddresses],
					function(rows){
						objHistoryRequest.last_stable_mci = rows[0].last_stable_mci || 0;
						handleResult(objHistoryRequest);
					}
				);*/
			});
		});
	}, 'wait');
}


function refreshLightClientHistory() {
	if (!conf.bLight)
		return;
	if (!network.light_vendor_url)
		return console.log('refreshLightClientHistory called too early: light_vendor_url not set yet');
	eventBus.emit('refresh_light_started');
	network.findOutboundPeerOrConnect(network.light_vendor_url, function onLocatedLightVendor(err, ws) {
		var finish = function (err) {
			if (err) {
				if (!changeLightVender) {
					eventBus.emit('refresh_light_timeout', function (msg) {
						console.log("changeLightVender: " + msg);
						if (!msg) {
							setTimeout(connecteToLightVendor, 5000);
							return;
						};
						changeLightVender = true;
						setLightVendorHost(msg);
					});
				}
				return;
			}
			if (ws)
				ws.bRefreshingHistory = false;
			eventBus.emit('refresh_light_done');
		};
		if (err)
			return finish("refreshLightClientHistory: " + err);
		console.log('refreshLightClientHistory connected');
		// handling the response may take some time, don't send new requests
		if (ws.bRefreshingHistory)
			return console.log("previous refresh not finished yet");
		ws.bRefreshingHistory = true;
		prepareRequestForHistory(function (objRequest) {
			if (!objRequest)
				return finish();
			network.sendRequest(ws, 'light/get_history', objRequest, false, function (ws, request, response) {
				if (response.error)
					return finish(response.error);
				ws.bLightVendor = true;
				var interval = setInterval(function () { // refresh UI periodically while we are processing history
					eventBus.emit('maybe_new_transactions');
				}, 500);
				light.processHistory(response, {
					ifError: function (err) {
						clearInterval(interval);
						network.sendError(ws, err);
						finish();
					},
					ifOk: function () {
						clearInterval(interval);
						finish();
						eventBus.emit('maybe_new_transactions');
					}
				});
			});
		});
	});
}


exports.setLightVendorHost = setLightVendorHost;
exports.refreshLightClientHistory = refreshLightClientHistory;