diff --git a/cmd/commands/commands.go b/cmd/commands/commands.go index 3746ddaac..f0ae1eac3 100644 --- a/cmd/commands/commands.go +++ b/cmd/commands/commands.go @@ -1463,15 +1463,19 @@ func walletBalance(ctx *cli.Context) error { return nil } -var channelBalanceCommand = cli.Command{ +var ChannelBalanceCommand = cli.Command{ Name: "channelbalance", Category: "Channels", Usage: "Returns the sum of the total available channel balance across " + "all open channels.", - Action: actionDecorator(channelBalance), + Action: actionDecorator(func(c *cli.Context) error { + return ChannelBalance(c, nil) + }), } -func channelBalance(ctx *cli.Context) error { +func ChannelBalance(ctx *cli.Context, + respDecorator ResponseDecorator[*lnrpc.ChannelBalanceResponse]) error { + ctxc := getContext() client, cleanUp := getClient(ctx) defer cleanUp() @@ -1482,6 +1486,13 @@ func channelBalance(ctx *cli.Context) error { return err } + if respDecorator != nil { + err = respDecorator(ctx, resp) + if err != nil { + return err + } + } + printRespJSON(resp) return nil } diff --git a/cmd/commands/main.go b/cmd/commands/main.go index 4aed13d7c..dea1f610d 100644 --- a/cmd/commands/main.go +++ b/cmd/commands/main.go @@ -458,7 +458,7 @@ func Main() { abandonChannelCommand, listPeersCommand, walletBalanceCommand, - channelBalanceCommand, + ChannelBalanceCommand, getInfoCommand, getDebugInfoCommand, encryptDebugPackageCommand,