device.go

package flags

import (
	"encoding/csv"
	"errors"
	"fmt"
	"strconv"
	"strings"

	"github.com/saucelabs/saucectl/internal/config"
	"github.com/saucelabs/saucectl/internal/msg"
)

// Device represents the RDC device configuration.
type Device struct {
	config.Device
	Changed bool
}

// String returns a string represenation of the device.
func (d Device) String() string {
	if !d.Changed {
		return ""
	}
	return fmt.Sprintf("%+v", d.Device)
}

// Set sets the device to the values present in s.
// The input has to be a comma separated string in the format of "key=value,key2=value2".
// This method is called by cobra when CLI flags are parsed.
func (d *Device) Set(s string) error {
	d.Changed = true

	rec, err := csv.NewReader(strings.NewReader(s)).Read()
	if err != nil {
		return err
	}

	for _, v := range rec {
		vs := strings.Split(v, "=")

		if len(vs) < 2 {
			msg.Error("--device must be specified using a key-value format, e.g. \"--device name=iPhone,private=true\"")
			return errors.New(msg.InvalidKeyValueInputFormat)
		}

		val := vs[1]
		switch vs[0] {
		case "id":
			d.ID = val
		case "name":
			d.Name = val
		case "platformName":
			d.PlatformName = val
		case "platformVersion":
			d.PlatformVersion = val
		case "carrierConnectivity":
			b, err := strconv.ParseBool(val)
			if err != nil {
				return err
			}
			d.Options.CarrierConnectivity = b
		case "deviceType":
			d.Options.DeviceType = val
		case "private":
			b, err := strconv.ParseBool(val)
			if err != nil {
				return err
			}
			d.Options.Private = b
		}
	}

	return nil
}

// Type returns the value type.
func (d Device) Type() string {
	return "device"
}