Update color utils

cmd/device.go

@@ -26,13 +26,23 @@ var deviceCmd = &cobra.Command{
 }
 
 // device will validate the IP address for a single device (if provided) otherwise
-// defaults to using all devices available on the local network. 
+// defaults to using all devices available on the local network.
 // The color is set for one or more device's using either an exact match for
 // a hex value or the name of a color preset.
 func device(ip string, hex string, name string, args []string) error {
-	color, err := lights.GetColor(hex, name)
-	if err != nil {
-		return err
+	var color lights.Color
+	var err error
+
+	if hex != "" {
+		color, err = lights.GetColorByHex(hex)
+		if err != nil {
+			return err
+		}
+	} else if name != "" {
+		color, err = lights.GetColorByName(name)
+		if err != nil {
+			return err
+		}
 	}
 
 	if ip != "" && !magichome.IsPrivateIpv4(ip) {

cmd/scene.go

@@ -25,7 +25,7 @@ var sceneCmd = &cobra.Command{
 	},
 }
 
-func createColorArray(c lights.Color, iterations int) []lights.Color {
+func createColors(c lights.Color, iterations int) ([]lights.Color, error) {
 	var arr []lights.Color
 
 	var r uint8 = c.R
@@ -40,15 +40,19 @@ func createColorArray(c lights.Color, iterations int) []lights.Color {
 	if c.B == 255 {
 		b = c.B - 200
 	}
+	var w uint8 = c.W
 
-	modifiedColor := lights.Color{R: r, G: g, B: b, W: c.W}
+	modifiedColor, err := lights.NewColor(r, g, b, w)
+	if err != nil {
+		return nil, err
+	}
 	for i := 0; i < iterations; i++ {
 		arr = append(arr, c)
 		arr = append(arr, modifiedColor)
 	}
 	arr = append(arr, c)
 
-	return arr
+	return arr, nil
 }
 
 func scene(ip string, colorName string, args []string) error {
@@ -57,13 +61,16 @@ func scene(ip string, colorName string, args []string) error {
 		return err
 	}
 
-	colorArray := createColorArray(c, 3)
+	colors, err := createColors(c, 3)
+	if err != nil {
+		return err
+	}
 
 	if ip != "" && !magichome.IsPrivateIpv4(ip) {
 		fmt.Printf("Error while validating ip: %s", ip)
 		return fmt.Errorf("error while validating ip: %s", ip)
 	} else if ip != "" {
-		for _, color := range colorArray {
+		for _, color := range colors {
 			d, err := magichome.NewDevice(net.ParseIP(ip), "", "", "")
 			if err != nil {
 				fmt.Println(err)
@@ -85,7 +92,7 @@ func scene(ip string, colorName string, args []string) error {
 		var iterations uint8 = 3
 		var delay time.Duration = 250 * time.Millisecond
 
-		s := magichome.NewScene(devices, colorArray, iterations, delay)
+		s := magichome.NewScene(devices, colors, iterations, delay)
 		if err := s.Play(); err != nil {
 			return err
 		}

examples/scene/main.go

@@ -16,15 +16,15 @@ func main() {
 		Timeout:       3,
 	})
 
-	var colorArray = []lights.Color{
+	var colors = []lights.Color{
 		lights.Blue(),
 		lights.Red(),
 		lights.Blue(),
 	}
 	var iterations uint8 = 3
 	var delay time.Duration = 250 * time.Millisecond
 
-	s := magichome.NewScene(devices, colorArray, iterations, delay)
+	s := magichome.NewScene(devices, colors, iterations, delay)
 	if err := s.Play(); err != nil {
 		fmt.Printf("Error %+v", err)
 		os.Exit(1)

pkg/lights/color.go

@@ -0,0 +1,54 @@
+package lights
+
+type InvalidColorError struct {
+	color string
+	msg   string
+}
+
+func (err *InvalidColorError) Error() string {
+	return err.msg
+}
+
+type InvalidColorValue struct {
+	value uint8
+	msg   string
+}
+
+func (err *InvalidColorValue) Error() string {
+	return err.msg
+}
+
+// Color represents a RGBW color
+type Color struct {
+	R uint8
+	G uint8
+	B uint8
+	W uint8
+}
+
+// NewColor initializes a new Color using valid RGBW values (between 0 - 255).
+func NewColor(r uint8, g uint8, b uint8, w uint8) (Color, error) {
+	var color Color
+
+	if r < 0 && r > 255 {
+		return color, &InvalidColorValue{r, "Red value is not within range. e.g. Value must be between 0 - 255"}
+	}
+	if g < 0 && g > 255 {
+		return color, &InvalidColorValue{r, "Green value is not within range. e.g. Value must be between 0 - 255"}
+	}
+	if b < 0 && b > 255 {
+		return color, &InvalidColorValue{r, "Blue value is not within range. e.g. Value must be between 0 - 255"}
+	}
+	if w < 0 && w > 255 {
+		return color, &InvalidColorValue{r, "White value is not within range. e.g. Value must be between 0 - 255"}
+	}
+
+	color = Color{
+		R: r,
+		G: g,
+		B: b,
+		W: w,
+	}
+
+	return color, nil
+}

pkg/lights/presets.go

@@ -1,13 +1,5 @@
 package lights
 
-// Color represents a RGBW color
-type Color struct {
-	R uint8
-	G uint8
-	B uint8
-	W uint8
-}
-
 // A White light
 func White() Color {
 	return Color{R: 255, G: 255, B: 255, W: 0}

pkg/lights/utils.go

@@ -8,15 +8,6 @@ import (
 	"gopkg.in/go-playground/colors.v1"
 )
 
-type invalidColorError struct {
-	color string
-	msg   string
-}
-
-func (ir *invalidColorError) Error() string {
-	return ir.msg
-}
-
 func reverseMap(m map[string]string) map[string]string {
 	n := make(map[string]string)
 	for k, v := range m {
@@ -39,7 +30,7 @@ func HexToColor(hex string) Color {
 	return Color{R: rgba.R, G: rgba.G, B: rgba.B, W: uint8(rgba.A * 100)}
 }
 
-func lookupColorByName(color string) (*Color, *invalidColorError) {
+func lookupColorByName(color string) (*Color, *InvalidColorError) {
 	content, err := ioutil.ReadFile("colors.min.json")
 
 	if err != nil {
@@ -51,16 +42,16 @@ func lookupColorByName(color string) (*Color, *invalidColorError) {
 	colorMap := reverseMap(hexMap)
 	hex, ok := colorMap[string(color)]
 	if !ok {
-		return nil, &invalidColorError{color, "Color is invalid"}
+		return nil, &InvalidColorError{color, "Color is invalid"}
 	}
 	h := HexToColor(hex)
 	return &h, nil
 }
 
-func lookupPrimaryColor(color string) (*Color, *invalidColorError) {
+func lookupPrimaryColor(color string) (*Color, *InvalidColorError) {
 	c, ok := ColorPresets[color]
 	if !ok {
-		return nil, &invalidColorError{color, "Color is invalid"}
+		return nil, &InvalidColorError{color, "Color is invalid"}
 	}
 	return &c, nil
 }
@@ -82,19 +73,11 @@ func GetColorByName(name string) (Color, error) {
 	return White(), nil
 }
 
-func GetColor(hex string, name string) (color Color, err error) {
+func GetColorByHex(hex string) (Color, error) {
+	var color Color
 	if hex != "" {
 		color = HexToColor(hex)
 		return color, nil
 	}
-
-	if color, err = GetColorByName(name); err != nil {
-		return color, err
-	}
-
-	if color != (Color{}) {
-		return color, nil
-	}
-
-	return White(), nil
+	return color, &InvalidColorError{hex, "Color is invalid"}
 }