1 files changed, 53 insertions, 29 deletions

diff --git a/lights.go b/lights.go
index 1138312..7c0fda2 100644
--- a/lights.go
+++ b/lights.go
@@ -1,12 +1,13 @@
 package hue
 
 import (
-    "log"
     "fmt"
     "os"
     "net/http"
     "io/ioutil"
     "encoding/json"
+    "strings"
+    "errors"
 )
 
 type Light struct {
@@ -21,7 +22,7 @@ type Light struct {
         alert       string    `json:"alert"`
         colormode   string    `json:"colormode"`
         reachable   bool      `json:"reachable"`
-    }
+    } `json:"state"`
     Type             string     `json:"type"`
     Name             string     `json:"name"`
     ModelID          string     `json:"modelid"`
@@ -30,35 +31,58 @@ type Light struct {
     SWVersion        string     `json:"swversion"`
 }
 
+
+
 //http://192.168.1.128/api/319b36233bd2328f3e40731b23479207/lights/
-// http://<bridge_ip>/api/<username>/lights/
 
 // GetAllLights retreives the state of all lights that the bridge is aware of.
-func GetAllLights(bridge *Bridge) {
-    response, error := http.Get(
-        fmt.Sprintf("http://%s/api/%s/lights/", bridge.IPAddress, bridge.Username)) 
-    if error != nil {
-        trace("", error)
-        os.Exit(1)
-    } else if response.StatusCode != 200 {
-        trace(fmt.Sprintf("Bridge status error %d", response.StatusCode), nil)
-        os.Exit(1)
-    }
-    
-    body, error := ioutil.ReadAll(response.Body)
-    defer response.Body.Close()
-    if error != nil {
-        trace("", error)
-        os.Exit(1)
+func GetAllLights(bridge *Bridge) []Light {
+    // Loop through all light indicies to see if they exist
+    // and parse their values. Supports 100 lights.
+    var lights []Light
+    for index := 1; index < 101; index++ {
+        response, err := http.Get(
+            fmt.Sprintf("http://%s/api/%s/lights/%d", bridge.IPAddress, bridge.Username, index))
+        if err != nil {
+            trace("", err)
+            os.Exit(1)
+        } else if response.StatusCode != 200 {
+            trace(fmt.Sprintf("Bridge status error %d", response.StatusCode), nil)
+            os.Exit(1)
+        }
+
+        // Read the response
+        body, err := ioutil.ReadAll(response.Body)
+        defer response.Body.Close()
+        if err != nil {
+            trace("", err)
+            os.Exit(1)
+        }
+        if strings.Contains(string(body), "not available") {
+            // Handle end of searchable lights
+            fmt.Printf("\n\n%d lights found.\n\n", index)
+            break
+        }
+
+        // Parse and load the response into the light array
+        data := Light{}
+        err = json.Unmarshal(body, &data)
+        if err != nil {
+            trace("", err)
+            os.Exit(1)
+        }
+        lights = append(lights, data)
     }
-    
-    data := Light{}
-    error = json.Unmarshal(body, &data)
-    if error != nil {
-        trace("", error)
-        os.Exit(1)
+    return lights
+}
+
+// GetLight will return a light struct containing data on a given name.
+func GetLight(bridge *Bridge, name string) (Light, error) {
+    lights := GetAllLights(bridge)
+    for index := 0; index < len(lights); index++ {
+        if lights[index].Name == name {
+            return lights[index], nil
+        }
     }
-    
-    log.Println(data)
-    
-}
\ No newline at end of file
+    return Light{}, errors.New("Light not found.")
+}