From 75b351d35866e71eeccfbd2f7d21587d4b97a472 Mon Sep 17 00:00:00 2001
From: Miguel Angel Ajo <majopela@redhat.com>
Date: Tue, 10 Oct 2023 17:37:12 +0200
Subject: [PATCH] Add set-config parameter

---
 cmd/power.go                            | 73 +++++++++++++++++++++++++
 cmd/setConfig.go                        | 48 ++++++++++++++++
 pkg/drivers/jumpstarter-board/config.go | 17 ++++++
 pkg/harness/device.go                   |  1 +
 4 files changed, 139 insertions(+)
 create mode 100644 cmd/power.go
 create mode 100644 cmd/setConfig.go

diff --git a/cmd/power.go b/cmd/power.go
new file mode 100644
index 0000000..5ce099c
--- /dev/null
+++ b/cmd/power.go
@@ -0,0 +1,73 @@
+/*
+Copyright © 2023 Miguel Angel Ajo Pelayo <majopela@redhat.com
+*/
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/fatih/color"
+	"github.com/redhat-et/jumpstarter/pkg/harness"
+	"github.com/spf13/cobra"
+)
+
+// powerCmd represents the listDevices command
+var powerCmd = &cobra.Command{
+	Use:   "power",
+	Short: "Powers control for devices",
+
+	Run: func(cmd *cobra.Command, args []string) {
+		if err := cobra.MinimumNArgs(2)(cmd, args); err != nil {
+			handleErrorAsFatal(err)
+		}
+
+		action := args[0]
+		device_id := args[1]
+
+		driver := cmd.Flag("driver").Value.String()
+		console, _ := cmd.Flags().GetBool("console")
+		reset, _ := cmd.Flags().GetBool("reset")
+		attach_storage, _ := cmd.Flags().GetBool("attach-storage")
+
+		device, err := harness.FindDevice(driver, device_id)
+		handleErrorAsFatal(err)
+
+		color.Set(COLOR_CMD_INFO)
+		fmt.Printf("🔌 Power action	%s on %s ... ", action, device_id)
+		color.Unset()
+
+		err = device.Power(action)
+		handleErrorAsFatal(err)
+
+		color.Set(COLOR_CMD_INFO)
+		fmt.Println("done")
+		color.Unset()
+
+		if attach_storage {
+			color.Set(COLOR_CMD_INFO)
+			fmt.Printf("💾 Attaching storage for %s ... ", device_id)
+			color.Unset()
+			err = device.AttachStorage(true)
+			handleErrorAsFatal(err)
+			color.Set(COLOR_CMD_INFO)
+			fmt.Println("done")
+			color.Unset()
+		}
+
+		if reset {
+			resetDevice(device, args[0])
+		}
+
+		if console {
+			serialConsole(device)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(powerCmd)
+	powerCmd.Flags().StringP("driver", "d", "", "Only list devices for the specified driver")
+	powerCmd.Flags().BoolP("console", "c", false, "Open console terminal after powering on")
+	powerCmd.Flags().BoolP("reset", "r", false, "Reset device after power up")
+	powerCmd.Flags().BoolP("attach-storage", "a", false, "Attach storage before powering on")
+}
diff --git a/cmd/setConfig.go b/cmd/setConfig.go
new file mode 100644
index 0000000..7adc8c4
--- /dev/null
+++ b/cmd/setConfig.go
@@ -0,0 +1,48 @@
+/*
+Copyright © 2023 Miguel Angel Ajo Pelayo <majopela@redhat.com
+*/
+package cmd
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/redhat-et/jumpstarter/pkg/harness"
+	"github.com/spf13/cobra"
+)
+
+// powerCmd represents the listDevices command
+var setConfig = &cobra.Command{
+	Use:   "set-config",
+	Short: "Changes a device config parameter",
+
+	Run: func(cmd *cobra.Command, args []string) {
+		if err := cobra.MinimumNArgs(3)(cmd, args); err != nil {
+			handleErrorAsFatal(err)
+		}
+
+		device_id := args[0]
+		k := strings.ToLower(args[1])
+		v := args[2]
+
+		driver := cmd.Flag("driver").Value.String()
+		device, err := harness.FindDevice(driver, device_id)
+		handleErrorAsFatal(err)
+		color.Set(COLOR_CMD_INFO)
+		fmt.Printf("✍ Changing device setting for %s > %s=%s ... ", device_id, k, v)
+		color.Unset()
+
+		err = device.SetConfig(k, v)
+		handleErrorAsFatal(err)
+
+		color.Set(COLOR_CMD_INFO)
+		fmt.Println("done")
+		color.Unset()
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(setConfig)
+	setConfig.Flags().StringP("driver", "d", "", "Only list devices for the specified driver")
+}
diff --git a/pkg/drivers/jumpstarter-board/config.go b/pkg/drivers/jumpstarter-board/config.go
index c4c4391..7b38434 100644
--- a/pkg/drivers/jumpstarter-board/config.go
+++ b/pkg/drivers/jumpstarter-board/config.go
@@ -86,6 +86,23 @@ func (d *JumpstarterDevice) readConfig() error {
 	return nil
 }
 
+func (d *JumpstarterDevice) SetConfig(k, v string) error {
+	k = strings.ToLower(k)
+	if err := d.ensureSerial(); err != nil {
+		return fmt.Errorf("SetConfig(%v, %v): %w", k, v, err)
+	}
+
+	if err := d.exitConsole(); err != nil {
+		return fmt.Errorf("SetConfig(%v, %v): %w", k, v, err)
+	}
+
+	if err := d.sendAndExpect("set-config "+k+" "+v, "Set "+k+" to "+v); err != nil {
+		return fmt.Errorf("SetConfig(%v, %v) %w", k, v, err)
+	}
+
+	return nil
+}
+
 func (d *JumpstarterDevice) SetName(name string) error {
 	if err := d.ensureSerial(); err != nil {
 		return fmt.Errorf("SetName(%v): %w", name, err)
diff --git a/pkg/harness/device.go b/pkg/harness/device.go
index 55fe0c5..7d8edbe 100644
--- a/pkg/harness/device.go
+++ b/pkg/harness/device.go
@@ -22,6 +22,7 @@ type Device interface {
 	SetName(name string) error       // set the name of the device, should be stored in config or flashed to device
 	SetUsbConsole(name string) error // set the substring of an out of band console name for this device
 	SetTags(tags []string) error
+	SetConfig(k, v string) error
 	Serial() (string, error)
 	SetDiskImage(path string, offset uint64) error
 	AttachStorage(connect bool) error