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

---
 config.go | 58 ++++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 52 insertions(+), 6 deletions(-)

(limited to 'config.go')

diff --git a/config.go b/config.go
index a12a240..1402c17 100644
--- a/config.go
+++ b/config.go
@@ -1,8 +1,10 @@
 package main
 
 import (
+	"errors"
 	"io/ioutil"
 	"log"
+	"strings"
 
 	"github.com/BurntSushi/xdg"
 	"gopkg.in/yaml.v2"
@@ -10,13 +12,15 @@ import (
 
 const XDG_CONFIG_NAME = "phlux"
 
+type BridgeConfig struct {
+	BridgeID string `yaml:"id"`
+	Token    string `yaml:"token"`
+}
+
 type PhluxConfig struct {
-	Latitude  float64 `yaml:"latitude"`
-	Longitude float64 `yaml:"longitude"`
-	Bridges   []struct {
-		BridgeID string `yaml:"id"`
-		Token    string `yaml:"token"`
-	} `yaml:"bridges"`
+	Latitude  float64        `yaml:"latitude"`
+	Longitude float64        `yaml:"longitude"`
+	Bridges   []BridgeConfig `yaml:"bridges"`
 }
 
 func (c *PhluxConfig) Read() {
@@ -37,3 +41,45 @@ func (c *PhluxConfig) Read() {
 		return
 	}
 }
+
+func (c *PhluxConfig) GetBridgeToken(id string) (string, error) {
+	lc := strings.ToLower(id)
+	for _, bridge := range c.Bridges {
+		if strings.ToLower(bridge.BridgeID) == lc {
+			return bridge.Token, nil
+		}
+	}
+	return "", errors.New("No token found for bridge " + id)
+}
+
+func (c *PhluxConfig) SetBridgeToken(id, token string) {
+	for _, bridge := range c.Bridges {
+		if bridge.BridgeID == id {
+			bridge.Token = token
+			return
+		}
+	}
+	c.Bridges = append(c.Bridges, BridgeConfig{
+		BridgeID: id,
+		Token:    token,
+	})
+}
+
+func (c *PhluxConfig) Save() (err error) {
+	out, err := yaml.Marshal(c)
+	if err != nil {
+		log.Printf("Error marshalling config: %s\n", err.Error())
+	}
+	var paths xdg.Paths
+	configFile, err := paths.ConfigFile(XDG_CONFIG_NAME)
+	if err != nil {
+		log.Printf("No config file found: %s\n", err.Error())
+		return
+	}
+	err = ioutil.WriteFile(configFile, out, 0600)
+	if err != nil {
+		log.Printf("Error writing config file %s: %s\n", configFile, err.Error())
+		return
+	}
+	return
+}
-- 
cgit v1.2.3