automation/main.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/amimof/huego"
	MQTT "github.com/eclipse/paho.mqtt.golang"
	"github.com/joho/godotenv"
	"github.com/kelvins/sunrisesunset"
)

type Location struct {
	Type      string   `json:"_type"`
	Longitude float32  `json:"lon"`
	Latitude  float32  `json:"lat"`
	Altitude  int      `json:"alt"`
	InRegions []string `json:"inregions"`
}

var defaultHandler MQTT.MessageHandler = func(client MQTT.Client, msg MQTT.Message) {
	fmt.Printf("TOPIC: %s\n", msg.Topic())
	fmt.Printf("MSG: %s\n", msg.Payload())
}

const groupId = 1
const brightness = 100

func getNextSunriseSunset() (time.Time, time.Time) {
	p := sunrisesunset.Parameters{
		Latitude:  51.9808334,
		Longitude: 4.347818,
		Date:      time.Now(),
	}

	sunrise, sunset, err := p.GetSunriseSunset()
	if err != nil {
		panic(err)
	}

	p2 := sunrisesunset.Parameters{
		Latitude:  51.9808334,
		Longitude: 4.347818,
		Date:      time.Now().Add(time.Hour * 24),
	}
	sunrise2, sunset2, err := p2.GetSunriseSunset()

	now := time.Now()
	if now.After(sunrise) {
		sunrise = sunrise2
	}

	if now.After(sunset) {
		sunset = sunset2
	}

	return sunrise, sunset
}

func isDay() bool {
	p := sunrisesunset.Parameters{
		Latitude:  51.9808334,
		Longitude: 4.347818,
		Date:      time.Now(),
	}

	sunrise, sunset, err := p.GetSunriseSunset()
	if err != nil {
		panic(err)
	}

	return time.Now().After(sunrise) && time.Now().Before(sunset)
}

func allLightsOff(bridge *huego.Bridge) {
	lights, _ := bridge.GetLights()
	for _, l := range lights {
		l.Off()
	}
}

func main() {
	_ = godotenv.Load()

	host, ok := os.LookupEnv("MQTT_HOST")
	if !ok {
		host = "localhost"
	}
	port, ok := os.LookupEnv("MQTT_PORT")
	if !ok {
		port = "1883"
	}
	user, ok := os.LookupEnv("MQTT_USER")
	if !ok {
		user = "test"
	}
	pass, ok := os.LookupEnv("MQTT_PASS")
	if !ok {
		pass = "test"
	}
	login, ok := os.LookupEnv("HUE_BRIDGE")

	fmt.Println(host, port, user, pass, login)

	halt := make(chan os.Signal, 1)
	signal.Notify(halt, os.Interrupt, syscall.SIGTERM)

	bridge, _ := huego.Discover()
	bridge = bridge.Login(login)
	livingRoom, _ := bridge.GetGroup(groupId)

	opts := MQTT.NewClientOptions().AddBroker(fmt.Sprintf("%s:%s", host, port))
	opts.SetClientID("automation")
	opts.SetDefaultPublishHandler(defaultHandler)
	opts.SetUsername(user)
	opts.SetPassword(pass)

	c := MQTT.NewClient(opts)
	if token := c.Connect(); token.Wait() && token.Error() != nil {
		panic(token.Error())
	}

	home := make(chan bool, 1)
	var locationHandler MQTT.MessageHandler = func(client MQTT.Client, msg MQTT.Message) {
		var location Location
		json.Unmarshal(msg.Payload(), &location)

		// Make sure that the message we get is a location update
		// @TODO Find a more elegant way of doing this
		if location.Type != "location" {
			return
		}

		fmt.Println(location)
		temp := false
		for _, region := range location.InRegions {
			if region == "home" {
				temp = true
			}
		}
		fmt.Println(temp)

		home <- temp
	}

	if token := c.Subscribe("owntracks/mqtt/apollo", 0, locationHandler); token.Wait() && token.Error() != nil {
		fmt.Println(token.Error())
		os.Exit(1)
	}

	// Setup initial states
	isHome := false

	sunrise, sunset := getNextSunriseSunset()
	sunriseTimer := time.NewTimer(sunrise.Sub(time.Now()))
	sunsetTimer := time.NewTimer(sunset.Sub(time.Now()))

	// Create the ticker, but stop it
	ticker := time.NewTicker(time.Second)
	ticker.Stop()

	var brightness uint8 = 0

	// Event loop
events:
	for {
		select {
		case temp := <-home:
			// Only do stuff if the state changes
			if temp == isHome {
				break
			}
			isHome = temp

			if isHome {
				fmt.Println("Coming home")
				if !isDay() {
					fmt.Println("\tTurning on lights in the living room")
					livingRoom.Bri(0xff)
				}
			} else {
				// Stop the ticker in case it is running
				ticker.Stop()

				fmt.Println("Leaving home")
				allLightsOff(bridge)
				break
			}

		case <-sunriseTimer.C:
			fmt.Println("Sun is rising, turning off all lights")
			allLightsOff(bridge)

			// Set new timer
			sunrise, _ := getNextSunriseSunset()
			sunriseTimer.Reset(sunrise.Sub(time.Now()))

		case <-sunsetTimer.C:
			fmt.Println("Sun is setting")
			if isHome {
				fmt.Println("\tGradually turning on lights in the living room")
				// Start the ticker to gradually turn on the living room lights
				ticker.Reset(1200 * time.Millisecond)
			}

			// Set new timer
			_, sunset := getNextSunriseSunset()
			sunsetTimer.Reset(sunset.Sub(time.Now()))

		case <-ticker.C:
			fmt.Println("Setting brightness:", brightness)
			livingRoom.Bri(brightness)

			brightness++
			if brightness == 0xff {
				fmt.Println("Lights are now on, stopping ticker")
				ticker.Stop()
				brightness = 0
			}

		case <-halt:
			break events
		}
	}

	// Cleanup
	if token := c.Unsubscribe("owntracks/mqtt/apollo"); token.Wait() && token.Error() != nil {
		fmt.Println(token.Error())
		os.Exit(1)
	}

	c.Disconnect(250)
}