From cbac86cb204bcc0b0f9f5eead5e50d8b31f080c8 Mon Sep 17 00:00:00 2001
From: Renan DelValle <commit@ridv.xyz>
Date: Fri, 28 Jun 2019 10:20:56 -0700
Subject: [PATCH] Adding job restart command.

---
 cmd/restart.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 cmd/restart.go

diff --git a/cmd/restart.go b/cmd/restart.go
new file mode 100644
index 0000000..bb84e41
--- /dev/null
+++ b/cmd/restart.go
@@ -0,0 +1,48 @@
+/**
+ * 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
+ *
+ *     http://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 cmd
+
+import (
+    "github.com/paypal/gorealis/v2/gen-go/apache/aurora"
+    "github.com/spf13/cobra"
+)
+
+func init() {
+    rootCmd.AddCommand(restartCmd)
+
+
+    restartCmd.AddCommand(restartJobCmd)
+    restartJobCmd.Flags().StringVarP(env, "environment", "e", "", "Aurora Environment")
+    restartJobCmd.Flags().StringVarP(role, "role", "r", "", "Aurora Role")
+    restartJobCmd.Flags().StringVarP(name, "name", "n", "", "Aurora Name")
+}
+
+var restartCmd = &cobra.Command{
+    Use:   "restart",
+    Short: "Restart an Aurora Job.",
+}
+
+var restartJobCmd = &cobra.Command{
+    Use:   "job",
+    Short: "Restart a Job.",
+    Run: restartJob,
+}
+
+func restartJob(cmd *cobra.Command, args []string) {
+    key := aurora.JobKey{Environment: *env, Role: *role, Name: *name}
+    if err := client.RestartJob(key); err != nil {
+        log.Fatal("unable to create Aurora job: ", err)
+    }
+}