97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package home
|
|
|
|
import (
|
|
"automation/config"
|
|
"automation/device"
|
|
"automation/integration/google"
|
|
"context"
|
|
"log"
|
|
|
|
"google.golang.org/api/homegraph/v1"
|
|
"google.golang.org/api/option"
|
|
)
|
|
|
|
type Home struct {
|
|
Service *google.Service
|
|
Username string
|
|
|
|
Devices map[device.InternalName]device.Basic
|
|
}
|
|
|
|
// Auto populate and update the device list
|
|
func New(username string, credentials config.Credentials) *Home {
|
|
home := &Home{Username: username, Devices: make(map[device.InternalName]device.Basic)}
|
|
|
|
homegraphService, err := homegraph.NewService(context.Background(), option.WithCredentialsJSON(credentials))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
home.Service = google.NewService(home, homegraphService)
|
|
|
|
return home
|
|
}
|
|
|
|
func (h *Home) AddDevice(d device.Basic) {
|
|
h.Devices[d.GetID()] = d
|
|
|
|
log.Printf("Added %s in %s (%s)\n", d.GetID().Name(), d.GetID().Room(), d.GetID())
|
|
}
|
|
|
|
func (h *Home) Sync(_ context.Context, _ string) ([]*google.Device, error) {
|
|
var devices []*google.Device
|
|
|
|
for _, device := range device.GetDevices[google.DeviceInterface](&h.Devices) {
|
|
devices = append(devices, device.Sync())
|
|
}
|
|
|
|
return devices, nil
|
|
}
|
|
|
|
func (h *Home) Query(_ context.Context, _ string, handles []google.DeviceHandle) (map[string]google.DeviceState, error) {
|
|
states := make(map[string]google.DeviceState)
|
|
|
|
for _, handle := range handles {
|
|
if device, err := device.GetDevice[google.DeviceInterface](&h.Devices, device.InternalName(handle.ID)); err == nil {
|
|
states[device.GetID().String()] = device.Query()
|
|
} else {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
return states, nil
|
|
}
|
|
|
|
func (h *Home) Execute(_ context.Context, _ string, commands []google.Command) (*google.ExecuteResponse, error) {
|
|
resp := &google.ExecuteResponse{
|
|
UpdatedState: google.NewDeviceState(true),
|
|
FailedDevices: make(map[string]struct{ Devices []string }),
|
|
}
|
|
|
|
for _, command := range commands {
|
|
for _, execution := range command.Execution {
|
|
for _, handle := range command.Devices {
|
|
if device, err := device.GetDevice[google.DeviceInterface](&h.Devices, device.InternalName(handle.ID)); err == nil {
|
|
errCode, online := device.Execute(execution, &resp.UpdatedState)
|
|
|
|
// Update the state
|
|
h.Devices[device.GetID()] = device
|
|
if !online {
|
|
resp.OfflineDevices = append(resp.OfflineDevices, handle.ID)
|
|
} else if len(errCode) == 0 {
|
|
resp.UpdatedDevices = append(resp.UpdatedDevices, handle.ID)
|
|
} else {
|
|
e := resp.FailedDevices[errCode]
|
|
e.Devices = append(e.Devices, handle.ID)
|
|
resp.FailedDevices[errCode] = e
|
|
}
|
|
} else {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return resp, nil
|
|
}
|