Skip to content

Commit

Permalink
feat(cli): rename channelbalance to balance
Browse files Browse the repository at this point in the history
  • Loading branch information
krrprr committed Aug 25, 2019
1 parent 328d35f commit d69cff1
Show file tree
Hide file tree
Showing 10 changed files with 364 additions and 279 deletions.
64 changes: 32 additions & 32 deletions docs/api.md

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

24 changes: 12 additions & 12 deletions lib/cli/commands/channelbalance.ts → lib/cli/commands/balance.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import { callback, loadXudClient } from '../command';
import { Arguments } from 'yargs';
import { ChannelBalanceRequest, ChannelBalanceResponse } from '../../proto/xudrpc_pb';
import { BalanceRequest, BalanceResponse } from '../../proto/xudrpc_pb';
import { satsToCoinsStr } from '../utils';
import Table, { HorizontalTable } from 'cli-table3';
import colors from 'colors/safe';
Expand All @@ -11,11 +11,11 @@ const HEADERS = [
colors.blue('Pending Balance'),
];

const formatChannels = (channels: ChannelBalanceResponse.AsObject) => {
const formatBalances = (balances: BalanceResponse.AsObject) => {
const formatted: any[] = [];
channels.balancesMap.forEach((channel) => {
balances.balancesMap.forEach((balance) => {
const element = [];
element.push(channel[0], `${satsToCoinsStr(channel[1].balance)}`, `${satsToCoinsStr(channel[1].pendingOpenBalance)}`);
element.push(balance[0], `${satsToCoinsStr(balance[1].balance)}`, `${satsToCoinsStr(balance[1].pendingOpenBalance)}`);
formatted.push(element);
});
return formatted;
Expand All @@ -28,17 +28,17 @@ const createTable = () => {
return table;
};

const displayChannels = (channels: ChannelBalanceResponse.AsObject) => {
const displayBalances = (balances: BalanceResponse.AsObject) => {
const table = createTable();
const formatted = formatChannels(channels);
formatted.forEach(channel => table.push(channel));
console.log(colors.underline(colors.bold('\nChannel balance:')));
const formatted = formatBalances(balances);
formatted.forEach(balance => table.push(balance));
console.log(colors.underline(colors.bold('\nBalance:')));
console.log(table.toString());
};

export const command = 'channelbalance [currency]';
export const command = 'balance [currency]';

export const describe = 'get total channel balance for a given currency';
export const describe = 'get total balance for a given currency';

export const builder = {
currency: {
Expand All @@ -48,9 +48,9 @@ export const builder = {
};

export const handler = (argv: Arguments) => {
const request = new ChannelBalanceRequest();
const request = new BalanceRequest();
if (argv.currency) {
request.setCurrency(argv.currency.toUpperCase());
}
loadXudClient(argv).channelBalance(request, callback(argv, displayChannels));
loadXudClient(argv).balance(request, callback(argv, displayBalances));
};
16 changes: 8 additions & 8 deletions lib/grpc/GrpcService.ts
Original file line number Diff line number Diff line change
Expand Up @@ -267,17 +267,17 @@ class GrpcService {
}

/**
* See [[Service.channelBalance]]
* See [[Service.balance]]
*/
public channelBalance: grpc.handleUnaryCall<xudrpc.ChannelBalanceRequest, xudrpc.ChannelBalanceResponse> = async (call, callback) => {
public balance: grpc.handleUnaryCall<xudrpc.BalanceRequest, xudrpc.BalanceResponse> = async (call, callback) => {
try {
const channelBalanceResponse = await this.service.channelBalance(call.request.toObject());
const response = new xudrpc.ChannelBalanceResponse();
const balanceResponse = await this.service.balance(call.request.toObject());
const response = new xudrpc.BalanceResponse();
const balancesMap = response.getBalancesMap();
channelBalanceResponse.forEach((channelBalance, currency) => {
const balance = new xudrpc.ChannelBalance();
balance.setBalance(channelBalance.balance);
balance.setPendingOpenBalance(channelBalance.pendingOpenBalance);
balanceResponse.forEach((balanceObj, currency) => {
const balance = new xudrpc.Balance();
balance.setBalance(balanceObj.balance);
balance.setPendingOpenBalance(balanceObj.pendingOpenBalance);
balancesMap.set(currency, balance);
});
callback(null, response);
Expand Down
Loading

0 comments on commit d69cff1

Please sign in to comment.