ziggs/main.go

package main

import (
	"context"
	"os"
	"time"

	"github.com/rs/zerolog"

	"git.tcp.direct/kayos/ziggs/config"
	"git.tcp.direct/kayos/ziggs/interactive"
	"git.tcp.direct/kayos/ziggs/lights"
)

var (
	log *zerolog.Logger
)

func init() {
	config.Init()
	log = config.StartLogger()
	log.Info().Msg("Logger started")
}

func TurnAll(Known []*lights.Controller, mode lights.ToggleMode) {
	for _, bridge := range Known {
		for _, l := range bridge.HueLights {
			go func(l *lights.HueLight) {
				l.Log().Debug().
					Str("caller", bridge.Host).
					Str("type", l.ProductName).
					Bool("on", l.IsOn()).Msg(l.ModelID)
				ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(5*time.Second))
				lights.Assert(ctx, l, mode)
				defer cancel()
			}(l)
		}
	}
}

func AggressiveStart() (Known []*lights.Controller) {
	for {
		Known, err := lights.Setup()
		if err != nil {
			log.Warn().Int("count", len(Known)).
				Msgf("%s", err.Error())
			continue
		}
		if len(Known) > 0 {
			return Known
		}
	}
}

func FindLights(ctx context.Context, c *lights.Controller) error {
	log.Trace().Msg("looking for lights...")
	resp, err := c.FindLights()
	if err != nil {
		c.Log().Fatal().Err(err).Msg("FUBAR")
	}
	for str, inter := range resp.Success {
		c.Log().Trace().Interface(str, inter).Msg(" ")
	}
	var count = 0
	for {
		select {
		case <-ctx.Done():
			return context.Canceled
		default:
			ls, err := c.GetLights()
			if err != nil {
				c.Log().Warn().Err(err).Msg(" ")
			}
			if len(ls) > count {
				count = len(ls)
				return nil
			}
		}
	}
}

func main() {
	Known := AggressiveStart()

	if len(Known) == 0 {
		panic("")
	}

	for _, arg := range os.Args {
		switch arg {
		case "on":
			log.Debug().Msg("turning all " + arg)
			TurnAll(Known, lights.ToggleOn)
		case "off":
			log.Debug().Msg("turning all " + arg)
			TurnAll(Known, lights.ToggleOff)
		case "rainbow":
			log.Debug().Msg("turning all " + arg)
			TurnAll(Known, lights.ToggleRainbow)
		case "scan":
			log.Debug().Msg("executing " + arg)
			if len(os.Args) < 2 {
				for _, k := range Known {
					ctx := context.TODO()
					FindLights(ctx, k)
				}
			}
		default:
			interactive.StartCLI(Known)
		}
	}

	done := make(chan struct{}, 1)
	<-done
}