-
Notifications
You must be signed in to change notification settings - Fork 3
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bridge Discovery via mDNS or URL (#5)
- Loading branch information
Showing
6 changed files
with
267 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,166 @@ | ||
| package openhue | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "github.com/grandcat/zeroconf" | ||
| "io" | ||
| "net/http" | ||
| "strings" | ||
| "time" | ||
| ) | ||
|
|
||
| const ( | ||
| defaultTimeout = 5 * time.Second | ||
| bridgeService = "_hue._tcp" | ||
| discoveryUrl = "https://discovery.meethue.com" | ||
| ) | ||
|
|
||
| type opt func(b *BridgeDiscovery) | ||
|
|
||
| type BridgeDiscovery struct { | ||
| timeout time.Duration | ||
| allowUrlDiscoveryFallback bool | ||
| } | ||
|
|
||
| func (e BridgeDiscoveryError) Error() string { | ||
| return string(e) | ||
| } | ||
|
|
||
| type BridgeInfo struct { | ||
| Instance string | ||
| HostName string | ||
| IpAddress string | ||
| } | ||
|
|
||
| func (b *BridgeInfo) String() string { | ||
| return fmt.Sprintf("Bridge{instance: \"%s\", host: \"%s\", ip: \"%s\"}", b.Instance, b.HostName, b.IpAddress) | ||
| } | ||
|
|
||
| type BridgeDiscoveryError string | ||
|
|
||
| const ( | ||
| TimeoutError BridgeDiscoveryError = "discovery via mDNS timeout" | ||
| NotFoundError BridgeDiscoveryError = "no bridge found" | ||
| TooManyAttempts BridgeDiscoveryError = "too many attempts to discover the bridge via URL" | ||
| ) | ||
|
|
||
| func NewBridgeDiscovery(opts ...opt) *BridgeDiscovery { | ||
| bd := &BridgeDiscovery{ | ||
| timeout: defaultTimeout, | ||
| allowUrlDiscoveryFallback: true, | ||
| } | ||
| for _, o := range opts { | ||
| o(bd) | ||
| } | ||
| return bd | ||
| } | ||
|
|
||
| func (d *BridgeDiscovery) Discover() (*BridgeInfo, error) { | ||
|
|
||
| bridgeInfo, err := mDNSDiscovery(d.timeout) | ||
| if err != nil { | ||
| if d.allowUrlDiscoveryFallback { | ||
| bridgeInfo, err = urlDiscovery() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| } else { | ||
| return nil, err | ||
| } | ||
| } | ||
|
|
||
| return bridgeInfo, nil | ||
| } | ||
|
|
||
| func mDNSDiscovery(timeout time.Duration) (*BridgeInfo, error) { | ||
| resolver, err := zeroconf.NewResolver() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| entries := make(chan *zeroconf.ServiceEntry) | ||
| foundEntry := make(chan *zeroconf.ServiceEntry) | ||
| var entry *zeroconf.ServiceEntry | ||
|
|
||
| ctx, cancel := context.WithTimeout(context.Background(), timeout) | ||
| defer cancel() | ||
|
|
||
| if err := resolver.Browse(ctx, bridgeService, "local", entries); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| go func(results <-chan *zeroconf.ServiceEntry, foundEntry chan *zeroconf.ServiceEntry) { | ||
| for e := range results { | ||
| if (len(e.AddrIPv4)) > 0 { | ||
| foundEntry <- e | ||
| cancel() | ||
| } | ||
| } | ||
| }(entries, foundEntry) | ||
|
|
||
| select { | ||
| // we only expect one hub to be found for now | ||
| case entry = <-foundEntry: | ||
| case <-ctx.Done(): | ||
| return nil, TimeoutError | ||
| } | ||
|
|
||
| if entry == nil { | ||
| return nil, NotFoundError | ||
| } | ||
|
|
||
| return &BridgeInfo{ | ||
| Instance: strings.ReplaceAll(entry.Instance, "\\", ""), | ||
| HostName: entry.HostName, | ||
| IpAddress: entry.AddrIPv4[0].String(), | ||
| }, nil | ||
| } | ||
|
|
||
| func urlDiscovery() (*BridgeInfo, error) { | ||
| resp, err := http.Get(discoveryUrl) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| if resp.StatusCode == http.StatusTooManyRequests { | ||
| return nil, TooManyAttempts | ||
| } | ||
|
|
||
| bridges := make([]map[string]interface{}, 0) | ||
| body, err := io.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| err = json.Unmarshal(body, &bridges) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| if len(bridges) == 0 { | ||
| return nil, NotFoundError | ||
| } | ||
|
|
||
| return &BridgeInfo{ | ||
| Instance: "N/A", | ||
| HostName: fmt.Sprintf("%s", bridges[0]["id"]), | ||
| IpAddress: fmt.Sprintf("%s", bridges[0]["internalipaddress"]), | ||
| }, nil | ||
| } | ||
|
|
||
| // WithTimeout specifies that timeout value for the Bridge Discovery. Default is 5 seconds. | ||
| func WithTimeout(timeout time.Duration) opt { | ||
| return func(b *BridgeDiscovery) { | ||
| b.timeout = timeout | ||
| } | ||
| } | ||
|
|
||
| // WithDisabledUrlDiscovery allows disabling the URL discovery process in case the mDNS one failed. | ||
| func WithDisabledUrlDiscovery() opt { | ||
| return func(b *BridgeDiscovery) { | ||
| b.allowUrlDiscoveryFallback = false | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "github.com/openhue/openhue-go" | ||
| "log" | ||
| "time" | ||
| ) | ||
|
|
||
| func main() { | ||
|
|
||
| bridge, err := openhue.NewBridgeDiscovery(openhue.WithTimeout(1 * time.Second)).Discover() | ||
| if err != nil { | ||
| log.Fatal("Bridge Discovery Error: ", err) | ||
| } | ||
|
|
||
| fmt.Println(bridge) | ||
| } |