go.charczuk.com@v0.0.0-20240327042549-bc490516bd1a/experiments/huectl/pkg/command/list_lights.go

/*

Copyright (c) 2023 - Present. Will Charczuk. All rights reserved.
Use of this source code is governed by a MIT license that can be found in the LICENSE file at the root of the repository.

*/

package command

import (
	"fmt"
	"os"

	"github.com/urfave/cli/v2"

	"go.charczuk.com/experiments/huectl/pkg/hue"
)

// ListLights returns a command.
func ListLights() *cli.Command {
	return &cli.Command{
		Name:  "list-lights",
		Usage: "List lights associated with a given bridge.",
		Flags: append(DefaultFlags, append(OutputFlags, GroupFlags...)...),
		Action: func(c *cli.Context) error {
			bridge, err := initHelper(c)
			if err != nil {
				return err
			}

			if c.Int(flagGroupID) > 0 && c.String(flagGroupName) != "" {
				return fmt.Errorf("please specify one of --group-id or --group-name")
			}
			var group *hue.Group
			if c.Int(flagGroupID) > 0 || c.String(flagGroupName) != "" {
				group, err = getGroup(c, &bridge, getGroupArgs{
					ID:   c.Int(flagGroupID),
					Name: c.String(flagGroupName),
				})
				if err != nil {
					return err
				}
			}

			lights, err := bridge.GetLights(c.Context)
			if err != nil {
				return err
			}

			if group != nil {
				var filtered []hue.Light
				for _, light := range lights {
					var include bool
					for _, lightID := range group.Lights {
						if lightID == fmt.Sprint(light.ID) {
							include = true
							break
						}
					}
					if include {
						filtered = append(filtered, light)
					}
				}
				output(c, os.Stdout, filtered)
				return nil
			}

			output(c, os.Stdout, lights)
			return nil
		},
	}
}