99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package google
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/jellydator/ttlcache/v3"
|
|
"google.golang.org/api/homegraph/v1"
|
|
)
|
|
|
|
type ExecuteResponse struct {
|
|
UpdatedState DeviceState
|
|
UpdatedDevices []string
|
|
OfflineDevices []string
|
|
// The key is the errorCode that is associated with the devices
|
|
FailedDevices map[string]struct {
|
|
Devices []string
|
|
}
|
|
}
|
|
|
|
type Provider interface {
|
|
Sync(context.Context, string) ([]*Device, error)
|
|
Query(context.Context, string, []DeviceHandle) (map[string]DeviceState, error)
|
|
Execute(context.Context, string, []Command) (*ExecuteResponse, error)
|
|
}
|
|
|
|
type Service struct {
|
|
provider Provider
|
|
deviceService *homegraph.DevicesService
|
|
cache *ttlcache.Cache[string, string]
|
|
}
|
|
|
|
func NewService(provider Provider, service *homegraph.Service) *Service {
|
|
s := Service{
|
|
provider: provider,
|
|
deviceService: homegraph.NewDevicesService(service),
|
|
cache: ttlcache.New(
|
|
ttlcache.WithTTL[string, string](30 * time.Minute),
|
|
),
|
|
}
|
|
|
|
go s.cache.Start()
|
|
|
|
return &s
|
|
}
|
|
|
|
func (s *Service) RequestSync(ctx context.Context, userID string) error {
|
|
call := s.deviceService.RequestSync(&homegraph.RequestSyncDevicesRequest{
|
|
AgentUserId: userID,
|
|
})
|
|
|
|
call.Context(ctx)
|
|
resp, err := call.Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.ServerResponse.HTTPStatusCode != http.StatusOK {
|
|
return errors.New(fmt.Sprintf("sync failed: %d", resp.ServerResponse.HTTPStatusCode))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) ReportState(ctx context.Context, userID string, states map[string]DeviceState) error {
|
|
j, err := json.Marshal(states)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
call := s.deviceService.ReportStateAndNotification(&homegraph.ReportStateAndNotificationRequest{
|
|
AgentUserId: userID,
|
|
EventId: uuid.New().String(),
|
|
RequestId: uuid.New().String(),
|
|
Payload: &homegraph.StateAndNotificationPayload{
|
|
Devices: &homegraph.ReportStateAndNotificationDevice{
|
|
States: j,
|
|
},
|
|
},
|
|
})
|
|
|
|
call.Context(ctx)
|
|
resp, err := call.Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.ServerResponse.HTTPStatusCode != http.StatusOK {
|
|
return errors.New(fmt.Sprintf("report failed: %d", resp.ServerResponse.HTTPStatusCode))
|
|
}
|
|
|
|
return nil
|
|
}
|