From 35b25dde9880436fe4a2394fc2bdf47e5733f271 Mon Sep 17 00:00:00 2001
From: Marc Di Luzio <marc.diluzio@gmail.com>
Date: Sun, 2 Aug 2020 12:49:15 +0100
Subject: [PATCH] Add the upgrade command to the cmdline client

---
 cmd/rove/main.go      | 23 +++++++++++++++++++++++
 cmd/rove/main_test.go |  2 ++
 2 files changed, 25 insertions(+)

diff --git a/cmd/rove/main.go b/cmd/rove/main.go
index 551cda1..f1e993d 100644
--- a/cmd/rove/main.go
+++ b/cmd/rove/main.go
@@ -44,6 +44,7 @@ func printUsage() {
 	fmt.Fprintln(os.Stderr, "\tbroadcast MSG  broadcast a simple ASCII triplet to nearby rovers")
 	fmt.Fprintln(os.Stderr, "\tsalvage        salvages a dormant rover for parts")
 	fmt.Fprintln(os.Stderr, "\ttransfer       transfer's control into a dormant rover")
+	fmt.Fprintln(os.Stderr, "\tupgrade SPEC   spends rover parts to upgrade one rover spec (capacity, range, integrity, charge")
 	fmt.Fprintln(os.Stderr, "\twait           waits before performing the next command")
 	fmt.Fprintf(os.Stderr, "\n")
 	fmt.Fprintln(os.Stderr, "Environment")
@@ -274,6 +275,28 @@ func InnerMain(command string, args ...string) error {
 					Command: roveapi.CommandType_broadcast,
 					Data:    []byte(args[i]),
 				}
+			case "upgrade":
+				i++
+				if len(args) == i {
+					return fmt.Errorf("upgrade command must be passed a spec to upgrade")
+				}
+				var u roveapi.RoverUpgrade
+				switch args[i] {
+				case "capacity":
+					u = roveapi.RoverUpgrade_Capacity
+				case "range":
+					u = roveapi.RoverUpgrade_Range
+				case "integrity":
+					u = roveapi.RoverUpgrade_MaximumIntegrity
+				case "charge":
+					u = roveapi.RoverUpgrade_MaximumCharge
+				default:
+					return fmt.Errorf("upgrade command must be passed a known upgrade spec")
+				}
+				cmd = &roveapi.Command{
+					Command: roveapi.CommandType_upgrade,
+					Upgrade: u,
+				}
 			default:
 				// By default just use the command literally
 				cmd = &roveapi.Command{
diff --git a/cmd/rove/main_test.go b/cmd/rove/main_test.go
index ca83947..250988a 100644
--- a/cmd/rove/main_test.go
+++ b/cmd/rove/main_test.go
@@ -54,6 +54,7 @@ func Test_InnerMain(t *testing.T) {
 	assert.NoError(t, InnerMain("command", "toggle"))
 	assert.NoError(t, InnerMain("command", "stash"))
 	assert.NoError(t, InnerMain("command", "repair"))
+	assert.NoError(t, InnerMain("command", "upgrade", "capacity"))
 	assert.NoError(t, InnerMain("command", "broadcast", "abc"))
 	assert.NoError(t, InnerMain("command", "wait", "10"))
 	assert.NoError(t, InnerMain("command", "wait", "1", "turn", "NW", "toggle", "broadcast", "zyx"))
@@ -61,5 +62,6 @@ func Test_InnerMain(t *testing.T) {
 	// Give it malformed commands
 	assert.Error(t, InnerMain("command", "unknown"))
 	assert.Error(t, InnerMain("command", "broadcast"))
+	assert.Error(t, InnerMain("command", "upgrade"))
 	assert.Error(t, InnerMain("command", "1"))
 }