99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/grandcat/zeroconf"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type Bridge struct {
|
|
IP net.IP
|
|
Name string
|
|
}
|
|
|
|
type RegisteredApplication struct {
|
|
bridge Bridge
|
|
applicationName string
|
|
deviceName string
|
|
token string
|
|
}
|
|
|
|
func discoverHueBridge() ([]Bridge, error) {
|
|
// Create a new resolver
|
|
resolver, err := zeroconf.NewResolver(nil)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize resolver: %v", err)
|
|
}
|
|
|
|
// Instantiate a new bridges slice
|
|
bridges := make([]Bridge, 0)
|
|
|
|
// Set up a context with a timeout
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
|
|
defer cancel()
|
|
|
|
// Channel to recive the results
|
|
entries := make(chan *zeroconf.ServiceEntry)
|
|
|
|
// Start the lookup for the Hue Bridge
|
|
go func() {
|
|
err = resolver.Browse(ctx, "_hue._tcp", "local.", entries)
|
|
if err != nil {
|
|
log.Fatalf("Failed to browse for Hue Bridge: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Process the results
|
|
go func() {
|
|
for entry := range entries {
|
|
bridges = append(bridges, Bridge{IP: entry.AddrIPv4[0], Name: entry.HostName})
|
|
}
|
|
}()
|
|
|
|
// Wait for the context to expire
|
|
<-ctx.Done()
|
|
fmt.Println("mDNS query finished")
|
|
fmt.Printf("Number of devices found: %d\n", len(bridges))
|
|
for _, device := range bridges {
|
|
fmt.Printf("Device: %s\n", device.Name)
|
|
fmt.Printf("IP Address: %s\n", device.IP)
|
|
}
|
|
return bridges, nil
|
|
}
|
|
|
|
func registerApplication(bridge Bridge) RegisteredApplication {
|
|
var applicationName string
|
|
var deviceName string
|
|
fmt.Printf("Enter name of your application: ")
|
|
fmt.Scan(&applicationName)
|
|
fmt.Printf("Enter name of your device: ")
|
|
fmt.Scan(&deviceName)
|
|
resp, err := http.Post("http://"+bridge.IP.String()+"/api", "application/json", bytes.NewBuffer([]byte(`{"devicetype:`+applicationName+` #`+deviceName+`"}`)))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
registeredApplication := RegisteredApplication{
|
|
bridge: bridge,
|
|
applicationName: applicationName,
|
|
deviceName: deviceName,
|
|
token: string(body)}
|
|
fmt.Println(registeredApplication)
|
|
return registeredApplication
|
|
}
|
|
|
|
func main() {
|
|
|
|
}
|