From a4001d255fecf814438468108da3321450f3c62f Mon Sep 17 00:00:00 2001
From: Drew Walters <andrew.walters@att.com>
Date: Mon, 30 Mar 2020 22:03:06 +0000
Subject: [PATCH] Add remote host reboot command

This change introduces a reboot command that allows the user to reboot a
remote host. The reboot command supports any airshipctl remote client
and simply uses the one defined in the remote options.

Relates #5

Change-Id: I6e11430aa60a09172eb3ac0f44d1a6ff17ca91b0
Signed-off-by: Drew Walters <andrew.walters@att.com>
---
 cmd/remote/remote.go        |  3 +++
 cmd/remote/remote_reboot.go | 47 +++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 cmd/remote/remote_reboot.go

diff --git a/cmd/remote/remote.go b/cmd/remote/remote.go
index 0e0a27654..366402763 100644
--- a/cmd/remote/remote.go
+++ b/cmd/remote/remote.go
@@ -28,5 +28,8 @@ func NewRemoteCommand(rootSettings *environment.AirshipCTLSettings) *cobra.Comma
 	powerStatusCmd := NewPowerStatusCommand(rootSettings)
 	remoteRootCmd.AddCommand(powerStatusCmd)
 
+	rebootCmd := NewRebootCommand(rootSettings)
+	remoteRootCmd.AddCommand(rebootCmd)
+
 	return remoteRootCmd
 }
diff --git a/cmd/remote/remote_reboot.go b/cmd/remote/remote_reboot.go
new file mode 100644
index 000000000..2acd0ec4b
--- /dev/null
+++ b/cmd/remote/remote_reboot.go
@@ -0,0 +1,47 @@
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package remote
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+
+	"opendev.org/airship/airshipctl/pkg/environment"
+	"opendev.org/airship/airshipctl/pkg/remote"
+)
+
+// NewRebootCommand provides a command with the capability to reboot hosts.
+func NewRebootCommand(rootSettings *environment.AirshipCTLSettings) *cobra.Command {
+	rebootCmd := &cobra.Command{
+		Use:   "reboot SYSTEM_ID",
+		Short: "Reboot a host",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			a, err := remote.NewAdapter(rootSettings)
+			if err != nil {
+				return err
+			}
+
+			if err = a.OOBClient.RebootSystem(a.Context, args[0]); err != nil {
+				return err
+			}
+
+			fmt.Fprintf(cmd.OutOrStdout(), "Rebooted remote host %s\n", args[0])
+
+			return nil
+		},
+	}
+
+	return rebootCmd
+}