balances.js

/*jslint node: true */
"use strict";
var _ = require('lodash');
var constants = require('./constants.js');
var db = require('./db');
var light = require('./light.js');

function readBalance(wallet, handleBalance) {
    var assocBalances = { base: { stable: light.stable(), pending: light.pending() } };
    handleBalance(assocBalances);
}

function readOutputsBalance(wallet, handleBalance) {
    var walletIsAddress = typeof wallet === 'string' && wallet.length === 32; // ValidationUtils.isValidAddress
    var join_my_addresses = walletIsAddress ? "" : "JOIN my_addresses USING(address)";
    var where_condition = walletIsAddress ? "address=?" : "wallet=?";
    var assocBalances = { base: { stable: 0, pending: 0 } };
    db.query(
        "SELECT asset, is_stable, SUM(amount) AS balance \n\
        FROM outputs "+ join_my_addresses + " CROSS JOIN units USING(unit) \n\
		WHERE is_spent=0 AND "+ where_condition + " AND sequence='good' \n\
		GROUP BY asset, is_stable",
        [wallet],
        function (rows) {
            for (var i = 0; i < rows.length; i++) {
                var row = rows[i];
                var asset = row.asset || "base";
                if (!assocBalances[asset])
                    assocBalances[asset] = { stable: 0, pending: 0 };
                assocBalances[asset][row.is_stable ? 'stable' : 'pending'] = row.balance;
            }
            handleBalance(assocBalances);
        }
    );
}

function readSharedAddressesOnWallet(wallet, handleSharedAddresses) {
    db.query("SELECT DISTINCT shared_address FROM my_addresses JOIN shared_address_signing_paths USING(address) WHERE wallet=?", [wallet], function (rows) {
        var arrSharedAddresses = rows.map(function (row) { return row.shared_address; });
        if (arrSharedAddresses.length === 0)
            return handleSharedAddresses([]);
        readSharedAddressesDependingOnAddresses(arrSharedAddresses, function (arrNewSharedAddresses) {
            handleSharedAddresses(arrSharedAddresses.concat(arrNewSharedAddresses));
        });
    });
}

function readSharedAddressesDependingOnAddresses(arrMemberAddresses, handleSharedAddresses) {
    var strAddressList = arrMemberAddresses.map(db.escape).join(', ');
    db.query("SELECT DISTINCT shared_address FROM shared_address_signing_paths WHERE address IN(" + strAddressList + ")", function (rows) {
        var arrSharedAddresses = rows.map(function (row) { return row.shared_address; });
        if (arrSharedAddresses.length === 0)
            return handleSharedAddresses([]);
        var arrNewMemberAddresses = _.difference(arrSharedAddresses, arrMemberAddresses);
        if (arrNewMemberAddresses.length === 0)
            return handleSharedAddresses([]);
        readSharedAddressesDependingOnAddresses(arrNewMemberAddresses, function (arrNewSharedAddresses) {
            handleSharedAddresses(arrNewMemberAddresses.concat(arrNewSharedAddresses));
        });
    });
}

function readSharedBalance(wallet, handleBalance) {
    var assocBalances = {};
    readSharedAddressesOnWallet(wallet, function (arrSharedAddresses) {
        if (arrSharedAddresses.length === 0)
            return handleBalance(assocBalances);
        var strAddressList = arrSharedAddresses.map(db.escape).join(', ');
        db.query(
            "SELECT asset, address, is_stable, SUM(amount) AS balance \n\
            FROM outputs CROSS JOIN units USING(unit) \n\
            WHERE is_spent=0 AND sequence='good' AND address IN("+ strAddressList + ") \n\
			GROUP BY asset, address, is_stable \n\
			UNION ALL \n\
			SELECT NULL AS asset, address, 1 AS is_stable, SUM(amount) AS balance FROM witnessing_outputs \n\
			WHERE is_spent=0 AND address IN("+ strAddressList + ") GROUP BY address \n\
			UNION ALL \n\
			SELECT NULL AS asset, address, 1 AS is_stable, SUM(amount) AS balance FROM headers_commission_outputs \n\
			WHERE is_spent=0 AND address IN("+ strAddressList + ") GROUP BY address",
            function (rows) {
                for (var i = 0; i < rows.length; i++) {
                    var row = rows[i];
                    var asset = row.asset || "base";
                    if (!assocBalances[asset])
                        assocBalances[asset] = {};
                    if (!assocBalances[asset][row.address])
                        assocBalances[asset][row.address] = { stable: 0, pending: 0 };
                    assocBalances[asset][row.address][row.is_stable ? 'stable' : 'pending'] += row.balance;
                }
                handleBalance(assocBalances);
            }
        );
    });
}

exports.readBalance = readBalance;
exports.readOutputsBalance = readOutputsBalance;
exports.readSharedBalance = readSharedBalance;