From 80ac8c1df08c09ea7a433d4df0738af23cee6a81 Mon Sep 17 00:00:00 2001
From: "Johan T. Halseth" <johanth@gmail.com>
Date: Thu, 13 Dec 2018 12:26:30 +0100
Subject: [PATCH] lncli: add lncli autopilot commands

This commit adds commands 'status', 'enable' and 'disable' for
autopilot.

The commands will only be enabled for autopilotrpc builds of lncli.
---
 cmd/lncli/autopilotrpc_active.go  | 113 ++++++++++++++++++++++++++++++
 cmd/lncli/autopilotrpc_default.go |  10 +++
 cmd/lncli/main.go                 |   3 +
 3 files changed, 126 insertions(+)
 create mode 100644 cmd/lncli/autopilotrpc_active.go
 create mode 100644 cmd/lncli/autopilotrpc_default.go

diff --git a/cmd/lncli/autopilotrpc_active.go b/cmd/lncli/autopilotrpc_active.go
new file mode 100644
index 00000000..542fb60c
--- /dev/null
+++ b/cmd/lncli/autopilotrpc_active.go
@@ -0,0 +1,113 @@
+// +build autopilotrpc
+
+package main
+
+import (
+	"context"
+
+	"github.com/lightningnetwork/lnd/lnrpc/autopilotrpc"
+	"github.com/urfave/cli"
+)
+
+func getAutopilotClient(ctx *cli.Context) (autopilotrpc.AutopilotClient, func()) {
+	conn := getClientConn(ctx, false)
+
+	cleanUp := func() {
+		conn.Close()
+	}
+
+	return autopilotrpc.NewAutopilotClient(conn), cleanUp
+}
+
+var getStatusCommand = cli.Command{
+	Name:        "status",
+	Usage:       "Get the active status of autopilot.",
+	Description: "",
+	Action:      actionDecorator(getStatus),
+}
+
+func getStatus(ctx *cli.Context) error {
+	ctxb := context.Background()
+	client, cleanUp := getAutopilotClient(ctx)
+	defer cleanUp()
+
+	req := &autopilotrpc.StatusRequest{}
+
+	resp, err := client.Status(ctxb, req)
+	if err != nil {
+		return err
+	}
+
+	printRespJSON(resp)
+	return nil
+}
+
+var enableCommand = cli.Command{
+	Name:        "enable",
+	Usage:       "Enable the autopilot.",
+	Description: "",
+	Action:      actionDecorator(enable),
+}
+
+var disableCommand = cli.Command{
+	Name:        "disable",
+	Usage:       "Disable the active autopilot.",
+	Description: "",
+	Action:      actionDecorator(disable),
+}
+
+func enable(ctx *cli.Context) error {
+	ctxb := context.Background()
+	client, cleanUp := getAutopilotClient(ctx)
+	defer cleanUp()
+
+	// We will enable the autopilot.
+	req := &autopilotrpc.ModifyStatusRequest{
+		Enable: true,
+	}
+
+	resp, err := client.ModifyStatus(ctxb, req)
+	if err != nil {
+		return err
+	}
+
+	printRespJSON(resp)
+	return nil
+}
+
+func disable(ctx *cli.Context) error {
+	ctxb := context.Background()
+	client, cleanUp := getAutopilotClient(ctx)
+	defer cleanUp()
+
+	// We will disable the autopilot.
+	req := &autopilotrpc.ModifyStatusRequest{
+		Enable: false,
+	}
+
+	resp, err := client.ModifyStatus(ctxb, req)
+	if err != nil {
+		return err
+	}
+
+	printRespJSON(resp)
+	return nil
+}
+
+// autopilotCommands will return the set of commands to enable for autopilotrpc
+// builds.
+func autopilotCommands() []cli.Command {
+	return []cli.Command{
+		{
+			Name:        "autopilot",
+			Category:    "Autopilot",
+			Usage:       "Interact with a running autopilot.",
+			Description: "",
+			Subcommands: []cli.Command{
+				getStatusCommand,
+				enableCommand,
+				disableCommand,
+			},
+		},
+	}
+}
diff --git a/cmd/lncli/autopilotrpc_default.go b/cmd/lncli/autopilotrpc_default.go
new file mode 100644
index 00000000..49061254
--- /dev/null
+++ b/cmd/lncli/autopilotrpc_default.go
@@ -0,0 +1,10 @@
+// +build !autopilotrpc
+
+package main
+
+import "github.com/urfave/cli"
+
+// autopilotCommands will return nil for non-autopilotrpc builds.
+func autopilotCommands() []cli.Command {
+	return nil
+}
diff --git a/cmd/lncli/main.go b/cmd/lncli/main.go
index b9c1c043..d5f994c1 100644
--- a/cmd/lncli/main.go
+++ b/cmd/lncli/main.go
@@ -293,6 +293,9 @@ func main() {
 		forwardingHistoryCommand,
 	}
 
+	// Add any extra autopilot commands determined by build flags.
+	app.Commands = append(app.Commands, autopilotCommands()...)
+
 	if err := app.Run(os.Args); err != nil {
 		fatal(err)
 	}