From d6b0483feb598a0c69c0d17a754545a9355e500c Mon Sep 17 00:00:00 2001
From: Ben Burwell <ben@benburwell.com>
Date: Thu, 24 May 2018 23:59:09 -0400
Subject: Authenticate to bridge from config

---
 main.go | 65 +++++++++++++++++++++++++++++++++++++++++++++++++----------------
 1 file changed, 49 insertions(+), 16 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 6253cbb..00850db 100644
--- a/main.go
+++ b/main.go
@@ -1,14 +1,15 @@
 package main
 
 import (
+	"errors"
+	"fmt"
 	"log"
-	"os"
 	"time"
 
 	hue "github.com/benburwell/gohue"
 )
 
-const username = "phlux"
+const USERNAME = "phlux"
 
 func main() {
 	var config PhluxConfig
@@ -23,29 +24,61 @@ func main() {
 	desiredColorTemp := getDesiredColorTemperature(time.Now(), config.Latitude, config.Longitude)
 	for _, bridge := range bridges {
 		log.Printf("Bridge: %s\n", bridge.IPAddress)
-		updateBridge(bridge, desiredColorTemp)
+		updateBridge(&bridge, desiredColorTemp, &config)
 	}
 }
 
-func updateBridge(bridge hue.Bridge, ct ColorTemperature) {
-	//username, err := bridge.CreateUser(username)
-	//if err != nil {
-	//	panic("Could not create user on bridge")
-	//}
-	//fmt.Printf("Made user %s for bridge %s\n", username, bridge.IPAddress)
-	err := bridge.Login(os.Getenv("HUE_LOGIN"))
+// In case we don't know the bridge's serial number for some reason, we won't
+// be able to look up the token, nor will we be able to save it. There is still
+// a chance we will be able to successfully log in though: if the link button
+// has been pressed, we should be able to create ourselves a temporary token.
+func authenticateOnce(bridge *hue.Bridge) error {
+	token, err := bridge.CreateUser(USERNAME)
 	if err != nil {
-		log.Fatalf("Could not log in to bridge: %s\n", err.Error())
+		return errors.New(fmt.Sprintf("Could not create temporary token: %s", err.Error()))
 	}
-	log.Println("Logged in to bridge")
+	log.Printf("Made token %s for bridge %s\n", token, bridge.IPAddress)
+	err = bridge.Login(token)
+	if err != nil {
+		return errors.New(fmt.Sprintf("Failed to log in with temporary token: %s", err.Error()))
+	}
+	log.Printf("Logged in to bridge %s\n", bridge.IPAddress)
+	return nil
+}
+
+func createToken(bridge *hue.Bridge, config *PhluxConfig) error {
+	if err := authenticateOnce(bridge); err != nil {
+		return err
+	}
+	config.SetBridgeToken(bridge.Info.Device.SerialNumber, bridge.Username)
+	config.Save()
+	return nil
+}
 
-	config, err := bridge.GetConfig()
+// Attempt to authenticate to the bridge using a variety of techniques,
+// including looking up a saved token for the bridge's serial number and
+// attempting to generate a new token for the bridge assuming the link button
+// has been pressed.
+func authenticate(bridge *hue.Bridge, config *PhluxConfig) error {
+	// get bridge info, which contains serial number
+	err := bridge.GetInfo()
 	if err != nil {
-		log.Fatalf("Could not get bridge info: %s\n", err.Error())
+		return authenticateOnce(bridge)
 	}
-	log.Printf("Bridge name: %s\n", config.Name)
-	log.Printf("Bridge ID: %s\n", config.BridgeID)
+	token, err := config.GetBridgeToken(bridge.Info.Device.SerialNumber)
+	if err != nil {
+		return createToken(bridge, config)
+	}
+	err = bridge.Login(token)
+	if err != nil {
+		log.Fatalf("Could not log in to bridge: %s\n", err.Error())
+	}
+	log.Println("Logged in to bridge")
+	return nil
+}
 
+func updateBridge(bridge *hue.Bridge, ct ColorTemperature, config *PhluxConfig) {
+	authenticate(bridge, config)
 	lights, err := bridge.GetAllLights()
 	if err != nil {
 		log.Fatalf("Error getting lights: %s\n", err.Error())
-- 
cgit v1.2.3