cmd/lncli: add key word argument support for new disconnect cmd

This commit is contained in:
Olaoluwa Osuntokun 2017-05-05 15:54:25 -07:00
parent 4b15310c08
commit ab007bb918
No known key found for this signature in database
GPG Key ID: 9CC5B105D03521A2
2 changed files with 17 additions and 5 deletions

View File

@ -254,7 +254,14 @@ var disconnectCommand = cli.Command{
Name: "disconnect",
Usage: "disconnect a remote lnd peer identified by public key",
ArgsUsage: "<pubkey>",
Action: disconnectPeer,
Flags: []cli.Flag{
cli.StringFlag{
Name: "node_key",
Usage: "The hex-encoded compressed public key of the peer " +
"to disconnect from",
},
},
Action: disconnectPeer,
}
func disconnectPeer(ctx *cli.Context) error {
@ -262,9 +269,14 @@ func disconnectPeer(ctx *cli.Context) error {
client, cleanUp := getClient(ctx)
defer cleanUp()
pubKey := ctx.Args().First()
if pubKey == "" {
return fmt.Errorf("target address expected in format: <pubkey>")
var pubKey string
switch {
case ctx.IsSet("node_key"):
pubKey = ctx.String("node_key")
case ctx.Args().Present():
pubKey = ctx.Args().First()
default:
return fmt.Errorf("must specify target public key")
}
req := &lnrpc.DisconnectPeerRequest{

View File

@ -57,6 +57,7 @@ func main() {
sendManyCommand,
sendCoinsCommand,
connectCommand,
disconnectCommand,
openChannelCommand,
closeChannelCommand,
listPeersCommand,
@ -78,7 +79,6 @@ func main() {
debugLevelCommand,
decodePayReqComamnd,
listChainTxnsCommand,
disconnectCommand,
}
if err := app.Run(os.Args); err != nil {