46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
// Copyright (c) 2024 Joshua Rich <joshua.rich@gmail.com>
|
|
//
|
|
// This software is released under the MIT License.
|
|
// https://opensource.org/licenses/MIT
|
|
|
|
//go:generate moq -out interfaces_mocks_test.go . Registry SensorTracker
|
|
package agent
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/joshuar/go-hass-agent/internal/agent/ui"
|
|
"github.com/joshuar/go-hass-agent/internal/hass"
|
|
"github.com/joshuar/go-hass-agent/internal/hass/sensor"
|
|
)
|
|
|
|
type Device interface {
|
|
DeviceName() string
|
|
DeviceID() string
|
|
Setup(ctx context.Context) context.Context
|
|
Updates() chan sensor.Details
|
|
}
|
|
|
|
// UI are the methods required for the agent to display its windows, tray
|
|
// and notifications.
|
|
type UI interface {
|
|
DisplayNotification(n ui.Notification)
|
|
DisplayTrayIcon(ctx context.Context, agent ui.Agent, trk ui.SensorTracker)
|
|
DisplayRegistrationWindow(ctx context.Context, input *hass.RegistrationInput, doneCh chan struct{})
|
|
Run(ctx context.Context, doneCh chan struct{})
|
|
}
|
|
|
|
type SensorTracker interface {
|
|
SensorList() []string
|
|
Process(ctx context.Context, reg sensor.Registry, sensorUpdates ...<-chan sensor.Details) error
|
|
Get(key string) (sensor.Details, error)
|
|
Reset()
|
|
}
|
|
|
|
type Registry interface {
|
|
SetDisabled(id string, state bool) error
|
|
SetRegistered(id string, state bool) error
|
|
IsDisabled(id string) bool
|
|
IsRegistered(id string) bool
|
|
}
|