11 changed files with 1096 additions and 23 deletions
-
2app/config/driver.go
-
12cmd/bridgetest/main.go
-
1go.mod
-
2go.sum
-
286internal/drivers/hue/bridge.go
-
196internal/drivers/hue/data.go
-
399internal/drivers/hue/driver.go
-
179internal/drivers/hue/state.go
-
12internal/drivers/nanoleaf/bridge.go
-
3models/device.go
-
3models/event.go
@ -0,0 +1,286 @@ |
|||
package hue |
|||
|
|||
import ( |
|||
"bytes" |
|||
"context" |
|||
"encoding/json" |
|||
"fmt" |
|||
"git.aiterp.net/lucifer/new-server/models" |
|||
"golang.org/x/sync/errgroup" |
|||
"io" |
|||
"net/http" |
|||
"strings" |
|||
"sync" |
|||
) |
|||
|
|||
type Bridge struct { |
|||
mu sync.Mutex |
|||
host string |
|||
token string |
|||
externalID int |
|||
lightStates []*hueLightState |
|||
sensorStates []*hueSensorState |
|||
} |
|||
|
|||
func (b *Bridge) Refresh(ctx context.Context) error { |
|||
lightMap, err := b.getLights(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
b.mu.Lock() |
|||
for index, light := range lightMap { |
|||
var state *hueLightState |
|||
for _, existingState := range b.lightStates { |
|||
if existingState.index == index { |
|||
state = existingState |
|||
} |
|||
} |
|||
|
|||
if state == nil { |
|||
state = &hueLightState{ |
|||
index: index, |
|||
uniqueID: light.Uniqueid, |
|||
externalID: -1, |
|||
info: light, |
|||
} |
|||
|
|||
b.lightStates = append(b.lightStates, state) |
|||
} else { |
|||
if light.Uniqueid != state.uniqueID { |
|||
state.uniqueID = light.Uniqueid |
|||
state.externalID = -1 |
|||
} |
|||
} |
|||
|
|||
state.CheckStaleness(light.State) |
|||
} |
|||
b.mu.Unlock() |
|||
|
|||
sensorMap, err := b.getSensors(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
b.mu.Lock() |
|||
for index, sensor := range sensorMap { |
|||
var state *hueSensorState |
|||
for _, existingState := range b.sensorStates { |
|||
if existingState.index == index { |
|||
state = existingState |
|||
} |
|||
} |
|||
|
|||
if state == nil { |
|||
state = &hueSensorState{ |
|||
index: index, |
|||
uniqueID: sensor.UniqueID, |
|||
externalID: -1, |
|||
} |
|||
|
|||
b.sensorStates = append(b.sensorStates, state) |
|||
} else { |
|||
if sensor.UniqueID != state.uniqueID { |
|||
state.uniqueID = sensor.UniqueID |
|||
state.externalID = -1 |
|||
} |
|||
} |
|||
} |
|||
b.mu.Unlock() |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func (b *Bridge) SyncStale(ctx context.Context) error { |
|||
indices := make([]int, 0, 4) |
|||
inputs := make([]LightStateInput, 0, 4) |
|||
|
|||
eg, ctx := errgroup.WithContext(ctx) |
|||
|
|||
b.mu.Lock() |
|||
for _, state := range b.lightStates { |
|||
if !state.stale { |
|||
continue |
|||
} |
|||
|
|||
indices = append(indices, state.index) |
|||
inputs = append(inputs, state.input) |
|||
} |
|||
b.mu.Unlock() |
|||
|
|||
if len(inputs) == 0 { |
|||
return nil |
|||
} |
|||
|
|||
for i, input := range inputs { |
|||
index := indices[i] |
|||
|
|||
eg.Go(func() error { return b.putLightState(ctx, index, input) }) |
|||
} |
|||
|
|||
return eg.Wait() |
|||
} |
|||
|
|||
func (b *Bridge) SyncSensors(ctx context.Context) ([]models.Event, error) { |
|||
sensorMap, err := b.getSensors(ctx) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
var events []models.Event |
|||
|
|||
b.mu.Lock() |
|||
for idx, sensorData := range sensorMap { |
|||
for _, state := range b.sensorStates { |
|||
if idx == state.index { |
|||
event := state.Update(sensorData) |
|||
if event != nil { |
|||
events = append(events, *event) |
|||
} |
|||
|
|||
break |
|||
} |
|||
} |
|||
} |
|||
b.mu.Unlock() |
|||
|
|||
return events, nil |
|||
} |
|||
|
|||
func (b *Bridge) StartDiscovery(ctx context.Context, model string) error { |
|||
return b.post(ctx, model, nil, nil) |
|||
} |
|||
|
|||
func (b *Bridge) putLightState(ctx context.Context, index int, input LightStateInput) error { |
|||
return b.put(ctx, fmt.Sprintf("lights/%d/state", index), input, nil) |
|||
} |
|||
|
|||
func (b *Bridge) getToken(ctx context.Context) (string, error) { |
|||
result := make([]CreateUserResponse, 0, 1) |
|||
err := b.post(ctx, "", CreateUserInput{DeviceType: "git.aiterp.net/lucifer"}, &result) |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
|
|||
if len(result) == 0 || result[0].Error != nil { |
|||
return "", errLinkButtonNotPressed |
|||
} |
|||
if result[0].Success == nil { |
|||
return "", models.ErrUnexpectedResponse |
|||
} |
|||
|
|||
return result[0].Success.Username, nil |
|||
} |
|||
|
|||
func (b *Bridge) getLights(ctx context.Context) (map[int]LightData, error) { |
|||
result := make(map[int]LightData, 16) |
|||
err := b.get(ctx, "lights", &result) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return result, nil |
|||
} |
|||
|
|||
func (b *Bridge) getSensors(ctx context.Context) (map[int]SensorData, error) { |
|||
result := make(map[int]SensorData, 16) |
|||
err := b.get(ctx, "sensors", &result) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return result, nil |
|||
} |
|||
|
|||
func (b *Bridge) get(ctx context.Context, resource string, target interface{}) error { |
|||
if b.token != "" { |
|||
resource = b.token + "/" + resource |
|||
} |
|||
|
|||
req, err := http.NewRequest("GET", fmt.Sprintf("http://%s/api/%s", b.host, resource), nil) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
res, err := http.DefaultClient.Do(req.WithContext(ctx)) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
defer res.Body.Close() |
|||
|
|||
return json.NewDecoder(res.Body).Decode(target) |
|||
} |
|||
|
|||
func (b *Bridge) post(ctx context.Context, resource string, body interface{}, target interface{}) error { |
|||
rb, err := reqBody(body) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
if b.token != "" { |
|||
resource = b.token + "/" + resource |
|||
} |
|||
|
|||
req, err := http.NewRequest("POST", fmt.Sprintf("http://%s/api/%s", b.host, resource), rb) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
res, err := http.DefaultClient.Do(req.WithContext(ctx)) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
defer res.Body.Close() |
|||
|
|||
return json.NewDecoder(res.Body).Decode(target) |
|||
} |
|||
|
|||
func (b *Bridge) put(ctx context.Context, resource string, body interface{}, target interface{}) error { |
|||
rb, err := reqBody(body) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
if b.token != "" { |
|||
resource = b.token + "/" + resource |
|||
} |
|||
|
|||
req, err := http.NewRequest("PUT", fmt.Sprintf("http://%s/api/%s", b.host, resource), rb) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
res, err := http.DefaultClient.Do(req.WithContext(ctx)) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
defer res.Body.Close() |
|||
|
|||
if target == nil { |
|||
return nil |
|||
} |
|||
|
|||
return json.NewDecoder(res.Body).Decode(target) |
|||
} |
|||
|
|||
func reqBody(body interface{}) (io.Reader, error) { |
|||
if body == nil { |
|||
return nil, nil |
|||
} |
|||
|
|||
switch v := body.(type) { |
|||
case []byte: |
|||
return bytes.NewReader(v), nil |
|||
case string: |
|||
return strings.NewReader(v), nil |
|||
case io.Reader: |
|||
return v, nil |
|||
default: |
|||
jsonData, err := json.Marshal(v) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return bytes.NewReader(jsonData), nil |
|||
} |
|||
} |
|||
@ -0,0 +1,196 @@ |
|||
package hue |
|||
|
|||
import ( |
|||
"encoding/xml" |
|||
"errors" |
|||
) |
|||
|
|||
type DiscoveryEntry struct { |
|||
Id string `json:"id"` |
|||
InternalIPAddress string `json:"internalipaddress"` |
|||
} |
|||
|
|||
type CreateUserInput struct { |
|||
DeviceType string `json:"devicetype"` |
|||
} |
|||
|
|||
type CreateUserResponse struct { |
|||
Success *struct { |
|||
Username string `json:"username"` |
|||
} `json:"success"` |
|||
Error *struct { |
|||
Type int `json:"type"` |
|||
Address string `json:"address"` |
|||
Description string `json:"description"` |
|||
} `json:"error"` |
|||
} |
|||
|
|||
type Overview struct { |
|||
Lights map[int]LightData `json:"lights"` |
|||
Config BridgeConfig `json:"config"` |
|||
Sensors map[int]SensorData `json:"sensors"` |
|||
} |
|||
|
|||
type BridgeConfig struct { |
|||
Name string `json:"name"` |
|||
ZigbeeChannel int `json:"zigbeechannel"` |
|||
BridgeID string `json:"bridgeid"` |
|||
Mac string `json:"mac"` |
|||
DHCP bool `json:"dhcp"` |
|||
IPAddress string `json:"ipaddress"` |
|||
NetMask string `json:"netmask"` |
|||
Gateway string `json:"gateway"` |
|||
ProxyAddress string `json:"proxyaddress"` |
|||
ProxyPort int `json:"proxyport"` |
|||
UTC string `json:"UTC"` |
|||
LocalTime string `json:"localtime"` |
|||
TimeZone string `json:"timezone"` |
|||
ModelID string `json:"modelid"` |
|||
DataStoreVersion string `json:"datastoreversion"` |
|||
SWVersion string `json:"swversion"` |
|||
APIVersion string `json:"apiversion"` |
|||
LinkButton bool `json:"linkbutton"` |
|||
PortalServices bool `json:"portalservices"` |
|||
PortalConnection string `json:"portalconnection"` |
|||
FactoryNew bool `json:"factorynew"` |
|||
StarterKitID string `json:"starterkitid"` |
|||
Whitelist map[string]WhitelistEntry `json:"whitelist"` |
|||
} |
|||
|
|||
type LightState struct { |
|||
On bool `json:"on"` |
|||
Bri int `json:"bri"` |
|||
Hue int `json:"hue"` |
|||
Sat int `json:"sat"` |
|||
Effect string `json:"effect"` |
|||
XY []float64 `json:"xy"` |
|||
CT int `json:"ct"` |
|||
Alert string `json:"alert"` |
|||
ColorMode string `json:"colormode"` |
|||
Mode string `json:"mode"` |
|||
Reachable bool `json:"reachable"` |
|||
} |
|||
|
|||
type LightStateInput struct { |
|||
On *bool `json:"on,omitempty"` |
|||
Bri *int `json:"bri,omitempty"` |
|||
Hue *int `json:"hue,omitempty"` |
|||
Sat *int `json:"sat,omitempty"` |
|||
Effect *string `json:"effect,omitempty"` |
|||
XY *[2]float64 `json:"xy,omitempty"` |
|||
CT *int `json:"ct,omitempty"` |
|||
Alert *string `json:"alert,omitempty"` |
|||
TransitionTime *int `json:"transitiontime,omitempty"` |
|||
} |
|||
|
|||
type LightData struct { |
|||
State LightState `json:"state"` |
|||
Type string `json:"type"` |
|||
Name string `json:"name"` |
|||
Modelid string `json:"modelid"` |
|||
Manufacturername string `json:"manufacturername"` |
|||
Productname string `json:"productname"` |
|||
Capabilities struct { |
|||
Certified bool `json:"certified"` |
|||
Control struct { |
|||
Mindimlevel int `json:"mindimlevel"` |
|||
Maxlumen int `json:"maxlumen"` |
|||
Colorgamuttype string `json:"colorgamuttype"` |
|||
Colorgamut [][]float64 `json:"colorgamut"` |
|||
CT struct { |
|||
Min int `json:"min"` |
|||
Max int `json:"max"` |
|||
} `json:"ct"` |
|||
} `json:"control"` |
|||
Streaming struct { |
|||
Renderer bool `json:"renderer"` |
|||
Proxy bool `json:"proxy"` |
|||
} `json:"streaming"` |
|||
} `json:"capabilities"` |
|||
Config struct { |
|||
Archetype string `json:"archetype"` |
|||
Function string `json:"function"` |
|||
Direction string `json:"direction"` |
|||
Startup struct { |
|||
Mode string `json:"mode"` |
|||
Configured bool `json:"configured"` |
|||
} `json:"startup"` |
|||
} `json:"config"` |
|||
Swupdate struct { |
|||
State string `json:"state"` |
|||
Lastinstall string `json:"lastinstall"` |
|||
} `json:"swupdate"` |
|||
Uniqueid string `json:"uniqueid"` |
|||
Swversion string `json:"swversion"` |
|||
Swconfigid string `json:"swconfigid"` |
|||
Productid string `json:"productid"` |
|||
} |
|||
|
|||
type SensorData struct { |
|||
State struct { |
|||
Daylight interface{} `json:"daylight"` |
|||
ButtonEvent int `json:"buttonevent"` |
|||
LastUpdated string `json:"lastupdated"` |
|||
Presence bool `json:"presence"` |
|||
} `json:"state"` |
|||
Config struct { |
|||
On bool `json:"on"` |
|||
Configured bool `json:"configured"` |
|||
Sunriseoffset int `json:"sunriseoffset"` |
|||
Sunsetoffset int `json:"sunsetoffset"` |
|||
} `json:"config"` |
|||
Name string `json:"name"` |
|||
Type string `json:"type"` |
|||
Modelid string `json:"modelid"` |
|||
Manufacturername string `json:"manufacturername"` |
|||
Productname string `json:"productname"` |
|||
Swversion string `json:"swversion"` |
|||
UniqueID string `json:"uniqueid"` |
|||
} |
|||
|
|||
type WhitelistEntry struct { |
|||
LastUseDate string `json:"last use date"` |
|||
CreateDate string `json:"create date"` |
|||
Name string `json:"name"` |
|||
} |
|||
|
|||
type BridgeDeviceInfo struct { |
|||
XMLName xml.Name `xml:"root"` |
|||
Text string `xml:",chardata"` |
|||
Xmlns string `xml:"xmlns,attr"` |
|||
SpecVersion struct { |
|||
Text string `xml:",chardata"` |
|||
Major string `xml:"major"` |
|||
Minor string `xml:"minor"` |
|||
} `xml:"specVersion"` |
|||
URLBase string `xml:"URLBase"` |
|||
Device struct { |
|||
Text string `xml:",chardata"` |
|||
DeviceType string `xml:"deviceType"` |
|||
FriendlyName string `xml:"friendlyName"` |
|||
Manufacturer string `xml:"manufacturer"` |
|||
ManufacturerURL string `xml:"manufacturerURL"` |
|||
ModelDescription string `xml:"modelDescription"` |
|||
ModelName string `xml:"modelName"` |
|||
ModelNumber string `xml:"modelNumber"` |
|||
ModelURL string `xml:"modelURL"` |
|||
SerialNumber string `xml:"serialNumber"` |
|||
UDN string `xml:"UDN"` |
|||
PresentationURL string `xml:"presentationURL"` |
|||
IconList struct { |
|||
Text string `xml:",chardata"` |
|||
Icon struct { |
|||
Text string `xml:",chardata"` |
|||
Mimetype string `xml:"mimetype"` |
|||
Height string `xml:"height"` |
|||
Width string `xml:"width"` |
|||
Depth string `xml:"depth"` |
|||
URL string `xml:"url"` |
|||
} `xml:"icon"` |
|||
} `xml:"iconList"` |
|||
} `xml:"device"` |
|||
} |
|||
|
|||
var buttonNames = []string{"On", "DimUp", "DimDown", "Off"} |
|||
|
|||
var errLinkButtonNotPressed = errors.New("link button not pressed") |
|||
@ -0,0 +1,399 @@ |
|||
package hue |
|||
|
|||
import ( |
|||
"context" |
|||
"encoding/json" |
|||
"encoding/xml" |
|||
"fmt" |
|||
"git.aiterp.net/lucifer/new-server/models" |
|||
"log" |
|||
"net/http" |
|||
"strconv" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
type Driver struct { |
|||
mu sync.Mutex |
|||
bridges []*Bridge |
|||
} |
|||
|
|||
func (d *Driver) SearchBridge(ctx context.Context, address string, dryRun bool) ([]models.Bridge, error) { |
|||
if address == "" { |
|||
if !dryRun { |
|||
return nil, models.ErrAddressOnlyDryRunnable |
|||
} |
|||
|
|||
res, err := http.Get("https://discovery.meethue.com") |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
defer res.Body.Close() |
|||
|
|||
entries := make([]DiscoveryEntry, 0, 8) |
|||
err = json.NewDecoder(res.Body).Decode(&entries) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
bridges := make([]models.Bridge, 0, len(entries)) |
|||
for _, entry := range entries { |
|||
bridges = append(bridges, models.Bridge{ |
|||
ID: -1, |
|||
Name: entry.Id, |
|||
Driver: models.DTHue, |
|||
Address: entry.InternalIPAddress, |
|||
Token: "", |
|||
}) |
|||
} |
|||
|
|||
return bridges, nil |
|||
} |
|||
|
|||
deviceInfo := BridgeDeviceInfo{} |
|||
res, err := http.Get(fmt.Sprintf("http://%s/description.xml", address)) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
defer res.Body.Close() |
|||
err = xml.NewDecoder(res.Body).Decode(&deviceInfo) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
bridge := models.Bridge{ |
|||
ID: -1, |
|||
Name: deviceInfo.Device.FriendlyName, |
|||
Driver: models.DTHue, |
|||
Address: address, |
|||
Token: "", |
|||
} |
|||
|
|||
if !dryRun { |
|||
b := &Bridge{host: address} |
|||
|
|||
timeout, cancel := context.WithTimeout(ctx, time.Second*30) |
|||
defer cancel() |
|||
|
|||
ticker := time.NewTicker(time.Second) |
|||
defer ticker.Stop() |
|||
|
|||
for range ticker.C { |
|||
token, err := b.getToken(timeout) |
|||
if err != nil { |
|||
if err == errLinkButtonNotPressed { |
|||
continue |
|||
} |
|||
|
|||
return nil, err |
|||
} |
|||
|
|||
bridge.Token = token |
|||
b.token = token |
|||
break |
|||
} |
|||
} |
|||
|
|||
return []models.Bridge{bridge}, nil |
|||
} |
|||
|
|||
func (d *Driver) SearchDevices(ctx context.Context, bridge models.Bridge, timeout time.Duration) ([]models.Device, error) { |
|||
b, err := d.ensureBridge(ctx, bridge) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
before, err := d.ListDevices(ctx, bridge) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
if timeout.Seconds() < 10 { |
|||
timeout = time.Second * 10 |
|||
} |
|||
halfTime := timeout / 2 |
|||
|
|||
err = b.StartDiscovery(ctx, "sensors") |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
select { |
|||
case <-time.After(halfTime): |
|||
case <-ctx.Done(): |
|||
return nil, ctx.Err() |
|||
} |
|||
|
|||
err = b.StartDiscovery(ctx, "lights") |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
select { |
|||
case <-time.After(halfTime): |
|||
case <-ctx.Done(): |
|||
return nil, ctx.Err() |
|||
} |
|||
|
|||
err = b.Refresh(ctx) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
after, err := d.ListDevices(ctx, bridge) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
intersection := make([]models.Device, 0, 4) |
|||
for _, device := range after { |
|||
found := false |
|||
for _, device2 := range before { |
|||
if device2.InternalID == device.InternalID { |
|||
found = true |
|||
break |
|||
} |
|||
} |
|||
|
|||
if !found { |
|||
intersection = append(intersection, device) |
|||
} |
|||
} |
|||
|
|||
return intersection, nil |
|||
} |
|||
|
|||
func (d *Driver) ListDevices(ctx context.Context, bridge models.Bridge) ([]models.Device, error) { |
|||
b, err := d.ensureBridge(ctx, bridge) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
devices := make([]models.Device, 0, 8) |
|||
|
|||
lightMap, err := b.getLights(ctx) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
for _, lightInfo := range lightMap { |
|||
device := models.Device{ |
|||
ID: -1, |
|||
BridgeID: b.externalID, |
|||
InternalID: lightInfo.Uniqueid, |
|||
Icon: "lightbulb", |
|||
Name: lightInfo.Name, |
|||
Capabilities: []models.DeviceCapability{ |
|||
models.DCPower, |
|||
}, |
|||
ButtonNames: nil, |
|||
DriverProperties: map[string]interface{}{ |
|||
"modelId": lightInfo.Modelid, |
|||
"productName": lightInfo.Productname, |
|||
"swVersion": lightInfo.Swversion, |
|||
"hueLightType": lightInfo.Type, |
|||
}, |
|||
UserProperties: nil, |
|||
State: models.DeviceState{}, |
|||
Tags: nil, |
|||
} |
|||
|
|||
hasDimming := false |
|||
hasCT := false |
|||
hasColor := false |
|||
|
|||
switch lightInfo.Type { |
|||
case "On/off light": |
|||
// Always take DCPower for granted anyway.
|
|||
case "Dimmable light": |
|||
hasDimming = true |
|||
case "Color temperature light": |
|||
hasDimming = true |
|||
hasCT = true |
|||
case "Color light": |
|||
hasDimming = true |
|||
hasColor = true |
|||
case "Extended color light": |
|||
hasDimming = true |
|||
hasColor = true |
|||
hasCT = true |
|||
} |
|||
|
|||
ctrl := lightInfo.Capabilities.Control |
|||
if hasDimming { |
|||
device.Capabilities = append(device.Capabilities, models.DCIntensity) |
|||
} |
|||
if hasCT { |
|||
device.Capabilities = append(device.Capabilities, models.DCColorKelvin) |
|||
device.DriverProperties["minKelvin"] = 1000000 / ctrl.CT.Max |
|||
device.DriverProperties["maxKelvin"] = 1000000 / ctrl.CT.Min |
|||
} |
|||
if hasColor { |
|||
device.Capabilities = append(device.Capabilities, models.DCColorHS) |
|||
device.DriverProperties["gamutType"] = ctrl.Colorgamuttype |
|||
device.DriverProperties["gamutData"] = ctrl.Colorgamut |
|||
} |
|||
device.DriverProperties["maxLumen"] = strconv.Itoa(ctrl.Maxlumen) |
|||
|
|||
devices = append(devices, device) |
|||
} |
|||
|
|||
sensorMap, err := b.getSensors(ctx) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
for _, sensorInfo := range sensorMap { |
|||
device := models.Device{ |
|||
ID: -1, |
|||
BridgeID: b.externalID, |
|||
InternalID: sensorInfo.UniqueID, |
|||
Name: sensorInfo.Name, |
|||
Capabilities: []models.DeviceCapability{}, |
|||
ButtonNames: []string{}, |
|||
DriverProperties: map[string]interface{}{ |
|||
"modelId": sensorInfo.Modelid, |
|||
"productName": sensorInfo.Productname, |
|||
"swVersion": sensorInfo.Swversion, |
|||
"hueLightType": sensorInfo.Type, |
|||
}, |
|||
UserProperties: nil, |
|||
State: models.DeviceState{}, |
|||
Tags: nil, |
|||
} |
|||
|
|||
switch sensorInfo.Type { |
|||
case "ZLLSwitch": |
|||
device.Capabilities = append(device.Capabilities, models.DCButtons) |
|||
device.ButtonNames = append(buttonNames[:0:0], buttonNames...) |
|||
device.Icon = "lightswitch" |
|||
case "ZLLPresence": |
|||
device.Capabilities = append(device.Capabilities, models.DCPresence) |
|||
device.Icon = "sensor" |
|||
case "Daylight": |
|||
continue |
|||
} |
|||
|
|||
devices = append(devices, device) |
|||
} |
|||
|
|||
return devices, nil |
|||
} |
|||
|
|||
func (d *Driver) Publish(ctx context.Context, bridge models.Bridge, devices []models.Device) error { |
|||
b, err := d.ensureBridge(ctx, bridge) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
err = b.Refresh(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
b.mu.Lock() |
|||
for _, device := range devices { |
|||
for _, state := range b.lightStates { |
|||
if device.InternalID == state.uniqueID { |
|||
state.externalID = device.ID |
|||
state.Update(device.State) |
|||
break |
|||
} |
|||
} |
|||
|
|||
for _, state := range b.sensorStates { |
|||
if device.InternalID == state.uniqueID { |
|||
state.externalID = device.ID |
|||
break |
|||
} |
|||
} |
|||
} |
|||
b.mu.Unlock() |
|||
|
|||
return b.SyncStale(ctx) |
|||
} |
|||
|
|||
func (d *Driver) Run(ctx context.Context, bridge models.Bridge, ch chan<- models.Event) error { |
|||
b, err := d.ensureBridge(ctx, bridge) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
fastTicker := time.NewTicker(time.Second / 10) |
|||
slowTicker := time.NewTicker(time.Second / 3) |
|||
selectedTicker := fastTicker |
|||
ticksUntilRefresh := 0 |
|||
ticksSinceChange := 0 |
|||
|
|||
for { |
|||
select { |
|||
case <-selectedTicker.C: |
|||
if ticksUntilRefresh <= 0 { |
|||
err := b.Refresh(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
err = b.SyncStale(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
ticksUntilRefresh = 60 |
|||
} |
|||
|
|||
events, err := b.SyncSensors(ctx) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
for _, event := range events { |
|||
ch <- event |
|||
ticksSinceChange = 0 |
|||
} |
|||
|
|||
if ticksSinceChange > 30 { |
|||
selectedTicker = slowTicker |
|||
} else if ticksSinceChange == 0 { |
|||
selectedTicker = fastTicker |
|||
} |
|||
case <-ctx.Done(): |
|||
return ctx.Err() |
|||
} |
|||
|
|||
ticksUntilRefresh -= 1 |
|||
ticksSinceChange += 1 |
|||
} |
|||
} |
|||
|
|||
func (d *Driver) ensureBridge(ctx context.Context, info models.Bridge) (*Bridge, error) { |
|||
d.mu.Lock() |
|||
for _, bridge := range d.bridges { |
|||
if bridge.host == info.Address { |
|||
d.mu.Unlock() |
|||
return bridge, nil |
|||
} |
|||
} |
|||
d.mu.Unlock() |
|||
|
|||
bridge := &Bridge{ |
|||
host: info.Address, |
|||
token: info.Token, |
|||
externalID: info.ID, |
|||
} |
|||
|
|||
// If this call succeeds, then the token is ok.
|
|||
lightMap, err := bridge.getLights(ctx) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
log.Printf("Found %d lights on bridge %d", len(lightMap), bridge.externalID) |
|||
|
|||
// To avoid a potential duplicate, try looking for it again before inserting
|
|||
d.mu.Lock() |
|||
for _, bridge := range d.bridges { |
|||
if bridge.host == info.Address { |
|||
d.mu.Unlock() |
|||
return bridge, nil |
|||
} |
|||
} |
|||
d.bridges = append(d.bridges, bridge) |
|||
d.mu.Unlock() |
|||
|
|||
return bridge, nil |
|||
} |
|||
@ -0,0 +1,179 @@ |
|||
package hue |
|||
|
|||
import ( |
|||
"git.aiterp.net/lucifer/new-server/models" |
|||
"strconv" |
|||
"time" |
|||
) |
|||
|
|||
type hueLightState struct { |
|||
index int |
|||
uniqueID string |
|||
externalID int |
|||
info LightData |
|||
input LightStateInput |
|||
|
|||
stale bool |
|||
} |
|||
|
|||
func (s *hueLightState) Update(state models.DeviceState) { |
|||
input := LightStateInput{} |
|||
if state.Power { |
|||
input.On = ptrBool(true) |
|||
if state.Color.IsKelvin() { |
|||
input.CT = ptrInt(1000000 / state.Color.Kelvin) |
|||
if s.input.CT == nil || *s.input.CT != *input.CT { |
|||
s.stale = true |
|||
} |
|||
} else { |
|||
input.Hue = ptrInt(int(state.Color.Hue*(65536/360)) % 65536) |
|||
if s.input.Hue == nil || *s.input.Hue != *input.Hue { |
|||
s.stale = true |
|||
} |
|||
|
|||
input.Sat = ptrInt(int(state.Color.Hue * 255)) |
|||
if *input.Sat > 254 { |
|||
*input.Sat = 254 |
|||
} |
|||
if *input.Sat < 0 { |
|||
*input.Sat = 0 |
|||
} |
|||
if s.input.Sat == nil || *s.input.Sat != *input.Sat { |
|||
s.stale = true |
|||
} |
|||
} |
|||
|
|||
input.Bri = ptrInt(int(state.Intensity * 255)) |
|||
if *input.Bri > 254 { |
|||
*input.Bri = 254 |
|||
} else if *input.Bri < 0 { |
|||
*input.Bri = 0 |
|||
} |
|||
|
|||
if s.input.Bri == nil || *s.input.Bri != *input.Bri { |
|||
s.stale = true |
|||
} |
|||
} else { |
|||
input.On = ptrBool(false) |
|||
} |
|||
|
|||
if s.input.On == nil || *s.input.On != *input.On { |
|||
s.stale = true |
|||
} |
|||
|
|||
input.TransitionTime = ptrInt(1) |
|||
|
|||
s.input = input |
|||
} |
|||
|
|||
func (s *hueLightState) CheckStaleness(state LightState) { |
|||
if state.ColorMode == "xy" { |
|||
s.stale = true |
|||
if s.input.CT == nil && s.input.Hue == nil { |
|||
s.input.Hue = ptrInt(state.Hue) |
|||
s.input.Sat = ptrInt(state.Sat) |
|||
s.input.Bri = ptrInt(state.Bri) |
|||
} |
|||
return |
|||
} else if state.ColorMode == "ct" { |
|||
if s.input.CT == nil || state.CT != *s.input.CT { |
|||
s.stale = true |
|||
} |
|||
} else { |
|||
if s.input.Hue == nil || state.Hue != *s.input.Hue || s.input.Sat == nil || state.Sat == *s.input.Sat { |
|||
s.stale = true |
|||
} |
|||
} |
|||
} |
|||
|
|||
type hueSensorState struct { |
|||
index int |
|||
externalID int |
|||
uniqueID string |
|||
prevData *SensorData |
|||
prevTime time.Time |
|||
} |
|||
|
|||
func (state *hueSensorState) Update(newData SensorData) *models.Event { |
|||
stateTime, err := time.ParseInLocation("2006-01-02T15:04:05", newData.State.LastUpdated, time.UTC) |
|||
if err != nil { |
|||
// Invalid time is probably "none".
|
|||
return nil |
|||
} |
|||
|
|||
defer func() { |
|||
state.prevData = &newData |
|||
state.prevTime = stateTime |
|||
}() |
|||
|
|||
if state.uniqueID != newData.UniqueID { |
|||
state.uniqueID = newData.UniqueID |
|||
} |
|||
|
|||
if state.prevData != nil && newData.Type != state.prevData.Type { |
|||
return nil |
|||
} |
|||
if time.Since(stateTime) > time.Second*3 { |
|||
return nil |
|||
} |
|||
|
|||
switch newData.Type { |
|||
case "ZLLSwitch": |
|||
{ |
|||
pe := state.prevData.State.ButtonEvent |
|||
ce := newData.State.ButtonEvent |
|||
|
|||
td := stateTime.Sub(state.prevTime) >= time.Second |
|||
pIdx := (pe / 1000) - 1 |
|||
cIdx := (ce / 1000) - 1 |
|||
pBtn := pe % 1000 / 2 // 0 = pressed, 1 = released
|
|||
cBtn := ce % 1000 / 2 // 0 = pressed, 1 = released
|
|||
|
|||
if pIdx == cIdx && cBtn == 1 && pBtn == 0 { |
|||
// Do not allow 4002 after 4000.
|
|||
// 4000 after 4002 is fine, though.
|
|||
break |
|||
} |
|||
|
|||
if cBtn != pBtn || cIdx != pIdx || td { |
|||
return &models.Event{ |
|||
Name: models.ENButtonPressed, |
|||
Payload: map[string]string{ |
|||
"buttonIndex": strconv.Itoa(cIdx), |
|||
"buttonName": buttonNames[cIdx], |
|||
"deviceId": strconv.Itoa(state.externalID), |
|||
"hueButtonEvent": strconv.Itoa(ce), |
|||
}, |
|||
} |
|||
} |
|||
} |
|||
case "ZLLPresence": |
|||
{ |
|||
// TODO: Test this!
|
|||
if state.prevData != nil && state.prevData.State.Presence != newData.State.Presence { |
|||
name := models.ENSensorPresenceStarted |
|||
if !newData.State.Presence { |
|||
name = models.ENSensorPresenceEnded |
|||
} |
|||
|
|||
return &models.Event{ |
|||
Name: name, |
|||
Payload: map[string]string{ |
|||
"deviceId": strconv.Itoa(state.externalID), |
|||
"deviceInternalId": newData.UniqueID, |
|||
}, |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func ptrBool(v bool) *bool { |
|||
return &v |
|||
} |
|||
|
|||
func ptrInt(v int) *int { |
|||
return &v |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue